TestParticleField.h

#pragma once
 
#ifdef FH_VISUALIZATION
#   pragma warning(push)
#   pragma warning(disable : 4251)
#   include <Ogre.h>
#   pragma warning(pop)
#   include <sfh/ogre/C3DArrow.h>
#endif
 
#include <fhsim/simobject/SimObject.h>
#include <fhsim_environment/EnvironmentProvider.h>
#include <string>
#include <vector>
 
namespace testmodels
{
 
class TestParticleField : public SimObject
{
 
public:
   TestParticleField(std::string SimObjectName, ISimObjectCreator* Creator);
 
   void OdeFcn(const double T, const double* const X, double* const XDot) const override;
 
   void FinalSetup(const double T, const double* const X, ISimObjectCreator* const Creator);
 
   const double* Position(const double T, const double* const X);
 
   const double* Velocity(const double T, const double* const X);
 
private:
   void CalcParticleForces(double t, const double pos[3], const double vel[3], double sumForces[3]) const;
 
   int m_PosIdx;
   int m_VelIdx;
   int m_numParticles;
   int m_numParticlesPerDimension[3];
   double m_particleSpacingPerDimension[3];
   double m_diameter;
   double m_mass;
   double m_dragCoeff;
   double m_rho_water;
   double m_rho_sphere;
   double m_gravityForce;
   double m_buoyancy;
   double m_netWeightWhenSubmerged;
   double R;
   environment::EnvironmentProvider* m_environment;
#ifdef FH_VISUALIZATION
public:
   void extracted(Ogre::SceneManager* sceneMgr, int& i);
 
   void RenderInit(Ogre::Root* const pOgreRoot, ISimObjectCreator* const pCreator);
 
   void RenderUpdate(const double dT, const double* const adX);
 
private:
   std::vector<Ogre::SceneNode*> m_Node;
   Ogre::SceneNode* m_lineEndNode;
   Ogre::String m_Material;
   std::string m_SphereMesh;
   C3DArrow* m_surfaceNormalArrow;
   std::vector<C3DArrow*> m_waterVelocityArrows;
   std::vector<C3DArrow*> m_objectVelocityArrows;
#endif
};
 
} // namespace testmodels