/* 
 * 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;
}