mirror of
https://github.com/nillerusr/source-engine.git
synced 2024-12-22 06:06:50 +00:00
481 lines
13 KiB
C++
481 lines
13 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
//
|
|
// Implementation of CDmMeshComp - CDmeMesh computation class
|
|
//
|
|
//=============================================================================
|
|
|
|
|
|
// Valve includes
|
|
#include "movieobjects/dmmeshcomp.h"
|
|
#include "movieobjects/dmefaceset.h"
|
|
#include "movieobjects/dmemesh.h"
|
|
#include "movieobjects/dmevertexdata.h"
|
|
|
|
|
|
// memdbgon must be the last include file in a .cpp file!!!
|
|
#include "tier0/memdbgon.h"
|
|
|
|
|
|
//=============================================================================
|
|
//
|
|
//=============================================================================
|
|
CDmMeshComp::CDmMeshComp( CDmeMesh *pMesh, CDmeVertexData *pPassedBase )
|
|
: m_pMesh( pMesh )
|
|
, m_pBase( NULL )
|
|
{
|
|
m_pBase = pPassedBase ? pPassedBase : pMesh->GetCurrentBaseState();
|
|
if ( !m_pBase )
|
|
return;
|
|
|
|
const FieldIndex_t pIndex( m_pBase->FindFieldIndex( CDmeVertexData::FIELD_POSITION ) );
|
|
if ( pIndex < 0 )
|
|
return;
|
|
|
|
const CUtlVector< Vector > &pPositionData( m_pBase->GetPositionData() );
|
|
const CUtlVector<int> &pPositionIndices( m_pBase->GetVertexIndexData( CDmeVertexData::FIELD_POSITION ) );
|
|
|
|
const int nVertices( pPositionData.Count() );
|
|
if ( nVertices <= 0 )
|
|
return;
|
|
|
|
// Create vertices
|
|
// TODO: check for duplicates in pPositionData - that would break this algorithm
|
|
m_verts.EnsureCapacity( nVertices );
|
|
for ( int i = 0; i < nVertices; ++i )
|
|
{
|
|
const CUtlVector< int > &vertexIndices = m_pBase->FindVertexIndicesFromDataIndex( CDmeVertexData::FIELD_POSITION, i );
|
|
m_verts.AddToTail( new CVert( i, &vertexIndices, &pPositionData[ i ] ) );
|
|
}
|
|
|
|
// Create edges and faces
|
|
const int nFaceSets( pMesh->FaceSetCount() );
|
|
for ( int i = 0; i < nFaceSets; ++i )
|
|
{
|
|
CDmeFaceSet *pFaceSet( pMesh->GetFaceSet( i ) );
|
|
const int nIndices( pFaceSet->NumIndices() );
|
|
if ( nIndices < 4 ) // At least a triangle and a -1
|
|
continue;
|
|
|
|
m_faces.EnsureCapacity( m_faces.Count() + nIndices / 4 ); // # new faces <= nIndices/4 (tri + -1)
|
|
m_edges.EnsureCapacity( m_edges.Count() + nIndices / 2 ); // # new edges <= 2*new faces
|
|
|
|
int facePosIndex( -1 );
|
|
int edgePosIndex0( -1 );
|
|
int edgePosIndex1( -1 );
|
|
|
|
CUtlVector< CVert * > verts;
|
|
CUtlVector< CEdge * > edges;
|
|
CUtlVector< bool > edgeReverseMap;
|
|
bool bReverse = false;
|
|
|
|
for ( int j( 0 ); j < nIndices; ++j )
|
|
{
|
|
const int faceVertexIndex( pFaceSet->GetIndex( j ) );
|
|
|
|
if ( faceVertexIndex < 0 )
|
|
{
|
|
// End of face
|
|
edgePosIndex0 = edgePosIndex1;
|
|
edgePosIndex1 = facePosIndex;
|
|
|
|
Assert( edgePosIndex0 >= 0 );
|
|
Assert( edgePosIndex1 >= 0 );
|
|
|
|
edges.AddToTail( FindOrCreateEdge( edgePosIndex0, edgePosIndex1, &bReverse ) );
|
|
edgeReverseMap.AddToTail( bReverse );
|
|
|
|
CreateFace( verts, edges, edgeReverseMap );
|
|
|
|
facePosIndex = -1;
|
|
verts.RemoveAll();
|
|
edges.RemoveAll();
|
|
edgeReverseMap.RemoveAll();
|
|
continue;
|
|
}
|
|
|
|
if ( facePosIndex < 0 )
|
|
{
|
|
// First vertex
|
|
facePosIndex = pPositionIndices[ faceVertexIndex ];
|
|
edgePosIndex1 = facePosIndex;
|
|
verts.AddToTail( m_verts[ edgePosIndex1 ] );
|
|
continue;
|
|
}
|
|
|
|
// 2nd through last vertex
|
|
edgePosIndex0 = edgePosIndex1;
|
|
edgePosIndex1 = pPositionIndices[ faceVertexIndex ];
|
|
verts.AddToTail( m_verts[ edgePosIndex1 ] );
|
|
|
|
Assert( edgePosIndex0 >= 0 );
|
|
Assert( edgePosIndex1 >= 0 );
|
|
|
|
edges.AddToTail( FindOrCreateEdge( edgePosIndex0, edgePosIndex1, &bReverse ) );
|
|
edgeReverseMap.AddToTail( bReverse );
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::~CDmMeshComp()
|
|
{
|
|
m_verts.PurgeAndDeleteElements();
|
|
m_edges.PurgeAndDeleteElements();
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::CVert::CVert( int nPositionIndex, const CUtlVector< int > *pVertexIndices, const Vector *pPosition )
|
|
: m_positionIndex( nPositionIndex )
|
|
, m_pVertexIndices( pVertexIndices )
|
|
, m_pPosition( pPosition )
|
|
, m_edges( 8, 8 )
|
|
{
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::CVert::CVert( const CVert &src )
|
|
: m_positionIndex( src.m_positionIndex )
|
|
, m_pVertexIndices( src.m_pVertexIndices )
|
|
, m_pPosition( src.m_pPosition )
|
|
, m_edges( 8, 8 )
|
|
{
|
|
m_edges.AddMultipleToTail( src.m_edges.Count(), src.m_edges.Base() );
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
int CDmMeshComp::CVert::PositionIndex() const
|
|
{
|
|
return m_positionIndex;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
const Vector *CDmMeshComp::CVert::Position() const
|
|
{
|
|
return m_pPosition;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
const CUtlVector< int > *CDmMeshComp::CVert::VertexIndices() const
|
|
{
|
|
return m_pVertexIndices;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
bool CDmMeshComp::CVert::operator==( const CVert &rhs ) const
|
|
{
|
|
return ( m_pPosition->DistToSqr( *rhs.m_pPosition ) < FLT_EPSILON );
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::CEdge::CEdge()
|
|
: m_pVert0( NULL )
|
|
, m_pVert1( NULL )
|
|
, m_faceCount( 0 )
|
|
{
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Returns the vertex position index given the edge relative vertex index
|
|
//-----------------------------------------------------------------------------
|
|
int CDmMeshComp::CEdge::GetVertPositionIndex( int edgeRelativeVertexIndex ) const
|
|
{
|
|
if ( edgeRelativeVertexIndex == 0 && m_pVert0 )
|
|
return m_pVert0->PositionIndex();
|
|
|
|
if ( edgeRelativeVertexIndex == 1 && m_pVert1 )
|
|
return m_pVert1->PositionIndex();
|
|
|
|
return -1;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::CVert *CDmMeshComp::CEdge::GetVert( int edgeRelativeVertexIndex ) const
|
|
{
|
|
if ( edgeRelativeVertexIndex == 0 )
|
|
return m_pVert0;
|
|
|
|
if ( edgeRelativeVertexIndex == 1 )
|
|
return m_pVert1;
|
|
|
|
return NULL;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Returns true if the edge starts and stops at the same position in space
|
|
// The order of the vertices is not checked
|
|
//-----------------------------------------------------------------------------
|
|
bool CDmMeshComp::CEdge::operator==( const CEdge &rhs ) const
|
|
{
|
|
return (
|
|
( *m_pVert0 == *rhs.m_pVert0 && *m_pVert1 == *rhs.m_pVert1 ) ||
|
|
( *m_pVert0 == *rhs.m_pVert1 && *m_pVert1 == *rhs.m_pVert0 ) );
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
Vector CDmMeshComp::CEdge::EdgeVector() const
|
|
{
|
|
if ( m_pVert0 && m_pVert1 )
|
|
return *m_pVert1->Position() - *m_pVert0->Position();
|
|
|
|
return vec3_origin;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Finds or Creates an edge... Can still return NULL if vertices do not exist
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::CEdge *CDmMeshComp::FindOrCreateEdge( int vIndex0, int vIndex1, bool *pReverse /* = NULL */ )
|
|
{
|
|
CEdge *pEdge = FindEdge( vIndex0, vIndex1, pReverse );
|
|
if ( pEdge )
|
|
return pEdge;
|
|
|
|
CVert *pVert0 = m_verts[ vIndex0 ];
|
|
if ( pVert0 == NULL )
|
|
return NULL;
|
|
|
|
CVert *pVert1 = m_verts[ vIndex1 ];
|
|
if ( pVert1 == NULL )
|
|
return NULL;
|
|
|
|
pEdge = m_edges[ m_edges.AddToTail( new CEdge() ) ];
|
|
pEdge->m_pVert0 = pVert0;
|
|
pEdge->m_pVert1 = pVert1;
|
|
pVert0->m_edges.AddToTail( pEdge );
|
|
if ( vIndex0 != vIndex1 )
|
|
pVert1->m_edges.AddToTail( pEdge );
|
|
|
|
if ( pReverse )
|
|
{
|
|
*pReverse = false;
|
|
}
|
|
|
|
return pEdge;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Returns the edge between vIndex0 & vIndex1 (or vice versa), NULL if not found
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::CEdge *CDmMeshComp::FindEdge( int vIndex0, int vIndex1, bool *pReverse /* = NULL */ )
|
|
{
|
|
CUtlVector< CEdge * > &edges = m_verts[ vIndex0 ]->m_edges;
|
|
for ( int i = 0; i < edges.Count(); i++ )
|
|
{
|
|
CEdge *e = edges[ i ];
|
|
|
|
if ( e->GetVertPositionIndex( 0 ) == vIndex0 && e->GetVertPositionIndex( 1 ) == vIndex1 )
|
|
{
|
|
if ( pReverse )
|
|
{
|
|
*pReverse = false;
|
|
}
|
|
return e;
|
|
}
|
|
|
|
if ( e->GetVertPositionIndex( 1 ) == vIndex0 && e->GetVertPositionIndex( 0 ) == vIndex1 )
|
|
{
|
|
if ( pReverse )
|
|
{
|
|
*pReverse = true;
|
|
}
|
|
return e;
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
CDmMeshComp::CFace *CDmMeshComp::CreateFace( const CUtlVector< CVert * > &verts, const CUtlVector< CEdge * > &edges, const CUtlVector< bool > &edgeReverseMap )
|
|
{
|
|
CFace *pFace = &m_faces[ m_faces.AddToTail() ];
|
|
|
|
pFace->m_verts.RemoveAll();
|
|
pFace->m_verts.AddVectorToTail( verts );
|
|
|
|
pFace->m_edges.RemoveAll();
|
|
pFace->m_edges.AddVectorToTail( edges );
|
|
|
|
pFace->m_edgeReverseMap.RemoveAll();
|
|
pFace->m_edgeReverseMap.AddVectorToTail( edgeReverseMap );
|
|
|
|
for ( int nEdgeIndex = edges.Count() - 1; nEdgeIndex >= 0; --nEdgeIndex )
|
|
{
|
|
edges[ nEdgeIndex ]->m_faceCount += 1;
|
|
}
|
|
|
|
return pFace;
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
int CDmMeshComp::FindFacesWithVert( int vIndex, CUtlVector< CFace * > &faces )
|
|
{
|
|
// TODO: optimize this by adding a vector of face pointers to each vertex
|
|
faces.RemoveAll();
|
|
|
|
for ( int fi( m_faces.Head() ); fi != m_faces.InvalidIndex(); fi = m_faces.Next( fi ) )
|
|
{
|
|
CFace &face( m_faces[ fi ] );
|
|
for ( int i = 0; i < face.m_verts.Count(); ++i )
|
|
{
|
|
if ( face.m_verts[ i ]->PositionIndex() == vIndex )
|
|
{
|
|
faces.AddToTail( &face );
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return faces.Count();
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
int CDmMeshComp::FindNeighbouringVerts( int vIndex, CUtlVector< CVert * > &verts )
|
|
{
|
|
verts.RemoveAll();
|
|
|
|
const CUtlVector< CEdge * > & edges = m_verts[ vIndex ]->m_edges;
|
|
|
|
for ( int i = 0; i < edges.Count(); ++i )
|
|
{
|
|
CEdge *e = edges[ i ];
|
|
if ( e->GetVertPositionIndex( 0 ) == vIndex )
|
|
{
|
|
verts.AddToTail( e->GetVert( 1 ) );
|
|
}
|
|
else
|
|
{
|
|
verts.AddToTail( e->GetVert( 0 ) );
|
|
}
|
|
}
|
|
|
|
return verts.Count();
|
|
}
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// Find all edges that are only used by 1 face
|
|
//-----------------------------------------------------------------------------
|
|
int CDmMeshComp::GetBorderEdges( CUtlVector< CUtlVector< CEdge * > > &borderEdgesList )
|
|
{
|
|
// TODO: optimize this by stepping from edge to edge to build chains, using CVert::m_edges
|
|
int retVal = 0;
|
|
|
|
borderEdgesList.RemoveAll();
|
|
|
|
bool connected;
|
|
|
|
for ( int ei = 0; ei < m_edges.Count(); ei++ )
|
|
{
|
|
CEdge *pEdge = m_edges[ ei ];
|
|
if ( pEdge->IsBorderEdge() )
|
|
{
|
|
++retVal;
|
|
connected = false;
|
|
|
|
for ( int i = borderEdgesList.Count() - 1; !connected && i >= 0; --i )
|
|
{
|
|
CUtlVector< CEdge * > &borderEdges = borderEdgesList[ i ];
|
|
for ( int j = borderEdges.Count() - 1; j >= 0; --j )
|
|
{
|
|
if ( borderEdges[ j ]->ConnectedTo( pEdge ) )
|
|
{
|
|
borderEdges.AddToTail( pEdge );
|
|
connected = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( !connected )
|
|
{
|
|
CUtlVector< CEdge * > &borderEdges = borderEdgesList[ borderEdgesList.AddToTail() ];
|
|
borderEdges.AddToTail( pEdge );
|
|
}
|
|
}
|
|
}
|
|
|
|
// Shrink the borderEdgesList to minimum number required
|
|
|
|
bool anyConnected = false;
|
|
do
|
|
{
|
|
anyConnected = false;
|
|
|
|
for ( int i = borderEdgesList.Count() - 1; i >= 0; --i )
|
|
{
|
|
CUtlVector< CEdge * > &srcBorderEdges = borderEdgesList[ i ];
|
|
for ( int j = srcBorderEdges.Count() - 1; j >= 0; --j )
|
|
{
|
|
CEdge *pSrcEdge = srcBorderEdges[ j ];
|
|
connected = false;
|
|
|
|
for ( int k = 0; !connected && k < i; ++k )
|
|
{
|
|
CUtlVector< CEdge * > &dstBorderEdges = borderEdgesList[ k ];
|
|
for ( int l = dstBorderEdges.Count() - 1; l >= 0; --l )
|
|
{
|
|
if ( dstBorderEdges[ l ]->ConnectedTo( pSrcEdge ) )
|
|
{
|
|
connected = true;
|
|
anyConnected = true;
|
|
dstBorderEdges.AddToTail( pSrcEdge );
|
|
srcBorderEdges.Remove( j );
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( srcBorderEdges.Count() == 0 )
|
|
{
|
|
borderEdgesList.Remove( i );
|
|
}
|
|
}
|
|
} while( anyConnected );
|
|
|
|
return retVal;
|
|
}
|