WaterVelocity.h

#pragma once
 
#include <marenv/wake/WakeTypes.h> // WakeComposition
 
namespace marenv
{
enum class WaterVelocityTerms
{
    Full, 
    NoWaves, 
    NoWakes, 
    CurrentOnly 
};
 
struct WaterVelocityParts
{
    bool current = true; 
    bool wakes   = true; 
    bool jets    = true; 
    bool waves   = true; 
};
 
constexpr WaterVelocityParts PartsOf(WaterVelocityTerms terms)
{
    switch (terms) {
        case WaterVelocityTerms::NoWaves:
            return {true, true, true, false};
        case WaterVelocityTerms::NoWakes:
            return {true, false, false, true};
        case WaterVelocityTerms::CurrentOnly:
            return {true, false, false, false};
        case WaterVelocityTerms::Full:
        default:
            return {true, true, true, true};
    }
}
 
inline void SumWaterVelocity(const WaterVelocityParts& parts, const double ambient[3], const wake::WakeComposition* wake, const double waveVel[3], double velOut[3])
{
    double vel[3] = {0.0, 0.0, 0.0};
    if (parts.current)
        for (int i = 0; i < 3; ++i)
            vel[i] = ambient[i];
    if (wake) {
        if (parts.wakes)
            for (int i = 0; i < 3; ++i)
                vel[i] += wake->deficit * (wake->sourceVelocity[i] - ambient[i]);
        if (wake->numJets > 0)
            for (int i = 0; i < 3; ++i)
                vel[i] += wake->jetVelocity[i];
    }
    for (int i = 0; i < 3; ++i)
        velOut[i] = parts.waves ? vel[i] + waveVel[i] : vel[i];
}
} // namespace marenv