#pragma once

#include <GL/glew.h>
#include <GLFW/glfw3.h>

#include "WallpaperEngine/Core/CWallpaper.h"
#include "WallpaperEngine/Core/CScene.h"
#include "WallpaperEngine/Core/CVideo.h"

#include "WallpaperEngine/Assets/CContainer.h"
#include "CFBO.h"
#include "CContext.h"

using namespace WallpaperEngine::Assets;

namespace WallpaperEngine::Render
{
    class CContext;

    class CWallpaper
    {
    public:
        template<class T> const T* as () const { assert (is<T> ()); return (const T*) this; }
        template<class T> T* as () { assert (is<T> ()); return (T*) this; }

        template<class T> bool is () { return this->m_type == T::Type; }

        CWallpaper (Core::CWallpaper* wallpaperData, std::string type, CContext* context);
        ~CWallpaper ();

        /**
         * Performs a render pass of the wallpaper
         */
        void render (glm::ivec4 viewport, bool renderFrame = true, bool newFrame = true);

        /**
         * @return The container to resolve files for this wallpaper
         */
        const CContainer* getContainer () const;

        /**
         * @return The current context rendering this wallpaper
         */
        CContext* getContext ();

        /**
         * @return The scene's framebuffer
         */
        GLuint getWallpaperFramebuffer () const;
        /**
         * @return The scene's texture
         */
        GLuint getWallpaperTexture () const;
        /**
         * Creates a new FBO for this wallpaper
         *
         * @param name The name of the FBO
         * @return
         */
        CFBO* createFBO (const std::string& name, ITexture::TextureFormat format, ITexture::TextureFlags flags, float scale, uint32_t realWidth, uint32_t realHeight, uint32_t textureWidth, uint32_t textureHeight);

        /**
         * @return The full FBO list to work with
         */
        const std::map<std::string, CFBO*>& getFBOs () const;
        /**
         * Searches the FBO list for the given FBO
         *
         * @param name
         * @return
         */
        CFBO* findFBO (const std::string& name) const;

        /**
         * @return The main FBO of this wallpaper
         */
        CFBO* getFBO () const;

        /**
         * Updates the texcoord used for drawing to the used framebuffer
         */
        void updateTexCoord (GLfloat* texCoords, GLsizeiptr size) const;

        /**
         * Updates the destination framebuffer for this wallpaper
         *
         * @param framebuffer
         */
        void setDestinationFramebuffer (GLuint framebuffer);

        /**
         * Creates a new instance of CWallpaper based on the information provided by the read backgrounds
         *
         * @param wallpaper
         *
         * @return
         */
        static CWallpaper* fromWallpaper (Core::CWallpaper* wallpaper, CContext* context);

    protected:
        /**
         * Renders a frame of the wallpaper
         */
        virtual void renderFrame (glm::ivec4 viewport) = 0;

        /**
         * Setups OpenGL's framebuffers for ping-pong and scene rendering
         */
        void setupFramebuffers ();

        Core::CWallpaper* m_wallpaperData;

        Core::CWallpaper* getWallpaperData ();

        /**
         * The FBO used for scene output
         */
        CFBO* m_sceneFBO;

    private:
        /**
         * The texture used for the scene output
         */
        GLuint m_texCoordBuffer;
        GLuint m_positionBuffer;
        GLuint m_shader;
        // shader variables
        GLint g_Texture0;
        GLint a_Position;
        GLint a_TexCoord;
        /**
         * The framebuffer to draw the background to
         */
        GLuint m_destFramebuffer;
        /**
         * Setups OpenGL's shaders for this wallpaper backbuffer
         */
        void setupShaders ();
        /**
         * The type of background this wallpaper is (used
         */
        std::string m_type;

        /**
         * List of FBOs registered for this wallpaper
         */
        std::map<std::string, CFBO*> m_fbos;
        /**
         * Context that is using this wallpaper
         */
        CContext* m_context;
    };
}