VelocityAbsProfile.h

//
// Created by Kruegener on 8/29/2018.
//
 
#ifndef MARDYN_TRUNK_VELOCITYABSPROFILE_H
#define MARDYN_TRUNK_VELOCITYABSPROFILE_H
 
#include "ProfileBase.h"
#include "plugins/SpatialProfile.h"
 
class DensityProfile;
 
class VelocityAbsProfile final : public ProfileBase {
public:
    VelocityAbsProfile(DensityProfile * dens) :
            _densityProfile(dens), _localProfile(), _globalProfile() {
    }
    ~~VelocityAbsProfile() final = default;
    void record(Molecule& mol, unsigned long uID) final  {
        double absV = 0.0;
        double v;
        for(unsigned short d = 0; d < 3; d++){
            v = mol.v(d);
            absV += v*v;
        }
        absV = sqrt(absV);
        _localProfile[uID] += absV;
    }
    void collectAppend(DomainDecompBase *domainDecomp, unsigned long uID) final {
        domainDecomp->collCommAppendDouble(_localProfile[uID]);
    }
    void collectRetrieve(DomainDecompBase *domainDecomp, unsigned long uID) final {
        _globalProfile[uID] = domainDecomp->collCommGetDouble();
    }
    void output(string prefix, long unsigned accumulatedDatasets) final;
    void reset(unsigned long uID) final  {
        _localProfile[uID] = 0.0;
        _globalProfile[uID] = 0.0;
    }
    // set correct number of communications needed for this profile
    int comms() final {return 1;}
 
private:
    DensityProfile * _densityProfile;
 
    // Local 1D Profile
    std::map<unsigned, double> _localProfile;
    // Global 1D Profile
    std::map<unsigned, double> _globalProfile;
 
    void writeDataEntry(unsigned long uID, ofstream &outfile) const final;
};
 
#endif //MARDYN_TRUNK_VELOCITYABSPROFILE_H