SourceSetWake.h

#pragma once
 
#include <marenv/marenv.h>
#include <marenv/wake/BlendedField.h>
#include <marenv/wake/Kernel.h>
#include <marenv/wake/LineKernel.h>
#include <marenv/wake/WakeBlend.h>
#include <marenv/wake/WakeField.h>
#include <marenv/wake/WakeTypes.h>
 
#include <memory>
#include <span>
#include <vector>
 
namespace marenv::wake
{
namespace detail
{
struct SourceSetSnapshot
{
    std::vector<ElementarySource> sources; 
    double velocity[3]; 
    KernelParams params; 
    double t0; 
    ImageSet images; 
    std::vector<KernelProfile> profiles; 
    std::vector<LineKernelProfile> lineProfiles; 
    std::vector<LineSource> lines; 
 
    bool Sample(double t, const double pos[3], WakeSample& out) const { return SampleExcluding(t, pos, {}, out); }
 
    bool SampleExcluding(double t, const double pos[3], std::span<const int> excludedTags, WakeSample& out) const;
 
    bool MayReach(const WakeQueryRegion& region) const;
 
    void Sketch(double t, FieldSketch& out) const;
 
    bool Stationary() const { return velocity[0] == 0.0 && velocity[1] == 0.0 && velocity[2] == 0.0; }
};
} // namespace detail
 
class SourceSetWake : public BlendedField<detail::SourceSetSnapshot>
{
public:
    SourceSetWake() = default;
 
    explicit SourceSetWake(const KernelParams& params);
 
    ~SourceSetWake() override = default;
 
    void Publish(SourceSet sources, const double velocity[3], double t0, double blendTime, double seabedDepth);
 
    Status SampleExcluding(double t, const double pos[3], std::span<const int> excludedTags, WakeSample& out) const override;
 
    WakeSampleParts SamplePartsExcluding(double t, const double pos[3], std::span<const int> excludedTags) const override;
 
private:
    using Snapshot = detail::SourceSetSnapshot;
 
    static std::shared_ptr<const Snapshot> MakeSnapshot(SourceSet sources, const double velocity[3], const KernelParams& params, double t0);
 
    KernelParams m_params; 
};
} // namespace marenv::wake