shells/utilities/MomentaEnergyFunctors.h

Go to the documentation of this file.

// -*- C++ -*-
//----------------------------------------------------------------------
//
//                           Fehmi Cirak
//                California Institute of Technology
//                   (C) 2004 All Rights Reserved
//
//----------------------------------------------------------------------
//
#ifndef MOMENTAENERGYFUNCTORS_H
#define MOMENTAENERGYFUNCTORS_H
#include "VArray.h"

#include "../driverCC/SVertexFunctors.h"

#include <functional>


namespace utilities {
    struct LMomentumFunctor;
    struct AMomentumFunctor; 
    struct KEnergyFunctor; 
}



struct utilities::LMomentumFunctor : 
    public std::binary_function<shells::SVertex *, 
                 utilities::VArray<shells::SVertexVelocity::numVar> *, void > {
    
    void operator()(shells::SVertex* vtx, 
                    utilities::VArray<shells::SVertexVelocity::numVar> *momentum) const 
        {
            shells::SVertexVelocity getVelocity;
            shells::SVertexMass getMass;
            
            const shells::SVertexVelocity::DataType *velocity = getVelocity(vtx);
            const shells::SVertexMass::DataType mass = *(getMass(vtx));
            
            for (unsigned i=0; i<shells::SVertexVelocity::numVar; ++i) {
                (*momentum)[i] += velocity[i]*mass;
            }
        }
};



struct utilities::AMomentumFunctor : 
    public std::binary_function<shells::SVertex *, 
                 utilities::VArray<shells::SVertexVelocity::numVar> *, void > { 
    
    void operator()(shells::SVertex* vtx, 
                    utilities::VArray<shells::SVertexVelocity::numVar> *amomentum) const 
        {
            // compute linear momentum
            utilities::LMomentumFunctor getLinearMomentum;
            utilities::VArray<3> lmomentum(0);

            getLinearMomentum(vtx, &lmomentum);
            
            shells::SVertexCoordinate getCoordinate;
            const shells::SVertexCoordinate::DataType *coor = getCoordinate(vtx);
            
            shells::SVertexDisplacement getDisplacement;
            const shells::SVertexDisplacement::DataType *disp = getDisplacement(vtx);

            const unsigned numVar=shells::SVertexDisplacement::numVar;
            shells::SVertexDisplacement::DataType deformed[numVar];

            for (unsigned i=0; i<numVar; ++i) {
                deformed[i] = disp[i]+coor[i];
            }

            (*amomentum)[0] += deformed[1]*lmomentum[2]-deformed[2]*lmomentum[1];
            (*amomentum)[1] += deformed[2]*lmomentum[0]-deformed[0]*lmomentum[2];
            (*amomentum)[2] += deformed[0]*lmomentum[1]-deformed[1]*lmomentum[0];  

        }
};



struct utilities::KEnergyFunctor : 
    public std::binary_function<shells::SVertex *, double *, void > { 
    
    void operator()(shells::SVertex* vtx, double *energy) const 
        {
            shells::SVertexVelocity getVelocity;
            shells::SVertexMass getMass;
            
            const shells::SVertexVelocity::DataType *velocity = getVelocity(vtx);
            const shells::SVertexMass::DataType mass = *(getMass(vtx));
            
            for (unsigned i=0; i<shells::SVertexVelocity::numVar; ++i) {
                *energy += 0.5*velocity[i]*velocity[i]*mass;
            }
        }
};

#endif

---