#include "common.h"
#include <iostream>
#include <unistd.h>

#include <X11/Xlib.h>
#include <X11/Xatom.h>
#include <X11/extensions/Xrandr.h>

#include <GL/glew.h>
#include <GL/glx.h>

#include "CRenderContext.h"
#include "CVideo.h"

using namespace WallpaperEngine::Render;

CRenderContext::CRenderContext (const COutput* output, CVideoDriver& driver, CInputContext& input, CWallpaperApplication& app) :
	m_defaultWallpaper (nullptr),
	m_output (output),
	m_driver (driver),
	m_app (app),
	m_input (input),
	m_textureCache (new CTextureCache (*this))
{
}

void CRenderContext::render ()
{
	bool firstFrame = true;
	bool renderFrame = true;

    for (const auto& cur : this->m_output->getViewports ())
    {
#if !NDEBUG
		std::string str = "Rendering to output " + cur.first;

		glPushDebugGroup (GL_DEBUG_SOURCE_APPLICATION, 0, -1, str.c_str ());
#endif /* DEBUG */

		// search the background in the viewport selection
		auto ref = this->m_wallpapers.find (cur.first);

		// render the background
		if (ref != this->m_wallpapers.end ())
			ref->second->render (cur.second.viewport, this->m_output->renderVFlip (), renderFrame, firstFrame);
		else
			this->m_defaultWallpaper->render (cur.second.viewport, this->m_output->renderVFlip (), renderFrame, firstFrame);
        // scenes need to render a new frame for each viewport as they produce different results
        // but videos should only be rendered once per group of viewports
        firstFrame = false;
#if !NDEBUG
		glPopDebugGroup ();
#endif /* DEBUG */
    }

    // read the full texture into the image
	if (this->m_output->haveImageBuffer ())
		glReadPixels (
			0, 0, this->m_output->getFullWidth (), this->m_output->getFullHeight (),
			GL_BGRA, GL_UNSIGNED_BYTE,
			this->m_output->getImageBuffer ()
		);

	// update the output with the given image
	this->m_output->updateRender ();
	// finally swap buffers
    this->m_driver.swapBuffers ();
}

void CRenderContext::setDefaultWallpaper (CWallpaper* wallpaper)
{
	this->m_defaultWallpaper = wallpaper;
}

void CRenderContext::setWallpaper (const std::string& display, CWallpaper* wallpaper)
{
	this->m_wallpapers.insert_or_assign (display, wallpaper);
}

CInputContext& CRenderContext::getInputContext () const
{
    return this->m_input;
}

const CWallpaperApplication& CRenderContext::getApp () const
{
    return this->m_app;
}

const COutput* CRenderContext::getOutput () const
{
	return this->m_output;
}

const ITexture* CRenderContext::resolveTexture (const std::string& name)
{
    return this->m_textureCache->resolve (name);
}