408 lines
11 KiB
C++
408 lines
11 KiB
C++
/********************************************************************************
|
|
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
|
|
|
|
|
|
class D3D11VertexShader;
|
|
|
|
|
|
//=============================================================================
|
|
|
|
class D3D11VertexBuffer : public VertexBuffer
|
|
{
|
|
friend class D3D11System;
|
|
friend class OBS;
|
|
|
|
ID3D11Buffer *vertexBuffer;
|
|
ID3D11Buffer *normalBuffer;
|
|
ID3D11Buffer *colorBuffer;
|
|
ID3D11Buffer *tangentBuffer;
|
|
List<ID3D11Buffer*> UVBuffers;
|
|
|
|
UINT vertexSize;
|
|
UINT normalSize;
|
|
UINT colorSize;
|
|
UINT tangentSize;
|
|
List<UINT> UVSizes;
|
|
|
|
BOOL bDynamic;
|
|
UINT numVerts;
|
|
VBData *data;
|
|
|
|
static VertexBuffer* CreateVertexBuffer(VBData *vbData, BOOL bStatic);
|
|
|
|
void MakeBufferList(D3D11VertexShader *vShader, List<ID3D11Buffer*> &bufferList, List<UINT> &strides) const;
|
|
|
|
public:
|
|
~D3D11VertexBuffer();
|
|
|
|
virtual void FlushBuffers();
|
|
virtual VBData* GetData();
|
|
};
|
|
|
|
//=============================================================================
|
|
|
|
class D3D11SamplerState : public SamplerState
|
|
{
|
|
friend class D3D11System;
|
|
friend class OBS;
|
|
|
|
ID3D11SamplerState *state;
|
|
|
|
static SamplerState* CreateSamplerState(SamplerInfo &info);
|
|
|
|
public:
|
|
~D3D11SamplerState();
|
|
};
|
|
|
|
//--------------------------------------------------
|
|
|
|
class D3D11Texture : public Texture
|
|
{
|
|
friend class D3D11System;
|
|
friend class OBS;
|
|
|
|
ID3D11Resource *texture;
|
|
ID3D11ShaderResourceView *resource;
|
|
ID3D11RenderTargetView *renderTarget;
|
|
|
|
UINT width, height;
|
|
GSColorFormat format;
|
|
IDXGISurface1 *surface;
|
|
bool bGDICompatible;
|
|
bool bDynamic;
|
|
|
|
static Texture* CreateTexture(unsigned int width, unsigned int height, GSColorFormat colorFormat, void *lpData, BOOL bGenMipMaps, BOOL bStatic);
|
|
static Texture* CreateFromFile(CTSTR lpFile, BOOL bBuildMipMaps);
|
|
static Texture* CreateRenderTarget(unsigned int width, unsigned int height, GSColorFormat colorFormat, BOOL bGenMipMaps);
|
|
static Texture* CreateGDITexture(unsigned int width, unsigned int height);
|
|
|
|
public:
|
|
~D3D11Texture();
|
|
|
|
virtual DWORD Width() const;
|
|
virtual DWORD Height() const;
|
|
virtual BOOL HasAlpha() const;
|
|
virtual void SetImage(void *lpData, GSImageFormat imageFormat, UINT pitch);
|
|
virtual bool Map(BYTE *&lpData, UINT &pitch);
|
|
virtual void Unmap();
|
|
virtual GSColorFormat GetFormat() const;
|
|
|
|
virtual bool GetDC(HDC &hDC);
|
|
virtual void ReleaseDC();
|
|
};
|
|
|
|
//=============================================================================
|
|
|
|
struct ShaderParam
|
|
{
|
|
ShaderParameterType type;
|
|
String name;
|
|
|
|
UINT samplerID;
|
|
UINT textureID;
|
|
|
|
int arrayCount;
|
|
|
|
List<BYTE> curValue;
|
|
List<BYTE> defaultValue;
|
|
BOOL bChanged;
|
|
|
|
inline ~ShaderParam() {FreeData();}
|
|
|
|
inline void FreeData()
|
|
{
|
|
name.Clear();
|
|
curValue.Clear();
|
|
defaultValue.Clear();
|
|
}
|
|
};
|
|
|
|
struct ShaderSampler
|
|
{
|
|
String name;
|
|
SamplerState *sampler;
|
|
|
|
inline ~ShaderSampler() {FreeData();}
|
|
|
|
inline void FreeData()
|
|
{
|
|
name.Clear();
|
|
delete sampler;
|
|
}
|
|
};
|
|
|
|
//--------------------------------------------------
|
|
|
|
struct ShaderProcessor : CodeTokenizer
|
|
{
|
|
BOOL ProcessShader(CTSTR input);
|
|
BOOL AddState(SamplerInfo &info, String &stateName, String &stateVal);
|
|
|
|
UINT nTextures;
|
|
List<ShaderSampler> Samplers;
|
|
List<ShaderParam> Params;
|
|
|
|
List<D3D11_INPUT_ELEMENT_DESC> generatedLayout;
|
|
|
|
bool bHasNormals;
|
|
bool bHasColors;
|
|
bool bHasTangents;
|
|
UINT numTextureCoords;
|
|
|
|
inline ShaderProcessor() {zero(this, sizeof(ShaderProcessor));}
|
|
inline ~ShaderProcessor() {FreeData();}
|
|
|
|
inline void FreeData()
|
|
{
|
|
UINT i;
|
|
for(i=0; i<Samplers.Num(); i++)
|
|
Samplers[i].FreeData();
|
|
Samplers.Clear();
|
|
for(i=0; i<Params.Num(); i++)
|
|
Params[i].FreeData();
|
|
Params.Clear();
|
|
}
|
|
|
|
inline UINT GetSamplerID(CTSTR lpSampler)
|
|
{
|
|
for(UINT i=0; i<Samplers.Num(); i++)
|
|
{
|
|
if(Samplers[i].name.Compare(lpSampler))
|
|
return i;
|
|
}
|
|
|
|
return INVALID;
|
|
}
|
|
};
|
|
|
|
//--------------------------------------------------
|
|
|
|
class D3D11Shader : public Shader
|
|
{
|
|
friend class D3D11System;
|
|
friend class OBS;
|
|
|
|
List<ShaderParam> Params;
|
|
List<ShaderSampler> Samplers;
|
|
|
|
ID3D11Buffer *constantBuffer;
|
|
UINT constantSize;
|
|
|
|
protected:
|
|
bool ProcessData(ShaderProcessor &processor, CTSTR lpFileName);
|
|
|
|
void UpdateParams();
|
|
|
|
public:
|
|
~D3D11Shader();
|
|
|
|
virtual int NumParams() const;
|
|
virtual HANDLE GetParameter(UINT parameter) const;
|
|
virtual HANDLE GetParameterByName(CTSTR lpName) const;
|
|
virtual void GetParameterInfo(HANDLE hObject, ShaderParameterInfo ¶mInfo) const;
|
|
|
|
virtual void LoadDefaults();
|
|
|
|
virtual void SetBool(HANDLE hObject, BOOL bValue);
|
|
virtual void SetFloat(HANDLE hObject, float fValue);
|
|
virtual void SetInt(HANDLE hObject, int iValue);
|
|
virtual void SetMatrix(HANDLE hObject, float *matrix);
|
|
virtual void SetVector(HANDLE hObject, const Vect &value);
|
|
virtual void SetVector2(HANDLE hObject, const Vect2 &value);
|
|
virtual void SetVector4(HANDLE hObject, const Vect4 &value);
|
|
virtual void SetTexture(HANDLE hObject, BaseTexture *texture);
|
|
virtual void SetValue(HANDLE hObject, const void *val, DWORD dwSize);
|
|
};
|
|
|
|
//--------------------------------------------------
|
|
|
|
class D3D11VertexShader : public D3D11Shader
|
|
{
|
|
friend class D3D11System;
|
|
friend class D3D11VertexBuffer;
|
|
friend class OBS;
|
|
|
|
ID3D11VertexShader *vertexShader;
|
|
ID3D11InputLayout *inputLayout;
|
|
|
|
bool bHasNormals;
|
|
bool bHasColors;
|
|
bool bHasTangents;
|
|
UINT nTextureCoords;
|
|
|
|
inline UINT NumBuffersExpected() const
|
|
{
|
|
UINT count = 1;
|
|
if(bHasNormals) count++;
|
|
if(bHasColors) count++;
|
|
if(bHasTangents) count++;
|
|
count += nTextureCoords;
|
|
|
|
return count;
|
|
}
|
|
|
|
static Shader* CreateVertexShader(CTSTR lpShader, CTSTR lpFileName);
|
|
|
|
public:
|
|
~D3D11VertexShader();
|
|
|
|
virtual ShaderType GetType() const {return ShaderType_Vertex;}
|
|
};
|
|
|
|
//--------------------------------------------------
|
|
|
|
class D3D11PixelShader : public D3D11Shader
|
|
{
|
|
friend class D3D11System;
|
|
|
|
ID3D11PixelShader *pixelShader;
|
|
|
|
static Shader* CreatePixelShader(CTSTR lpShader, CTSTR lpFileName);
|
|
|
|
public:
|
|
~D3D11PixelShader();
|
|
|
|
virtual ShaderType GetType() const {return ShaderType_Pixel;}
|
|
};
|
|
|
|
//=============================================================================
|
|
|
|
struct SavedBlendState
|
|
{
|
|
GSBlendType srcFactor, destFactor;
|
|
ID3D11BlendState *blendState;
|
|
};
|
|
|
|
class D3D11System : public GraphicsSystem
|
|
{
|
|
friend class OBS;
|
|
|
|
ID3D11Device *d3d;
|
|
ID3D11DeviceContext *d3dContext;
|
|
IDXGISwapChain *swap;
|
|
ID3D11RenderTargetView *swapRenderView;
|
|
|
|
ID3D11DepthStencilState *depthState;
|
|
ID3D11RasterizerState *rasterizerState;
|
|
|
|
//---------------------------
|
|
|
|
D3D11Texture *curRenderTarget;
|
|
D3D11Texture *curTextures[8];
|
|
D3D11SamplerState *curSamplers[8];
|
|
D3D11VertexBuffer *curVertexBuffer;
|
|
D3D11VertexShader *curVertexShader;
|
|
D3D11PixelShader *curPixelShader;
|
|
|
|
D3D11_PRIMITIVE_TOPOLOGY curTopology;
|
|
|
|
List<SavedBlendState> blends;
|
|
ID3D11BlendState *curBlendState;
|
|
ID3D11BlendState *disabledBlend;
|
|
BOOL bBlendingEnabled;
|
|
|
|
//---------------------------
|
|
|
|
VertexBuffer *spriteVertexBuffer, *boxVertexBuffer;
|
|
|
|
//---------------------------
|
|
|
|
float curProjMatrix[16];
|
|
float curViewMatrix[16];
|
|
float curViewProjMatrix[16];
|
|
|
|
float curBlendFactor[4];
|
|
|
|
virtual void ResetViewMatrix();
|
|
|
|
virtual void ResizeView();
|
|
virtual void UnloadAllData();
|
|
|
|
public:
|
|
D3D11System();
|
|
~D3D11System();
|
|
|
|
virtual LPVOID GetDevice();
|
|
virtual LPVOID GetDeviceContext();
|
|
|
|
virtual void Init();
|
|
|
|
|
|
////////////////////////////
|
|
//Texture Functions
|
|
virtual Texture* CreateTexture(unsigned int width, unsigned int height, GSColorFormat colorFormat, void *lpData, BOOL bBuildMipMaps, BOOL bStatic);
|
|
virtual Texture* CreateTextureFromFile(CTSTR lpFile, BOOL bBuildMipMaps);
|
|
virtual Texture* CreateRenderTarget(unsigned int width, unsigned int height, GSColorFormat colorFormat, BOOL bGenMipMaps);
|
|
virtual Texture* CreateGDITexture(unsigned int width, unsigned int height);
|
|
|
|
virtual bool GetTextureFileInfo(CTSTR lpFile, TextureInfo &info);
|
|
|
|
virtual SamplerState* CreateSamplerState(SamplerInfo &info);
|
|
|
|
|
|
////////////////////////////
|
|
//Shader Functions
|
|
virtual Shader* CreateVertexShader(CTSTR lpShader, CTSTR lpFileName);
|
|
virtual Shader* CreatePixelShader(CTSTR lpShader, CTSTR lpFileName);
|
|
|
|
|
|
////////////////////////////
|
|
//Vertex Buffer Functions
|
|
virtual VertexBuffer* CreateVertexBuffer(VBData *vbData, BOOL bStatic=1);
|
|
|
|
|
|
////////////////////////////
|
|
//Main Rendering Functions
|
|
virtual void LoadVertexBuffer(VertexBuffer* vb);
|
|
virtual void LoadTexture(Texture *texture, UINT idTexture=0);
|
|
virtual void LoadSamplerState(SamplerState *sampler, UINT idSampler=0);
|
|
virtual void LoadVertexShader(Shader *vShader);
|
|
virtual void LoadPixelShader(Shader *pShader);
|
|
|
|
virtual Shader* GetCurrentPixelShader();
|
|
virtual Shader* GetCurrentVertexShader();
|
|
|
|
virtual void SetRenderTarget(Texture *texture);
|
|
virtual void Draw(GSDrawMode drawMode, DWORD startVert=0, DWORD nVerts=0);
|
|
|
|
|
|
////////////////////////////
|
|
//Drawing mode functions
|
|
virtual void EnableBlending(BOOL bEnable);
|
|
virtual void BlendFunction(GSBlendType srcFactor, GSBlendType destFactor, float fFactor);
|
|
|
|
virtual void ClearColorBuffer(DWORD color=0xFF000000);
|
|
|
|
|
|
////////////////////////////
|
|
//Other Functions
|
|
void Ortho(float left, float right, float top, float bottom, float znear, float zfar);
|
|
void Frustum(float left, float right, float top, float bottom, float znear, float zfar);
|
|
|
|
virtual void SetViewport(float x, float y, float width, float height);
|
|
|
|
virtual void DrawSpriteEx(Texture *texture, float x, float y, float x2 = -1.0f, float y2 = -1.0f, float u = -1.0f, float v = -1.0f, float u2 = -1.0f, float v2 = -1.0f);
|
|
virtual void DrawBox(const Vect2 &upperLeft, const Vect2 &size);
|
|
};
|
|
|
|
inline ID3D11Device* GetD3D() {return static_cast<ID3D11Device*>(GS->GetDevice());}
|
|
inline ID3D11DeviceContext* GetD3DContext() {return static_cast<ID3D11DeviceContext*>(GS->GetDeviceContext());}
|