#pragma once

#include "ofMain.h"
#include "utils/MeshGenerator.h"
#include "Bullet.h"
#include "vp/VP.h"
#include <onnxruntime_cxx_api.h>
#include "onnx/Onnx.h"
#include "onnx/Yolo.h"
#include "onnx/ModelThread.h"
#include "network/Request.h"
#include "network/Server.h"
#include "ofxTSNE.h"
#include "utils/KDTree.hpp"
#include "utils/QuadSource.h"
#include "ofxPiMapper.h"

class ofApp : public ofBaseApp{

	public:
		void setup();
		void update();
		void draw();
		void getNearestImages();
		void createNodes(std::string json_path);
		std::vector<std::vector<double>> createDoubleVectorFromNodes(const std::vector<Node>& nodes);
		void buildKDTree();
		void queryKD(glm::vec3& _position, int k);
		void buildMeshes();
		void drawPortrait();
		void drawPortraitZ();
		void keyPressed(int key);
		void keyReleased(int key);
		void mousePressed(int x, int y, int button);
		void mouseReleased(int x, int y, int button);
		void mouseDragged(int x, int y, int button);
		void exit();

		ofEasyCam portrait_camera;
		ofImage img;
		ofMesh mesh;
		MeshGenerator mesh_generator;
		Bullet bullet;
		vector<ofImage> images;
		
		ofFbo map_fbo;
		ofFbo map_fbo_alpha;
		ofFbo portrait_fbo;
		ofFbo portrait_pre_fbo;
		ofFbo portrait_pre_fbo_alpha;
		ofFbo comp_fbo;
		ofFbo model_outptut_fbo;
		ofFbo model_esp_out_fbo;
		ofFbo model_portrait_out_fbo;

		ofFbo esp_comp_fbo;
		ofFbo kd_out;
		ofImage model_image_esp;
		ofImage model_image_portrait;

		ofShader shaders;
		ofShader esp_shader;
		ofShader vert_snap;
		ofShader p_depth;

		ofPlanePrimitive plane;
		ofTexture bayer;

		VP vp_tree;

		/* onnx */
		Onnx depth_onnx;
		Onnx depth_onnx_esp;
		Onnx depth_onnx_portrait;
		ModelThread depth_thread;
		ModelThread depth_esp;
		ModelThread depth_portrait;
		ofImage model_image;
		ofPixels map_pixels;

		/* the current embedding */
		Embedding embed;

		/* server */
		std::unique_ptr<Server> server;

		/* esp_image fbos */
		vector<ofFbo> esp_images;
		ofPixels esp_comp_pixels;
		float last_updated_time;

		/* nodes */
		ofJson json;
		vector<Node> nodes;

		ofxTSNE tsne;
		int dims = 2;
		float perplexity = 20;
		float theta = 0.2;
		bool normalise = true;
		bool runManually = false;
		float tsne_scale = 1000;

		/* kd tree */
		std::unique_ptr<KDTree> kd_tree;
		std::vector<pointIndex> kd_result;

		float g_size = 10;
		float plane_size = 100;

		/* pi mapper */
		ofxPiMapper mapper;
		QuadSource projection_src;
		int map_h = 960;
		int map_w = 1920;
		
};