NetCdfSeafloor.h

#pragma once
 
#include "SimObject.h"
#include "fhsim_environment/EnvironmentProvider.h"
 
#include <marenv/seafloor/BathymetryField.h>
#include <memory>
#include <string>
#include <vector>
 
#ifdef NETCDFCPP_AVAILABLE
#   include "util/NetCDFParser.h"
#endif
 
namespace environmentmodels
{
 
class NetCdfSeafloorBathymetry : public marenv::seafloor::BathymetryField
{
public:
   explicit NetCdfSeafloorBathymetry(const std::string& filename);
   ~NetCdfSeafloorBathymetry() override;
 
   [[nodiscard]] marenv::Status GetDepth(const double pos[2], double& depthOut) override;
 
   [[nodiscard]] marenv::Status ComputeSurfaceUnitNormal(const double pos[2], double unitNormalOut[3]) override;
 
   [[nodiscard]] marenv::Status GetSeabedSigma(const double pos[3], double& sigmaOut) override;
 
   [[nodiscard]] marenv::Status GetSeabedTau(const double pos[3], double& tauOut) override;
 
   [[nodiscard]] marenv::Status GetSeabedRho(const double pos[3], double& rhoOut) override;
 
   [[nodiscard]] marenv::Status ComputeRayIntersection(const double origin[3], const double direction[3],
         double& distanceOut, double maxSearchDistance, double startSearchDistance,
         double stepSize, double convergenceTolerance) override;
 
#ifdef NETCDFCPP_AVAILABLE
   const netcdf_sinmod::Grid& GetGrid() const { return m_grid; }
 
   const double* GetDepthMap() const { return m_depthMap; }
 
   double GetLengthX() const { return m_length_x; }
 
   double GetLengthY() const { return m_length_y; }
#endif
 
private:
   double GetDepthInternal(const double pos[2]) const;
 
#ifdef NETCDFCPP_AVAILABLE
   netcdf_sinmod::Grid m_grid {};
   double*            m_depthMap = nullptr;
   double             m_length_x = 0.0;
   double             m_length_y = 0.0;
#endif
};
 
class NetCdfSeafloor : public SimObject
{
public:
   NetCdfSeafloor(std::string simObjectName, ISimObjectCreator* creator);
   ~NetCdfSeafloor() override = default;
 
   void FinalSetup(const double T, const double* const adX, ISimObjectCreator* creator) override;
 
   void OdeFcn(const double dT, const double* const adX, double* const adXDot, const bool bIsMajorTimeStep) override;
 
#ifdef FH_VISUALIZATION
   void RenderInit(Ogre::Root* ogreRoot, ISimObjectCreator* creator) override;
 
   void RenderUpdate(double T, const double* X) override;
#endif
 
private:
   std::string                               m_filename;
   std::shared_ptr<NetCdfSeafloorBathymetry> m_bathymetry;
   environment::EnvironmentProvider*         m_pEnvironment = nullptr;
 
#ifdef FH_VISUALIZATION
   Ogre::SceneNode* m_pSeabedNode = nullptr;
#endif
};
 
} // namespace environmentmodels