NetNodeStates.h

#pragma once
 
#include <vector>
 
struct NetNodeLayout
{
   std::vector<int> posIndex; 
   std::vector<int> velIndex; 
};
 
struct NetNodeView
{
   const double*        X         = nullptr; 
   const NetNodeLayout* layout    = nullptr; 
   int                  stateBase = 0;       
 
   const double* Position(int node) const { return X + layout->posIndex[node] - stateBase; }
   const double* Velocity(int node) const { return X + layout->velIndex[node] - stateBase; }
};
 
class NetNodeStates
{
public:
   virtual ~NetNodeStates() = default;
 
   virtual NetNodeLayout Layout() const = 0;
 
   virtual void AddExternalForces(double T, const double* X, double* nodeForce) const = 0;
 
   virtual void TakeNodeForces(double T, const double* X, const double* nodeForce, double* XDot) const = 0;
};