TestWakeCluster.h

#pragma once
 
#include <fhsim/simobject/SimObject.h>
#include <fhsim_environment/BodyWake.h>
#include <fhsim_environment/hydrodynamics/SphereHydro.h>
#include <fhsim_environment/WakeSourceSet.h>
#include <memory>
#include <string>
#include <vector>
 
namespace environment
{
class EnvironmentProvider;
} // namespace environment
 
namespace testmodels
{
 
class TestWakeCluster : public SimObject, public environment::wake::IndexedWakeCastingObject
{
public:
   TestWakeCluster(std::string SimObjectName, ISimObjectCreator* Creator);
   ~TestWakeCluster() override;
 
   void FinalSetup(double T, const double* X, ISimObjectCreator* Creator) override;
 
   void PreOdeFcn(double T, const double* X, IStateUpdater* updater) override;
 
   void OdeFcn(double T, const double* X, double* XDot) const override;
 
   const double* Position(double T, const double* X);
   const double* WaterVelocity(double T, const double* X);
   const double* Force(double T, const double* X);
 
   const environment::wake::BodyWake* Wake() const { return m_wake.get(); }
 
   int                              NumBodies() const override { return m_numBodies; }
   environment::wake::WakeCastState Body(const double* X, int body) const override;
   double                           BodyRadius(int /*body*/) const override { return 0.5 * m_sphere.diameter; }
   environment::wake::SourceLoad    BodyLoad(const double* X, int body, const double bodyVel[3], const double waterVel[3]) const override;
 
#ifdef FH_VISUALIZATION
   void RenderInit(Ogre::Root* ogreRoot, ISimObjectCreator* creator) override { }
   void RenderUpdate(double T, const double* X) override { }
#endif
 
private:
   int                                          m_positionIndex = 0;                
   int                                          m_numBodies     = 3;                
   environment::sphere_hydro::Sphere            m_sphere;                           
   double                                       m_towVelocity[3] = {0.0, 0.0, 0.0}; 
   environment::wake::BodyWakeSettings          m_wakeSettings;                     
   std::unique_ptr<environment::wake::BodyWake> m_wake;                             
   std::vector<double>                          m_waterVelocity;                    
   std::vector<double>                          m_force;                            
};
 
} // namespace testmodels