WaveEnergySpectrum.h

#pragma once
 
#include "marenv/marenv.h"
#include "marenv/calc/LcgRandom.h"
#include <vector>
#include <marenv/wave/WaveComponents.h>
 
namespace marenv::wave
{
 
 
class WaveEnergySpectrum : private marenv::MoveNoCopy
{
public:
 
    enum class DirectionSpectrum
    {
        None = 0,
        CosineSquared, 
    };
 
    enum class DirectionBinning
    {
        Random, 
        Sequential, 
 
    };
 
    enum class FrequencyBinning
    {
        EqualEnergy, 
        EqualSqrtEnergy 
    };
 
    struct SpectrumParameters
    {
        double Hs                         = 1.0; 
        double T1                         = 6.0; 
        double Theta                      = 0.0; 
        double cosinePower                = 2.0; 
        double omegaNDimLowLim            = 0.1; 
        double omegaNDimUppLim            = 10.0; 
        SpectrumType spectrumType         = SpectrumType::JONSWAP; 
        DirectionBinning directionBinning = DirectionBinning::Sequential; 
        FrequencyBinning frequencyBinning = FrequencyBinning::EqualSqrtEnergy; 
    };
    WaveEnergySpectrum(const SpectrumParameters& spectrumParameters, int numWaves, uint32_t randomSeed);
    ~WaveEnergySpectrum() = default;
 
    std::shared_ptr<WaveComponents> GetWaveComponents();
 
protected:
 
    using NormalizedProbabilityDistribution = double (WaveEnergySpectrum::*)(const double);
    void CalcWaveDirections(calc::LcgRandom& random);
    [[nodiscard]] double IntegrateSpectrum(double omegaLow, double omegaHigh, double step) const;
    [[nodiscard]] double BinFrequency(double omegaLow, double omegaHigh, double step) const;
    [[nodiscard]] std::vector<double> CalcBinEdges() const;
    void CalcWaveFrequencies();
 
    void CalcCalmSeaFrequencies();
    [[nodiscard]] double CosPower(double theta) const;
    [[nodiscard]] double SpectralEnergy(double omega) const;
    [[nodiscard]] double JONSWAP(double omega) const;
    [[nodiscard]] double Issc(double omega) const;
    SpectrumParameters m_param; 
 
private:
    struct WaveComponentSpec
    {
        double zetaA; 
        double omega; 
        double theta; 
        double energy; 
        bool operator<(const WaveComponentSpec& a)
        {
            return zetaA < a.zetaA;
        }
    };
    std::shared_ptr<WaveComponents> m_waveComponents;
    int m_numWaves;
    uint32_t m_randomSeed;
    std::vector<WaveComponentSpec> m_waveComponentSpecs;
};
 
} // namespace marenv::wave