package com.imptt.proptt.core;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.channels.FileChannel;

/**
 * Created by jeonhyunggue on 2016. 7. 25..
 */
public class WaveContainer {

    final String TAG = this.getClass().getSimpleName();

    private String path;
    private FileOutputStream fos = null;
    private int channelCount ,sampleRate, byteRate,bit_per_sample, totalAudioLen;


    public void open(String dirPath)
    {
        this.path = dirPath;
        if(fos == null){

            try {

                // please check directory is already exist.

                fos = new FileOutputStream(path);
                FileChannel channel = fos.getChannel();
                channel.position(44);       // wav header size is 44 byte
            } catch (FileNotFoundException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            } catch (IOException e2)
            {
                e2.printStackTrace();
            }

        }
    }

    public void close()
    {
        if(fos != null)
        {
            try {
                FileChannel channel = fos.getChannel();
                channel.position(0);
                writeHeader();
                fos.close();
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            fos = null;
            totalAudioLen = 0;
        }
    }
    public void initialize(int channelCount, int sampleRate,int bit_per_sample, int byteRate)
    {
        this.channelCount = channelCount;
        this.sampleRate = sampleRate;

        this.bit_per_sample = bit_per_sample;
        this.byteRate = byteRate;
    }

    public int writeFrame(byte[] buffer, int bufferLength)
    {
        try {
            if(fos!= null){
                fos.write(buffer);
            }
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        totalAudioLen += bufferLength;
        return bufferLength;
    }

    private int writeHeader()
    {
        byte[] header = null;

        header = new byte[44];

        header[0] = 'R';  // RIFF/WAVE header
        header[1] = 'I';
        header[2] = 'F';
        header[3] = 'F';
        header[4] = (byte) (totalAudioLen+36 & 0xff);
        header[5] = (byte) ((totalAudioLen+36 >> 8) & 0xff);
        header[6] = (byte) ((totalAudioLen+36 >> 16) & 0xff);
        header[7] = (byte) ((totalAudioLen+36 >> 24) & 0xff);
        header[8] = 'W';
        header[9] = 'A';
        header[10] = 'V';
        header[11] = 'E';
        header[12] = 'f';  // 'fmt ' chunk
        header[13] = 'm';
        header[14] = 't';
        header[15] = ' ';
        header[16] = 16;  // 4 bytes: size of 'fmt ' chunk
        header[17] = 0;
        header[18] = 0;
        header[19] = 0;
        header[20] = 1;  // format = 1
        header[21] = 0;
        header[22] = (byte) channelCount;
        header[23] = 0;
        header[24] = (byte) (sampleRate & 0xff);
        header[25] = (byte) ((sampleRate >> 8) & 0xff);
        header[26] = (byte) ((sampleRate >> 16) & 0xff);
        header[27] = (byte) ((sampleRate >> 24) & 0xff);
        header[28] = (byte) (byteRate & 0xff);
        header[29] = (byte) ((byteRate >> 8) & 0xff);
        header[30] = (byte) ((byteRate >> 16) & 0xff);
        header[31] = (byte) ((byteRate >> 24) & 0xff);
        header[32] = (byte) (channelCount * bit_per_sample / 8);  // block align
        header[33] = 0;
        header[34] = (byte) bit_per_sample;  // bits per sample
        header[35] = 0;
        header[36] = 'd';
        header[37] = 'a';
        header[38] = 't';
        header[39] = 'a';
        header[40] = (byte) (totalAudioLen & 0xff);
        header[41] = (byte) ((totalAudioLen >> 8) & 0xff);
        header[42] = (byte) ((totalAudioLen >> 16) & 0xff);
        header[43] = (byte) ((totalAudioLen >> 24) & 0xff);

        try {
            if(fos != null) fos.write(header);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }

        return header.length;
    }

}