WakeAccumulator.h

#pragma once
 
#include <marenv/wake/Superposition.h>
#include <marenv/wake/WakeField.h>
#include <marenv/wake/WakeTypes.h>
 
namespace marenv::wake
{
struct WakeComposition
{
    double deficit; 
    double sourceVelocity[3]; 
    int numSources; 
 
    void Apply(double vel[3]) const
    {
        for (int i = 0; i < 3; ++i)
            vel[i] += deficit * (sourceVelocity[i] - vel[i]);
    }
};
 
class WakeAccumulator
{
public:
    double Add(const WakeField& field, double t, const double pos[3]);
 
    double Add(const WakeSample& sample);
 
    double Add(const WakeSampleParts& parts, double t);
 
    double AddOwn(const WakeField& field, double t, const double pos[3]);
 
    double AddOwn(const WakeSample& sample);
 
    double AddOwn(const WakeSampleParts& parts, double t);
 
    WakeComposition Result() const;
 
private:
    double Weigh(const WakeSample& sample);
 
    MomentumFluxSum m_registered; 
    double m_own            = 0.0; 
    bool m_hasOwn           = false; 
    double m_weightSum      = 0.0; 
    double m_weightedVel[3] = {0.0, 0.0, 0.0}; 
    int m_numSources        = 0; 
};
} // namespace marenv::wake