/*
 * Copyright (C) 2020 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.
 */

#ifndef MEDIACTSNATIVE_NATIVE_CODEC_TEST_BASE_H
#define MEDIACTSNATIVE_NATIVE_CODEC_TEST_BASE_H

#include <media/NdkMediaCodec.h>
#include <zlib.h>

#include <cmath>
#include <cstdint>
#include <list>
#include <mutex>
#include <vector>

#include "NativeMediaCommon.h"

#define RETURN_IF_FAIL(status, str)                                          \
    {                                                                        \
        media_status_t val = (status);                                       \
        if (AMEDIA_OK != val) {                                              \
            std::string msg = StringFormat("%s with error %d \n", str, val); \
            ALOGE("%s", msg.c_str());                                        \
            mErrorLogs.append((msg));                                        \
            return false;                                                    \
        }                                                                    \
    }

#define RETURN_IF_TRUE(cond, msg)              \
    if ((cond)) {                              \
        ALOGE("%s", (msg).c_str());            \
        mErrorLogs.append((msg)).append("\n"); \
        return false;                          \
    }

#define RETURN_IF_FALSE(cond, msg)             \
    if (!(cond)) {                             \
        ALOGE("%s", (msg).c_str());            \
        mErrorLogs.append((msg)).append("\n"); \
        return false;                          \
    }

#define RETURN_IF_NULL(var, msg)               \
    if ((var) == nullptr) {                    \
        ALOGE("%s", (msg).c_str());            \
        mErrorLogs.append((msg)).append("\n"); \
        return false;                          \
    }

struct callbackObject {
    AMediaCodecBufferInfo bufferInfo;
    int32_t bufferIndex;
    bool isInput;

    callbackObject(int32_t index, AMediaCodecBufferInfo* info)
        : bufferInfo{*info}, bufferIndex{index}, isInput{false} {}

    callbackObject(int32_t index) : bufferIndex{index}, isInput{true} {}

    callbackObject() : bufferIndex{-1}, isInput{false} {}
};

class CodecAsyncHandler {
  private:
    std::mutex mMutex;
    std::condition_variable mCondition;
    std::list<callbackObject> mCbInputQueue;
    std::list<callbackObject> mCbOutputQueue;
    AMediaFormat* mOutFormat;
    volatile bool mSignalledOutFormatChanged;
    volatile bool mSignalledError;
    std::string mErrorMsg;

  public:
    CodecAsyncHandler();
    ~CodecAsyncHandler();
    void pushToInputList(callbackObject element);
    void pushToOutputList(callbackObject element);
    callbackObject getInput();
    callbackObject getOutput();
    callbackObject getWork();
    bool isInputQueueEmpty();
    void clearQueues();
    void setOutputFormat(AMediaFormat* format);
    AMediaFormat* getOutputFormat();
    bool hasOutputFormatChanged();
    bool waitOnFormatChange();
    void setError(bool status, std::string& msg);
    bool getError() const;
    void resetContext();
    std::string getErrorMsg();
    media_status_t setCallBack(AMediaCodec* codec, bool isCodecInAsyncMode);
};

class OutputManager {
  private:
    std::vector<int64_t> inpPtsArray;
    std::vector<int64_t> outPtsArray;
    std::vector<uint8_t> memory;
    uLong crc32value = 0U;
    std::string mErrorLogs;
    std::shared_ptr<std::string> mSharedErrorLogs;

  public:
    OutputManager(std::shared_ptr<std::string> log = std::make_shared<std::string>())
            : mSharedErrorLogs(log) {}

    void saveInPTS(int64_t pts) {
        // Add only Unique timeStamp, discarding any duplicate frame / non-display frame
        if (0 == std::count(inpPtsArray.begin(), inpPtsArray.end(), pts)) {
            inpPtsArray.push_back(pts);
        }
    }
    void saveOutPTS(int64_t pts) { outPtsArray.push_back(pts); }
    bool isPtsStrictlyIncreasing(int64_t lastPts);
    bool isOutPtsListIdenticalToInpPtsList(bool requireSorting);
    void saveToMemory(uint8_t* buf, AMediaCodecBufferInfo* info) {
        memory.insert(memory.end(), buf, buf + info->size);
    }
    void updateChecksum(uint8_t* buf, AMediaCodecBufferInfo* info) {
        updateChecksum(buf, info, 0, 0, 0, 0);
    }

    void updateChecksum(uint8_t* buf, AMediaCodecBufferInfo* info, int width, int height,
                        int stride, int bytesPerSample);
    uLong getChecksum() { return crc32value; }
    void reset() {
        inpPtsArray.clear();
        outPtsArray.clear();
        memory.clear();
        crc32value = 0U;
        mErrorLogs.clear();
        mSharedErrorLogs->clear();
    }
    bool equalsPtsList(OutputManager* that);
    bool equalsDequeuedOutput(OutputManager* that);
    bool equals(OutputManager* that);
    float getRmsError(uint8_t* refData, int length);
    std::string getErrorMsg() { return mErrorLogs + *mSharedErrorLogs; }
    int getOutStreamSize() { return memory.size(); }
    std::shared_ptr<std::string> getSharedErrorLogs() { return mSharedErrorLogs; }
};

class CodecTestBase {
  protected:
    const char* mMediaType;
    bool mIsAudio;
    bool mIsVideo;
    CodecAsyncHandler mAsyncHandle;
    bool mIsCodecInAsyncMode;
    bool mSawInputEOS;
    bool mSawOutputEOS;
    bool mSignalEOSWithLastFrame;
    int mInputCount;
    int mOutputCount;
    int64_t mPrevOutputPts;
    bool mSignalledOutFormatChanged;
    AMediaFormat* mOutFormat;
    int mBytesPerSample;

    bool mSaveToMem;
    OutputManager* mOutputBuff;
    OutputManager* mRefBuff;
    OutputManager* mTestBuff;
    OutputManager* mReconfBuff;

    AMediaCodec* mCodec;
    std::string mTestEnv;
    std::string mErrorLogs;

    CodecTestBase(const char* mediaType);
    ~CodecTestBase();
    virtual bool configureCodec(AMediaFormat* format, bool isAsync, bool signalEOSWithLastFrame,
                                bool isEncoder);
    virtual bool flushCodec();
    bool reConfigureCodec(AMediaFormat* format, bool isAsync, bool signalEOSWithLastFrame,
                          bool isEncoder);
    virtual void resetContext(bool isAsync, bool signalEOSWithLastFrame);
    virtual bool enqueueInput(size_t bufferIndex) = 0;
    virtual bool dequeueOutput(size_t bufferIndex, AMediaCodecBufferInfo* bufferInfo) = 0;
    virtual bool isTestStateValid();
    bool enqueueEOS(size_t bufferIndex);
    virtual bool doWork(int frameLimit);
    bool queueEOS();
    bool waitForAllOutputs();
    static int getWidth(AMediaFormat* format);
    static int getHeight(AMediaFormat* format);
    static bool isFormatSimilar(AMediaFormat* inpFormat, AMediaFormat* outFormat);
    bool hasSeenError() { return mAsyncHandle.getError(); }
  public:
    std::string getErrorMsg() {
        return mTestEnv +
                "###################       Error Details         #####################\n" +
                mErrorLogs;
    }
};

#endif // MEDIACTSNATIVE_NATIVE_CODEC_TEST_BASE_H