71 lines
2.0 KiB
C++
71 lines
2.0 KiB
C++
#include "mpeg-ps.h"
|
|
#include "mpeg-ts-proto.h"
|
|
#include <stdio.h>
|
|
#include <assert.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
static FILE *vfp;
|
|
static FILE *afp;
|
|
|
|
inline const char *ftimestamp(int64_t t, char *buf)
|
|
{
|
|
if (PTS_NO_VALUE == t) {
|
|
sprintf(buf, "(null)");
|
|
} else {
|
|
t /= 90;
|
|
sprintf(buf, "%d:%02d:%02d.%03d", (int)(t / 36000000), (int)((t / 60000) % 60), (int)((t / 1000) % 60), (int)(t % 1000));
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
static void onpacket(void * /*param*/, int /*stream*/, int avtype, int flags, int64_t pts, int64_t dts, const void *data, size_t bytes)
|
|
{
|
|
static char s_pts[64], s_dts[64];
|
|
|
|
if (PSI_STREAM_AAC == avtype) {
|
|
static int64_t a_pts = 0, a_dts = 0;
|
|
if (PTS_NO_VALUE == dts) {
|
|
dts = pts;
|
|
}
|
|
//assert(0 == a_dts || dts >= a_dts);
|
|
printf("[A] pts: %s(%lld), dts: %s(%lld), diff: %03d/%03d\n", ftimestamp(pts, s_pts), pts, ftimestamp(dts, s_dts), dts, (int)(pts - a_pts) / 90, (int)(dts - a_dts) / 90);
|
|
a_pts = pts;
|
|
a_dts = dts;
|
|
|
|
fwrite(data, 1, bytes, afp);
|
|
} else if (PSI_STREAM_H264 == avtype) {
|
|
static int64_t v_pts = 0, v_dts = 0;
|
|
assert(0 == v_dts || dts >= v_dts);
|
|
printf("[V] pts: %s(%lld), dts: %s(%lld), diff: %03d/%03d\n", ftimestamp(pts, s_pts), pts, ftimestamp(dts, s_dts), dts, (int)(pts - v_pts) / 90, (int)(dts - v_dts) / 90);
|
|
v_pts = pts;
|
|
v_dts = dts;
|
|
|
|
fwrite(data, 1, bytes, vfp);
|
|
} else {
|
|
//assert(0);
|
|
}
|
|
}
|
|
|
|
static uint8_t s_packet[2 * 1024 * 1024];
|
|
|
|
void mpeg_ps_dec_test(const char *file)
|
|
{
|
|
FILE *fp = fopen(file, "rb");
|
|
vfp = fopen("v.h264", "wb");
|
|
afp = fopen("a.aac", "wb");
|
|
|
|
size_t n, i = 0;
|
|
ps_demuxer_t *ps = ps_demuxer_create(onpacket, NULL);
|
|
while ((n = fread(s_packet + i, 1, sizeof(s_packet) - i, fp)) > 0) {
|
|
size_t r = ps_demuxer_input(ps, s_packet, n + i);
|
|
memmove(s_packet, s_packet + r, n + i - r);
|
|
i = n + i - r;
|
|
}
|
|
ps_demuxer_destroy(ps);
|
|
|
|
fclose(fp);
|
|
fclose(vfp);
|
|
fclose(afp);
|
|
}
|