DtCDBVector.h

Go to the documentation of this file.

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

#pragma once

#include <osgEarth/Map>
#include <osgEarth/Progress>
#include <osgEarth/MapNodeObserver>

#include <vrvOsg/DtRenderingMethod.h>

#include <osgdb_osgearth_model_cdb/DtCDBClassRow.h>
#include <osgdb_osgearth_model_cdb/DtCDBPagerConfig.h>
#include <osgdb_osgearth_model_cdb/DtCDBAttributeIndex.h>

#include <osgearth_cdb_utils/DtCDBSigSizeTable.h>
#include <osgearth_cdb_utils/DTCDBDatabase.h>
#include <osgearth_cdb_utils/DtCDBDatasetInfo.h>
#include <Vect/Cee_VectFeature.h>
#include <Cee_CdbDB.h>
#include <vrvUtil/DtProfiler.h>
//#include <osgEarth/ElevationQuery>
#include <boost/thread/condition_variable.hpp>
#include <tbb/atomic.h>
#include <tbb/tbb.h>
#include <thread>
//#define NotTBBGetElevation 1

#include <vector>

namespace osgEarth
{
   class SpatialReference;

   namespace Util
   {
      class ElevationQuery;
   }
}

namespace makVrv { 
   class DtIndirectGeometryRegistrar;
   class DtIndirectGeometry;
   namespace oe {
      namespace CDB
{
   class DatasetInfo;
   class CDBTileKey;
   class CDBTileVerbose;
   class CDBModelOptions;
   class CdbGroupNode;
   class CDBDatasets;

   // get
   class DtTbbGetElevationHelper
   {
   public:
      DtTbbGetElevationHelper();
      virtual ~DtTbbGetElevationHelper();
   public:
      virtual void getElevations(std::vector<osg::Vec3d>& modelPoints
         , int nbFeature
         , osgEarth::Util::ElevationQuery& wElevationQuery
         , const osgEarth::SpatialReference* pointsSRS
         , bool ignoreZ
      );
   };
      
   class DtCDBVector : public osg::Referenced
   {
   public:
      DtCDBVector(const DatasetInfo& CurrentDataset, osg::ref_ptr<DtCDBDatabase> cdbDb, const osgDB::Options* dbOptions, const CDBModelOptions* cdbModelOptions, DtDe& de);
      ~DtCDBVector();

      virtual void initialize(const std::string& shpInstanceFile, DtCDBInstanceIndex& instanceAttrIndexes,
                              const std::string& dbfClassFile,    DtCDBClassIndex& classAttrIndexes, 
                              const std::string& dbfExtendedFile) = 0;
      virtual bool isInitialized() const { return myIsInitialized; };
      virtual void readInstance(const std::string& dbfClassFile, DtCDBInstanceIndex& instanceAttrIndexes, const CDBTileKey& key, const std::string& shpInstanceFile, VectFile* shpVector,
         const osgEarth::Util::ElevationQuery& elevationQuery, VecInstancesDbfRow& vecInstanceRow, const CDBTileVerbose& verboseKey,
         const MapStringToClassDbfRow& classMap, const std::string* datasetCachePath = NULL) = 0;
      virtual void readClass(const std::string& dbfClassFile, DtCDBClassIndex& classAttrIndexes, const CDBTileKey& key, const CDBTileVerbose& verboseKey, MapStringToClassDbfRow& classMap) = 0;
      //virtual void readExtended() = 0;
      virtual void processFeature(const std::string& dbfClassFile, const CDBTileKey& key, CdbGroupNode* groupNode, VecInstancesDbfRow& vecInstanceRow, 
         osgEarth::ProgressCallback* progress, bool& wasModelFound) = 0;
      bool getModelPoint(const VectFeature*  feature, osg::Vec3d& modelPoint);
      bool isValidName(std::string &filename) const;
      void setMapNode(osgEarth::MapNode* mapNode) { myMapNode = mapNode; };

      virtual DtIndirectGeometryRegistrar * indirectGeometryRegistrar() { return NULL; }
      
      virtual void finishTileLoadAndWait(CdbGroupNode* groupNode) { ; }

      virtual void finishLoadingIndirectAndWait(DtIndirectGeometry * indirectGeometry);

      osg::ref_ptr<osgDB::Options> & getDbOptions() { return myDbOptions; }

      virtual void markTaskFinished();

   protected:
      bool myIsInitialized;
      osg::ref_ptr< osgEarth::SpatialReference > myWgs84;
      osg::observer_ptr<osgEarth::MapNode> myMapNode;
      static const DtCDBPagerConfig myVectorConfig;
      DatasetInfo myCurrentDataset;
      std::string myShpInstanceFile;
      std::string myDbfClassFile; 
      std::string myDbfExtendedFile;
      osg::ref_ptr<DtCDBDatabase> myCdbDatabase;
      const CDBModelOptions* myCdbModelOptions;
      osg::ref_ptr<osgDB::Options>  myDbOptions;
      CDBSigSizeTable mySigSizeTable;
      std::string myCDBRootPath;
      std::string myFACC;
      std::string myFSC;
      std::string myCNAM;
      unsigned int myCMIX;

      //int myClassIndexCNAM;
      //int myClassIndexFACC;
      //int myClassIndexFSC;
      //int myClassIndexCMIX;
      
      int myTileLod;

      CDBDatasets* myCDBDatasets;

      RenderingMethod myRenderingMethod;

      static tbb::atomic<int> theTotalNumActiveAsyncTasks;
      tbb::atomic<int> myActiveAsyncTasks;
      boost::condition_variable myWaitCondition;
      boost::mutex myWaitMutex;

      DtDe& myDe;

      static const std::thread::id theMainThreadId;

      const bool myEnableIntersections = true;
   };
}}} // namespace makVrv::oe::CDB