/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

//package org.drrickorang.loopback;

package com.android.cts.verifier.audio;

import android.media.AudioFormat;
import android.media.AudioManager;
import android.media.AudioTrack;
//import android.media.MediaPlayer;
import android.media.AudioRecord;
import android.media.MediaRecorder;
import android.util.Log;

import android.os.Handler;
import  android.os.Message;

/**
 * A thread/audio track based audio synth.
 */
public class NativeAudioThread extends Thread {

    public boolean isRunning = false;
    double twoPi = 6.28318530718;

    public int mSessionId;

    public double[] mvSamples; //captured samples
    int mSamplesIndex;

    private final int mSecondsToRun = 2;
    public int mSamplingRate = 48000;
    private int mChannelConfigIn = AudioFormat.CHANNEL_IN_MONO;
    private int mAudioFormat = AudioFormat.ENCODING_PCM_16BIT;

    int mMinPlayBufferSizeInBytes = 0;
    int mMinRecordBuffSizeInBytes = 0;
    private int mChannelConfigOut = AudioFormat.CHANNEL_OUT_MONO;

    int mMicSource = 0;

    int mNumFramesToIgnore;

//    private double [] samples = new double[50000];

    boolean isPlaying = false;
    private Handler mMessageHandler;
    boolean isDestroying = false;
    boolean hasDestroyingErrors = false;

    static final int NATIVE_AUDIO_THREAD_MESSAGE_REC_STARTED = 892;
    static final int NATIVE_AUDIO_THREAD_MESSAGE_REC_ERROR = 893;
    static final int NATIVE_AUDIO_THREAD_MESSAGE_REC_COMPLETE = 894;
    static final int NATIVE_AUDIO_THREAD_MESSAGE_REC_COMPLETE_ERRORS = 895;

    public void setParams(int samplingRate, int playBufferInBytes, int recBufferInBytes,
                          int micSource, int numFramesToIgnore) {
        mSamplingRate = samplingRate;
        mMinPlayBufferSizeInBytes = playBufferInBytes;
        mMinRecordBuffSizeInBytes = recBufferInBytes;
        mMicSource = micSource;
        mNumFramesToIgnore = numFramesToIgnore;
    }

    //JNI load
    static {
        try {
            System.loadLibrary("audioloopback_jni");
        } catch (UnsatisfiedLinkError e) {
            log("Error loading loopback JNI library");
            e.printStackTrace();
        }

        /* TODO: gracefully fail/notify if the library can't be loaded */
    }

    //jni calls
    public native long slesInit(int samplingRate, int frameCount, int micSource,
                                int numFramesToIgnore);
    public native int slesProcessNext(long sles_data, double[] samples, long offset);
    public native int slesDestroy(long sles_data);

    public void run() {

        setPriority(Thread.MAX_PRIORITY);
        isRunning = true;

        //erase output buffer
        if (mvSamples != null)
            mvSamples = null;

        //resize
        int nNewSize = (int)(1.1* mSamplingRate * mSecondsToRun ); //10% more just in case
        mvSamples = new double[nNewSize];
        mSamplesIndex = 0; //reset index

        //clear samples
        for (int i=0; i<nNewSize; i++) {
            mvSamples[i] = 0;
        }

        //start playing
        isPlaying = true;


        log(" Started capture test");
        if (mMessageHandler != null) {
            Message msg = Message.obtain();
            msg.what = NATIVE_AUDIO_THREAD_MESSAGE_REC_STARTED;
            mMessageHandler.sendMessage(msg);
        }



        log(String.format("about to init, sampling rate: %d, buffer:%d", mSamplingRate,
                mMinPlayBufferSizeInBytes/2 ));
        long sles_data = slesInit(mSamplingRate, mMinPlayBufferSizeInBytes/2, mMicSource,
                                  mNumFramesToIgnore);
        log(String.format("sles_data = 0x%X",sles_data));

        if (sles_data == 0 ) {
            log(" ERROR at JNI initialization");
            if (mMessageHandler != null) {
                Message msg = Message.obtain();
                msg.what = NATIVE_AUDIO_THREAD_MESSAGE_REC_ERROR;
                mMessageHandler.sendMessage(msg);
            }
        }  else {

            //wait a little bit...
            try {
                sleep(10); //just to let it start properly?
            } catch (InterruptedException e) {
                e.printStackTrace();
            }



            mSamplesIndex = 0;
            int totalSamplesRead = 0;
            long offset = 0;
            for (int ii = 0; ii < mSecondsToRun; ii++) {
                log(String.format("block %d...", ii));
                int samplesRead = slesProcessNext(sles_data, mvSamples,offset);
                totalSamplesRead += samplesRead;

                offset += samplesRead;
                log(" [" + ii + "] jni samples read:" + samplesRead + "  currentOffset:" + offset);
            }

            log(String.format(" samplesRead: %d, sampleOffset:%d", totalSamplesRead, offset));
            log(String.format("about to destroy..."));

            runDestroy(sles_data);

            int maxTry = 20;
            int tryCount = 0;
            //isDestroying = true;
            while (isDestroying) {

                try {
                    sleep(40);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }

                tryCount++;

                log("destroy try: " + tryCount);

                if (tryCount >= maxTry) {
                    hasDestroyingErrors = true;
                    log("WARNING: waited for max time to properly destroy JNI.");
                    break;
                }
            }
            log(String.format("after destroying. TotalSamplesRead = %d", totalSamplesRead));

            if (totalSamplesRead==0)
            {
                hasDestroyingErrors = true;
            }

            endTest();
        }
    }

    public void setMessageHandler(Handler messageHandler) {
        mMessageHandler = messageHandler;
    }

    private void runDestroy(final long sles_data ) {
        isDestroying = true;

        //start thread

        final long local_sles_data = sles_data;
        ////
        Thread thread = new Thread(new Runnable() {
            public void run() {
                isDestroying = true;
                log("**Start runnable destroy");

                int status = slesDestroy(local_sles_data);
                log(String.format("**End runnable destroy sles delete status: %d", status));
                isDestroying = false;
            }
        });

        thread.start();



        log("end of runDestroy()");


    }

    public void togglePlay() {

    }

    public void runTest() {


    }

   public void endTest() {
       log("--Ending capture test--");
       isPlaying = false;


       if (mMessageHandler != null) {
           Message msg = Message.obtain();
           if (hasDestroyingErrors)
               msg.what = NATIVE_AUDIO_THREAD_MESSAGE_REC_COMPLETE_ERRORS;
           else
               msg.what = NATIVE_AUDIO_THREAD_MESSAGE_REC_COMPLETE;
           mMessageHandler.sendMessage(msg);
       }

   }

    public void finish() {

        if (isRunning) {
            isRunning = false;
            try {
                sleep(20);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    private static void log(String msg) {
        Log.v("Loopback", msg);
    }

    double [] getWaveData () {
        return mvSamples;
    }

}  //end thread.