mkvtoolnix/p_pcm.cpp

140 lines
3.5 KiB
C++
Raw Normal View History

2003-02-24 13:12:27 +00:00
/*
mkvmerge -- utility for splicing together matroska files
from component media subtypes
p_pcm.cpp
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_pcm.cpp,v 1.16 2003/04/29 16:23:12 mosu Exp $
2003-02-24 13:12:27 +00:00
\brief PCM output module
\author Moritz Bunkus <moritz @ bunkus.org>
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include "mkvmerge.h"
2003-02-24 13:12:27 +00:00
#include "common.h"
#include "pr_generic.h"
2003-02-24 13:12:27 +00:00
#include "p_pcm.h"
#include "matroska.h"
2003-02-24 13:12:27 +00:00
#ifdef DMALLOC
#include <dmalloc.h>
#endif
pcm_packetizer_c::pcm_packetizer_c(unsigned long nsamples_per_sec,
2003-02-24 13:12:27 +00:00
int nchannels, int nbits_per_sample,
track_info_t *nti) throw (error_c):
generic_packetizer_c(nti) {
2003-02-24 13:12:27 +00:00
packetno = 0;
bps = nchannels * nbits_per_sample * nsamples_per_sec / 8;
tempbuf = (unsigned char *)malloc(bps + 128);
2003-02-24 13:12:27 +00:00
if (tempbuf == NULL)
die("malloc");
tempbuf_size = bps;
2003-02-24 13:12:27 +00:00
samples_per_sec = nsamples_per_sec;
channels = nchannels;
bits_per_sample = nbits_per_sample;
bytes_output = 0;
2003-02-28 13:00:26 +00:00
remaining_sync = 0;
2003-02-24 13:12:27 +00:00
set_header();
}
pcm_packetizer_c::~pcm_packetizer_c() {
if (tempbuf != NULL)
free(tempbuf);
}
void pcm_packetizer_c::set_header() {
using namespace LIBMATROSKA_NAMESPACE;
set_serial(-1);
set_track_type(track_audio);
set_codec_id(MKV_A_PCM);
set_audio_sampling_freq((float)samples_per_sec);
set_audio_channels(channels);
set_audio_bit_depth(bits_per_sample);
2003-02-24 13:12:27 +00:00
if (ti->default_track)
set_as_default_track('a');
generic_packetizer_c::set_header();
2003-02-24 13:12:27 +00:00
}
2003-04-11 11:23:54 +00:00
int pcm_packetizer_c::process(unsigned char *buf, int size,
int64_t, int64_t, int64_t, int64_t) {
2003-02-24 13:12:27 +00:00
int i, bytes_per_packet, remaining_bytes, complete_packets;
unsigned char *new_buf;
2003-02-24 13:12:27 +00:00
if (size > tempbuf_size) {
tempbuf = (unsigned char *)realloc(tempbuf, size + 128);
if (tempbuf == NULL)
die("realloc");
tempbuf_size = size;
2003-02-24 13:12:27 +00:00
}
2003-02-28 13:00:26 +00:00
new_buf = buf;
if (ti->async.displacement != 0) {
if (ti->async.displacement > 0) {
2003-02-28 13:00:26 +00:00
// Add silence.
int pad_size;
pad_size = bps * ti->async.displacement / 1000;
new_buf = (unsigned char *)malloc(size + pad_size);
2003-02-28 13:00:26 +00:00
if (new_buf == NULL)
die("malloc");
memset(new_buf, 0, pad_size);
memcpy(&new_buf[pad_size], buf, size);
size += pad_size;
} else
// Skip bytes.
remaining_sync = -1 * bps * ti->async.displacement / 1000;
ti->async.displacement = 0;
2003-02-28 13:00:26 +00:00
}
if (remaining_sync > 0) {
if (remaining_sync > size) {
remaining_sync -= size;
return EMOREDATA;
}
memmove(buf, &buf[remaining_sync], size - remaining_sync);
size -= remaining_sync;
remaining_sync = 0;
}
2003-02-24 13:12:27 +00:00
bytes_per_packet = bps / pcm_interleave;
complete_packets = size / bytes_per_packet;
remaining_bytes = size % bytes_per_packet;
for (i = 0; i < complete_packets; i++) {
2003-02-28 13:00:26 +00:00
add_packet(new_buf + i * bytes_per_packet, bytes_per_packet,
(int64_t)((bytes_output * 1000 / bps) * ti->async.linear));
2003-02-24 13:12:27 +00:00
bytes_output += bytes_per_packet;
2003-02-28 13:00:26 +00:00
packetno++;
2003-02-24 13:12:27 +00:00
}
if (remaining_bytes != 0) {
2003-02-28 13:00:26 +00:00
add_packet(new_buf + complete_packets * bytes_per_packet, remaining_bytes,
(int64_t)((bytes_output * 1000 / bps) * ti->async.linear));
2003-02-24 13:12:27 +00:00
bytes_output += remaining_bytes;
2003-02-28 13:00:26 +00:00
packetno++;
2003-02-24 13:12:27 +00:00
}
2003-02-28 13:00:26 +00:00
if (new_buf != buf)
free(new_buf);
2003-04-11 11:23:54 +00:00
return EMOREDATA;
2003-02-24 13:12:27 +00:00
}