mirror of https://github.com/nu774/fdkaac.git
122 lines
3.3 KiB
C
122 lines
3.3 KiB
C
/*
|
|
* Copyright (C) 2013 nu774
|
|
* For conditions of distribution and use, see copyright notice in COPYING
|
|
*/
|
|
#if HAVE_CONFIG_H
|
|
# include "config.h"
|
|
#endif
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#if HAVE_STDINT_H
|
|
# include <stdint.h>
|
|
#endif
|
|
#include <assert.h>
|
|
#include <fdk-aac/aacenc_lib.h>
|
|
#include "pcm_reader.h"
|
|
|
|
typedef struct pcm_sint16_converter_t {
|
|
pcm_reader_vtbl_t *vtbl;
|
|
pcm_reader_t *src;
|
|
pcm_sample_description_t format;
|
|
void *pivot;
|
|
unsigned capacity;
|
|
} pcm_sint16_converter_t;
|
|
|
|
static inline pcm_reader_t *get_source(pcm_reader_t *reader)
|
|
{
|
|
return ((pcm_sint16_converter_t *)reader)->src;
|
|
}
|
|
|
|
static const
|
|
pcm_sample_description_t *get_format(pcm_reader_t *reader)
|
|
{
|
|
return &((pcm_sint16_converter_t *)reader)->format;
|
|
}
|
|
|
|
static int64_t get_length(pcm_reader_t *reader)
|
|
{
|
|
return pcm_get_length(get_source(reader));
|
|
}
|
|
|
|
static int64_t get_position(pcm_reader_t *reader)
|
|
{
|
|
return pcm_get_position(get_source(reader));
|
|
}
|
|
|
|
static int read_frames(pcm_reader_t *reader, void *buffer, unsigned nframes)
|
|
{
|
|
unsigned i, count;
|
|
pcm_sint16_converter_t *self = (pcm_sint16_converter_t *)reader;
|
|
const pcm_sample_description_t *sfmt = pcm_get_format(self->src);
|
|
unsigned bytes = nframes * sfmt->bytes_per_frame;
|
|
if (self->capacity < bytes) {
|
|
void *p = realloc(self->pivot, bytes);
|
|
if (!p) return -1;
|
|
self->pivot = p;
|
|
self->capacity = bytes;
|
|
}
|
|
nframes = pcm_read_frames(self->src, self->pivot, nframes);
|
|
count = nframes * sfmt->channels_per_frame;
|
|
if (PCM_IS_FLOAT(sfmt)) {
|
|
float *ip = self->pivot;
|
|
INT_PCM *op = buffer;
|
|
#if SAMPLE_BITS == 16
|
|
for (i = 0; i < count; ++i)
|
|
op[i] = (int16_t)pcm_clip(ip[i] * 32768.0, -32768.0, 32767.0);
|
|
#else
|
|
for (i = 0; i < count; ++i)
|
|
op[i] = (int32_t)pcm_clip(ip[i] * 2147483648.0, -2147483648.0, 2147483647.0);
|
|
#endif
|
|
} else {
|
|
int32_t *ip = self->pivot;
|
|
INT_PCM *op = buffer;
|
|
#if SAMPLE_BITS == 16
|
|
if (sfmt->bits_per_channel <= 16) {
|
|
for (i = 0; i < count; ++i)
|
|
op[i] = ip[i] >> 16;
|
|
} else {
|
|
for (i = 0; i < count; ++i) {
|
|
int n = ((ip[i] >> 15) + 1) >> 1;
|
|
op[i] = (n == 0x8000) ? 0x7fff : n;
|
|
}
|
|
}
|
|
#else
|
|
for (i = 0; i < count; ++i)
|
|
op[i] = ip[i];
|
|
#endif
|
|
}
|
|
return nframes;
|
|
}
|
|
|
|
static void teardown(pcm_reader_t **reader)
|
|
{
|
|
pcm_sint16_converter_t *self = (pcm_sint16_converter_t *)*reader;
|
|
pcm_teardown(&self->src);
|
|
free(self->pivot);
|
|
free(self);
|
|
*reader = 0;
|
|
}
|
|
|
|
static pcm_reader_vtbl_t my_vtable = {
|
|
get_format, get_length, get_position, read_frames, teardown
|
|
};
|
|
|
|
pcm_reader_t *pcm_open_sint16_converter(pcm_reader_t *reader)
|
|
{
|
|
pcm_sint16_converter_t *self = 0;
|
|
pcm_sample_description_t *fmt;
|
|
|
|
assert((SAMPLE_BITS>>3) == sizeof(INT_PCM));
|
|
|
|
if ((self = calloc(1, sizeof(pcm_sint16_converter_t))) == 0)
|
|
return 0;
|
|
self->src = reader;
|
|
self->vtbl = &my_vtable;
|
|
memcpy(&self->format, pcm_get_format(reader), sizeof(self->format));
|
|
fmt = &self->format;
|
|
fmt->bits_per_channel = SAMPLE_BITS;
|
|
fmt->sample_type = PCM_TYPE_SINT;
|
|
fmt->bytes_per_frame = sizeof(INT_PCM) * fmt->channels_per_frame;
|
|
return (pcm_reader_t *)self;
|
|
}
|