Implement TSpriteShaderHAL::Invalidate()

Author: AdventureT

Toshi/Plugins/Source/PGUIRenderer/PGUITRDisplayContext.cpp

@@ -79,8 +79,8 @@ void PGUITRDisplayContext::PopAdditiveColour()
 // $PGUIRenderer: FUNCTION 100012e0
 void PGUITRDisplayContext::BeginScene()
 {
-	m_pShader->BeginMeshGeneration();
 	m_pShader->SetMaterial(TNULL);
+	m_pShader->BeginMeshGeneration();
 }
 
 // $PGUIRenderer: FUNCTION 10005a50

Toshi/Shaders/TSpriteShader/Include/D3D/TSpriteShaderD3D.h

@@ -76,6 +76,7 @@ class TSPRITESHADERD3D_EXPORTS TSpriteShaderHAL : public TSpriteShader
 	virtual void             Flush();
 	virtual void             Render(TRenderPacket *a_pRenderPacket);
 	virtual TBOOL            Validate();
+	virtual void             Invalidate();
 	virtual void             SetMaterial(TSpriteMaterial *a_pMaterial);
 	virtual void             BeginMeshGeneration();
 	virtual void             EndMeshGeneration();

Toshi/Shaders/TSpriteShader/Source/D3D/TSpriteShaderD3D.cpp

@@ -152,7 +152,7 @@ TBOOL TSpriteShaderHAL::OnResetDevice()
 	}
 	m_pVertexPool = static_cast<TVertexPoolResource *>(pVertexFactory->CreatePoolResource(m_usMaxStaticVertices, uiFlags));
 	// Same with IndexPool
-	TIMPLEMENT();
+	TIMPLEMENT("Same with IndexPool");
 	for (auto it = m_aMeshes.Begin(); it != m_aMeshes.End(); it++) {
 		it->Get()->SetVertexPool(m_pVertexPool);
 		// Same with IndexPool
@@ -201,8 +201,6 @@ void TSpriteShaderHAL::Render(TRenderPacket *a_pRenderPacket)
 	if (pMesh->IsDying()) {
 		return;
 	}
-	TIMPLEMENT();
-	return;
 	pMesh->GetMaterial()->PreRender();
 	if (m_bVertexShaderSuccess) {
 		D3DXVECTOR4 vec4(1.0f, 1.0f, 1.0f, 1.0f);
@@ -273,6 +271,18 @@ TBOOL TSpriteShaderHAL::Validate()
 	return TResource::Validate();
 }
 
+void TSpriteShaderHAL::Invalidate()
+{
+	if (!TResource::IsValid()) {
+		return;
+	}
+	if (m_bVertexShaderSuccess) {
+		GetRenderer()->GetD3DDevice()->DeleteVertexShader(m_dwVertexShaderHandle);
+	}
+	m_dwVertexShaderHandle = INVALIDSHADERHANDLE;
+	TResource::Invalidate();
+}
+
 void TSpriteShaderHAL::SetMaterial(TSpriteMaterial *a_pMaterial)
 {
 	TSpriteShader::SetMaterial(a_pMaterial);