//
// Created by Grishka on 12.08.2018.
//

#include "VideoRendererAndroid.h"
#include "../../PrivateDefines.h"
#include "../../logging.h"
#include "JNIUtilities.h"

using namespace tgvoip;
using namespace tgvoip::video;

jmethodID VideoRendererAndroid::resetMethod = nullptr;
jmethodID VideoRendererAndroid::decodeAndDisplayMethod = nullptr;
jmethodID VideoRendererAndroid::setStreamEnabledMethod = nullptr;
jmethodID VideoRendererAndroid::setRotationMethod = nullptr;
std::vector<std::uint32_t> VideoRendererAndroid::availableDecoders;
int VideoRendererAndroid::maxResolution;

extern JavaVM* sharedJVM;

VideoRendererAndroid::VideoRendererAndroid(jobject jobj)
    : queue(50)
{
    this->jobj = jobj;
}

VideoRendererAndroid::~VideoRendererAndroid()
{
    running = false;
    Request req {
        Buffer(0),
        Request::Type::Shutdown};
    queue.Put(std::move(req));
    thread->Join();
    delete thread;
    /*decoderThread.Post([this]{
		decoderEnv->DeleteGlobalRef(jobj);
	});*/
}

void VideoRendererAndroid::Reset(std::uint32_t codec, unsigned int width, unsigned int height, std::vector<Buffer>& _csd)
{
    csd.clear();
    for (Buffer& b : _csd)
    {
        csd.push_back(Buffer::CopyOf(b));
    }
    this->codec = codec;
    this->width = width;
    this->height = height;
    Request req {
        Buffer(0),
        Request::Type::ResetDecoder};
    queue.Put(std::move(req));
    Request req2 {
        Buffer(0),
        Request::Type::UpdateStreamState};
    queue.Put(std::move(req2));

    if (!thread)
    {
        thread = new Thread(std::bind(&VideoRendererAndroid::RunThread, this));
        thread->Start();
    }
}

void VideoRendererAndroid::DecodeAndDisplay(Buffer frame, std::uint32_t pts)
{
    /*decoderThread.Post(std::bind([this](Buffer frame){
	}, std::move(frame)));*/
    //LOGV("2 before decode %u", (unsigned int)frame.Length());
    Request req {
        std::move(frame),
        Request::Type::DecodeFrame};
    queue.Put(std::move(req));
}

void VideoRendererAndroid::RunThread()
{
    JNIEnv* env;
    sharedJVM->AttachCurrentThread(&env, nullptr);

    constexpr std::size_t bufferSize = 200 * 1024;
    std::uint8_t* buf = reinterpret_cast<std::uint8_t*>(std::malloc(bufferSize));
    jobject jbuf = env->NewDirectByteBuffer(buf, bufferSize);
    std::uint16_t lastSetRotation = 0;

    while (running)
    {
        //LOGV("before get from queue");
        Request request = queue.GetBlocking();
        //LOGV("1 before decode %u", (unsigned int)request.Length());
        if (request.type == Request::Type::Shutdown)
        {
            LOGI("Shutting down video decoder thread");
            break;
        }
        else if (request.type == Request::Type::DecodeFrame)
        {
            if (request.buffer.Length() > bufferSize)
            {
                LOGE("Frame data is too long (%u, max %u)", static_cast<int>(request.buffer.Length()), static_cast<int>(bufferSize));
            }
            else
            {
                if (lastSetRotation != rotation)
                {
                    lastSetRotation = rotation;
                    env->CallVoidMethod(jobj, setRotationMethod, (jint)rotation);
                }
                std::memcpy(buf, *request.buffer, request.buffer.Length());
                env->CallVoidMethod(jobj, decodeAndDisplayMethod, jbuf, (jint)request.buffer.Length(), 0);
            }
        }
        else if (request.type == Request::Type::ResetDecoder)
        {
            jobjectArray jcsd = nullptr;
            if (!csd.empty())
            {
                jcsd = env->NewObjectArray((jsize)csd.size(), env->FindClass("[B"), nullptr);
                jsize i = 0;
                for (Buffer& b : csd)
                {
                    env->SetObjectArrayElement(jcsd, i, jni::BufferToByteArray(env, b));
                    i++;
                }
            }
            std::string codecStr = "";
            switch (codec)
            {
            case CODEC_AVC:
                codecStr = "video/avc";
                break;
            case CODEC_HEVC:
                codecStr = "video/hevc";
                break;
            case CODEC_VP8:
                codecStr = "video/x-vnd.on2.vp8";
                break;
            case CODEC_VP9:
                codecStr = "video/x-vnd.on2.vp9";
                break;
            }
            env->CallVoidMethod(jobj, resetMethod, env->NewStringUTF(codecStr.c_str()), (jint)width, (jint)height, jcsd);
        }
        else if (request.type == Request::Type::UpdateStreamState)
        {
            env->CallVoidMethod(jobj, setStreamEnabledMethod, streamEnabled, streamPaused);
        }
    }
    std::free(buf);
    sharedJVM->DetachCurrentThread();
    LOGI("==== decoder thread exiting ====");
}

void VideoRendererAndroid::SetStreamEnabled(bool enabled)
{
    LOGI("Video stream state: %d", enabled);
    streamEnabled = enabled;
    Request req {
        Buffer(0),
        Request::Type::UpdateStreamState};
    queue.Put(std::move(req));
}

void VideoRendererAndroid::SetRotation(std::uint16_t rotation)
{
    this->rotation = rotation;
}

void VideoRendererAndroid::SetStreamPaused(bool paused)
{
    streamPaused = paused;
    Request req {
        Buffer(0),
        Request::Type::UpdateStreamState};
    queue.Put(std::move(req));
}