#include "TextureViewer.h"
#include "ui_TextureViewer.h"

#include "FlowLayout.h"

#include "Code/Core.h"

#if defined(__linux__)
#include <QX11Info>
#include <X11/Xlib.h>
#include <GL/glx.h>
#endif

TextureViewer::TextureViewer(Core *core, QWidget *parent) :
QFrame(parent),
ui(new Ui::TextureViewer),
m_Core(core)
{
	ui->setupUi(this);

	m_Core->AddLogViewer(this);

	ui->render->SetOutput(NULL);
	ui->pixelContext->SetOutput(NULL);
	m_Output = NULL;

	QWidget *renderContainer = ui->renderContainer;

	ui->dockarea->addToolWindow(ui->renderContainer, ToolWindowManager::EmptySpace);
	ui->dockarea->setToolWindowProperties(renderContainer, ToolWindowManager::DisallowUserDocking |
	                                                       ToolWindowManager::HideCloseButton |
	                                                       ToolWindowManager::DisableDraggableTab);

	ToolWindowManager::AreaReference ref(ToolWindowManager::AddTo, ui->dockarea->areaOf(renderContainer));

	QWidget *lockedTabTest = new QWidget(this);
	lockedTabTest->setWindowTitle(tr("Locked Tab #1"));

	ui->dockarea->addToolWindow(lockedTabTest, ref);
	ui->dockarea->setToolWindowProperties(lockedTabTest, ToolWindowManager::DisallowUserDocking | ToolWindowManager::HideCloseButton);

	lockedTabTest = new QWidget(this);
	lockedTabTest->setWindowTitle(tr("Locked Tab #2"));
	
	ui->dockarea->addToolWindow(lockedTabTest, ref);
	ui->dockarea->setToolWindowProperties(lockedTabTest, ToolWindowManager::DisallowUserDocking | ToolWindowManager::HideCloseButton);

	lockedTabTest = new QWidget(this);
	lockedTabTest->setWindowTitle(tr("Locked Tab #3"));
	
	ui->dockarea->addToolWindow(lockedTabTest, ref);
	ui->dockarea->setToolWindowProperties(lockedTabTest, ToolWindowManager::DisallowUserDocking | ToolWindowManager::HideCloseButton);

	lockedTabTest = new QWidget(this);
	lockedTabTest->setWindowTitle(tr("Locked Tab #4"));
	
	ui->dockarea->addToolWindow(lockedTabTest, ref);
	ui->dockarea->setToolWindowProperties(lockedTabTest, ToolWindowManager::DisallowUserDocking | ToolWindowManager::HideCloseButton);
	
	ui->dockarea->addToolWindow(ui->resourceThumbs, ToolWindowManager::AreaReference(ToolWindowManager::RightOf, ui->dockarea->areaOf(renderContainer)));
	ui->dockarea->setToolWindowProperties(ui->resourceThumbs, ToolWindowManager::HideCloseButton);
	
	ui->dockarea->addToolWindow(ui->targetThumbs, ToolWindowManager::AreaReference(ToolWindowManager::AddTo, ui->dockarea->areaOf(ui->resourceThumbs)));
	ui->dockarea->setToolWindowProperties(ui->targetThumbs, ToolWindowManager::HideCloseButton);
	
	// need to add a way to make this less than 50% programmatically
	ui->dockarea->addToolWindow(ui->pixelContextLayout, ToolWindowManager::AreaReference(ToolWindowManager::BottomOf, ui->dockarea->areaOf(ui->targetThumbs)));
	ui->dockarea->setToolWindowProperties(ui->pixelContextLayout, ToolWindowManager::HideCloseButton);
	
	ui->dockarea->setAllowFloatingWindow(false);
	ui->dockarea->setRubberBandLineWidth(50);

	renderContainer->setWindowTitle(tr("OM RenderTarget 0 - GBuffer Colour"));
	ui->pixelContextLayout->setWindowTitle(tr("Pixel Context"));
	ui->targetThumbs->setWindowTitle(tr("OM Targets"));
	ui->resourceThumbs->setWindowTitle(tr("PS Resources"));

	QVBoxLayout *vertical = new QVBoxLayout(this);

	vertical->setSpacing(3);
	vertical->setContentsMargins(0, 0, 0, 0);

	QWidget *flow1widget = new QWidget(this);
	QWidget *flow2widget = new QWidget(this);

	FlowLayout *flow1 = new FlowLayout(flow1widget, 0, 3, 3);
	FlowLayout *flow2 = new FlowLayout(flow2widget, 0, 3, 3);

	flow1widget->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Minimum);
	flow2widget->setSizePolicy(QSizePolicy::Preferred, QSizePolicy::Minimum);

	flow1->addWidget(ui->channelsToolbar);
	flow1->addWidget(ui->subresourceToolbar);
	flow1->addWidget(ui->actionToolbar);

	flow2->addWidget(ui->zoomToolbar);
	flow2->addWidget(ui->overlayToolbar);
	flow2->addWidget(ui->rangeToolbar);

	vertical->addWidget(flow1widget);
	vertical->addWidget(flow2widget);
	vertical->addWidget(ui->dockarea);

	Ui_TextureViewer *u = ui;
	u->pixelcontextgrid->setAlignment(u->pushButton, Qt::AlignCenter);
	u->pixelcontextgrid->setAlignment(u->pushButton_2, Qt::AlignCenter);
}

TextureViewer::~TextureViewer()
{
	m_Core->RemoveLogViewer(this);
	delete ui;
}

void TextureViewer::OnLogfileLoaded()
{
#if defined(WIN32)
	HWND wnd = (HWND)ui->render->winId();
#elif defined(__linux__)
	Display *display = QX11Info::display();
	GLXDrawable drawable = (GLXDrawable)ui->render->winId();

	void *displayAndDrawable[2] = { (void *)display, (void *)drawable };
	void *wnd = displayAndDrawable;
#else
#error "Unknown platform"
#endif

	m_Core->Renderer()->BlockInvoke([wnd, this](IReplayRenderer *r) {
		m_Output = r->CreateOutput(wnd);
		ui->render->SetOutput(m_Output);

		OutputConfig c = { eOutputType_TexDisplay };
		m_Output->SetOutputConfig(c);
	});
}

void TextureViewer::OnLogfileClosed()
{
	m_Output = NULL;
	ui->render->SetOutput(NULL);
}

void TextureViewer::OnEventSelected(uint32_t frameID, uint32_t eventID)
{
	m_Core->Renderer()->AsyncInvoke([this](IReplayRenderer *) {
		TextureDisplay d;
		if(m_Core->APIProps().pipelineType == ePipelineState_D3D11)
			d.texid = m_Core->CurD3D11PipelineState.m_OM.RenderTargets[0].Resource;
		else
			d.texid = m_Core->CurGLPipelineState.m_FB.m_DrawFBO.Color[0];
		d.mip = 0;
		d.sampleIdx = ~0U;
		d.overlay = eTexOverlay_None;
		d.CustomShader = ResourceId();
		d.HDRMul = -1.0f;
		d.linearDisplayAsGamma = true;
		d.FlipY = false;
		d.rangemin = 0.0f;
		d.rangemax = 1.0f;
		d.scale = -1.0f;
		d.offx = 0.0f;
		d.offy = 0.0f;
		d.sliceFace = 0;
		d.rawoutput = false;
		d.lightBackgroundColour = d.darkBackgroundColour =
			FloatVector(0.0f, 0.0f, 0.0f, 0.0f);
		d.Red = d.Green = d.Blue = true;
		d.Alpha = false;
		m_Output->SetTextureDisplay(d);

		GUIInvoke::call([this]() { ui->render->update(); });
	});
}