PropellerJet.h

#pragma once
 
#include <marenv/marenv.h>
#include <marenv/wake/JetField.h>
#include <marenv/wake/Kernel.h> // ImageSet
#include <marenv/wake/WakeBlend.h>
 
#include <memory>
 
namespace marenv::wake
{
struct JetCentreline
{
    double amplitude; 
    double sigma; 
};
 
class JetProfile
{
public:
    explicit JetProfile(const JetSourceInput& input);
 
    bool Active() const { return m_active; }
 
    double AdvanceSpeed() const { return m_advanceSpeed; }
 
    double Efflux() const { return m_efflux; }
 
    double InitialSigma() const { return m_initialSigma; }
 
    double ZoneEnd() const { return m_zoneEnd; }
 
    double RampLength() const { return m_rampLength; }
 
    double VelocityScale() const { return m_advanceSpeed + m_efflux; }
 
    JetCentreline At(double x) const;
 
    double CentrelineExcess(double sigma) const;
 
    double DistanceAt(double sigma) const;
 
    double SigmaAt(double x) const;
 
    double PeakAmplitude(double xFrom, double xTo) const;
 
private:
    bool m_active; 
    double m_advanceSpeed; 
    double m_momentum; 
    double m_efflux; 
    double m_initialSigma; 
    double m_zoneEnd; 
    double m_rampLength; 
};
 
class PropellerJet : public JetField
{
public:
    PropellerJet()           = default;
    ~PropellerJet() override = default;
 
    void Publish(const JetSourceInput& input, const double velocity[3], double t0, double blendTime, double seabedDepth, bool surfaceImage);
 
    Status Sample(double t, const double pos[3], JetSample& out) const override;
 
    JetSampleParts SampleParts(double t, const double pos[3]) const override;
 
    bool MayReach(const WakeQueryRegion& region) const override;
 
private:
    struct Snapshot
    {
        double origin[3]; 
        double axis[3]; 
        double velocity[3]; 
        double t0; 
        ImageSet images; 
        JetProfile profile; 
 
        bool Sample(double t, const double pos[3], JetSample& out) const;
 
        bool MayReach(const WakeQueryRegion& region) const;
    };
 
    using Blend = WakeBlend<Snapshot, JetSample, JetBlendSum, JetBlendSamples>;
 
    static Status SampleState(const Blend::State* state, double t, const double pos[3], JetSample& out);
 
    Blend m_blend; 
};
} // namespace marenv::wake