WakeRegistry.h

#pragma once
 
#include <marenv/marenv.h>
#include <marenv/wake/WakeAccumulator.h> // WakeComposition
#include <marenv/wake/WakeField.h>
 
#include <memory>
#include <string>
#include <utility>
#include <vector>
 
namespace marenv::wake
{
struct WakeQuery
{
    const WakeField* exclude = nullptr; 
    const WakeField* own     = nullptr; 
};
 
class WakeRegistry : private marenv::NoCopyNoMove
{
public:
    void Add(const std::string& id, std::shared_ptr<WakeField> field);
 
    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::shared_ptr<const WakeField>> SelectReaching(const WakeQueryRegion& region, const WakeField* exclude = nullptr) const;
 
    WakeComposition Compose(double time, const double pos[3], const WakeQuery& query = {}) const;
 
private:
    using Entry = std::pair<std::string, std::shared_ptr<WakeField>>;
 
    std::vector<Entry>::const_iterator Find(const std::string& id) const;
 
    std::vector<Entry> m_fields; 
};
 
} // namespace marenv::wake