int krad_container_read_packet (krad_container_t *container, uint32_t *track,
uint64_t *timecode, uint8_t *buffer) {
if (container->type == OGG) {
return krad_ogg_read_packet ( container->ogg, track, timecode, buffer );
} else {
return kr_mkv_read_packet ( container->mkv, track, timecode, NULL, buffer );
}
}
static void remux (kr_mkv_t *mkv, char *file) {
int bytes_read;
uint32_t track;
uint64_t timecode;
uint8_t *buffer;
uint32_t out_track;
kr_mkv_t *new_mkv;
int keyframe;
uint8_t flags;
int packets;
packets = 0;
buffer = malloc (10000000);
new_mkv = kr_mkv_create_file (file);
if (new_mkv == NULL) {
fprintf (stderr, "Could not open %s\n", file);
exit (1);
}
printf ("Created file: %s\n", file);
out_track = kr_mkv_add_video_track (new_mkv, VP8,
30,
1,
mkv->tracks[VIDEO_TRACK].width,
mkv->tracks[VIDEO_TRACK].height);
printf ("Added new track: %d\n", out_track);
printf ("\n");
while ((bytes_read = kr_mkv_read_packet (mkv, &track, &timecode, &flags, buffer)) > 0) {
printf ("\rRead packet %d track %d %d bytes\t\t", packets++, track, bytes_read);
fflush (stdout);
if (flags == 0x80) {
keyframe = 1;
} else {
keyframe = 0;
}
if (track == 1) {
kr_mkv_add_video (new_mkv, out_track, buffer, bytes_read, keyframe);
}
}
printf ("\nDone.\n");
kr_mkv_destroy (&new_mkv);
free (buffer);
}
void mkv_frames_print (kr_mkv_t *mkv) {
int ret;
uint32_t track;
uint64_t timecode;
uint8_t *buffer;
buffer = malloc (10000000);
do {
ret = kr_mkv_read_packet (mkv, &track, &timecode, NULL, buffer);
if (ret > 0) {
printf ("Got packet for track %d, size %d, timecode %"PRIu64"\n",
track, ret, timecode);
}
} while (ret > 0);
free (buffer);
}
void krad_transmitter_mkv_test2 (int port, char *filename1, char *filename2) {
krad_transmitter_t *transmitter;
kr_mkv_t *mkv_tx;
kr_mkv_t *in[2];
krad_ticker_t *ticker;
int i;
int ret;
uint32_t track;
int bytes_read;
uint32_t out_track;
uint64_t start_tc;
uint64_t timecode;
uint64_t last_tc;
uint64_t sleeptime;
uint8_t *buffer;
int keyframe;
uint8_t flags;
int packets;
start_tc = 0;
last_tc = 0;
packets = 0;
char *stream_name = "stream.webm";
char *content_type = "video/webm";
buffer = malloc (10000000);
for (i = 0; i < 2; i++) {
if (i == 0) {
in[i] = kr_mkv_open_file (filename1);
} else {
in[i] = kr_mkv_open_file (filename2);
}
printf ("File %d: FPS %d/%d RES %dx%d\n",
i,
in[i]->tracks[VIDEO_TRACK].fps_numerator,
in[i]->tracks[VIDEO_TRACK].fps_denominator,
in[i]->tracks[VIDEO_TRACK].width,
in[i]->tracks[VIDEO_TRACK].height);
if ((i == 0) && (in[i] == NULL)) {
fprintf (stderr, "Could not open input file1: %s\n", filename1);
exit (1);
}
if ((i == 1) && (in[i] == NULL)) {
fprintf (stderr, "Could not open input file2: %s\n", filename2);
exit (1);
}
}
if ((in[0]->tracks[VIDEO_TRACK].width != in[0]->tracks[VIDEO_TRACK].width) ||
(in[1]->tracks[VIDEO_TRACK].height != in[1]->tracks[VIDEO_TRACK].height)) {
fprintf (stderr, "Resolutions are not equal\n");
exit (1);
}
ticker = krad_ticker_throttle_create ();
transmitter = krad_transmitter_create ();
ret = krad_transmitter_listen_on (transmitter, port);
if (ret != 0) {
fprintf (stderr, "transmitter could not listen on port %d\n", port);
exit (1);
}
mkv_tx = kr_mkv_create_transmission (transmitter, stream_name, content_type);
if (mkv_tx == NULL) {
fprintf (stderr, "Could not create mkv transmission\n");
exit (1);
}
out_track = kr_mkv_add_video_track (mkv_tx, VP8,
30,
1,
in[0]->tracks[VIDEO_TRACK].width,
in[0]->tracks[VIDEO_TRACK].height);
printf ("Added new track: %d\n", out_track);
for (i = 0; i < 2; i++) {
while ((bytes_read = kr_mkv_read_packet (in[i], &track, &timecode, &flags, buffer)) > 0) {
if (start_tc == 0) {
start_tc = timecode;
krad_ticker_start (ticker);
}
if (flags == 0x80) {
keyframe = 1;
} else {
keyframe = 0;
}
printf ("\rFile %d Packet %10d track %d sync %d timecode %14"PRIu64"s",
i, packets++, track, keyframe, timecode / 1000);
fflush (stdout);
if (track == 1) {
kr_mkv_add_video (mkv_tx, out_track, buffer, bytes_read, keyframe);
}
if ((last_tc) && (last_tc < timecode)) {
sleeptime = (timecode - start_tc) - (last_tc - start_tc);
krad_ticker_throttle (ticker, sleeptime);
}
if (timecode > last_tc) {
last_tc = timecode;
}
}
}
krad_ticker_destroy (ticker);
kr_mkv_destroy (&in[0]);
kr_mkv_destroy (&in[1]);
kr_mkv_destroy (&mkv_tx);
krad_transmitter_destroy (transmitter);
free (buffer);
}
static void kraise_splice (char *file1, char *file2, char *fileout) {
kr_mkv_t *in[2];
kr_mkv_t *out;
int i;
int bytes_read;
uint32_t track;
uint64_t timecode;
uint8_t *buffer[2];
uint32_t out_track;
int keyframe;
uint8_t flags;
int packets;
int have_buffer[2];
int have_buffer_bytes[2];
int inframes;
have_buffer_bytes[0] = 0;
have_buffer_bytes[1] = 0;
have_buffer[1] = 0;
inframes = 0;
have_buffer[0] = 0;
have_buffer[1] = 0;
packets = 0;
buffer[0] = malloc (10000000);
buffer[1] = malloc (10000000);
for (i = 0; i < 2; i++) {
if (i == 0) {
in[i] = kr_mkv_open_file (file1);
} else {
in[i] = kr_mkv_open_file (file2);
}
printf ("File %d: FPS %d/%d RES %dx%d\n",
i,
in[i]->tracks[VIDEO_TRACK].fps_numerator,
in[i]->tracks[VIDEO_TRACK].fps_denominator,
in[i]->tracks[VIDEO_TRACK].width,
in[i]->tracks[VIDEO_TRACK].height);
if (in[0] == NULL) {
fprintf (stderr, "Could not open input file: %s\n", file1);
exit (1);
}
}
if ((in[0]->tracks[VIDEO_TRACK].width != in[0]->tracks[VIDEO_TRACK].width) ||
(in[1]->tracks[VIDEO_TRACK].height != in[1]->tracks[VIDEO_TRACK].height)) {
fprintf (stderr, "Resolutions are not equal\n");
exit (1);
}
out = kr_mkv_create_file (fileout);
if (out == NULL) {
fprintf (stderr, "Could not open output file: %s\n", fileout);
exit (1);
}
printf ("Created file: %s\n", fileout);
out_track = kr_mkv_add_video_track (out, VP8,
30,
1,
in[0]->tracks[VIDEO_TRACK].width,
in[0]->tracks[VIDEO_TRACK].height);
printf ("Added new track: %d\n", out_track);
while ((bytes_read = kr_mkv_read_packet (in[0], &track, &timecode, &flags, buffer[0])) > 0) {
have_buffer_bytes[0] = bytes_read;
printf ("Read file %d packet %d track %d %d bytes\n",
1, packets++, track, bytes_read);
fflush (stdout);
if (flags == 0x80) {
keyframe = 1;
} else {
keyframe = 0;
}
if (keyframe == 1) {
inframes = 0;
while ((have_buffer[1] == 1) ||
(((bytes_read = kr_mkv_read_packet (in[1], &track, &timecode, &flags, buffer[1])) > 0))) {
if (have_buffer[1] == 1) {
keyframe = 1;
have_buffer[1] = 0;
} else {
printf ("Read file %d packet %d track %d %d bytes\n",
2, packets++, track, bytes_read);
fflush (stdout);
if (flags == 0x80) {
keyframe = 1;
} else {
keyframe = 0;
}
have_buffer_bytes[1] = bytes_read;
if ((keyframe == 1) && (inframes > 0)) {
have_buffer[1] = 1;
break;
}
}
if (track == 1) {
kr_mkv_add_video (out, out_track, buffer[1], have_buffer_bytes[1], keyframe);
}
inframes++;
}
}
if (track == 1) {
kr_mkv_add_video (out, out_track, buffer[0], have_buffer_bytes[0], keyframe);
}
}
printf ("\nDone.\n");
free (buffer[0]);
free (buffer[1]);
kr_mkv_destroy (&in[0]);
kr_mkv_destroy (&in[1]);
kr_mkv_destroy (&out);
}
static void remuxcode (kr_mkv_t *mkv, char *file, char *file2) {
int32_t ret;
int32_t bytes_read;
uint32_t track;
uint64_t timecode;
uint8_t *buffer;
uint8_t *vbuffer;
uint8_t *vbuffer2;
uint32_t out_track;
kr_mkv_t *new_mkv;
int32_t keyframe;
uint8_t flags;
int32_t packets;
krad_vpx_decoder_t *vpx_dec;
krad_vpx_encoder_t *vpx_enc;
kr_codec_hdr_t header;
kr_medium_t *medium;
kr_codeme_t *codeme;
krad_vorbis_t *vorbis_dec;
krad_vorbis_t *vorbis_enc;
packets = 0;
//vpxtest ();
buffer = malloc (20000000);
vbuffer = malloc (20000000);
vbuffer2 = malloc (20000000);
new_mkv = kr_mkv_create_file (file);
if (new_mkv == NULL) {
fprintf (stderr, "Could not open %s\n", file);
exit (1);
}
printf ("Created file: %s\n", file);
vpx_dec = krad_vpx_decoder_create ();
vpx_enc = krad_vpx_encoder_create (1280, 720,
30000,
1000,
2200);
out_track = kr_mkv_add_video_track (new_mkv, VP8,
30000,
1000,
mkv->tracks[VIDEO_TRACK].width,
mkv->tracks[VIDEO_TRACK].height);
header.count = mkv->tracks[AUDIO_TRACK].headers;
header.data[0] = mkv->tracks[AUDIO_TRACK].header[0];
header.sz[0] = mkv->tracks[AUDIO_TRACK].header_len[0];
header.data[1] = mkv->tracks[AUDIO_TRACK].header[1];
header.sz[1] = mkv->tracks[AUDIO_TRACK].header_len[1];
header.data[2] = mkv->tracks[AUDIO_TRACK].header[2];
header.sz[2] = mkv->tracks[AUDIO_TRACK].header_len[2];
vorbis_dec = krad_vorbis_decoder_create (&header);
vorbis_enc = krad_vorbis_encoder_create (2, 48000, 0.4);
kr_mkv_add_audio_track (new_mkv, VORBIS, 48000, 2,
vorbis_enc->hdrdata,
3 +
vorbis_enc->header.sz[0] +
vorbis_enc->header.sz[1] +
vorbis_enc->header.sz[2]);
printf ("\n");
while ((bytes_read = kr_mkv_read_packet (mkv, &track, &timecode, &flags, buffer)) > 0) {
printk ("Read packet %d track %d %d bytes", packets++, track, bytes_read);
//fflush (stdout);
if (flags == 0x80) {
keyframe = 1;
} else {
keyframe = 0;
}
if (track == 1) {
/*
medium = kr_medium_kludge_create ();
codeme = kr_codeme_kludge_create ();
codeme->sz = bytes_read;
memcpy (codeme->data, buffer, codeme->sz);
ret = kr_vpx_decode (vpx_dec, medium, codeme);
//printf ("Vpx dec ret was %d", ret);
kr_codeme_kludge_destroy (&codeme);
codeme = kr_codeme_kludge_create ();
ret = kr_vpx_encode (vpx_enc, codeme, medium);
//printf ("Vpx enc ret was %d", ret);
kr_medium_kludge_destroy (&medium);
bytes_read = codeme->sz;
memcpy (buffer, codeme->data, codeme->sz);
kr_codeme_kludge_destroy (&codeme);
*/
kr_mkv_add_video (new_mkv, out_track, buffer, bytes_read, keyframe);
}
if (track == 2) {
medium = kr_medium_kludge_create ();
codeme = kr_codeme_kludge_create ();
codeme->sz = bytes_read;
memcpy (codeme->data, buffer, codeme->sz);
kr_vorbis_decode (vorbis_dec, medium, codeme);
kr_codeme_kludge_destroy (&codeme);
if (medium->a.count > 0) {
do {
codeme = kr_codeme_kludge_create ();
ret = kr_vorbis_encode (vorbis_enc, codeme, medium);
if (ret == 1) {
kr_mkv_add_audio (new_mkv, 2, codeme->data, codeme->sz, codeme->count);
}
kr_medium_kludge_destroy (&medium);
kr_codeme_kludge_destroy (&codeme);
} while (ret == 1);
}
//}
}
}
//-----------------------------
krad_vorbis_decoder_destroy (&vorbis_dec);
//new_mkv->tracks[VIDEO_TRACK].fps_numerator = 30000;
kr_mkv_destroy (&mkv);
mkv = kr_mkv_open_file (file2);
printf ("FPS %d/%d RES: %dx%d\n",
mkv->tracks[VIDEO_TRACK].fps_numerator,
mkv->tracks[VIDEO_TRACK].fps_denominator,
mkv->tracks[VIDEO_TRACK].width,
mkv->tracks[VIDEO_TRACK].height);
if (mkv == NULL) {
fprintf (stderr, "Could not open %s\n", file2);
exit (1);
}
header.count = mkv->tracks[AUDIO_TRACK].headers;
header.data[0] = mkv->tracks[AUDIO_TRACK].header[0];
header.sz[0] = mkv->tracks[AUDIO_TRACK].header_len[0];
header.data[1] = mkv->tracks[AUDIO_TRACK].header[1];
header.sz[1] = mkv->tracks[AUDIO_TRACK].header_len[1];
header.data[2] = mkv->tracks[AUDIO_TRACK].header[2];
header.sz[2] = mkv->tracks[AUDIO_TRACK].header_len[2];
vorbis_dec = krad_vorbis_decoder_create (&header);
while ((bytes_read = kr_mkv_read_packet (mkv, &track, &timecode, &flags, buffer)) > 0) {
printk ("Read packet %d track %d %d bytes", packets++, track, bytes_read);
//fflush (stdout);
if (flags == 0x80) {
keyframe = 1;
} else {
keyframe = 0;
}
if (track == 1) {
/*
medium = kr_medium_kludge_create ();
codeme = kr_codeme_kludge_create ();
codeme->sz = bytes_read;
memcpy (codeme->data, buffer, codeme->sz);
ret = kr_vpx_decode (vpx_dec, medium, codeme);
//printf ("Vpx dec ret was %d", ret);
kr_codeme_kludge_destroy (&codeme);
codeme = kr_codeme_kludge_create ();
ret = kr_vpx_encode (vpx_enc, codeme, medium);
//printf ("Vpx enc ret was %d", ret);
kr_medium_kludge_destroy (&medium);
bytes_read = codeme->sz;
memcpy (buffer, codeme->data, codeme->sz);
kr_codeme_kludge_destroy (&codeme);
*/
kr_mkv_add_video (new_mkv, out_track, buffer, bytes_read, keyframe);
}
if (track == 2) {
medium = kr_medium_kludge_create ();
codeme = kr_codeme_kludge_create ();
codeme->sz = bytes_read;
memcpy (codeme->data, buffer, codeme->sz);
kr_vorbis_decode (vorbis_dec, medium, codeme);
kr_codeme_kludge_destroy (&codeme);
if (medium->a.count > 0) {
do {
codeme = kr_codeme_kludge_create ();
ret = kr_vorbis_encode (vorbis_enc, codeme, medium);
if (ret == 1) {
kr_mkv_add_audio (new_mkv, 2, codeme->data, codeme->sz, codeme->count);
}
kr_medium_kludge_destroy (&medium);
kr_codeme_kludge_destroy (&codeme);
} while (ret == 1);
}
//}
}
}
medium = kr_medium_kludge_create ();
codeme = kr_codeme_kludge_create ();
ret = kr_vorbis_encode (vorbis_enc, codeme, medium);
if (codeme->sz > 0) {
printk ("Got 1 another sized %zu with %d samples",
codeme->sz, codeme->count);
kr_mkv_add_audio (new_mkv, 2, codeme->data, codeme->sz, codeme->count);
}
kr_codeme_kludge_destroy (&codeme);
kr_medium_kludge_destroy (&medium);
do {
codeme = kr_codeme_kludge_create ();
ret = kr_vorbis_encode (vorbis_enc, codeme, medium);
if (ret == 1) {
kr_mkv_add_audio (new_mkv, 2, codeme->data, codeme->sz, codeme->count);
}
kr_codeme_kludge_destroy (&codeme);
} while (ret == 1);
//----------------------------
printf ("\nDone.\n");
krad_vpx_encoder_destroy (&vpx_enc);
krad_vpx_decoder_destroy (&vpx_dec);
krad_vorbis_decoder_destroy (&vorbis_dec);
krad_vorbis_encoder_destroy (&vorbis_enc);
kr_mkv_destroy (&new_mkv);
kr_mkv_destroy (&mkv);
free (buffer);
free (vbuffer);
free (vbuffer2);
}
