81 lines
2.2 KiB
C++
81 lines
2.2 KiB
C++
#include "flv-writer.h"
|
|
#include "flv-muxer.h"
|
|
#include "mpeg4-aac.h"
|
|
#include "mpeg-ts.h"
|
|
#include "mpeg-types.h"
|
|
#include <stdio.h>
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
static int on_flv_packet(void* flv, int type, const void* data, size_t bytes, uint32_t timestamp)
|
|
{
|
|
return flv_writer_input(flv, type, data, bytes, timestamp);
|
|
}
|
|
|
|
static int on_ts_packet(void* param, int program, int /*stream*/, int avtype, int flags, int64_t pts, int64_t dts, const void* data, size_t bytes)
|
|
{
|
|
static int64_t s_pts = PTS_NO_VALUE;
|
|
if (PTS_NO_VALUE == s_pts)
|
|
s_pts = pts;
|
|
pts -= s_pts;
|
|
dts -= s_pts;
|
|
|
|
flv_muxer_t* muxer = (flv_muxer_t*)param;
|
|
if (PSI_STREAM_AAC == avtype)
|
|
{
|
|
int len = mpeg4_aac_adts_frame_length((const uint8_t*)data, bytes);
|
|
while (len > 0 && len <= bytes)
|
|
{
|
|
flv_muxer_aac(muxer, data, len, (uint32_t)(pts / 90), (uint32_t)(pts / 90));
|
|
|
|
mpeg4_aac_t aac;
|
|
memset(&aac, 0, sizeof(aac));
|
|
if(mpeg4_aac_adts_load((const uint8_t*)data, bytes, &aac) > 0 && aac.sampling_frequency > 0)
|
|
pts += 1024 /*frame*/ * 1000 / aac.sampling_frequency;
|
|
data = (const uint8_t*)data + len;
|
|
bytes -= len;
|
|
len = mpeg4_aac_adts_frame_length((const uint8_t*)data, bytes);
|
|
}
|
|
assert(0 == bytes);
|
|
}
|
|
else if (PSI_STREAM_MP3 == avtype)
|
|
{
|
|
flv_muxer_mp3(muxer, data, bytes, (uint32_t)(pts / 90), (uint32_t)(dts / 90));
|
|
}
|
|
else if (PSI_STREAM_H264 == avtype)
|
|
{
|
|
flv_muxer_avc(muxer, data, bytes, (uint32_t)(pts / 90), (uint32_t)(dts / 90));
|
|
}
|
|
else if (PSI_STREAM_H265 == avtype)
|
|
{
|
|
flv_muxer_hevc(muxer, data, bytes, (uint32_t)(pts / 90), (uint32_t)(dts / 90));
|
|
}
|
|
else if (PSI_STREAM_VIDEO_AVS3 == avtype)
|
|
{
|
|
flv_muxer_avs3(muxer, data, bytes, (uint32_t)(pts / 90), (uint32_t)(dts / 90));
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void ts2flv_test(const char* inputTS, const char* outputFLV)
|
|
{
|
|
void* f = flv_writer_create(outputFLV);
|
|
flv_muxer_t* m = flv_muxer_create(on_flv_packet, f);
|
|
|
|
unsigned char ptr[188];
|
|
FILE* fp = fopen(inputTS, "rb");
|
|
ts_demuxer_t *ts = ts_demuxer_create(on_ts_packet, m);
|
|
while (1 == fread(ptr, sizeof(ptr), 1, fp))
|
|
{
|
|
ts_demuxer_input(ts, ptr, sizeof(ptr));
|
|
}
|
|
ts_demuxer_flush(ts);
|
|
ts_demuxer_destroy(ts);
|
|
fclose(fp);
|
|
|
|
flv_muxer_destroy(m);
|
|
flv_writer_destroy(f);
|
|
}
|