/* aac_encode.c */

#include <stdio.h>

#include <faac.h>

#include <stdint.h>

#include <stdlib.h>

#include <string.h>

int main (int argc, char **argv)

{

    unsigned long sampleRate = 8000;

    unsigned int numChannels = 1;

    unsigned long inputSample = 0;

    unsigned long maxOutputBytes = 0;

    faacEncHandle encoder;

    faacEncConfigurationPtr config;

    FILE *rfile = NULL;

    FILE *wfile = NULL;

    int16_t *pcm_input = NULL;

    uint8_t *aac_output = NULL;

    int readcount = 0;

    int writecount = 0;

    encoder = faacEncOpen(sampleRate, numChannels, &inputSample, &maxOutputBytes);

    config = faacEncGetCurrentConfiguration(encoder);

    config->aacObjectType = MAIN;

    config->mpegVersion = MPEG4;

    config->useLfe = 0;

    config->useTns = 1;

    config->allowMidside = 1;

    /*RAW_STREAM=0, ADTS_STREAM=1*/

    config->outputFormat = 1;

    config->bitRate = sampleRate;

    config->inputFormat = FAAC_INPUT_16BIT;

    faacEncSetConfiguration(encoder, config);

    printf("sampleRate:%ld, numChannels:%d, inputSample:%ld, maxOutputBytes:%ld\n",

            sampleRate, numChannels, inputSample, maxOutputBytes);

    if (argv[1]) {

        rfile = fopen(argv[1], "rb");

    } else {

        printf("try to open input.pcm\n");

        rfile = fopen("input.pcm", "rb");

    }

    if (!rfile) {

        printf("open error\n");

        goto end;

    }

    if (argv[2]) {

        wfile = fopen(argv[2], "wb");

    } else {

        printf("try to open output.aac\n");

        wfile = fopen("output.aac", "wb");

    }

    if (!wfile) {

        printf("open error\n");

        goto end;

    }

    pcm_input = (int16_t *)malloc(inputSample * sizeof(int16_t));

    aac_output = (uint8_t *)malloc(maxOutputBytes * sizeof(uint8_t));

    /* encode */

    while (1) {

        int readlen = 0;

        int ret = 0;

        readlen = fread(pcm_input, sizeof(int16_t), inputSample, rfile);

        ret = faacEncEncode(encoder, (int32_t *)pcm_input, readlen, aac_output, maxOutputBytes);

        if (ret > 0) {

            fwrite(aac_output, sizeof(uint8_t), ret, wfile);

        } else if (ret < 0) {

            printf("encode error\n");

            break;

        }

        readcount += readlen * 2;

        writecount += ret;

        if (!readlen && !ret) {

            printf("encode complete, from %d bytes to %d bytes\n", readcount, writecount);

            break;

        }

    }

    free(pcm_input);

    free(aac_output);

  end:

    if (wfile) fclose(wfile);

    if (rfile) fclose(rfile);

    faacEncClose(encoder);

    return 0;

}