86 lines
1.7 KiB
C++
86 lines
1.7 KiB
C++
#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_
|