TAPsModelDefBasedOnFEMTet.cpp

Go to the documentation of this file.

/******************************************************************************
TAPsModelDefBasedOnFEMTet.cpp
******************************************************************************/
/******************************************************************************
SUKITTI PUNAK   (03/16/2010)
UPDATE          (12/05/2010)
******************************************************************************/
#include "TAPsModelDefBasedOnFEMTet.hpp"
// Using Inclusion Model (i.e. definitions are included in declarations)
//                       (this name.cpp is included in name.hpp)
// Each friend is defined directly inside its declaration.

BEGIN_NAMESPACE_TAPs
//=============================================================================
//-----------------------------------------------------------------------------
template <typename T, typename DATA>
ModelDefBasedOnFEMTet<T,DATA>::ModelDefBasedOnFEMTet ( std::string const & modelFile )
    : ModelDefBasedOnFEM<T,DATA>( modelFile )
{}
//-----------------------------------------------------------------------------
//template <typename T, typename DATA>
//ModelDefBasedOnFEMTet<T,DATA>::ModelDefBasedOnFEMTet ( ModelDefBasedOnFEMTet<T,DATA> const &orig )
//  :
//{}
//-----------------------------------------------------------------------------
template <typename T, typename DATA>
ModelDefBasedOnFEMTet<T,DATA>::~ModelDefBasedOnFEMTet ()
{}
//-----------------------------------------------------------------------------
template <typename T, typename DATA>
std::string ModelDefBasedOnFEMTet<T,DATA>::StrInfo () const
{
    std::ostringstream ss;
    ss << "ModelDefBasedOnFEMTet<" << typeid(T).name() << ">";
    ss << "\n";
    return ss.str();
}
//-----------------------------------------------------------------------------
//=============================================================================
// Assignment Operator
//-----------------------------------------------------------------------------
//template <typename T, typename DATA>
//inline ModelDefBasedOnFEMTet<T,DATA> & ModelDefBasedOnFEMTet<T,DATA>::operator= ( ModelDefBasedOnFEMTet<T,DATA> const &orig )
//{ 
//    return *this;
//}
//-----------------------------------------------------------------------------
//=============================================================================









//-----------------------------------------------------------------------------
template <typename T, typename DATA>
bool ModelDefBasedOnFEMTet<T,DATA>::GenerateGridsWithDimension (
    T iGridWidth,       
    T iGridHeight,      
    T iGridDepth,       
    T   mass,           
    T   YoungsModulus,  
    T   PoissonsRatio,  
    bool bGenElementsInsideVolume   
)
{
    std::cout << __FILE__ << " (LINE#" << __LINE__ << ") NOT IMPLEMENTED YET!\n";
    return false;
}


//-----------------------------------------------------------------------------
template <typename T, typename DATA>
bool ModelDefBasedOnFEMTet<T,DATA>::GenerateGridsWithSize (
    int iNumOfXSlices,  
    int iNumOfYSlices,  
    int iNumOfZSlices,  
    T   mass,           
    T   YoungsModulus,  
    T   PoissonsRatio,  
    bool bGenElementsInsideVolume   
)
{
    if ( !m_pHEModel ) {
        return false;
    }

    // Generate grids
    //GridGenerator<T,DATA> GridGen;
    m_GridGen.SetModelUseToGenGrid( m_pHEModel );
    if ( m_GridGen.GenerateGridsWithSize( iNumOfXSlices, iNumOfYSlices, iNumOfZSlices ) ) {
        // Generate FEM mesh from the generated grids
        if ( GenerateMeshFromGridGenerator( m_GridGen, mass, YoungsModulus, PoissonsRatio, bGenElementsInsideVolume ) ) {
            // Clear the generated grids
            m_GridGen.DeleteGrids();

            // Set transformation support
            m_FEMMesh.SetTransformationSupportPointer( &(m_pHEModel->GetTransform()) );

            return true;
        }
        else {
            std::cout << "ModelDefBasedOnFEMTet<T,DATA>::GenerateGridsWithSize(...) FAILED to generate FEM mesh!\n";
            // Clear the generated grids
            m_GridGen.DeleteGrids();
            return false;
        }
    }
    else {
        std::cout << "ModelDefBasedOnFEMTet<T,DATA>::GenerateGridsWithSize(...) FAILED to generate grids!\n";
        return false;
    }
}


//-----------------------------------------------------------------------------
template <typename T, typename DATA>
bool ModelDefBasedOnFEMTet<T,DATA>::GenerateMeshFromGridGenerator (
    GridGenerator<T,DATA> & GridGen,    
    T   mass,                           
    T   YoungsModulus,                  
    T   PoissonsRatio,                  
    bool bGenElementsInsideVolume       
)
{
    if ( !m_pHEModel ) {
        return false;
    }
    if ( !GridGen.IsGridGenerated() ) {
        return false;
    }

    // Clear data
    // other data are clear by MeshTetrahedraGen function
    m_BaryCoords.clear();
    //m_ElementID.clear();

    // Use the static function of Meshexahedra Genetator to generate the hexahedral mesh from the grid generator
    //if ( !FEM::MeshTetrahedraGen<T,DATA>::MeshFromGridGenerator( m_FEMMesh, m_BaryCoords, m_ElementID, GridGen, mass, YoungsModulus, PoissonsRatio, bGenElementsInsideVolume ) ) {
    if ( !FEM::MeshTetrahedraGen<T,DATA>::MeshFromGridGenerator( m_FEMMesh, m_BaryCoords, GridGen, mass, YoungsModulus, PoissonsRatio, bGenElementsInsideVolume ) ) {
        return false;
    }

    // Set the link between HEVertex and FEM element
    // HEVertex'ID <--> FEM element location in the element list
    SetHEVertexIDToElementLocation();

    return true;
}


//-----------------------------------------------------------------------------
//template <typename T, typename DATA>
//bool ModelDefBasedOnFEMTet<T,DATA>::GenerateBaryCoordsFromGridGenerator ( GridGenerator<T,DATA> & GridGen )
//{
//  int C = GridGen.GetGridSizeX();
//  int R = GridGen.GetGridSizeY();
//  int D = GridGen.GetGridSizeZ();
//
//  // Set the size of the 2D vector data structure for the baracentric coordinates
//  m_BaryCoords.resize( D*R*C );
//
//  return true;
//}


//-----------------------------------------------------------------------------
template <typename T, typename DATA>
void ModelDefBasedOnFEMTet<T,DATA>::AdvanceSimulationStatic ()
{
    CDUnit().MoveAllGrabbedParts();
    m_FEMMesh.AdvanceSimulationStatic();
    UpdateSurfaceMesh();
}

//-----------------------------------------------------------------------------
template <typename T, typename DATA>
void ModelDefBasedOnFEMTet<T,DATA>::AdvanceSimulationStatic_woFixedDisplacementNodes ()
{
    CDUnit().MoveAllGrabbedParts();
    m_FEMMesh.AdvanceSimulationStatic_woFixedDisplacementNodes();
    UpdateSurfaceMesh();
}

//-----------------------------------------------------------------------------
#ifdef  TAPs_SIM_DYNAMIC_SYSTEM
template <typename T, typename DATA>
void ModelDefBasedOnFEMTet<T,DATA>::AdvanceSimulationDynamic ()
{
    CDUnit().MoveAllGrabbedParts();
    m_FEMMesh.AdvanceSimulationDynamic();
    UpdateSurfaceMesh();
}
#endif//TAPs_SIM_DYNAMIC_SYSTEM

//-----------------------------------------------------------------------------
#ifdef  TAPs_SIM_DYNAMIC_SYSTEM
template <typename T, typename DATA>
void ModelDefBasedOnFEMTet<T,DATA>::AdvanceSimulationDynamic_woFixedDisplacementNodes ()
{
    CDUnit().MoveAllGrabbedParts();
    m_FEMMesh.AdvanceSimulationDynamic_woFixedDisplacementNodes();
    UpdateSurfaceMesh();
}
#endif//TAPs_SIM_DYNAMIC_SYSTEM

//-----------------------------------------------------------------------------
template <typename T, typename DATA>
void ModelDefBasedOnFEMTet<T,DATA>::UpdateSurfaceMesh ()
{
    // Iterate all vertices of the surface mesh
    for ( unsigned int v = 0; v < m_BaryCoords.size(); ++v ) {
        m_BaryCoords[v].ComputeNewVertexLocation();
    }

    // Update all extra vertices
    TetBarycentricCoordsForVertexRef<T> * pTetBC;
    for ( unsigned int v = 0; v < m_SetOfExtraVertexCoords.size(); ++v ) {
        pTetBC = dynamic_cast< TetBarycentricCoordsForVertexRef<T>* >( m_SetOfExtraVertexCoords[v] );
        pTetBC->ComputeNewVertexLocation();
    }

    // Update BVH tree
    // The update of normals is not done by this function.
    // The update of normals must be done after this but before rendering.
    ModelDefBasedOnFEM<T,DATA>::UpdateSurfaceMesh();

    // Adjust the surface mesh, i.e. the vertices that is collided with another object.
    //CDUnit().AdjustSurfaceMeshByCDObjs();

    if ( m_CD.GetCollisionStatus() ) {
        m_CD.SetCollisionStatus( false );
    }
}

//-----------------------------------------------------------------------------
template <typename T, typename DATA>
void ModelDefBasedOnFEMTet<T,DATA>::SetHEVertexIDToElementLocation ()
{
    if ( !m_pHEModel )  return;

    std::list< HEVertex<T> * > pVList;
    // Store all vertices for linking
    HEVertex<T> * V = m_pHEModel->GetVertexList()->Head();
    while ( V ) {
        pVList.push_back( &(*V) );
        V = V->Next();
    }
    

    //std::cout << __FILE__ << "\n";

    // Do the link
    int c = 0;
    for ( unsigned int i = 0; i < m_BaryCoords.size(); ++i ) {
        std::list< HEVertex<T> * >::iterator it = pVList.begin();
        while ( it != pVList.end() ) {
            if ( (void *)(m_BaryCoords[i].RetVertexPtr()) == (void *)((**it).RefToPosition()) ) {
                //(**it).SetID( m_ElementID[i] );
                (**it).SetID( m_BaryCoords[i].GetElementID() );
                pVList.erase( it );

                //std::cout << ++c << ": " << m_ElementID[i] << "\n";

                break;
            }
            ++it;
        }
    }
}

//-----------------------------------------------------------------------------
template <typename T, typename DATA>
bool ModelDefBasedOnFEMTet<T,DATA>::InitCD ()
{
    if ( ModelDefBasedOnFEM<T,DATA>::InitCD() ) {
        m_CD.Init( m_pHEModel, &m_FEMMesh );
        return true;
    }
    return false;
}

//-----------------------------------------------------------------------------
template <typename T, typename DATA>
HEVertex<T> * ModelDefBasedOnFEMTet<T,DATA>::AddExtraVertexInLocalSpace (
    Vector3<T> location,                    
    int elementID,                          
    FEM::Element<T> * elementPtr,           
    Vector3<unsigned int> gridLocations,    
    Vector3<T> coordinates                  
)
{
    PrtNotImplementedYet( "AddExtraVertexInLocalSpace" );
    return NULL;
}

//-----------------------------------------------------------------------------
template <typename T, typename DATA>
bool ModelDefBasedOnFEMTet<T,DATA>::FindElementThatContainsPointInLocalSpace (
    Vector3<T> const & point,               
    unsigned int & elementID,               
    Vector3<unsigned int> & gridLocations,  
    Vector3<T> & coordinates                
)
{
    PrtNotImplementedYet( "FindElementThatContainsPointInLocalSpace" );
    return false;
}



//=============================================================================
#if defined(__gl_h_) || defined(__GL_H__)
//-----------------------------------------------------------------------------
    template <typename T, typename DATA>
    void ModelDefBasedOnFEMTet<T,DATA>::DrawUndeformedFEMMesh ()
    {
        m_FEMMesh.DrawUndeformedMesh();
    }

    template <typename T, typename DATA>
    void ModelDefBasedOnFEMTet<T,DATA>::DrawDeformedFEMMesh ()
    {
        m_FEMMesh.DrawDeformedMesh();
    }

    template <typename T, typename DATA>
    void ModelDefBasedOnFEMTet<T,DATA>::DrawCollidedMeshParts ()
    {
        glPushMatrix();
        m_pHEModel->GetTransform().TransformByOpenGLForDrawing();
        for ( unsigned int i = 0; i < m_CD.RefToCollidedMeshParts().size(); ++i ) {
            m_CD.RefToCollidedMeshParts()[i].pHEVertex->Draw();
        }
        glPopMatrix();
    }
//-----------------------------------------------------------------------------
#endif // #if defined(__gl_h_) || defined(__GL_H__)
//=============================================================================

//=============================================================================
END_NAMESPACE_TAPs
//34567890123456789012345678901234567890123456789012345678901234567890123456789
//--+----1----+----2----+----3----+----4----+----5----+----6----+----7----+----