shells/driverCC/PrescribeVarFunctor.h

Go to the documentation of this file.

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


namespace shells {
    template <typename T1, typename T2>
    class CheckPrescribeSVtxField;
    template <typename EXTR>
    class CheckSVtxFieldInBox;
    template <typename EXTR>
    class PrescribeSVtxField;
    template <typename EXTR>
    class CheckSVtxFieldNotInBox;
} 


namespace shells {

template <typename EXTR>
class PrescribeSVtxField : 
    public std::unary_function<shells::SVertexS *, void> {
// prescribe a vertex field (e.g. displacements) to _pVar 

public:    
    typedef typename EXTR::DataType    _DataType;

    PrescribeSVtxField(const _DataType * const pVar)
        : _pVar(pVar){}
    
    void operator()(shells::SVertexS * const vtx) const
        {
            EXTR extract;
            _DataType *vtxVar = extract(vtx);
            
            for (std::size_t i=0; i<EXTR::numVar ; ++i) {
                vtxVar[i] = _pVar[i]; 
            }     
        }
    
private:
    const _DataType * const     _pVar; 
};



template <typename EXTR>
class CheckSVtxFieldInBox :
    public std::unary_function<shells::SVertexS*, bool> {
// check if a vertex field (e.g. coordinates) is in a domain 
// (_low x _upp)

public:
    typedef typename EXTR::DataType    _DataType;

    CheckSVtxFieldInBox(const _DataType * const low, 
                        const _DataType * const upp)
        :_low(low), _upp(upp){}

    bool operator()(shells::SVertex * const vtx) const 
        {
            EXTR extract;
            _DataType *pos = extract(vtx);
            
            for (std::size_t i=0; i<EXTR::numVar ; ++i) {
                if (pos[i] < _low[i] || _upp[i] < pos[i]) return false;
            }       
            return true;
        }

private:
    const _DataType * const     _low;
    const _DataType * const     _upp;
};



template <typename EXTR>
class CheckSVtxFieldNotInBox  :
    public std::unary_function<shells::SVertexS*, bool> {
// check if a vertex field (e.g. coordinates) is outside the domain 
// (_low x _upp), i.e., not(CheckSVtxFieldInBox)

public:
    typedef typename EXTR::DataType    _DataType;

    CheckSVtxFieldNotInBox(const _DataType * const low, 
                           const _DataType * const upp) 
        : _checker(CheckSVtxFieldInBox<EXTR>(low, upp)){}
    
    bool operator()(shells::SVertex * const vtx) const {
        return !(_checker(vtx));
    }
    
private:
    const CheckSVtxFieldInBox<EXTR>     _checker;
};



template <typename T1, typename T2>
class CheckPrescribeSVtxField :
    public std::unary_function<shells::SVertexS *, void>  {
// this functor will be used in combination with the previous three functors

public:
    CheckPrescribeSVtxField(const T1& setVar, const T2& check):
        _setVar(setVar), _check(check){}
    
    void operator()(shells::SVertexS *vtx) const
        {
            if (_check(vtx)) {
                _setVar(vtx);
            }
        }
    
private:
    const T1&     _setVar;
    const T2&     _check;
};

}  // namespace shells

#endif

---