TAPsAdvSimSupport_DS.hpp

Go to the documentation of this file.

/******************************************************************************
TAPsAdvSimSupport_DS.hpp
******************************************************************************/
/******************************************************************************
SUKITTI PUNAK   (09/29/2008)
UPDATE          (08/01/2010)
******************************************************************************/
#ifndef TAPs_ADV_SIM_SUPPORT_DS_HPP
#define TAPs_ADV_SIM_SUPPORT_DS_HPP

#include <TAPs/Core/TAPsLib.hpp>

BEGIN_NAMESPACE_TAPs
//=============================================================================
//typedef unsigned int t_TAPsAdvSimID;  //!< typedef for advanced simulation ID (currently used with CUDA only)
//=============================================================================


END_NAMESPACE_TAPs

#include "TAPsListOfAdvSimConsts.hpp"
#include <TAPs/GeometricDataStructure/HalfEdge/TAPsHalfEdgeDataStructure.hpp>

//#include <TAPs/Support/TAPsAdvSimConstraint.hpp>
#include "AdvSimConstraints/TAPsAdvSimConstraint_HEVertexVsHEVertex.hpp"
#include "AdvSimConstraints/TAPsAdvSimConstraint_ERSutureVsHEVertex.hpp"

#include <TAPs/OpenGLModel/TAPsModelSuture.hpp>
#include <TAPs/ModelSpecific/TAPsModelForSurgery.hpp>
#include <TAPs/Model/TAPsModelElasticRod.hpp>
#include <TAPs/Model/TAPsModelDefBasedOnFEMTet.hpp>
#include <TAPs/Model/TAPsModelDefBasedOnFEMHex.hpp>


BEGIN_NAMESPACE_TAPs
//=============================================================================
template <typename T, typename DATA = unsigned char>
class AdvSimSupport_DS {
//=============================================================================
public:
    //=========================================================================
    typedef unsigned int    AdvSimID_ERSuture;  
    typedef unsigned int    AdvSimID_Model;     
    enum ModelType {
        MODEL_BASED_ON_MSS,
        MODEL_BASED_ON_ER,
        MODEL_BASED_ON_FEM,
        MODEL_TYPE_UNDEFINED
    };
    //=========================================================================

    // Member Functions -------------------------------------------------------
    friend std::ostream & operator<< ( std::ostream &output, AdvSimSupport_DS<T,DATA> const &obj )
    {
        output << obj.StrInfo();
        return output;
    }
    //-------------------------------------------------------------------------
    AdvSimSupport_DS ();
    ~AdvSimSupport_DS ();
    //-------------------------------------------------------------------------
    std::string StrInfo () const;
    //-------------------------------------------------------------------------

    //static int AddModel ( OpenGL::ModelSuture<T> * pModel );
    int AddModel ( ModelElasticRod<T> * pModel );
    //static int AddModel ( ModelForSurgery<T> * pModel );
    int AddModel( ModelDefBasedOnFEM<T,DATA> * pModel );

    void AddVertexConnectionModelAndModel ( 
        unsigned int modelA,                        
        HEVertex<T> * pVertexA,                     
        enum Enum::AddOn::SimConstraints flagsA,    
        unsigned int modelB,                        
        HEVertex<T> * pVertexB,                     
        enum Enum::AddOn::SimConstraints flagsB,    
        T forceRatio,           
        T forceScaleForModelA,  
        T forceScaleForModelB,  
        T forceThresholdForModelA = 1.0,    
        T forceThresholdForModelB = 1.0     
    );

    void AddVertexConnectionERSutureAndModel ( 
        unsigned int sutureID,                          
        unsigned int vertexIDSuture,                    
        //DS::SimulationFlags * simFlagsSuture,         //!< simulation flags of suture
        //Vector3<T> * vertexSuture,                        //!< vertex of suture
        enum Enum::AddOn::SimConstraints flagsSuture,   
        unsigned int modelID,                           
        HEVertex<T> * pVvertexModel,                    
        enum Enum::AddOn::SimConstraints flagsModel,    
        T forceRatio,           
        T forceScaleForSuture,  
        T forceScaleForModel,   
        T forceThresholdForSuture = 1.0,    
        T forceThresholdForModel = 1.0      
    );

    //-------------------------------------------------------------------------

    inline void EnforceAllConstraints ();
    inline void EnforceAllConstraintsOfTwoModels ( int modelA, int modelB );
    inline void EnforceAllConstraintsOfERSutureAndModel ( int suture, int model );
    inline void EnforceTheConstraintNumberOfTwoModels (
        int modelA,                         
        Matrix4x4<T> const & TrxModelA,     
        Matrix4x4<T> const & InvTrxModelA,  
        int modelB,                         
        Matrix4x4<T> const & TrxModelB,     
        Matrix4x4<T> const & InvTrxModelB,  
        unsigned int constraintNumber       
    );
    inline void EnforceTheConstraintNumberOfERSutureAndModel (
        int suture,                         
        int model,                          
        Matrix4x4<T> const & TrxModel,      
        Matrix4x4<T> const & InvTrxModel,   
        unsigned int constraintNumber       
    );

    void EnforceAllConsistentPositions ();
    inline void EnforceAllConsistentPositionsOfTwoModels ( int modelA, int modelB );
    inline void EnforceAllConsistentPositionsOfERSutureAndModel ( int suture, int model );

    inline void ClearAllConstraints ();
    inline void ClearAllConstraintsOfTwoModels ( int modelA, int modelB );
    inline void ClearAllConstraintsOfERSutureAndModel ( int suture, int model );
    inline void ClearTheConstraintNumberOfTwoModels ( int modelA, int modelB, unsigned int constraintNumber );
    inline void ClearTheConstraintNumberOfERSutureAndModel ( int suture, int model, unsigned int constraintNumber );

    //-------------------------------------------------------------------------

    unsigned int GetNumOfSutureModelsBasedOnMSS ()          { return static_cast<unsigned int>( m_ListOfSutureModelsBasedOnMSS.size() ); }
    unsigned int GetNumOfSutureModelsBasedOnER ()           { return static_cast<unsigned int>( m_ListOfSutureModelsBasedOnER.size() ); }
    unsigned int GetNumOfModelsForSurgeryBasedOnMSS ()      { return static_cast<unsigned int>( m_ListOfModelsForSurgeryBasedOnMSS.size() ); }
    unsigned int GetNumOfSubModelsForSurgeryBasedOnMSS ()   { return static_cast<unsigned int>( m_ListOfHETriMeshOneModelMultiParts.size() ); }
    unsigned int GetNumOfModelsForSurgeryBasedOnFEM ()      { return static_cast<unsigned int>( m_ListOfModelsForSurgeryBasedOnFEM.size() ); }

    std::vector< OpenGL::ModelSuture<T> * > &       GetListOfSutureModelsBasedOnMSS ()      { return m_ListOfSutureModelsBasedOnMSS; }
    std::vector< ModelElasticRod<T> * > &           GetListOfSutureModelsBasedOnER ()       { return m_ListOfSutureModelsBasedOnER; }
    std::vector< ModelForSurgery<T> * > &           GetListOfModelsForSurgeryBasedOnMSS ()  { return m_ListOfModelsForSurgeryBasedOnMSS; }
    std::vector< ModelDefBasedOnFEM<T,DATA> * > &   GetListOfModelsForSurgeryBasedOnFEM ()  { return m_ListOfModelsForSurgeryBasedOnFEM; }

    inline std::vector< AdvSimConstraint_HEVertexVsHEVertex<T> * > &    RefToInteractionModelVsModel (
        unsigned int modelA, unsigned int modelB )
    {
        return m_ListOfInteractionModelVsModel[modelA][modelB];
    }

    inline std::vector< AdvSimConstraint_ERSutureVsHEVertex<T> * > &    RefToInteractionERBasedSutureVsModel ( 
        unsigned int sutureNo )
    { return m_ListOfInteractionERSutureVsModel[sutureNo]; }

    //-------------------------------------------------------------------------
    void AdvSimAllModels (
        T timestep,                                 
        bool bEnforceAllConstraints = true,         
        bool bEnforceAllConsistentPositions = true  
    );

    inline void AdvSimSutureModelBasedOnMSS ( unsigned int sutureID, T timestep );
    inline void AdvSimSutureModelBasedOnER ( unsigned int sutureID, T timestep );
    inline void AdvSimModelForSurgeryBasedOnMSS ( unsigned int modelID, T timestep );
    inline void AdvSimModelForSurgeryBasedOnFEM ( unsigned int modelID, T timestep );

    // Data Members -----------------------------------------------------------
//=============================================================================
protected:
    // Member Functions -------------------------------------------------------


    // Data Members -----------------------------------------------------------

    typedef std::list< PointForce<T> * >    ListOfPointForces;

    // SEMI-EXCLUSIVELY FOR MODELS VS MODELS INTERACTIONS.
    // This three std::vector must be managed together! 
    // ALSO m_ModelType AND m_ModelLocation ARE USED FOR INTERACTION WITH SUTURES!
    std::vector< ModelType >    m_ModelType;        
    std::vector< unsigned int > m_ModelLocation;    
    std::vector< std::vector< std::vector< AdvSimConstraint_HEVertexVsHEVertex<T> * > > >   m_ListOfInteractionModelVsModel;

    ListOfPointForces   m_ListOfPointForces;    


    // SEMI-EXCLUSIVELY FOR ER-BASED-SUTURES VS MODELS INTERACTIONS.
    std::vector< std::vector< std::vector< AdvSimConstraint_ERSutureVsHEVertex<T> * > > >   m_ListOfInteractionERSutureVsModel; 

    // List of models
    std::vector< OpenGL::ModelSuture<T> * > m_ListOfSutureModelsBasedOnMSS;     
    std::vector< ModelElasticRod<T> * >     m_ListOfSutureModelsBasedOnER;      
    std::vector< ModelForSurgery<T> * >     m_ListOfModelsForSurgeryBasedOnMSS; 
    std::vector< OpenGL::HETriMeshOneModelMultiParts<T> * > m_ListOfHETriMeshOneModelMultiParts;    
    std::vector< ModelDefBasedOnFEM<T,DATA> * > m_ListOfModelsForSurgeryBasedOnFEM; 
//=============================================================================
private:
    // Member Functions -------------------------------------------------------

    int AddModel ();

    int AddERBasedSuture ();

    // Data Members -----------------------------------------------------------
//=============================================================================

#if defined(__gl_h_) || defined(__GL_H__)
public:
    void DrawAllModels ();
    void DrawAddedInteractions ();
private:
    void DrawAddedInteractionsBetweenTwoModels ( int modelA, int modelB );
    void DrawAddedInteractionsBetweenERSutureAndModel ( int suture, int model );
#endif

//=============================================================================
}; // END CLASS AdvSimSupport_DS
//=============================================================================
END_NAMESPACE_TAPs
//-----------------------------------------------------------------------------
// Include definition if TAPs_USE_EXPORT is not defined
//#if !defined( TAPs_USE_EXPORT )
    #include "TAPsAdvSimSupport_DS.cpp"
//#endif
//-----------------------------------------------------------------------------
#endif//TAPs_ADV_SIM_SUPPORT_HPP_DS
//34567890123456789012345678901234567890123456789012345678901234567890123456789
//--+----1----+----2----+----3----+----4----+----5----+----6----+----7----+----