#include "RtspNativeCodecJNI.h"
#include <PipeLine.h>
#include <PL_RTSPClient.h>
#include <PL_AndroidMediaCodecDecoder.h>
#include <PL_AndroidSurfaceViewSink.h>
#include <logger.h>
#include <Logger/src/logger.hpp>
#include <logger.h>

#include <jni.h>
#include "looper.h"
#include "media/NdkMediaCodec.h"
#include "media/NdkMediaExtractor.h"

std::stringstream logss;
Logger g_logger(logss);

class CameraWrapper
{
	PipeLine* pipeLine;
	
	PL_RTSPClient_Config rtspConfig;
	PL_AndroidMediaCodecDecoder_Config amcdConfig;
	PL_AndroidSurfaceViewSink asvsConfig;
	
	jmethodID faceCallback;
	
	CameraWrapper() : pipeLine(nullptr), rtspConfig(), amcdConfig(), asvsConfig(), faceCallback(0)
	{
	}
	
	~CameraWrapper()
	{
		delete pipeLine;
	}
	
	bool start()
	{
	}
	
	void stop()
	{
	}
};

CameraWrapper g_CameraWrappers[CAMERA_COUNT];

extern "C"
{

void Java_RtspNativeCodec_init(JNIEnv *env, jclass clazz)
{
	g_logger.set_level(VERBOSE);
	
	PipeLine::register_global_elem_creator("PL_RTSPClient", create_PL_RTSPClient);
	PipeLine::register_global_elem_creator("PL_AndroidMediaCodecDecoder", create_PL_AndroidMediaCodecDecoder);
	PipeLine::register_global_elem_creator("PL_AndroidSurfaceViewSink", create_PL_AndroidSurfaceViewSink);
	
	for (size_t i = 0; i < CAMERA_COUNT; i++)
	{
		g_CameraWrappers[i].pipeLine = new PipeLine;
		PipeLine& pipeLine(*(g_CameraWrappers[i].pipeLine));
	}
}

// set the surface
void Java_RtspNativeCodec_setSurface(JNIEnv *env, jclass clazz, jint cameraIdx, jobject surface)
{
    LOGV("@@@ Java_RtspNativeCodec_setSurface");
	assert(cameraIdx <= CAMERA_COUNT);

    // obtain a native window from a Java surface
    if (asvsConfig.window)
	{
        ANativeWindow_release((ANativeWindow)(asvsConfig.window));
        asvsConfig.window = NULL;
    }
    asvsConfig.window = ANativeWindow_fromSurface(env, surface);
    LOGV("@@@ setsurface %p", asvsConfig.window);
}

jboolean Java_RtspNativeCodec_createPlayer(JNIEnv* env, jclass clazz, jobject assetMgr, jint cameraIdx, jstring uri)
{
	LOGV("@@@ Java_RtspNativeCodec_createPlayer");
	assert(cameraIdx <= CAMERA_COUNT);
	
	CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]);
	
	{
		const char *utf8Uri = env->GetStringUTFChars(uri, NULL);
		cameraWrapper.rtspConfig = utf8Uri;
		env->ReleaseStringUTFChars(uri, utf8Uri);

		cameraWrapper.rtspConfig.progName = "RtspNativeCodec";
		cameraWrapper.rtspConfig.rtspURL = utf8;
		cameraWrapper.rtspConfig.aux = true; // ffmpeg need aux, but live555 not
		cameraWrapper.rtspConfig.verbosityLevel = 1;
		cameraWrapper.rtspConfig.tunnelOverHTTPPortNum = 0;
		cameraWrapper.rtspConfig.args = nullptr;
		
		PL_RTSPClient* rtspClient = (PL_RTSPClient*)cameraWrapper.pipeLine.push_elem("PL_RTSPClient");
		bool ret = rtspClient->init(&(cameraWrapper.rtspConfig));
		if (!ret)
		{
			LOG_ERROR << "rtspClient.init error" << std::endl;
			return JNI_FALSE;
		}
	}
	
	{
		PL_AndroidMediaCodecDecoder* amcDecoder = (PL_AndroidMediaCodecDecoder*)cameraWrapper.pipeLine.push_elem("PL_AndroidMediaCodecDecoder");
		cameraWrapper.amcdConfig.ak_mime = "";
		cameraWrapper.amcdConfig.ak_width = 1920;
		cameraWrapper.amcdConfig.ak_height = 1080;
		bool ret = amcDecoder->init(&(cameraWrapper.amcdConfig));
		if (!ret)
		{
			LOG_ERROR << "amcDecoder.init error" << std::endl;
			return JNI_FALSE;
		}
	}
	
	{
		PL_AndroidSurfaceViewSink* asvSink = (PL_AndroidSurfaceViewSink*)cameraWrapper.pipeLine.push_elem("PL_AndroidSurfaceViewSink");
		cameraWrapper.asvsConfig.
		bool ret = asvSink->init(&(cameraWrapper.asvsConfig));
		if (!ret)
		{
			LOG_ERROR << "asvSink.init error" << std::endl;
			return JNI_FALSE;
		}
	}

	return (cameraWrapper.start() ? JNI_TRUE : JNI_FALSE);
}

// shut down the native media system
void Java_RtspNativeCodec_shutdown(JNIEnv* env, jclass clazz, jint cameraIdx)
{
	LOGV("@@@ Java_RtspNativeCodec_shutdown");
	assert(cameraIdx <= CAMERA_COUNT);
	
	CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]);
	cameraWrapper.stop();
}

jboolean Java_RtspNativeCodec_setFaceCallback(JNIEnv* env, jclass clazz, jobject assetMgr, jint cameraIdx, jstring func)
{
	LOGV("@@@ Java_RtspNativeCodec_setFaceCallback");
	assert(cameraIdx <= CAMERA_COUNT);
	
	CameraWrapper& cameraWrapper(g_CameraWrappers[cameraIdx]);
	
	LOGE("jni callback (0)"); 
    jclass cls = env->GetObjectClass(clazz);  
    cameraWrapper.faceCallback = env->GetMethodID(cls, func, "(II)V"); // Java_FaceCallback_func
	
	// call:
    //env->CallVoidMethod(obj, callback, 5 , 10);
}

jboolean Java_RtspNativeCodec_getFaceList(JNIEnv* env, jclass clazz, jobject assetMgr, jint cameraIdx, jbyteArray faceListPb)
{
	assert(cameraIdx <= CAMERA_COUNT);
}

}