Add FLV file output code

This doesn't add FLV file output to the user interface yet, but we'll
get around to that eventually.  This just adds an FLV output type.

Also, removed ftello/fseeko because off_t is a really annoying data
type, and I'd rather have a firm int64_t for large sizes, so I named it
to os_fseeki64 and os_ftelli64 instead, and changed the file size
function to return an int64_t.

plugins/obs-outputs/CMakeLists.txt

@@ -32,10 +32,12 @@ set(obs-outputs_HEADERS
 	obs-output-ver.h
 	rtmp-helpers.h
 	flv-mux.h
+	flv-output.h
 	librtmp)
 set(obs-outputs_SOURCES
 	obs-outputs.c
 	rtmp-stream.c
+	flv-output.c
 	flv-mux.c)
 	
 add_library(obs-outputs MODULE

plugins/obs-outputs/flv-mux.c

@@ -16,6 +16,7 @@
 ******************************************************************************/
 
 #include <obs.h>
+#include <stdio.h>
 #include <util/array-serializer.h>
 #include "flv-mux.h"
 #include "obs-output-ver.h"
@@ -28,7 +29,6 @@
 //#define WRITE_FLV_HEADER
 
 #define VIDEO_HEADER_SIZE 5
-#define MILLISECOND_DEN   1000
 
 static inline double encoder_bitrate(obs_encoder_t encoder)
 {
@@ -39,6 +39,22 @@ static inline double encoder_bitrate(obs_encoder_t encoder)
 	return bitrate;
 }
 
+#define FLV_INFO_SIZE_OFFSET 42
+
+void write_file_info(FILE *file, int64_t duration_ms, int64_t size)
+{
+	char buf[64];
+	char *enc = buf;
+	char *end = enc + sizeof(buf);
+
+	fseek(file, FLV_INFO_SIZE_OFFSET, SEEK_SET);
+
+	enc_num_val(&enc, end, "duration", (double)duration_ms / 1000.0);
+	enc_num_val(&enc, end, "fileSize", (double)size);
+
+	fwrite(buf, 1, enc - buf, file);
+}
+
 static void build_flv_meta_data(obs_output_t context,
 		uint8_t **output, size_t *size)
 {
@@ -83,7 +99,8 @@ static void build_flv_meta_data(obs_output_t context,
 	*output = bmemdup(buf, *size);
 }
 
-void flv_meta_data(obs_output_t context, uint8_t **output, size_t *size)
+void flv_meta_data(obs_output_t context, uint8_t **output, size_t *size,
+		bool write_header)
 {
 	struct array_output_data data;
 	struct serializer s;
@@ -95,13 +112,13 @@ void flv_meta_data(obs_output_t context, uint8_t **output, size_t *size)
 
 	build_flv_meta_data(context, &meta_data, &meta_data_size);
 
-#ifdef WRITE_FLV_HEADER
-	s_write(&s, "FLV", 3);
-	s_w8(&s, 1);
-	s_w8(&s, 5);
-	s_wb32(&s, 9);
-	s_wb32(&s, 0);
-#endif
+	if (write_header) {
+		s_write(&s, "FLV", 3);
+		s_w8(&s, 1);
+		s_w8(&s, 5);
+		s_wb32(&s, 9);
+		s_wb32(&s, 0);
+	}
 
 	start_pos = serializer_get_pos(&s);
 
@@ -121,11 +138,6 @@ void flv_meta_data(obs_output_t context, uint8_t **output, size_t *size)
 	bfree(meta_data);
 }
 
-static uint32_t get_ms_time(struct encoder_packet *packet, int64_t val)
-{
-	return (uint32_t)(val * MILLISECOND_DEN / packet->timebase_den);
-}
-
 #ifdef DEBUG_TIMESTAMPS
 static int32_t last_time = 0;
 #endif

plugins/obs-outputs/flv-mux.h

@@ -19,6 +19,16 @@
 
 #include <obs.h>
 
-extern void flv_meta_data(obs_output_t context, uint8_t **output, size_t *size);
+#define MILLISECOND_DEN   1000
+
+static uint32_t get_ms_time(struct encoder_packet *packet, int64_t val)
+{
+	return (uint32_t)(val * MILLISECOND_DEN / packet->timebase_den);
+}
+
+extern void write_file_info(FILE *file, int64_t duration_ms, int64_t size);
+
+extern void flv_meta_data(obs_output_t context, uint8_t **output, size_t *size,
+		bool write_header);
 extern void flv_packet_mux(struct encoder_packet *packet,
 		uint8_t **output, size_t *size, bool is_header);

plugins/obs-outputs/flv-output.c

@@ -0,0 +1,211 @@
+/******************************************************************************
+    Copyright (C) 2014 by Hugh Bailey <obs.jim@gmail.com>
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 2 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+******************************************************************************/
+
+#include <stdio.h>
+#include <obs.h>
+#include <obs-avc.h>
+#include <util/platform.h>
+#include <util/dstr.h>
+#include <util/threading.h>
+#include <inttypes.h>
+#include "flv-mux.h"
+
+struct flv_output {
+	obs_output_t output;
+	struct dstr  path;
+	FILE         *file;
+	bool         active;
+	int64_t      last_packet_ts;
+};
+
+static const char *flv_output_getname(const char *locale)
+{
+	UNUSED_PARAMETER(locale);
+	return "FLV File Output";
+}
+
+static void flv_output_stop(void *data);
+
+static void flv_output_destroy(void *data)
+{
+	struct flv_output *stream = data;
+
+	if (stream->active)
+		flv_output_stop(data);
+
+	dstr_free(&stream->path);
+	bfree(stream);
+}
+
+static void *flv_output_create(obs_data_t settings, obs_output_t output)
+{
+	struct flv_output *stream = bzalloc(sizeof(struct flv_output));
+	stream->output = output;
+
+	UNUSED_PARAMETER(settings);
+	return stream;
+}
+
+static void flv_output_stop(void *data)
+{
+	struct flv_output *stream = data;
+
+	if (stream->active) {
+		if (stream->file)
+			write_file_info(stream->file, stream->last_packet_ts,
+					os_ftelli64(stream->file));
+
+		fclose(stream->file);
+		obs_output_end_data_capture(stream->output);
+		stream->active = false;
+	}
+}
+
+static int write_packet(struct flv_output *stream,
+		struct encoder_packet *packet, bool is_header)
+{
+	uint8_t *data;
+	size_t  size;
+	int     ret = 0;
+
+	stream->last_packet_ts = get_ms_time(packet, packet->dts);
+
+	flv_packet_mux(packet, &data, &size, is_header);
+	fwrite(data, 1, size, stream->file);
+	bfree(data);
+	obs_free_encoder_packet(packet);
+
+	return ret;
+}
+
+static void write_meta_data(struct flv_output *stream)
+{
+	uint8_t *meta_data;
+	size_t  meta_data_size;
+
+	flv_meta_data(stream->output, &meta_data, &meta_data_size, true);
+	fwrite(meta_data, 1, meta_data_size, stream->file);
+	bfree(meta_data);
+}
+
+static void write_audio_header(struct flv_output *stream)
+{
+	obs_output_t  context  = stream->output;
+	obs_encoder_t aencoder = obs_output_get_audio_encoder(context);
+	uint8_t       *header;
+
+	struct encoder_packet packet   = {
+		.type         = OBS_ENCODER_AUDIO,
+		.timebase_den = 1
+	};
+
+	obs_encoder_get_extra_data(aencoder, &header, &packet.size);
+	packet.data = bmemdup(header, packet.size);
+	write_packet(stream, &packet, true);
+}
+
+static void write_video_header(struct flv_output *stream)
+{
+	obs_output_t  context  = stream->output;
+	obs_encoder_t vencoder = obs_output_get_video_encoder(context);
+	uint8_t       *header;
+	size_t        size;
+
+	struct encoder_packet packet   = {
+		.type         = OBS_ENCODER_VIDEO,
+		.timebase_den = 1,
+		.keyframe     = true
+	};
+
+	obs_encoder_get_extra_data(vencoder, &header, &size);
+	packet.size = obs_parse_avc_header(&packet.data, header, size);
+	write_packet(stream, &packet, true);
+}
+
+static void write_headers(struct flv_output *stream)
+{
+	write_meta_data(stream);
+	write_audio_header(stream);
+	write_video_header(stream);
+}
+
+static bool flv_output_start(void *data)
+{
+	struct flv_output *stream = data;
+	obs_data_t settings;
+	const char *path;
+
+	if (!obs_output_can_begin_data_capture(stream->output, 0))
+		return false;
+	if (!obs_output_initialize_encoders(stream->output, 0))
+		return false;
+
+	/* get path */
+	settings = obs_output_get_settings(stream->output);
+	path = obs_data_getstring(settings, "path");
+	dstr_copy(&stream->path, path);
+	obs_data_release(settings);
+
+	stream->file = os_fopen(stream->path.array, "wb");
+	if (!stream->file) {
+		blog(LOG_WARNING, "Unable to open FLV file '%s'",
+				stream->path.array);
+		return false;
+	}
+
+	/* write headers and start capture */
+	stream->active = true;
+	write_headers(stream);
+	obs_output_begin_data_capture(stream->output, 0);
+
+	return true;
+}
+
+static void flv_output_data(void *data, struct encoder_packet *packet)
+{
+	struct flv_output     *stream = data;
+	struct encoder_packet parsed_packet;
+
+	if (packet->type == OBS_ENCODER_VIDEO) {
+		obs_parse_avc_packet(&parsed_packet, packet);
+		write_packet(stream, &parsed_packet, false);
+		obs_free_encoder_packet(&parsed_packet);
+	} else {
+		write_packet(stream, packet, false);
+	}
+}
+
+static obs_properties_t flv_output_properties(const char *locale)
+{
+	obs_properties_t props = obs_properties_create(locale);
+
+	/* TODO: locale */
+	obs_properties_add_text(props, "path", "File Path", OBS_TEXT_DEFAULT);
+	return props;
+}
+
+struct obs_output_info flv_output_info = {
+	.id             = "flv_output",
+	.flags          = OBS_OUTPUT_AV | OBS_OUTPUT_ENCODED,
+	.getname        = flv_output_getname,
+	.create         = flv_output_create,
+	.destroy        = flv_output_destroy,
+	.start          = flv_output_start,
+	.stop           = flv_output_stop,
+	.encoded_packet = flv_output_data,
+	.properties     = flv_output_properties
+};

plugins/obs-outputs/obs-outputs.c

@@ -8,6 +8,7 @@
 OBS_DECLARE_MODULE()
 
 extern struct obs_output_info rtmp_output_info;
+extern struct obs_output_info flv_output_info;
 
 bool obs_module_load(uint32_t libobs_ver)
 {
@@ -17,6 +18,7 @@ bool obs_module_load(uint32_t libobs_ver)
 #endif
 
 	obs_register_output(&rtmp_output_info);
+	obs_register_output(&flv_output_info);
 
 	UNUSED_PARAMETER(libobs_ver);
 	return true;

plugins/obs-outputs/rtmp-stream.c

@@ -255,7 +255,7 @@ static void send_meta_data(struct rtmp_stream *stream)
 	uint8_t *meta_data;
 	size_t  meta_data_size;
 
-	flv_meta_data(stream->output, &meta_data, &meta_data_size);
+	flv_meta_data(stream->output, &meta_data, &meta_data_size, false);
 #ifdef FILE_TEST
 	fwrite(meta_data, 1, meta_data_size, stream->test);
 #else