#include "panel.h"
#include "app.h"
#include "util.h"

#include <X11/Xutil.h>
#include <glm/common.hpp>
#include <glm/glm.hpp>

Panel::Panel(App *app, vr::HmdMatrix34_t start_pose, int index, int x, int y, int width, int height)
	: _app(app),
	  _index(index),
	  _x(x),
	  _y(y),
	  _width(width),
	  _height(height)
{
	_name = "screen_view_" + std::to_string(index);
	_alpha = 1.0f;
	_meters = 1.0f;
	_active_hand = -1;
	glGenTextures(1, &_gl_texture);
	glBindTexture(GL_TEXTURE_2D, _gl_texture);
	glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
	glTexImage2D(
		GL_TEXTURE_2D, 0, GL_RGB,
		_width, _height, 0,
		GL_BGRA, GL_UNSIGNED_BYTE, 0);

	_texture.eColorSpace = vr::ColorSpace_Auto;
	_texture.eType = vr::TextureType_OpenGL;
	_texture.handle = (void *)(uintptr_t)_gl_texture;

	// create overlay
	{
		auto overlay_create_err = _app->vr_overlay->CreateOverlay(_name.c_str(), _name.c_str(), &_id);
		assert(overlay_create_err == 0);
		_app->vr_overlay->ShowOverlay(_id);
		uint8_t col[4] = {20, 50, 50, 255};
		_app->vr_overlay->SetOverlayRaw(_id, &col, 1, 1, 4);
		printf("Created overlay instance %d\n", _index);

		// (flipping uv on y axis because opengl and xorg are opposite)
		vr::VRTextureBounds_t bounds{0, 1, 1, 0};
		_app->vr_overlay->SetOverlayTextureBounds(_id, &bounds);
		_app->vr_overlay->SetOverlayTransformAbsolute(_id, _app->_tracking_origin, &start_pose);
	}
}

void Panel::Update()
{
	Render();
	UpdateCursor();

	if (!_is_held)
	{
		for (auto controller : _app->GetControllers())
		{
			if (_app->IsInputJustPressed(controller, _app->_input_handles.grab))
			{
				vr::HmdMatrix34_t overlay_pose;
				vr::ETrackingUniverseOrigin tracking_universe;
				_app->vr_overlay->GetOverlayTransformAbsolute(_id, &tracking_universe, &overlay_pose);

				auto controller_pos = GetPos(_app->GetTrackerPose(controller));

				auto local_pos = glm::inverse(ConvertMat(overlay_pose)) * glm::vec4(controller_pos - GetPos(overlay_pose), 0);

				float grab_area_thickness = 0.3f;
				bool close_enough = glm::abs(local_pos.z) < grab_area_thickness;
				close_enough &= glm::abs(local_pos.x) < _meters / 2.0f;
				close_enough &= glm::abs(local_pos.y) < _meters / 2.0f;

				if (close_enough)
				{
					ControllerGrab(controller);
				}
			}
		}
	}
	else
	{
		if (!_app->GetControllerInputDigital(_active_hand, _app->_input_handles.grab).bState)
		{
			ControllerRelease();
		}
		// auto state = _app->GetControllerState(_active_hand);
		// auto touchpad = state.rAxis[0];
		// if (touchpad.x != 0.0f)
		// {
		// 	// TODO take into account the current framerate
		// 	_alpha += touchpad.x * 0.05;
		// 	_alpha = glm::clamp(_alpha, 0.1f, 1.0f);
		// 	_app->vr_overlay->SetOverlayAlpha(_id, _alpha);
		// }
	}
}

void Panel::Render()
{
	glCopyImageSubData(
		_app->_gl_frame, GL_TEXTURE_2D, 0,
		_x, _y, 0,
		_gl_texture, GL_TEXTURE_2D, 0,
		0, 0, 0,
		_width, _height, 1);

	auto set_texture_err = _app->vr_overlay->SetOverlayTexture(_id, &_texture);
	assert(set_texture_err == 0);
}

void Panel::SetHidden(bool state)
{
	_hidden = state;
	if (state)
		_app->vr_overlay->HideOverlay(_id);
	else
		_app->vr_overlay->ShowOverlay(_id);
}

void Panel::UpdateCursor()
{
	auto global_pos = _app->GetCursorPosition();
	if (global_pos.x < _x || global_pos.x >= _x + _width || global_pos.y < _y || global_pos.y >= _y + _height)
	{
		_app->vr_overlay->ClearOverlayCursorPositionOverride(_id);
		return;
	}
	int local_x = global_pos.x - _x;
	int local_y = global_pos.y - _y;

	// TODO: make this work when aspect ratio is >1 (root window is taller than it is wide)
	float ratio = (float)_height / (float)_width;
	float top_edge = 0.5f - ratio / 2.0f;
	float x = local_x / (float)_width;
	float y = 1.0f - (local_y / (float)_width + top_edge);
	auto pos = vr::HmdVector2_t{x, y};
	_app->vr_overlay->SetOverlayCursorPositionOverride(_id, &pos);
}

void Panel::ControllerGrab(TrackerID controller)
{
	// printf("Grabbed panel %d\n", _index);
	_is_held = true;
	_active_hand = controller;

	_app->vr_overlay->SetOverlayColor(_id, 0.6f, 1.0f, 1.0f);

	vr::HmdMatrix34_t abs_pose;
	vr::ETrackingUniverseOrigin tracking_universe;

	_app->vr_overlay->GetOverlayTransformAbsolute(_id, &tracking_universe, &abs_pose);
	auto abs_mat = ConvertMat(abs_pose);

	auto controller_mat = _app->GetTrackerPose(controller);

	vr::HmdMatrix34_t relative_pose = ConvertMat(glm::inverse(controller_mat) * (abs_mat));

	_app->vr_overlay->SetOverlayTransformTrackedDeviceRelative(_id, controller, &relative_pose);
}

void Panel::ControllerRelease()
{
	// printf("Released panel %d\n", _index);
	_is_held = false;

	_app->vr_overlay->SetOverlayColor(_id, 1.0f, 1.0f, 1.0f);

	vr::HmdMatrix34_t relative_pose;
	_app->vr_overlay->GetOverlayTransformTrackedDeviceRelative(_id, &_active_hand, &relative_pose);
	auto relative_mat = ConvertMat(relative_pose);

	auto controller_mat = _app->GetTrackerPose(_active_hand);

	vr::HmdMatrix34_t new_pose = ConvertMat(controller_mat * relative_mat);

	_app->vr_overlay->SetOverlayTransformAbsolute(_id, _app->_tracking_origin, &new_pose);
}