#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_