/********************************************************************************
 Copyright (C) 2012 Hugh Bailey <obs.jim@gmail.com>

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
********************************************************************************/


#pragma  once


/*=========================================================
    Enums
==========================================================*/

//---------------------------------
//drawing types
enum GSDrawMode {GS_POINTS, GS_LINES, GS_LINESTRIP, GS_TRIANGLES, GS_TRIANGLESTRIP};


//---------------------------------
//texture formats
enum GSColorFormat {GS_UNKNOWNFORMAT, GS_ALPHA, GS_GRAYSCALE, GS_RGB, GS_RGBA, GS_BGR, GS_BGRA, GS_RGBA16F, GS_RGBA32F, GS_B5G5R5A1, GS_B5G6R5, GS_R10G10B10A2, GS_DXT1, GS_DXT3, GS_DXT5};


//---------------------------------
//index size
enum GSIndexType {GS_UNSIGNED_SHORT, GS_UNSIGNED_LONG};


//---------------------------------
//blend functions
enum GSBlendType {GS_BLEND_ZERO, GS_BLEND_ONE, GS_BLEND_SRCCOLOR, GS_BLEND_INVSRCCOLOR, GS_BLEND_SRCALPHA, GS_BLEND_INVSRCALPHA, GS_BLEND_DSTCOLOR, GS_BLEND_INVDSTCOLOR, GS_BLEND_DSTALPHA, GS_BLEND_INVDSTALPHA, GS_BLEND_FACTOR, GS_BLEND_INVFACTOR};


//---------------------------------
//sampling filters
enum GSSampleFilter
{
    GS_FILTER_LINEAR,
    GS_FILTER_POINT,
    GS_FILTER_ANISOTROPIC,
    GS_FILTER_MIN_MAG_POINT_MIP_LINEAR,
    GS_FILTER_MIN_POINT_MAG_LINEAR_MIP_POINT,
    GS_FILTER_MIN_POINT_MAG_MIP_LINEAR,
    GS_FILTER_MIN_LINEAR_MAG_MIP_POINT,
    GS_FILTER_MIN_LINEAR_MAG_POINT_MIP_LINEAR,
    GS_FILTER_MIN_MAG_LINEAR_MIP_POINT,

    GS_FILTER_MIN_MAG_MIP_POINT=GS_FILTER_POINT,
    GS_FILTER_MIN_MAG_MIP_LINEAR=GS_FILTER_LINEAR
};

//---------------------------------
//sampling address mode
enum GSAddressMode
{
    GS_ADDRESS_CLAMP,
    GS_ADDRESS_WRAP,
    GS_ADDRESS_MIRROR,
    GS_ADDRESS_BORDER,
    GS_ADDRESS_MIRRORONCE,

    GS_ADDRESS_NONE=GS_ADDRESS_CLAMP,
    GS_ADDRESS_REPEAT=GS_ADDRESS_WRAP
};


/*=========================================================
    Vertex Buffer Data struct
==========================================================*/

class UVCoordList : public List<UVCoord> {};

struct VBData
{
    List<Vect>      VertList;
    List<Vect>      NormalList;
    List<DWORD>     ColorList;
    List<Vect>      TangentList;
    List<UVCoordList> UVList;

    inline VBData() {}
    inline ~VBData() {Clear();}

    inline void Clear()
    {
        VertList.Clear();
        NormalList.Clear();
        ColorList.Clear();
        TangentList.Clear();

        for(DWORD i=0;i<UVList.Num();i++)
            UVList[i].Clear();

        UVList.Clear();
    }

    inline void Serialize(Serializer &s)
    {
        Vect::SerializeList(s, VertList);
        Vect::SerializeList(s, NormalList);
        s << ColorList;
        Vect::SerializeList(s, TangentList);

        DWORD dwSize;

        if(s.IsLoading())
        {
            //FIXME: ???
            s << dwSize;
            UVList.SetSize(dwSize);
        }
        else
        {
            dwSize = UVList.Num();
            s << dwSize;
        }

        for(DWORD i=0; i<dwSize; i++)
            s << UVList[i];
    }
};


/*=========================================================
    SamplerState class
==========================================================*/

struct SamplerInfo
{
    inline SamplerInfo()
    {
        zero(this, sizeof(SamplerInfo));
        maxAnisotropy = 16;
    }

    GSSampleFilter filter;
    GSAddressMode addressU;
    GSAddressMode addressV;
    GSAddressMode addressW;
    UINT maxAnisotropy;
    Color4 borderColor;
};

class BASE_EXPORT SamplerState
{
protected:
    SamplerInfo info;

public:
    virtual ~SamplerState() {}

    const SamplerInfo& GetSamplerInfo() const {return info;}
};


/*=========================================================
    Texture class
==========================================================*/

class BASE_EXPORT BaseTexture
{
public:
    virtual ~BaseTexture() {}
};


/*=========================================================
    Texture class
==========================================================*/

enum GSImageFormat
{
    GS_IMAGEFORMAT_A8,
    GS_IMAGEFORMAT_L8,
    GS_IMAGEFORMAT_RGB,
    GS_IMAGEFORMAT_RGBX,
    GS_IMAGEFORMAT_RGBA,
    GS_IMAGEFORMAT_RGBA16F,
    GS_IMAGEFORMAT_RGBA32F,
    GS_IMAGEFORMAT_BGR,
    GS_IMAGEFORMAT_BGRX,
    GS_IMAGEFORMAT_BGRA,
};

class BASE_EXPORT Texture : public BaseTexture
{
public:
    virtual DWORD Width() const=0;
    virtual DWORD Height() const=0;
    virtual BOOL HasAlpha() const=0;
    virtual void SetImage(void *lpData, GSImageFormat imageFormat, UINT pitch)=0;
    virtual bool Map(BYTE *&lpData, UINT &pitch)=0;
    virtual void Unmap()=0;
    virtual GSColorFormat GetFormat() const=0;

    virtual bool GetDC(HDC &hDC)=0;
    virtual void ReleaseDC()=0;

    virtual LPVOID GetD3DTexture()=0;
};


/*=========================================================
    Shader class
==========================================================*/

enum ShaderParameterType
{
    Parameter_Unknown,
    Parameter_Bool,
    Parameter_Float,
    Parameter_Int,
    Parameter_String,
    Parameter_Vector2,
    Parameter_Vector3,
    Parameter_Vector=Parameter_Vector3,
    Parameter_Vector4,
    Parameter_Matrix3x3,
    Parameter_Matrix,
    Parameter_Texture
};

struct ShaderParameterInfo
{
    String name;
    ShaderParameterType type;
};

enum ShaderType
{
    ShaderType_Vertex,
    ShaderType_Pixel,
    ShaderType_Geometry
};

class BASE_EXPORT Shader
{
protected:
    HANDLE hViewProj;

    inline HANDLE  GetViewProj() const {return hViewProj;}

public:
    virtual ~Shader() {}

    virtual ShaderType GetType() const=0;

    virtual int    NumParams() const=0;
    virtual HANDLE GetParameter(UINT parameter) const=0;
    virtual HANDLE GetParameterByName(CTSTR lpName) const=0;
    virtual void   GetParameterInfo(HANDLE hObject, ShaderParameterInfo &paramInfo) const=0;

    virtual void   SetBool(HANDLE hObject, BOOL bValue)=0;
    virtual void   SetFloat(HANDLE hObject, float fValue)=0;
    virtual void   SetInt(HANDLE hObject, int iValue)=0;
    virtual void   SetMatrix(HANDLE hObject, float *matrix)=0;
    virtual void   SetVector(HANDLE hObject, const Vect &value)=0;
    virtual void   SetVector2(HANDLE hObject, const Vect2 &value)=0;
    virtual void   SetVector4(HANDLE hObject, const Vect4 &value)=0;
    virtual void   SetTexture(HANDLE hObject, BaseTexture *texture)=0;
    virtual void   SetValue(HANDLE hObject, const void *val, DWORD dwSize)=0;

    inline  void   SetColor(HANDLE hObject, const Color4 &value)
    {
        SetVector4(hObject, value);
    }
    inline  void   SetColor(HANDLE hObject, float fR, float fB, float fG, float fA=1.0f)
    {
        SetVector4(hObject, Color4(fR, fB, fG, fA));
    }
    inline  void   SetColor(HANDLE hObject, DWORD color)
    {
        SetVector4(hObject, RGBA_to_Vect4(color));
    }

    inline  void   SetColor3(HANDLE hObject, const Color3 &value)
    {
        SetVector(hObject, value);
    }
    inline  void   SetColor3(HANDLE hObject, float fR, float fB, float fG)
    {
        SetVector(hObject, Color3(fR, fB, fG));
    }
    inline  void   SetColor3(HANDLE hObject, DWORD color)
    {
        SetVector(hObject, RGB_to_Vect(color));
    }

    inline  void   SetVector4(HANDLE hObject, float fX, float fY, float fZ, float fW)
    {
        SetVector4(hObject, Vect4(fX, fY, fZ, fW));
    }

    inline  void   SetVector(HANDLE hObject, float fX, float fY, float fZ)
    {
        SetVector(hObject, Vect(fX, fY, fZ));
    }

    inline void SetMatrix(HANDLE hObject, const Matrix &mat)
    {
        float out[16];
        Matrix4x4Convert(out, mat);
        SetMatrix(hObject, out);
    }

    inline void SetMatrixIdentity(HANDLE hObject)
    {
        float out[16];
        Matrix4x4Identity(out);
        SetMatrix(hObject, out);
    }
};


/*=========================================================
    Output Duplication class
==========================================================*/

enum DuplicatorInfo
{
    DuplicatorInfo_Error,
    DuplicatorInfo_Timeout,
    DuplicatorInfo_Lost,
    DuplicatorInfo_Acquired
};

class BASE_EXPORT OutputDuplicator
{
public:
    virtual ~OutputDuplicator() {}

    virtual DuplicatorInfo AcquireNextFrame(UINT timeout)=0;
    virtual Texture* GetCopyTexture()=0;
    virtual Texture* GetCursorTex(POINT* pos)=0;
};

/*=========================================================
    Vertex Buffer class
==========================================================*/

class BASE_EXPORT VertexBuffer
{
public:
    virtual ~VertexBuffer() {}

    virtual void FlushBuffers()=0;
    virtual VBData* GetData()=0;
};


/*=========================================================
    Graphics System Class
==========================================================*/

struct TextureInfo
{
    GSColorFormat type;
    int width, height;
};

class BASE_EXPORT GraphicsSystem
{
    friend class OBS;

    virtual void ResizeView()=0;
    virtual void UnloadAllData()=0;

public:
    //----------------------------------------------------
    //Initialization/Destruction
    GraphicsSystem();
    virtual ~GraphicsSystem() {}

    virtual LPVOID GetDevice()=0;

    virtual void Init();


    //----------------------------------------------------
    //Matrix Functions
    void  MatrixPush();
    void  MatrixPop();
    void  MatrixSet(const Matrix &m);
    void  MatrixGet(Vect &v, Quat &q);
    void  MatrixGet(Matrix &m);

    void  MatrixIdentity();

    void  MatrixMultiply(const Matrix &m);
    void  MatrixRotate(float x, float y, float z, float a);
    void  MatrixRotate(const AxisAngle &aa);
    void  MatrixRotate(const Quat &q);

    void  MatrixTranslate(float x, float y);
    void  MatrixTranslate(const Vect2 &pos);
    void  MatrixScale(const Vect2 &scale);
    void  MatrixScale(float x, float y);
    void  MatrixTranspose();


    //----------------------------------------------------
    //Texture Functions
    virtual Texture*        CreateTextureFromSharedHandle(unsigned int width, unsigned int height, GSColorFormat colorFormat, HANDLE handle)=0;
    virtual Texture*        CreateTexture(unsigned int width, unsigned int height, GSColorFormat colorFormat, void *lpData, BOOL bBuildMipMaps, BOOL bStatic)=0;
    virtual Texture*        CreateTextureFromFile(CTSTR lpFile, BOOL bBuildMipMaps)=0;
    virtual Texture*        CreateRenderTarget(unsigned int width, unsigned int height, GSColorFormat colorFormat, BOOL bGenMipMaps)=0;
    virtual Texture*        CreateGDITexture(unsigned width, unsigned int height)=0;

    virtual bool            GetTextureFileInfo(CTSTR lpFile, TextureInfo &info)=0;

    virtual SamplerState*   CreateSamplerState(SamplerInfo &info)=0;

    virtual UINT            GetNumOutputs()=0;
    virtual OutputDuplicator *CreateOutputDuplicator(UINT outputID)=0;


    //----------------------------------------------------
    //Shader Functions
    virtual Shader*         CreateVertexShader(CTSTR lpShader, CTSTR lpFileName)=0;
    virtual Shader*         CreatePixelShader(CTSTR lpShader, CTSTR lpFileName)=0;
    Shader*                 CreateVertexShaderFromFile(CTSTR lpFileName);
    Shader*                 CreatePixelShaderFromFile(CTSTR lpFileName);


    //----------------------------------------------------
    //Vertex Buffer Functions
    virtual VertexBuffer* CreateVertexBuffer(VBData *vbData, BOOL bStatic=1)=0;


    //----------------------------------------------------
    //Main Rendering Functions
    virtual void  LoadVertexBuffer(VertexBuffer* vb)=0;
    virtual void  LoadTexture(Texture *texture, UINT idTexture=0)=0;
    virtual void  LoadSamplerState(SamplerState *sampler, UINT idSampler=0)=0;
    virtual void  LoadVertexShader(Shader *vShader)=0;
    virtual void  LoadPixelShader(Shader *pShader)=0;

    virtual Shader* GetCurrentPixelShader()=0;
    virtual Shader* GetCurrentVertexShader()=0;

    virtual void  SetRenderTarget(Texture *texture)=0;
    virtual void  Draw(GSDrawMode drawMode, DWORD startVert=0, DWORD nVerts=0)=0;


    //----------------------------------------------------
    // Drawing mode functions

    virtual void  EnableBlending(BOOL bEnable)=0;
    virtual void  BlendFunction(GSBlendType srcFactor, GSBlendType destFactor, float fFactor=1.0f)=0;

    virtual void  ClearColorBuffer(DWORD color=0xFF000000)=0;


    //----------------------------------------------------

    void  StartVertexBuffer();
    VertexBuffer* SaveVertexBuffer();
    void  Vertex(float x, float y, float z=0);
    void  Vertex(const Vect &v);
    void  Normal(float x, float y, float z);
    void  Normal(const Vect &v);
    void  Color(DWORD dwRGBA);
    void  Color(const Color4 &v);
    void  TexCoord(float u, float v, int idTexture=0);
    void  TexCoord(const UVCoord &uv, int idTexture=0);

    void  Vertex(const Vect2 &v2) {Vertex(Vect(v2));}

    void DrawSprite(Texture *texture, DWORD color, float x, float y, float x2 = -998.0f, float y2 = -998.0f);
    virtual void DrawSpriteEx(Texture *texture, DWORD color, float x, float y, float x2 = -998.0f, float y2 = -998.0f, float u = -998.0f, float v = -998.0f, float u2 = -998.0f, float v2 = -998.0f)=0;
    virtual void DrawBox(const Vect2 &upperLeft, const Vect2 &size)=0;


    //----------------------------------------------------
    //Other Functions
    virtual void  Ortho(float left, float right, float top, float bottom, float znear, float zfar)=0;
    virtual void  Frustum(float left, float right, float top, float bottom, float znear, float zfar)=0;

    virtual void  SetViewport(float x, float y, float width, float height)=0;

    virtual void  SetScissorRect(XRect *pRect=NULL)=0;

protected:
    //manual coordinate generation
    VBData *vbd;
    DWORD dwCurPointVert;
    DWORD dwCurTexVert;
    DWORD dwCurColorVert;
    DWORD dwCurNormVert;
    BitList TexCoordSetList;
    BOOL bNormalSet;
    BOOL bColorSet;

    //matrix stuff
    List<Matrix> MatrixStack;
    int curMatrix;

    virtual void ResetViewMatrix()=0;

public:
    virtual void CopyTexture(Texture *texDest, Texture *texSrc)=0;
};



//-----------------------------------------
//main extern
BASE_EXPORT extern GraphicsSystem *GS;



//-----------------------------------------
//C inline refs
inline void  MatrixPush()                                       {GS->MatrixPush();}
inline void  MatrixPop()                                        {GS->MatrixPop();}
inline void  MatrixSet(const Matrix &m)                         {GS->MatrixSet(m);}
inline void  MatrixGet(Matrix &m)                               {GS->MatrixGet(m);}
inline void  MatrixMultiply(const Matrix &m)                    {GS->MatrixMultiply(m);}
inline void  MatrixRotate(float x, float y, float z, float a)   {GS->MatrixRotate(x, y, z, a);}  //axis angle
inline void  MatrixRotate(const AxisAngle &aa)                  {GS->MatrixRotate(aa);}
inline void  MatrixRotate(const Quat &q)                        {GS->MatrixRotate(q);}
inline void  MatrixTranslate(float x, float y)                  {GS->MatrixTranslate(x, y);}
inline void  MatrixTranslate(const Vect2 &pos)                  {GS->MatrixTranslate(pos);}
inline void  MatrixScale(const Vect2 &scale)                    {GS->MatrixScale(scale);}
inline void  MatrixScale(float x, float y)                      {GS->MatrixScale(x, y);}
inline void  MatrixTranspose()                                  {GS->MatrixTranspose();}
inline void  MatrixIdentity()                                   {GS->MatrixIdentity();}


inline Texture* CreateTexture(unsigned int width, unsigned int height, GSColorFormat colorFormat, void *lpData=NULL, BOOL bGenMipMaps=1, BOOL bStatic=1)
    {return GS->CreateTexture(width, height, colorFormat, lpData, bGenMipMaps, bStatic);}

inline Texture* CreateTextureFromFile(CTSTR lpFile, BOOL bBuildMipMaps)
    {return GS->CreateTextureFromFile(lpFile, bBuildMipMaps);}

inline Texture* CreateRenderTarget(unsigned int width, unsigned int height, GSColorFormat colorFormat, BOOL bGenMipMaps)
    {return GS->CreateRenderTarget(width, height, colorFormat, bGenMipMaps);}

inline Texture* CreateGDITexture(unsigned int width, unsigned int height)
    {return GS->CreateGDITexture(width, height);}


inline SamplerState* CreateSamplerState(SamplerInfo &info)          {return GS->CreateSamplerState(info);}

inline Shader* CreateVertexShader(CTSTR lpShader, CTSTR lpFileName) {return GS->CreateVertexShader(lpShader, lpFileName);}
inline Shader* CreatePixelShader(CTSTR lpShader, CTSTR lpFileName)  {return GS->CreatePixelShader(lpShader, lpFileName);}
inline Shader* CreateVertexShaderFromFile(CTSTR lpFileName)         {return GS->CreateVertexShaderFromFile(lpFileName);}
inline Shader* CreatePixelShaderFromFile(CTSTR lpFileName)          {return GS->CreatePixelShaderFromFile(lpFileName);}


inline VertexBuffer* CreateVertexBuffer(VBData *vbData, BOOL bStatic=1)     {return GS->CreateVertexBuffer(vbData, bStatic);}

inline void  LoadVertexBuffer(VertexBuffer* vb)                 {GS->LoadVertexBuffer(vb);}
inline void  LoadTexture(Texture *texture, UINT idTexture=0)    {GS->LoadTexture(texture, idTexture);}
inline void  LoadSamplerState(SamplerState *sampler, UINT idSampler=0) {GS->LoadSamplerState(sampler, idSampler);}
inline void  LoadVertexShader(Shader *vShader)                  {GS->LoadVertexShader(vShader);}
inline void  LoadPixelShader(Shader *pShader)                   {GS->LoadPixelShader(pShader);}


inline Shader* GetCurrentPixelShader()                          {return GS->GetCurrentPixelShader();}
inline Shader* GetCurrentVertexShader()                         {return GS->GetCurrentVertexShader();}

inline void  SetRenderTarget(Texture *texture) {GS->SetRenderTarget(texture);}


inline void  Draw(GSDrawMode drawMode, DWORD StartVert=0, DWORD nVerts=0) {GS->Draw(drawMode, StartVert, nVerts);}

inline void  EnableBlending(BOOL bEnabled)                      {GS->EnableBlending(bEnabled);}
inline void  BlendFunction(GSBlendType srcFactor, GSBlendType destFactor, float fFactor=1.0f) {GS->BlendFunction(srcFactor, destFactor, fFactor);}

inline void  ClearColorBuffer(DWORD color=0xFF000000)
    {GS->ClearColorBuffer(color);}

inline void  StartVertexBuffer()                                {GS->StartVertexBuffer();}
inline VertexBuffer* SaveVertexBuffer()                         {return GS->SaveVertexBuffer();}
inline void  Vertex(float x, float y, float z=0)                {GS->Vertex(x, y, z);}
inline void  Vertex(const Vect &v)                              {GS->Vertex(v);}
inline void  Vertex(const Vect2 &v2)                            {GS->Vertex(v2);}
inline void  Normal(float x, float y, float z)                  {GS->Normal(x, y, z);}
inline void  Normal(const Vect &v)                              {GS->Normal(v);}
inline void  Color(DWORD dwRGBA)                                {GS->Color(dwRGBA);}
inline void  Color(const Color4 &v)                             {GS->Color(v);}
inline void  Color(float R, float G, float B, float A)          {Color4 rgba(R,G,B,A); GS->Color(rgba);}
inline void  TexCoord(float u, float v, int idTexture=0)        {GS->TexCoord(u, v, idTexture);}
inline void  TexCoord(const UVCoord &uv, int idTexture=0)       {GS->TexCoord(uv, idTexture);}

inline void  Ortho(float left, float right, float top, float bottom, float znear, float zfar)
    {GS->Ortho(left, right, top, bottom, znear, zfar);}
inline void  Frustum(float left, float right, float top, float bottom, float znear, float zfar)
    {GS->Frustum(left, right, top, bottom, znear, zfar);}

inline void  SetViewport(float x, float y, float width, float height) {GS->SetViewport(x, y, width, height);}
inline void  SetScissorRect(XRect *pRect=NULL)                        {GS->SetScissorRect(pRect);}

inline void DrawSprite(Texture *texture, DWORD color, float x, float y, float x2 = -998.0f, float y2 = -998.0f)
    {GS->DrawSprite(texture, color, x, y, x2, y2);}
inline void DrawSpriteEx(Texture *texture, DWORD color, float x, float y, float x2 = -998.0f, float y2 = -998.0f, float u = -998.0f, float v = -998.0f, float u2 = -998.0f, float v2 = -998.0f)
    {GS->DrawSpriteEx(texture, color, x, y, x2, y2, u, v, u2, v2);}
inline void DrawBox(const Vect2 &upperLeft, const Vect2 &size)
    {GS->DrawBox(upperLeft, size);}