ZLKMedia/3rdpart/media-server/libflv/test/flv-read-write-test.cpp

#include "flv-reader.h"
#include "flv-writer.h"
#include "flv-demuxer.h"
#include "flv-muxer.h"
#include "flv-proto.h"
#include <assert.h>
#include <string.h>
#include <stdio.h>

static int flv_onmuxer(void* flv, int type, const void* data, size_t bytes, uint32_t timestamp)
{
	return flv_writer_input(flv, type, data, bytes, timestamp);
}

static int flv_ondemuxer(void* param, int codec, const void* data, size_t bytes, uint32_t pts, uint32_t dts, int format)
{
	flv_muxer_t* muxer = (flv_muxer_t*)param;
	switch (codec)
	{
	case FLV_AUDIO_AAC:
		flv_muxer_aac(muxer, data, bytes, pts, dts);
		break;

	case FLV_AUDIO_MP3:
		flv_muxer_mp3(muxer, data, bytes, pts, dts);
		break;

	case FLV_VIDEO_H264:
		flv_muxer_avc(muxer, data, bytes, pts, dts);
		break;

	case FLV_VIDEO_H265:
		flv_muxer_hevc(muxer, data, bytes, pts, dts);
		break;

	case FLV_VIDEO_AV1:
		flv_muxer_av1(muxer, data, bytes, pts, dts);
		break;

	default:
		break;
	}
	return 0;
}

void flv_read_write_test(const char* flv)
{
	static char packet[2 * 1024 * 1024];
	snprintf(packet, sizeof(packet), "%s.flv", flv);

	void* r = flv_reader_create(flv);
	void* w = flv_writer_create(packet);
	flv_muxer_t* e = flv_muxer_create(flv_onmuxer, w);
	flv_demuxer_t* d = flv_demuxer_create(flv_ondemuxer, e);

	//struct flv_metadata_t metadata;
	//metadata.audiocodecid = 4;
	//metadata.audiodatarate = 16.1;
	//metadata.audiosamplerate = 48000;
	//metadata.audiosamplesize = 16;
	//metadata.stereo = true;
	//metadata.videocodecid = 7;
	//metadata.videodatarate = 64.0;
	//metadata.framerate = 25;
	//metadata.width = 1920;
	//metadata.height = 1080;
	//flv_muxer_metadata(e, &metadata);

	int ret, tag;
	size_t taglen;
	uint32_t timestamp;
	while (1 == flv_reader_read(r, &tag, &timestamp, &taglen, packet, sizeof(packet)))
	{
		ret = flv_demuxer_input(d, tag, packet, taglen, timestamp);
		assert(0 == ret);
	}

	flv_muxer_destroy(e);
	flv_demuxer_destroy(d);
	flv_reader_destroy(r);
	flv_writer_destroy(w);
}