WakeRegistry.h

#pragma once
 
#include <marenv/marenv.h>
#include <marenv/wake/JetField.h>
#include <marenv/wake/WakeField.h>
#include <marenv/wake/WakeSelection.h>
 
#include <array>
#include <cstdint>
#include <memory>
#include <span>
#include <string>
#include <utility>
#include <vector>
 
namespace marenv::wake
{
class WakeRegistry : private marenv::NoCopyNoMove
{
public:
    void Add(const std::string& id, std::shared_ptr<WakeField> field);
 
    void AddJet(const std::string& id, std::shared_ptr<JetField> field);
 
    std::shared_ptr<JetField> GetJet(const std::string& id) const;
 
    void Remove(const std::string& id);
 
    bool Has(const std::string& id) const;
 
    std::shared_ptr<WakeField> Get(const std::string& id) const;
 
    std::vector<std::string> Ids() const;
 
    std::vector<std::string> JetIds() const;
 
    WakeSelection Select(const WakeQueryRegion& region, const WakeQuery& query = {}, const WaterVelocityParts& parts = {}) const;
 
    WakeComposition Compose(double time, const double pos[3], const WakeQuery& query = {}, const WaterVelocityParts& parts = {}) const;
 
    void ComposeMany(double time, std::span<const std::array<double, 3>> positions, const WakeQuery& query, std::span<WakeComposition> out, const WaterVelocityParts& parts = {}) const;
 
    std::uint64_t Revision() const;
 
    bool StaticFrom(double t) const;
 
private:
    using Entry    = std::pair<std::string, std::shared_ptr<WakeField>>;
    using JetEntry = std::pair<std::string, std::shared_ptr<JetField>>;
 
    std::vector<Entry>::const_iterator Find(const std::string& id) const;
 
    std::vector<JetEntry>::const_iterator FindJet(const std::string& id) const;
 
    bool IsTaken(const std::string& id) const;
 
    std::vector<Entry> m_fields; 
    std::vector<JetEntry> m_jets; 
    std::vector<const WakeField*> m_fieldPointers; 
    std::vector<const JetField*> m_jetPointers; 
    std::uint64_t m_registrations = 0; 
};
 
} // namespace marenv::wake