#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "audio_encode.h"
#include "mono.h"
#include "Aencode.h"

long aenc_ReadSamp(void *data)
{
	FILE *f = (FILE *)data;
	int a,b;
	a = getc(f);
	if (a==EOF)
		return ReadSampleEof;
	
	b = getc(f);
	if (b==EOF)
		return ReadSampleEof;
	return (short)((b<<8) + a);
}

int aenc_Compress(char *input_filename,char *output_filename,int *input_levels,int *input_samples,int *input_rate,int *input_channels,float *input_factor,float *input_volscale)
{
    FILE *in, *out;
    long result;

    int levels, samples_per_subband;
    unsigned sample_rate, channels;
    float factor, volume_scale;
    int levels_set=0, samples_per_subband_set=0, sample_rate_set=0,
        channels_set=0, factor_set=0, volume_scale_set=0;


    in = fopen(input_filename, "rb");
    if (!in){
		mprintf((0,"AENC: Unable to open %s for input.\n", input_filename));
        return 0;
	}

	if(input_levels){
		levels = *input_levels;	//Levels (default 7 or for 2k total)
		levels_set = 1;
		if (levels < 0 || levels > 16){
			mprintf((0,"AENC: Warning: levels outside of the range 0 to 16\n"));
		}
	}

	if(input_samples){
		samples_per_subband = *input_samples;	//Samples per subband (default 16 or for 2k total)
		samples_per_subband_set = 1;
		if (samples_per_subband < 1 || samples_per_subband > 1024){
			mprintf((0,"AENC: Warning: samples per subband not in the range 1 to 1024\n"));
		}
	}

	if(input_rate){
		sample_rate = *input_rate;	//Sample rate (default 22K)
		sample_rate_set = 1;
		if (sample_rate != 11025 && sample_rate != 22050 && sample_rate != 44100){
			mprintf((0,"AENC: Warning: sample rate not 11025, 22050, or 44100\n"));
		}
	}

	if(input_channels){
		channels = *input_channels;
		channels_set = 1;
		if (channels!=1 && channels!=2){
			mprintf((0,"AENC: Warning: /C channels not 1 or 2\n"));
		}
	}

	if(input_factor){
		factor = *input_factor;// Factor of compression (default 4 for 22K, 8 for 44K)
		factor_set = 1;

	  if (factor < 1.0f && factor)
		  factor = 1.0f/factor;
	  
	  if (factor <= 0.0f){
		  mprintf((0,"AENC: Warning: compression factor <= 0.0\n"));
		  factor = 1.0f;
	  }
	}

	if(input_volscale){
		volume_scale = *input_volscale;// Volume scaling (slightly <= 1.0, default ,97)
		volume_scale_set = 1;
	}

    if (!levels_set && !samples_per_subband_set)
	{
		levels=7, samples_per_subband=16;
	}
	else if (!samples_per_subband_set)
	{
		samples_per_subband = 2048 / (1<<levels);

		if (samples_per_subband < 8)
			samples_per_subband = 8;
	}
	else if (!levels_set)
	{ 		
		unsigned subbands = (2048 / samples_per_subband)>>1;

		for (levels=0; subbands; subbands>>=1, ++levels);
	}

    if (!sample_rate_set) sample_rate = 22050;
    if (!channels_set) channels = 1;
    if (!factor_set) factor = sample_rate<=22050?4.0f:8.0f;
    if (!volume_scale_set) volume_scale = .97f;
    
    out = fopen(output_filename, "wb");
    if (!out){
		mprintf((0,"AENC: Unable to open %s for output.\n", output_filename));
		return 0;
	}

    result = AudioEncode( aenc_ReadSamp, in, channels, sample_rate, volume_scale, out, levels, samples_per_subband, 1.0f/factor );

    fclose(out);
    fclose(in);

	return 1;
}