86 lines
2.1 KiB
C++
86 lines
2.1 KiB
C++
|
#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 flags)
|
||
|
|
{
|
||
|
|
printf("[%d] pts: %u, dts: %u, bytes: %u%s\n", codec, pts, dts, (unsigned int)bytes, flags?" [I]":"");
|
||
|
|
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, ×tamp, &taglen, packet, sizeof(packet)))
|
||
|
|
{
|
||
|
|
if(FLV_TYPE_SCRIPT == tag)
|
||
|
|
flv_writer_input(w, tag, packet, taglen, timestamp);
|
||
|
|
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);
|
||
|
|
}
|