mirror of
https://github.com/ireader/media-server.git
synced 2026-01-12 00:18:51 +08:00
92 lines
2.2 KiB
C++
92 lines
2.2 KiB
C++
#include "mpeg-ps.h"
|
|
#include "mpeg-ts.h"
|
|
#include <assert.h>
|
|
#include <string.h>
|
|
#include <stdio.h>
|
|
#include <map>
|
|
|
|
static void* ts_alloc(void* /*param*/, size_t bytes)
|
|
{
|
|
static char s_buffer[188];
|
|
assert(bytes <= sizeof(s_buffer));
|
|
return s_buffer;
|
|
}
|
|
|
|
static void ts_free(void* /*param*/, void* /*packet*/)
|
|
{
|
|
return;
|
|
}
|
|
|
|
static int ts_write(void* param, const void* packet, size_t bytes)
|
|
{
|
|
return 1 == fwrite(packet, bytes, 1, (FILE*)param) ? 0 : ferror((FILE*)param);
|
|
}
|
|
|
|
inline const char* ts_type(int type)
|
|
{
|
|
switch (type)
|
|
{
|
|
case PSI_STREAM_MP3: return "MP3";
|
|
case PSI_STREAM_AAC: return "AAC";
|
|
case PSI_STREAM_H264: return "H264";
|
|
case PSI_STREAM_H265: return "H265";
|
|
case PSI_STREAM_VIDEO_AVS3: return "AVS3";
|
|
default: return "*";
|
|
}
|
|
}
|
|
|
|
static int ts_stream(void* ts, int stream, int codecid)
|
|
{
|
|
static std::map<int, int> streams;
|
|
std::map<int, int>::const_iterator it = streams.find(stream);
|
|
if (streams.end() != it)
|
|
return it->second;
|
|
|
|
int i = mpeg_ts_add_stream(ts, codecid, NULL, 0);
|
|
streams[stream] = i;
|
|
return i;
|
|
}
|
|
|
|
static int on_ts_packet(void* ts, int program, int stream, int avtype, int flags, int64_t pts, int64_t dts, const void* data, size_t bytes)
|
|
{
|
|
printf("[%d:%d][%s] pts: %08lu, dts: %08lu%s\n", program, stream, ts_type(avtype), (unsigned long)pts, (unsigned long)dts, flags ? " [I]":"");
|
|
|
|
return mpeg_ts_write(ts, ts_stream(ts, stream, avtype), flags, pts, dts, data, bytes);
|
|
}
|
|
|
|
static void mpeg_ts_file(const char* file, void* muxer)
|
|
{
|
|
unsigned char ptr[188];
|
|
struct ts_demuxer_t *ts;
|
|
FILE* fp = fopen(file, "rb");
|
|
|
|
ts = ts_demuxer_create(on_ts_packet, muxer);
|
|
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);
|
|
}
|
|
|
|
//mpeg_ts_test("test/fileSequence0.ts", "test/apple.ts")
|
|
void mpeg_ts_test(const char* input)
|
|
{
|
|
char output[256] = { 0 };
|
|
snprintf(output, sizeof(output) - 1, "%s.ts", input);
|
|
|
|
struct mpeg_ts_func_t tshandler;
|
|
tshandler.alloc = ts_alloc;
|
|
tshandler.write = ts_write;
|
|
tshandler.free = ts_free;
|
|
|
|
FILE* fp = fopen(output, "wb");
|
|
void* ts = mpeg_ts_create(&tshandler, fp);
|
|
|
|
mpeg_ts_file(input, ts);
|
|
|
|
mpeg_ts_destroy(ts);
|
|
fclose(fp);
|
|
}
|