added video recorder based on SDL_ffmpeg

1 files changed, 125 insertions, 0 deletions

diff --git a/src/recorder.cpp b/src/recorder.cpp
new file mode 100644
index 0000000..13e2b13
--- /dev/null
+++ b/src/recorder.cpp
@@ -0,0 +1,125 @@
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <stdlib.h>
+#include <stdio.h>
+
+#include <SDL.h>
+#include <SDL/SDL_ffmpeg.h>
+
+#include "osc_graphics.h"
+#include "osc_server.h"
+
+#include "recorder.h"
+
+/*
+ * Macros
+ */
+#define SDL_FFMPEGFREE_SAFE(FILE) do {			\
+	if (FILE) {					\
+		SDL_ffmpegFree(FILE);			\
+		FILE = NULL;				\
+	}						\
+} while (0)
+
+#define SDL_FFMPEG_ERROR(FMT, ...) do {			\
+	fprintf(stderr, "%s(%d): " FMT ": %s\n",	\
+		__FILE__, __LINE__, ##__VA_ARGS__,	\
+		SDL_ffmpegGetError());			\
+} while (0)
+
+extern "C" {
+
+static int start_handler(const char *path, const char *types,
+		       	 lo_arg **argv, int argc,
+		       	 void *data, void *user_data);
+static int stop_handler(const char *path, const char *types,
+			lo_arg **argv, int argc,
+		       	void *data, void *user_data);
+
+}
+
+void
+Recorder::register_methods()
+{
+	osc_server.add_method("s", start_handler, this, "/recorder/start");
+	osc_server.add_method("", stop_handler, this, "/recorder/stop");
+}
+
+static int
+start_handler(const char *path, const char *types,
+	      lo_arg **argv, int argc, void *data, void *user_data)
+{
+	Recorder *obj = (Recorder *)user_data;
+
+	obj->start(&argv[0]->s);
+
+	return 0;
+}
+
+void
+Recorder::start(const char *filename)
+{
+	SDL_ffmpegCodec codec = {
+		-1,			/* video codec based on file name */
+		screen->w, screen->h,
+		1, 20 /* FIXME */,	/* framerate */
+		6000000, -1, -1,
+		-1, 0, -1, -1, -1, -1	/* no audio */
+	};
+
+	lock();
+
+	SDL_FFMPEGFREE_SAFE(file);
+
+	file = SDL_ffmpegCreate(filename);
+	if (!file) {
+		SDL_FFMPEG_ERROR("SDL_ffmpegCreate");
+		exit(EXIT_FAILURE);
+	}
+
+	SDL_ffmpegAddVideoStream(file, codec);
+	SDL_ffmpegSelectVideoStream(file, 0);
+
+	unlock();
+}
+
+static int
+stop_handler(const char *path, const char *types,
+	     lo_arg **argv, int argc, void *data, void *user_data)
+{
+	Recorder *obj = (Recorder *)user_data;
+
+	obj->stop();
+
+	return 0;
+}
+
+void
+Recorder::stop()
+{
+	lock();
+	SDL_FFMPEGFREE_SAFE(file);
+	unlock();
+}
+
+void
+Recorder::record(SDL_Surface *surf)
+{
+	lock();
+
+	if (file)
+		SDL_ffmpegAddVideoFrame(file, surf);
+
+	unlock();
+}
+
+Recorder::~Recorder()
+{
+	osc_server.del_method("s", "/recorder/start");
+	osc_server.del_method("",  "/recorder/stop");
+
+	SDL_FFMPEGFREE_SAFE(file);
+	SDL_DestroyMutex(mutex);
+}