shells/driverCC/Checkpointer.h

Go to the documentation of this file.

// -*- C++ -*- 
//
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
//                                   Fehmi Cirak
//                        California Institute of Technology
//                           (C) 2004 All Rights Reserved
//
// <LicenseText>
//
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
//
#ifndef CHECKPOINTER_H
#define CHECKPOINTER_H
#include "MShell.h"
#include "SVertexFunctors.h"
#include "SElementFunctors.h"
#include "../fem/selement.h"


#include <vector>
#include <iterator>  
#include <functional>
#include <iosfwd>


namespace shells {
    class CheckpointData;
    class CheckpointIFaceData;
    class CheckpointVertexData;
    class CheckpointElementData;
}



class shells::CheckpointElementData {
public:
    CheckpointElementData(MShell *const mshell);
    ~CheckpointElementData();
    
    // dump/read element data
    template <typename INS>
    void dumpElementData(INS itb);
    template <typename IT>
    void readAndDistributeElementData(IT itb, IT ite);

private:
    MShell      *_mshell;    
};


class shells::CheckpointVertexData {
public:
    CheckpointVertexData(MShell * const mshell);
    ~CheckpointVertexData();
    
    // dump/read vertex data
    template <typename INS>
    void dumpVertexData(INS itb);
    template <typename IT>
    void readAndDistributeVertexData(IT itb, IT ite);
 
private:
    MShell      *_mshell;  
};



class shells::CheckpointData {
public:
    CheckpointData(MShell *const mshell):
        _cpVtxData(CheckpointVertexData(mshell)),
        _cpElemData(CheckpointElementData(mshell)){}

    void dumpData(std::ostream& ofs);
    void readAndDistributeData(std::istream& ifs);

private:
    CheckpointVertexData     _cpVtxData;
    CheckpointElementData    _cpElemData;

    typedef std::vector<double>     _VecDCont; 
    _VecDCont     _bufferElem;
    _VecDCont     _bufferVtx;
};



// templated methods
namespace shells {
    
    template <typename INS>
    void shells::CheckpointVertexData::dumpVertexData(INS itbs) 
    {
        // collect displacements
        SVertexCollector<INS, SVertexDisplacement> getDisp;
        _mshell->iterateOverVertices(std::bind2nd(getDisp, itbs));
        
        // collect velocities
        SVertexCollector<INS, SVertexVelocity> getVel;
        _mshell->iterateOverVertices(std::bind2nd(getVel, itbs));
        
        // collect accelerations
        SVertexCollector<INS, SVertexAcceleration> getAcc;
        _mshell->iterateOverVertices(std::bind2nd(getAcc, itbs));
        
        return;
    }    


    template <typename IT>
    void CheckpointVertexData::readAndDistributeVertexData(IT itb, IT ite) 
    {
        const unsigned size = std::distance(itb, ite);
        const unsigned dofPnode = SVertexDisplacement::numVar;

        //  size = (dofPnode+dofPnode+dofPnode)*nodes
        const unsigned nodes = size/(dofPnode+dofPnode+dofPnode);

        // displacements
        IT itBegin = itb;
        IT itEnd = itBegin;
        std::advance(itEnd, nodes*dofPnode);
        SVertexDistributor<IT, SVertexDisplacement> setDisp(itBegin, itEnd);
        _mshell->iterateOverVertices(setDisp);

        // velocities
        itBegin = itEnd;        
        itEnd = itBegin;
        std::advance(itEnd, nodes*dofPnode);
        SVertexDistributor<IT, SVertexVelocity> setVel(itBegin, itEnd);
        _mshell->iterateOverVertices(setVel);

        // accelerations
        itBegin = itEnd;
        itEnd = itBegin;
        std::advance(itEnd, nodes*dofPnode);
        SVertexDistributor<IT, SVertexAcceleration> setAcc(itBegin, itEnd);
        _mshell->iterateOverVertices(setAcc);

    }
    
    
    template <typename INS>
    void shells::CheckpointElementData::dumpElementData(INS itbs)
    {
        SELementInternalDataCollect<INS> getInternalData;
        
        _mshell->iterateOverElements(std::bind2nd(getInternalData, itbs), shells::MShell::active);
                
        return;
    }

    
    
    template <typename IT>
    void CheckpointElementData::readAndDistributeElementData(IT itb, IT ite) 
    {
        SELementInternalDataDistrib<IT> setInternalData(itb, ite);
        
        _mshell->iterateOverElements(setInternalData, shells::MShell::active);
    }   
}

#endif

---