DtModelMergerCell.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2025 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvUtil/DtGeoReference.h>
#include <vrvUtil/DtCoordinateSystem.h>

#include <osgEarth/GeoData>

#include <array>

namespace makVrv
{
   class DT_DLL_VRVOSG DtModelMergerCell
   {
   public:

      struct GroupedModelInfo
      {
         // CTOR
         GroupedModelInfo()
         {

         };
         GroupedModelInfo( osg::Matrix mat ) :
            myGroupedModelMatrix( mat )
         {

         };
         osg::Matrix myGroupedModelMatrix;
      };

      struct ModelInfo
      {
         // These should come from the shape file and be unconverted
         // We want to compare models at LODs, conversions will introduce
         // floating point issues
         // DEGREES
         double lat;
         double lon;
         double alt;

         int lod;
         unsigned int index;

         std::string modelFilePath;
         std::string componentName;
         osg::Matrix transform;
      };

      DtModelMergerCell( double lat, double lon, double maxLat, double maxLon, double alt, int lod, 
         DtCoordinateSystem& coordinateSystem, osg::ref_ptr< osgEarth::SpatialReference > wgs84 );

      virtual ~DtModelMergerCell();

      virtual bool addModel( ModelInfo& model );

      virtual void setMaxModelCount( size_t count );
      virtual size_t maxModelCount() const;
      
      virtual void setMinimumCellSize( double size);
      virtual double mimimumCellSize() const;

      virtual double lat()const;
      virtual double lon()const;
      virtual double maxLat()const;
      virtual double maxLon()const;
      virtual double alt()const;
      virtual int lod()const;
      
      virtual std::array< DtModelMergerCell*, 4>& subCells();
      virtual std::vector<ModelInfo>& models();

      virtual osg::ref_ptr< osg::Group > modelsGroupNode();
      virtual osg::ref_ptr<osg::MatrixTransform> cellTransform();

      virtual void setCacheDirectory( std::string& cacheDir );
      virtual void setMappingsCacheDirectory( std::string& mappingsDir );
      virtual void setLatLonLodString( std::string& str );
      virtual const std::string& cacheDirectory() const;

      virtual void runDiskWriteOps();

      virtual void setOverwrite( bool overwrite );

      virtual void setIsRootCell( bool isRoot );

      virtual void setMappingsMap( std::unordered_map<std::string, std::string>* map );
      virtual void setInfosMap( std::unordered_map<std::string, GroupedModelInfo>* map );

      virtual void setDbOptions( osg::ref_ptr<osgDB::Options> options );

      virtual void printMinCellStats();
   protected:
      virtual void initCellCoordinates();

      virtual void createSubCells();

      virtual bool isPointInCell( const osg::Vec2d& point );

      virtual void buildTransformForCell();

      virtual void writeModelsGroupNodeToDisk();

      virtual void writeGroupedModelInfosToDisk();

      virtual void writeMappingsToDisk();

      virtual void runOptimisationsOnNode( osg::ref_ptr<osg::Group> group );
   protected:
      double myLat;
      double myLon;
      double myMaxLat;
      double myMaxLon;
      double myAlt;
      double myMinSize;

      int myLod;
      
      bool mySubCellsCreated;
      bool myBeenVisualised;
      bool myOverwrite;
      bool myIsRootCell;
      bool myIsMinSize;

      std::array< DtModelMergerCell*, 4> mySubCells;

      osg::ref_ptr< osgEarth::SpatialReference > myWgs84;
      DtCoordinateSystem& myCoordinateSystem;
      DtGeoReference myShapeFileGeoRef;

      std::vector<osg::Vec2> myCellCoordinates;
      osg::Vec2 myCellSizeInMetres;

      std::vector<ModelInfo> myModels;

      osg::ref_ptr< osg::Group > myModelsGroupNode;

      osg::ref_ptr<osg::MatrixTransform> myCellTransform;

      size_t myMaxModelCount;

      std::string myCacheDirectory;
      std::string myMappingsCacheDirectory;
      std::string myLatLonLodString;

      std::unordered_map<std::string, std::string>* myDatabaseFileToGroupNodeFileMap;
      std::unordered_map<std::string, GroupedModelInfo>* myGroupedModelToInfoMap;

      osg::ref_ptr<osgDB::Options> myDbOptions;
   };
}