shells/driverCC/PressureFunctor.h

Go to the documentation of this file.

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

#include "SElementFunctors.h"
          
#include <functional>


namespace shells {
    template <typename OP>
    class ComputePressureFunctor; 
    class ComputeConstantPressureFunctor;
} 



class shells::ComputeConstantPressureFunctor : 
    public std::binary_function<shells::SElementS *, void, void > {
public:
    ComputeConstantPressureFunctor(const unsigned int& numSurf, 
                                   const double& pressure):
        _numSurf(numSurf),_pressure(pressure){}

    void operator()(shells::SElementS  * const element);
    
private:
    double                  _xref[3][MAXVAL];
    double                  _xcur[3][MAXVAL];
    double                  _loadElem[3][MAXVAL];           

    const unsigned int&     _numSurf;
    const double&           _pressure;
};


namespace shells {

// general version build on top of the ComputeConstantPressureFunctor 
template<typename OP> 
class ComputePressureFunctor : 
    public std::binary_function<shells::SElementS *, double, void > {
public:
    ComputePressureFunctor(const unsigned int& numSurf, const OP& pressureAt):
        _numSurf(numSurf), _pressureAt(pressureAt){}

    void operator()(shells::SElementS  * const element, const double& time) const {
        // compute element center
        SVertexCoordinate::DataType xyz[SVertexCoordinate::numVar];
        SElementTriangleAverage<SVertexCoordinate::DataType *, SVertexCoordinate> getCoordinate; 
        getCoordinate(element, xyz);
                        
        // compute element center displacements
        SVertexDisplacement::DataType disp[SVertexDisplacement::numVar];
        SElementTriangleAverage<SVertexDisplacement::DataType *, SVertexDisplacement> getDisplacement; 
        getDisplacement(element, disp);

        assert(SVertexCoordinate::numVar==SVertexDisplacement::numVar);
        
        // compute deformed position
        for(unsigned i=0; i<SVertexCoordinate::numVar; ++i) {
            xyz[i] += disp[i];
        }

        // prescribed pressure at the element center
        double pressure = _pressureAt(time, xyz[0], xyz[1], xyz[2]);

        // call the specialized version
        ComputeConstantPressureFunctor constantPressure(_numSurf, pressure);
        constantPressure(element);
    }
    
private:
    const unsigned int&     _numSurf;
    const OP&               _pressureAt;
};

} // namespace shells

#endif

---