123

RenderDll/XRenderD3D8/StaticVB.h

@@ -0,0 +1,85 @@
+#ifndef _StaticVB_H_
+#define _StaticVB_H_
+
+template <class VertexType> class StaticVB 
+{
+  private :
+
+    LPDIRECT3DVERTEXBUFFER8 mpVB;
+    uint mVertexCount;
+    bool    mbLocked;
+
+  public :
+
+    uint GetVertexCount() const 
+    { 
+      return mVertexCount; 
+    }
+
+    StaticVB( const LPDIRECT3DDEVICE8 pD3D, const DWORD& theFVF, const unsigned int& theVertexCount )
+    {
+      mpVB = 0;
+
+      mbLocked = false;
+
+      mVertexCount = theVertexCount;
+    
+      HRESULT hr = pD3D->CreateVertexBuffer( mVertexCount * sizeof( VertexType ), D3DUSAGE_WRITEONLY | D3DUSAGE_DYNAMIC, theFVF, D3DPOOL_DEFAULT, &mpVB);
+      ASSERT( ( hr == D3D_OK ) && ( mpVB ) );
+    }
+
+    LPDIRECT3DVERTEXBUFFER8 GetInterface() const { return mpVB; }
+
+    VertexType* Lock( const unsigned int& theLockCount, unsigned int& theStartVertex )
+    {
+      theStartVertex = 0;
+      VertexType* pLockedData = 0;
+
+      // Ensure there is enough space in the VB for this data
+      ASSERT ( theLockCount <= mVertexCount );
+
+      if (mbLocked)
+        Unlock();
+
+      if ( mpVB )
+      {
+#ifndef _XBOX
+        DWORD dwFlags = D3DLOCK_DISCARD;
+#else
+        DWORD dwFlags = 0;
+#endif
+
+        DWORD dwSize = 0;
+
+        HRESULT hr = mpVB->Lock( 0, 0, reinterpret_cast< BYTE** >( &pLockedData ), dwFlags );
+
+        ASSERT( hr == D3D_OK );
+        ASSERT( pLockedData != 0 );
+        mbLocked = true;
+      }
+
+      return pLockedData;
+    }
+
+    void Unlock()
+    {
+      if ( ( mbLocked ) && ( mpVB ) )
+      {
+        HRESULT hr = mpVB->Unlock();        
+        ASSERT( hr == D3D_OK );
+        mbLocked = false;
+      }
+    }
+
+    ~StaticVB()
+    {
+      Unlock();
+      if ( mpVB )
+      {
+        mpVB->Release();
+      }
+    }
+  
+};
+
+#endif  _StaticVB_H_