/*
 * Copyright (C) 2017 Amlogic Corporation.
 *
 * 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.
 */

#ifndef _DOLBY_MS12_H_
#define _DOLBY_MS12_H_

#include "DolbyMS12ConfigParams.h"

#ifdef __cplusplus

typedef enum  {
    MS12_CONFIG_MAIN_VOLUME,
    MS12_CONFIG_MAT_STREAM_PROFILE,
    MS12_CONFIG_GAME_MODE,
    MS12_CONFIG_ATMOS_DROP,    /*drop the first 2 frames for atmos case*/
    MS12_CONFIG_MIXER_MAX_SIZE_ENABLED, /*enable the mixer max size to 1536, this can save cpu bandwidth*/
    MS12_CONFIG_STEREO_OUT_LATENCY,
    MS12_CONFIG_MULTICHANNEL_OUT_LATENCY,
    MS12_CONFIG_DAP_SPEAKER_OUT_LATENCY,
    MS12_CONFIG_DAP_HEADPHONE_OUT_LATENCY,
    MS12_CONFIG_DDP_OUT_LATENCY,
    MS12_CONFIG_DD_OUT_LATENCY,
    MS12_CONFIG_MAT_OUT_LATENCY,
    MS12_CONFIG_COMPRESSION_FORMAT,
    MS12_CONFIG_SCHEDULER_STATE,
    MS12_CONFIG_AC4DEC_GET_ACTIVE_PRESENTATION,
    MS12_CONFIG_AC4DEC_CHECK_THE_PGI_IS_PRESENT,
    MS12_CONFIG_ALSA_DELAY_FRAME,
    MS12_CONFIG_ALSA_LIMIT_FRAME,
    MS12_CONFIG_SCHEDULER_SLEEP,
    MS12_CONFIG_FADEIN_MAX_DETECT_TIME_MS,
    MS12_CONFIG_SYSTEM_BUFFER_FRAME,
    MS12_CONFIG_FLUSH_SYSTEM_BUFFER,
    MS12_CONFIG_FLUSH_APP_BUFFER,
    MS12_CONFIG_APP_BUFFER_FRAME
}ms12_config_type_t;

typedef union ms12_config {
    float main_volume;
    int mat_stream_profile;
}ms12_config_t;

typedef enum  {
    MAT_ENC_CONFIG_MIXER_LEVEL, //runtime param
    MAT_ENC_CONFIG_OUT_BITDEPTH, //static param
    MAT_ENC_CONFIG_OUT_CH, //static param
} mat_enc_config_type_t;

typedef union mat_config {
    int mixer_level;
    int out_bitdepth;
    int out_ch;
} mat_enc_config_t;

struct aml_audio_info{
    int is_dolby_atmos;
    int sample_rate;
    int acmod;
    int b_lfe;
};

namespace android
{
typedef int (*output_callback)(void *buffer, void *priv, size_t size);
typedef int (*ms12sync_callback)(void *priv_data, unsigned long long , int, int);
typedef int (*scaletempo_callback)(void *priv, void *info);
typedef int (*clipmeta_callback)(void *priv, void *info, int offset);

class DolbyMS12
{

public:
    // static DolbyMS12* getInstance();

    DolbyMS12();
    virtual ~DolbyMS12();
    virtual int     GetLibHandle(char *dolby_ms12_path);
    virtual void    ReleaseLibHandle(void);
    virtual int     GetMS12OutputMaxSize(void);
    virtual void *  DolbyMS12Init(int configNum, char **configParams);
    virtual void    DolbyMS12Release(void *dolbyMS12_pointer);
    virtual int     DolbyMs12EncoderOpen(void *DolbyMS12Pointer, int configNum, char **configParams);
    virtual int     DolbyMs12EncoderClose(void *DolbyMS12Pointer);
    virtual int     DolbyMS12InputSystem(
        void *dolbyMS12_pointer
        , const void *audio_stream_out_buffer //ms12 input buffer
        , size_t audio_stream_out_buffer_size //ms12 input buffer size
        , int audio_stream_out_format
        , int audio_stream_out_channel_num
        , int audio_stream_out_sample_rate
    );
    virtual int     DolbyMS12InputDeepBuffer(
        void *dolbyMS12_pointer
        , const void *audio_stream_out_buffer //ms12 input buffer
        , size_t audio_stream_out_buffer_size //ms12 input buffer size
        , int audio_stream_out_format
        , int audio_stream_out_channel_num
        , int audio_stream_out_sample_rate
    );
    virtual int     DolbyMS12InputApp(
        void *dolbyMS12_pointer
        , const void *audio_stream_out_buffer //ms12 input buffer
        , size_t audio_stream_out_buffer_size //ms12 input buffer size
        , int audio_stream_out_format
        , int audio_stream_out_channel_num
        , int audio_stream_out_sample_rate
    );
    virtual int     DolbyMS12DapProcess(
        void *dolbyMS12_pointer
        , const void *audio_stream_out_buffer //ms12 input buffer
        , size_t audio_stream_out_buffer_size //ms12 input buffer size
        , int audio_stream_out_format
        , int audio_stream_out_channel_num
        , int audio_stream_out_sample_rate
    );

#ifdef REPLACE_OUTPUT_BUFFER_WITH_CALLBACK

    virtual int     DolbyMS12RegisterOutputCallback(output_callback callback, void *priv_data);

#else

    virtual int     DolbyMS12Output(
        void *dolbyMS12_pointer
        , const void *ms12_out_buffer //ms12 output buffer
        , size_t request_out_buffer_size //ms12 output buffer size
    );

#endif
    virtual int MS12ContinuousRegisterCallback(void *dolbyMS12_pointer, int callback_type, void *callback, void *priv_data);
    virtual int MS12ContinuousUnregisterCallback(void *dolbyMS12_pointer, int callback_type);

    virtual int     DolbyMS12UpdateRuntimeParams(
        void *DolbyMS12Pointer
        , int configNum
        , char **configParams);

    virtual int     DolbyMS12UpdateRuntimeParamsNoLock(
        void *DolbyMS12Pointer
        , int configNum
        , char **configParams);

    virtual int     DolbyMS12SchedulerRun(void *DolbyMS12Pointer);

    virtual void    DolbyMS12SetQuitFlag(int is_quit);

    virtual void    DolbyMS12FlushAppInputBuffer(void);

    virtual void    DolbyMS12GetBitstreamOutputSize(unsigned long long *all_output_size, unsigned long long *ms12_generate_zero_size);

    virtual int     DolbyMS12GetSystemBufferAvail(int * max_size);

    virtual int     DolbyMS12GetDeepBufferAvailFrames(int * max_size);

    virtual int     DolbyMS12GetGain(int);
    virtual int     DolbyMS12SetMATStreamProfile(int stream_profile);
    virtual int     DolbyMS12SetAtmosDrop(int atmos_drop);
    virtual int     DolbyMS12GetInputISDolbyAtmos();
    virtual int     DolbyMS12GetMATDecLatency();
    virtual int     DolbyMS12EnableMixerMaxSize(int enable);
    virtual int     DolbyMS12SetCompressionFormat(int compression_format);

    virtual unsigned long long DolbyMS12GetContinuousNFramesPcmOutput(void *ms12_pointer, int index);


    virtual void DolbyMS12SetDebugLevel(int);
    virtual void DumpDolbyMS12Info(int);

    virtual unsigned long long DolbyMS12GetNBytesConsumedSysSound(void);
    virtual unsigned long long DolbyMS12GetFramesConsumedDeepBufferAudio(void);

    virtual int DolbyMS12GetTotalNFramesDelay(void *);

    virtual int     DolbyMS12GetLatencyForStereoOut(int *latency);
    virtual int     DolbyMS12GetLatencyForMultiChannelOut(int *latency);
    virtual int     DolbyMS12GetLatencyForDAPSpeakerOut(int *latency);
    virtual int     DolbyMS12GetLatencyForDAPHeadphoneOut(int *latency);
    virtual int     DolbyMS12GetLatencyForDDPOut(int *latency);
    virtual int     DolbyMS12GetLatencyForDDOut(int *latency);
    virtual int     DolbyMS12GetLatencyForMATOut(int *latency);
    virtual int     DolbyMS12GetSystemBufferFrame(int *frame);
    virtual int     DolbyMS12GetAppBufferFrame(int *frame);
    virtual int     DolbyMS12MATEncoderInit
        (int b_lfract_precision
        , int b_chmod_locking
        , unsigned int *p_matenc_maxoutbufsize
        , int b_iec_header
        , int dbg_enable
        , void **mat_enc_handle
        );

    virtual void     DolbyMS12MatEncoderCleanup(void *mat_enc_handle);

    virtual int      DolbyMS12MATEncoderProcess
        (void *mat_enc_handle
        , const unsigned char *in_buf
        , int n_bytes_in_buf
        , const unsigned char *out_buf
        , int *n_bytes_out_buf
        , int out_buf_max_size
        , int *nbytes_consumed
        );


    virtual int      DolbyMS12MATEncoderConfig
        (void *mat_enc_handle
        , mat_enc_config_type_t config_type
        , mat_enc_config_t *config
        );

    virtual char *  DolbMS12GetVersion(void);

    virtual int DolbyMS12GetAC4ActivePresentation(int *presentation_group_index);

    virtual int DolbyMS12AC4DecCheckThePgiIsPresent(int presentation_group_index);

    virtual int DolbyMS12SetAlsaDelayFrame(int delay_frame);

    virtual int DolbyMS12SetAlsaLimitFrame(int limit_frame);

    virtual int DolbyMS12SetAppBufferFlush(int enable_flush);

    virtual int DolbyMS12SetSystemBufferFlush(int enable_flush);

    virtual int DolbyMS12SetSchedulerSleep(int enable_sleep);

    virtual int MS12DeocderOpen(void *dolbyMS12_pointer, int *ms12_decid, void *codec_info);
    virtual int MS12DeocderClose(void *dolbyMS12_pointer, int ms12_decid);
    virtual int MS12DecoderProcess(void *dolbyMS12_pointer, int ms12_decid);
    virtual int MS12DeocderPause(void *dolbyMS12_pointer, int ms12_decid);
    virtual int MS12DeocderResume(void *dolbyMS12_pointer, int  ms12_decid);
    virtual int MS12DeocderFlush(void *dolbyMS12_pointer, int ms12_decid);
    virtual int MS12DecoderMainWrite(void *dolbyMS12_pointer, int ms12_decid, void *buffer, int size, void *pcm_info);
    virtual int MS12DecoderAssociateWrite(void *dolbyMS12_pointer, int ms12_decid, void *buffer, int size, void *pcm_info);
    virtual int MS12DecoderSetparameter(void *dolbyMS12_pointer, int ms12_decid, int parameter_type, void *parameter, int size);
    virtual int MS12DecoderGetparameter(void *dolbyMS12_pointer, int ms12_decid, int parameter_type, void *parameter, int size);
    virtual int MS12DecoderRegisterCallback(void *dolbyMS12_pointer, int ms12_decid, int callback_type, void *callback, void *priv_data);
    virtual int MS12DecoderUnregisterCallback(void *dolbyMS12_pointer, int ms12_decid, int callback_type);

    // protected:


private:
    // DolbyMS12(const DolbyMS12&);
    // DolbyMS12& operator = (const DolbyMS12&);
    // static android::Mutex mLock;
    // static DolbyMS12 *gInstance;
    void *mDolbyMS12LibHandle;
};  // class DolbyMS12

}   // android
#endif


#endif  // _DOLBY_MS12_H_