#include "FhGAACEncoder.h"
#include "mp4FastAAClib.h"
#include <malloc.h>
#include "config.h"
#include "../nsutil/pcm.h"

FhGAACEncoder *FhGAACEncoder::CreateDecoder(const AACConfiguration *cfg, int nch, int srate, int bps)
{
	MPEG4ENC_ERROR err;

	MPEG4ENC_SETUP setup;
	setup.aot =	AACConfig_GetAOT(cfg);
	setup.nBitRate = AACConfig_GetBitrate(cfg, nch); 
	setup.bitrateMode = AACConfig_GetBitrateMode(cfg);
	setup.quality = MP4_QUAL_HIGH;
	setup.chMode = AACConfig_GetChannelMode(cfg, nch);
	setup.sbrSignaling = MP4_SBRSIG_EXPL_BC;
	setup.nSampleRateIn = srate;
	setup.transportFormat = MP4_TT_RAW;
	setup.nGranuleLength = MP4_GRANULE_1024;
	setup.metadataMode = MP4_METADATA_NONE;

	HANDLE_MPEG4ENC_ENCODER encoder=0;
	err = MPEG4ENC_Configure(&encoder, &setup);
	if (err != MPEG4ENC_NO_ERROR)
	{
		return 0;
	}

	unsigned int first_samples;
	err = MPEG4ENC_Open(&encoder, &first_samples);
	if (err != MPEG4ENC_NO_ERROR)
	{
		
		MPEG4ENC_Close(&encoder);
		return 0;
	}
		float *sample_buffer = (float *)malloc(first_samples * sizeof(float));
	if (!sample_buffer)
	{
		MPEG4ENC_Close(&encoder);
		return 0;
	}

	FhGAACEncoder *fhg_enc = new FhGAACEncoder(encoder, &setup, nch, srate, bps, sample_buffer, first_samples);
	if (!fhg_enc)
	{
		free(sample_buffer);
		MPEG4ENC_Close(&encoder);
	}

	AACConfig_GetToolString(&setup, fhg_enc->tool, sizeof(fhg_enc->tool));
	return fhg_enc;
}

FhGAACEncoder::FhGAACEncoder(HANDLE_MPEG4ENC_ENCODER encoder, const MPEG4ENC_SETUP *setup, int nch, int srate, int bps, float *sample_buffer, unsigned int next_samples)
: encoder(encoder), channels(nch), sample_rate(srate), bits_per_sample(bps), sample_buffer(sample_buffer), next_samples(next_samples)
{
	MPEG4ENC_INFO info;
	MPEG4ENC_GetInfo(encoder, &info);
	samples_per_frame = info.nSamplesFrame[0];

	if (info.nSamplingRate[0] != srate)
		resampling = true;
	else
		resampling = false;
	finishing=false;
	total_samples=0;
	decodable_samples=0;
	// TODO: move this somewhere where we can error-check
	mp4_writer.AddAudioTrack(encoder, setup);
}

FhGAACEncoder::~FhGAACEncoder()
{
	free(sample_buffer);
	MPEG4ENC_Close(&encoder);
}

int FhGAACEncoder::Encode(int framepos, void *in, int in_avail, int *in_used, void *out, int out_avail)
{
	if (!in_avail && !finishing)
		return 0;

	size_t num_samples = in_avail / (bits_per_sample/8);
	num_samples = min(num_samples, next_samples);

	nsutil_pcm_IntToFloat_Interleaved(sample_buffer, in, bits_per_sample, num_samples);

	int samples_consumed=0;
	int out_used = 0;
	MPEG4ENC_AUINFO *info;
	MPEG4ENC_ERROR err = MPEG4ENC_Encode(encoder, sample_buffer, num_samples, &samples_consumed, &next_samples,	(unsigned char * const)out, out_avail, &out_used,	&info);
	if (err == MPEG4ENC_NO_ERROR && out_used)
	{
		decodable_samples += samples_per_frame;
		mp4_writer.Write(out, out_used, samples_per_frame);
	}
	else if (err != MPEG4ENC_NO_ERROR)
	{
		return 0;
	}
	
	if (!finishing)
	{
		*in_used = samples_consumed * (bits_per_sample/8);
		total_samples += samples_consumed / channels;
	}

	return out_used;
}

void FhGAACEncoder::PrepareToFinish()
{
	finishing=true;
}

void FhGAACEncoder::Finish(const wchar_t *filename)
{
	MPEG4ENC_INFO info;
	MPEG4ENC_GetInfo(encoder, &info);
	if (!resampling)
		mp4_writer.WriteGaps(info.nDelay, decodable_samples-total_samples-info.nDelay, total_samples);

	mp4_writer.WriteTool(tool);

	mp4_writer.CloseTo(filename);
}