AttractorBase.h

 
 
#pragma once
 
#include "sfh/sim/simulation.h"
 
#include <functional>
#include <string>
 
class AttractorBase
{
public:
   struct Motion
   {
      double pos[3]      = {0.0, 0.0, 0.0};
      double vel[3]      = {0.0, 0.0, 0.0};
      double relStrength = 1.0;
   };
 
   using PosAndVelSource = std::function<void(double T, const double* X, double pos[3], double vel[3])>;
 
   using StrengthSource = std::function<double(double T)>;
 
   AttractorBase();
 
   void SetPosAndVelSource(PosAndVelSource source);
 
   void SetStrengthSource(StrengthSource source);
 
   Motion MotionAt(double T, const double* X) const;
 
   virtual void GetForce(const Motion& motion, double* const adForce, const double* const adNodePos, const double* const adNodeVel)                        = 0;
   virtual void AddForce(const Motion& motion, double* const adForce, const double nodeMass, const double* const adNodePos, const double* const adNodeVel) = 0;
   virtual void AddForce(const Motion& motion, double* const adForce, const double* const adNodePos, const double* const adNodeVel)                        = 0;
   virtual void SetForceIfActive(const Motion& motion, double* const adForce, const double* const adNodePos, const double* const adNodeVel) { };
   virtual bool IsActive(const Motion& motion, const double* const adNodePos) = 0;
 
private:
   PosAndVelSource m_posAndVel; 
   StrengthSource  m_strength;  
};