TAPsSimPropForMultiPartMeshModel_HalfEdge.hpp

Go to the documentation of this file.

/******************************************************************************
TAPsSimPropForMultiPartMeshModel_HalfEdge.hpp

SimPropForMultiPartMeshModel_HalfEdge class (hpp file).

SUKITTI PUNAK   (07/28/2008)
UPDATE          (10/07/2009)
******************************************************************************/
#ifndef TAPs_SIM_PROP_FOR_MULTI_PART_MESH_MODEL_HALF_EDGE_HPP
#define TAPs_SIM_PROP_FOR_MULTI_PART_MESH_MODEL_HALF_EDGE_HPP

#include "TAPsSimPropForMultiPartMeshModel.hpp"

#include "TAPsHEVertexPtrWithExtraInfo.hpp"
#include "../../Simulation/TAPsListOfODESolvers.hpp"

BEGIN_NAMESPACE_TAPs
//=============================================================================
template <typename T>
class SimPropForMultiPartMeshModel_HalfEdge : public /*virtual*/ SimPropForMultiPartMeshModel<T> {
//=============================================================================
    // Member Functions -------------------------------------------------------
    friend std::ostream & operator<< ( 
        std::ostream &output, SimPropForMultiPartMeshModel_HalfEdge<T> const &obj )
    {
        output << obj.StrInfo();
        return output;
    }

//=============================================================================
public:
    // Member Functions -------------------------------------------------------

    SimPropForMultiPartMeshModel_HalfEdge ( 
        T massOfPoint       = 1,            
        T springStiffness   = 1,            
        T springDamper      = 1,            
        TAPs::Enum::ModelType modelType = TAPs::Enum::MODEL_DEFORM, 
        T predefinedTimeStep    = 0.01,     
        int numSimSubSteps      = 10,       
        bool enableColDet       = true,     
        bool enableSimulation   = true      
    );

    //SimPropForMultiPartMeshModel_HalfEdge ( SimPropForMultiPartMeshModel_HalfEdge<T> const &obj );

    ~SimPropForMultiPartMeshModel_HalfEdge ();

    //inline SimPropForMultiPartMeshModel_HalfEdge<T> & operator= ( 
    //  SimPropForMultiPartMeshModel_HalfEdge<T> const &obj );

    virtual std::string StrInfo () const;

    // Get number of states
    int Size () const   { return m_iStateSize; }

    std::vector< HEVertexPtrWithExtraInfo<T> * > const &    GetVertexList () const  { return m_HEVertexPtrList; }
    std::vector< HEVertexPtrWithExtraInfo<T> * > &          GetVertexList ()        { return m_HEVertexPtrList; }

    std::vector< SpringRef<T> * > const &   GetSpringList () const  { return m_vpSpringRef; }
    std::vector< SpringRef<T> * > &         GetSpringList ()        { return m_vpSpringRef; }

    //===============================================================
    // For simulation by ODE solver
    //---------------------------------------------------------------
    void ClearAllForces ();
    bool SetODESolver ( TAPs::Enum::ODESolver solverMethod );
    void AdvSim ( T tCurrent, T tNext );
    void AdvSim ( T tCurrent )
    { AdvSim( tCurrent, tCurrent +  GetPredefinedTimeStep() ); }
    //---------------------------------------------------------------

    void UpdateStateToArray ()
    {
        if ( GetModelType() == TAPs::Enum::MODEL_DEFORM ) {
            StateToArray( &(*xEnd)[0] );
        }
    }

    // CUDA ---------------------------------------------------------
#ifdef  TAPs_USE_CUDA

    bool CUDA_Initialize ();
    void CUDA_Cleanup ();
    void CUDA_CopyVertexToMem ();
    void CUDA_CopyMemToVertex ();
#endif//TAPs_USE_CUDA
    // CUDA ---------------------------------------------------------

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

    //===============================================================
    // For simulation by ODE solver
    //---------------------------------------------------------------
    // REMARK:  DxDt has to be static so that it can be passed to 
    // ODESolver Fn.
    // But we don't want StateToArray and ArrayToState be static.
    // Therefore, we need to pass userData, i.e. this pointer
    // as a void pointer.
    static bool DxDt (  
        T                       dt,         // i/p: time step
        Simulation::VectorSet<T> &  x,      // i/p: array  x = {pos, vel}
        Simulation::VectorSet<T> &  xdot,   // o/p: array xdot = {vel, accel}
        void *                  userData    // o/p: array of user data
    );
    //---------------------------------------------------------------
    void StateToArray ( T *dst );           // copy states to dst array
    void ArrayToState ( T *src );           // copy src array to states
    void DdtStateToArray ( T *xdot, T dt ); // copy d/dt states to dst (xdot) array
    //---------------------------------------------------------------

    //void Clear ();

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

    //-----------------------------------------------------
    std::vector< SpringRef<T> * >                   m_vpSpringRef;      
    std::vector< HEVertexPtrWithExtraInfo<T> * >    m_HEVertexPtrList;  

#ifdef  TAPs_USE_CUDA
    float *             m_cudaVertexList;       
    //unsigned int * 
#endif//TAPs_USE_CUDA

    //===============================================================
    // For simulation by ODE solver
    //---------------------------------------------------------------
    TAPs::Simulation::ODESolver<T> *    m_prODESolver;  
    Simulation::VectorSet<T> *          x0;             
    Simulation::VectorSet<T> *          xEnd;           
    Simulation::VectorSet<T>            xData_1;        
    Simulation::VectorSet<T>            xData_2;        
    int m_iStateSize;       
    int m_iLimitStepCount;  
    //---------------------------------------------------------------

//=============================================================================
private:
    // Member Functions -------------------------------------------------------
    // Data Members -----------------------------------------------------------
//=============================================================================
}; // END CLASS SimPropForMultiPartMeshModel_HalfEdge
//=============================================================================
//-----------------------------------------------------------------------------
typedef SimPropForMultiPartMeshModel_HalfEdge<int>    SimPropForMultiPartMeshModel_HalfEdgei;
typedef SimPropForMultiPartMeshModel_HalfEdge<float>  SimPropForMultiPartMeshModel_HalfEdgef;
typedef SimPropForMultiPartMeshModel_HalfEdge<double> SimPropForMultiPartMeshModel_HalfEdged;
//=============================================================================
END_NAMESPACE_TAPs
//-----------------------------------------------------------------------------
// Include definition if TAPs_USE_EXPORT is not defined
// If the class is template, use this
//#if !defined( TAPs_USE_EXPORT )
    #include "TAPsSimPropForMultiPartMeshModel_HalfEdge.cpp"
//#endif

// If the class is NOT template, use this
//#if !defined( TAPs_USE_EXPORT )
//  #include "TAPsSimPropForMultiPartMeshModel_HalfEdge.cpp"
//#endif
//-----------------------------------------------------------------------------
#endif
//34567890123456789012345678901234567890123456789012345678901234567890123456789
//--+----1----+----2----+----3----+----4----+----5----+----6----+----7----+----