refactor to remove duplicate code

1 files changed, 22 insertions, 2 deletions

diff --git a/src/OffscreenContext.h b/src/OffscreenContext.h
index a1792b4..3494898 100644
--- a/src/OffscreenContext.h
+++ b/src/OffscreenContext.h
@@ -2,13 +2,33 @@
 #define OFFSCREENCONTEXT_H_
 
 #include <iostream>
+#include <fstream>
 #include <string>
 
 struct OffscreenContext *create_offscreen_context(int w, int h);
-void bind_offscreen_context(OffscreenContext *ctx);
 bool teardown_offscreen_context(OffscreenContext *ctx);
-bool save_framebuffer(OffscreenContext *ctx, const char *filename);
 bool save_framebuffer(OffscreenContext *ctx, std::ostream &output);
 std::string offscreen_context_getinfo(OffscreenContext *ctx);
 
+void bind_offscreen_context(OffscreenContext *ctx)
+{
+	if (ctx) fbo_bind(ctx->fbo);
+}
+
+/*
+  Capture framebuffer from OpenGL and write it to the given filename as PNG.
+*/
+inline bool save_framebuffer(OffscreenContext *ctx, const char *filename)
+{
+  std::ofstream fstream(filename);
+  if (!fstream.is_open()) {
+    std::cerr << "Can't open file " << filename << " for writing";
+    return false;
+  } else {
+    save_framebuffer(ctx, fstream);
+    fstream.close();
+  }
+  return true;
+}
+
 #endif