57 lines
1.9 KiB
C++
57 lines
1.9 KiB
C++
|
#include "fmp4-writer.h"
|
||
|
|
#include "mov-format.h"
|
||
|
|
#include "mov-reader.h"
|
||
|
|
#include "mpeg4-aac.h"
|
||
|
|
#include "mov-file-buffer.h"
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <assert.h>
|
||
|
|
|
||
|
|
static uint8_t s_buffer[2 * 1024 * 1024];
|
||
|
|
static int s_audio_track = -1;
|
||
|
|
static int s_video_track = -1;
|
||
|
|
|
||
|
|
static void mov_onread(void* param, uint32_t track, const void* buffer, size_t bytes, int64_t pts, int64_t dts, int flags)
|
||
|
|
{
|
||
|
|
fmp4_writer_t* fmp4 = (fmp4_writer_t*)param;
|
||
|
|
int r = fmp4_writer_write(fmp4, track-1, buffer, bytes, pts, dts, flags);
|
||
|
|
assert(0 == r);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void mov_video_info(void* param, uint32_t track, uint8_t object, int width, int height, const void* extra, size_t bytes)
|
||
|
|
{
|
||
|
|
fmp4_writer_t* fmp4 = (fmp4_writer_t*)param;
|
||
|
|
s_video_track = fmp4_writer_add_video(fmp4, object, width, height, extra, bytes);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void mov_audio_info(void* param, uint32_t track, uint8_t object, int channel_count, int bit_per_sample, int sample_rate, const void* extra, size_t bytes)
|
||
|
|
{
|
||
|
|
fmp4_writer_t* fmp4 = (fmp4_writer_t*)param;
|
||
|
|
s_audio_track = fmp4_writer_add_audio(fmp4, object, channel_count, bit_per_sample, sample_rate, extra, bytes);
|
||
|
|
}
|
||
|
|
|
||
|
|
void fmp4_writer_test2(const char* mp4, const char* outmp4)
|
||
|
|
{
|
||
|
|
struct mov_file_cache_t file, wfile;
|
||
|
|
memset(&file, 0, sizeof(file));
|
||
|
|
memset(&wfile, 0, sizeof(wfile));
|
||
|
|
file.fp = fopen(mp4, "rb");
|
||
|
|
wfile.fp = fopen(outmp4, "wb");
|
||
|
|
mov_reader_t* mov = mov_reader_create(mov_file_cache_buffer(), &file);
|
||
|
|
fmp4_writer_t* fmp4 = fmp4_writer_create(mov_file_cache_buffer(), &wfile, MOV_FLAG_SEGMENT);
|
||
|
|
|
||
|
|
struct mov_reader_trackinfo_t info = { mov_video_info, mov_audio_info };
|
||
|
|
mov_reader_getinfo(mov, &info, fmp4);
|
||
|
|
fmp4_writer_init_segment(fmp4);
|
||
|
|
|
||
|
|
while (mov_reader_read(mov, s_buffer, sizeof(s_buffer), mov_onread, fmp4) > 0)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
fmp4_writer_destroy(fmp4);
|
||
|
|
mov_reader_destroy(mov);
|
||
|
|
fclose(wfile.fp);
|
||
|
|
fclose(file.fp);
|
||
|
|
}
|