mirror of
https://gitlab.com/mbunkus/mkvtoolnix.git
synced 2024-12-26 04:42:04 +00:00
145 lines
4.1 KiB
C++
145 lines
4.1 KiB
C++
/*
|
|
mkvmerge -- utility for splicing together matroska files
|
|
from component media subtypes
|
|
|
|
p_video.h
|
|
|
|
Written by Moritz Bunkus <moritz@bunkus.org>
|
|
|
|
Distributed under the GPL
|
|
see the file COPYING for details
|
|
or visit http://www.gnu.org/copyleft/gpl.html
|
|
*/
|
|
|
|
/*!
|
|
\file
|
|
\version \$Id: p_video.cpp,v 1.13 2003/03/05 13:51:20 mosu Exp $
|
|
\brief video output module
|
|
\author Moritz Bunkus <moritz @ bunkus.org>
|
|
*/
|
|
|
|
#include <ctype.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
|
|
#include "common.h"
|
|
#include "queue.h"
|
|
#include "p_video.h"
|
|
|
|
#include "KaxTracks.h"
|
|
#include "KaxTrackVideo.h"
|
|
|
|
#ifdef DMALLOC
|
|
#include <dmalloc.h>
|
|
#endif
|
|
|
|
video_packetizer_c::video_packetizer_c(char *ncodec, double nfps, int nwidth,
|
|
int nheight, int nbpp,
|
|
int nmax_frame_size,
|
|
int navi_compat_mode, track_info_t *nti)
|
|
throw (error_c) : q_c(nti) {
|
|
packetno = 0;
|
|
memcpy(codec, ncodec, 4);
|
|
codec[4] = 0;
|
|
fps = nfps;
|
|
width = nwidth;
|
|
height = nheight;
|
|
bpp = nbpp;
|
|
max_frame_size = nmax_frame_size;
|
|
tempbuf = (unsigned char *)malloc(max_frame_size + 1);
|
|
if (tempbuf == NULL)
|
|
die("malloc");
|
|
avi_compat_mode = navi_compat_mode;
|
|
frames_output = 0;
|
|
avi_compat_mode = 1;
|
|
last_id = 1;
|
|
set_header();
|
|
}
|
|
|
|
void video_packetizer_c::set_header() {
|
|
using namespace LIBMATROSKA_NAMESPACE;
|
|
int set_codec_private = 0;
|
|
char *codecstr;
|
|
|
|
if (kax_last_entry == NULL)
|
|
track_entry =
|
|
&GetChild<KaxTrackEntry>(static_cast<KaxTracks &>(*kax_tracks));
|
|
else
|
|
track_entry =
|
|
&GetNextChild<KaxTrackEntry>(static_cast<KaxTracks &>(*kax_tracks),
|
|
static_cast<KaxTrackEntry &>(*kax_last_entry));
|
|
kax_last_entry = track_entry;
|
|
|
|
if (serialno == -1)
|
|
serialno = track_number++;
|
|
KaxTrackNumber &tnumber =
|
|
GetChild<KaxTrackNumber>(static_cast<KaxTrackEntry &>(*track_entry));
|
|
*(static_cast<EbmlUInteger *>(&tnumber)) = serialno;
|
|
|
|
*(static_cast<EbmlUInteger *>
|
|
(&GetChild<KaxTrackType>(static_cast<KaxTrackEntry &>(*track_entry)))) =
|
|
track_video;
|
|
|
|
KaxCodecID &codec_id =
|
|
GetChild<KaxCodecID>(static_cast<KaxTrackEntry &>(*track_entry));
|
|
if (!avi_compat_mode) {
|
|
codecstr = map_video_codec_fourcc(codec, &set_codec_private);
|
|
codec_id.CopyBuffer((binary *)codecstr, countof(codecstr));
|
|
} else
|
|
codec_id.CopyBuffer((binary *)"V_MS/VFW/FOURCC",
|
|
countof("V_MS/VFW/FOURCC"));
|
|
if (set_codec_private || avi_compat_mode) {
|
|
KaxCodecPrivate &codec_private =
|
|
GetChild<KaxCodecPrivate>(static_cast<KaxTrackEntry &>(*track_entry));
|
|
codec_private.CopyBuffer((binary *)ti->private_data, ti->private_size);
|
|
}
|
|
|
|
KaxTrackVideo &track_video =
|
|
GetChild<KaxTrackVideo>(static_cast<KaxTrackEntry &>(*track_entry));
|
|
|
|
KaxVideoPixelHeight &pheight = GetChild<KaxVideoPixelHeight>(track_video);
|
|
*(static_cast<EbmlUInteger *>(&pheight)) = height;
|
|
|
|
KaxVideoPixelWidth &pwidth = GetChild<KaxVideoPixelWidth>(track_video);
|
|
*(static_cast<EbmlUInteger *>(&pwidth)) = width;
|
|
|
|
KaxVideoFrameRate &frate = GetChild<KaxVideoFrameRate>(track_video);
|
|
*(static_cast<EbmlFloat *>(&frate)) = fps;
|
|
}
|
|
|
|
int video_packetizer_c::process(unsigned char *buf, int size, int num_frames,
|
|
int key, int last_frame,
|
|
int64_t old_timecode) {
|
|
u_int64_t timecode;
|
|
|
|
if (size > max_frame_size) {
|
|
fprintf(stderr, "FATAL: p_video: size (%d) > max_frame_size (%d)\n",
|
|
size, max_frame_size);
|
|
exit(1);
|
|
}
|
|
|
|
if (old_timecode == -1)
|
|
timecode = (u_int64_t)(1000.0 * frames_output / fps);
|
|
else
|
|
timecode = old_timecode;
|
|
|
|
if (key)
|
|
// Add a key frame and save its ID so that we can reference it later.
|
|
last_id = add_packet(buf, size, timecode);
|
|
else
|
|
// This is a P frame - let's reference the last frame.
|
|
last_id = add_packet(buf, size, timecode, last_id);
|
|
frames_output += num_frames;
|
|
|
|
packetno++;
|
|
|
|
return EMOREDATA;
|
|
}
|
|
|
|
video_packetizer_c::~video_packetizer_c() {
|
|
if (tempbuf != NULL)
|
|
free(tempbuf);
|
|
}
|