CollarChain.h

#pragma once
#include <Eigen/Eigen>
 
#include "SupergridCable.h"
#include "LineStructureGravityHydro.h"
 
 
 
 
class CollarChain : public CoRiBoDynamics::Structures::PointRetractableCable {
public:
    CollarChain(CoRiBoDynamics::ConstraintSolver* solver, ISimObjectCreator* creator, double beta, string name);
 
    void setEnvironment(CEnvironment*e) { env_wrapper = EnvironmentWrapper(e); }
 
    virtual void ComputeConstraints(const double T, const double * const X);
 
protected:
    static int GetElementNum(ISimObjectCreator* creator);
 
    class EnvironmentWrapper : public CoRiBoDynamics::Environment_Interface
    {
    public:
        EnvironmentWrapper(CEnvironment* environment) { m_environment = environment; }
 
        Eigen::Vector3d Current(double T, const double* X, const double* Position) { Eigen::Vector3d C; m_environment->GetCurrentVelocity(T, X, Position, C.data()); return C; }
        double OceanSurfaceElevation(double T, const double* X, const double* Position) { return m_environment->GetSurfaceElevation(T, X, Position); }
 
        void CurrentQuery(double T, const double* X, const Eigen::Vector3d& Position, Eigen::Vector3d& current, double& surface_elevation, double& density) { auto result = m_environment->PointEnvironmentQuery(T, X, Position.data()); current = Eigen::Vector3d(result.current_velocity); surface_elevation = result.surface_elevation; density = result.density; }
 
        double SeaDepth(const double* Position) { return m_environment->GetSeadepth(Position); }
 
    private:
        CEnvironment* m_environment;
    };
    EnvironmentWrapper env_wrapper;
};