DtCDBTileCache.h

Go to the documentation of this file.

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

#pragma once

#include <string>

#include <osg/ref_ptr>
#include <osg/Image>
#include <osg/Shape>
#include <osgEarth/GeoData>
#include <vrvOsg/DtOsgLRUCache.h>
#include <vrvOsg/DtOsgThreadingUtils.h>
#include <osgearth_cdb_utils/DtCDBExport.h>
#include <osgearth_cdb_utils/DtCDBDatasetInfo.h>

#include <map>
#include <queue>
#include <vector>

namespace makVrv { namespace oe { namespace CDB
{
    class DT_DLL_OSGEARTH_CDB_UTILS CDBTileCache : public osg::Referenced
    {
    public:

        CDBTileCache(const std::string& CdbRootPath);

        virtual ~CDBTileCache();

        void createCdbGeoCellIndex(const DatasetInfo& datasetInfo, osgEarth::GeoExtent& CdbGeoExtent, bool readMaxLod);

        bool isCdbGeocellAvailable(const std::string& latitude, const std::string& longitude) const;

        bool isCdbGeocellAvailable(const int latitude, const int longitude) const;

        void unregister(const DatasetInfo& datasetInfo);

        osg::Image* getImage(const std::string& CdbTile);

        void addImage(const std::string& CdbTile, osg::ref_ptr< osg::Image > rbgImage); 

        float imageHitPercentage();

        osg::HeightField* getHeightField(const std::string& CdbTile);

        void  addHeightField(const std::string& CdbTile, osg::ref_ptr< osg::HeightField > elevation); 

        float heightFieldHitPercentage();

        void setCacheSize(int cacheSize);

        int cacheSize();

        int maxElevationLod() const;

        int maxImageryLod() const;

        int maxRasterMaterialLod() const;

        int maxT2dGeometryLod() const;

        int minT2dGeometryLod() const;

        typedef std::map<std::string, bool> DtLogitudeGeocellMap;

        typedef std::map<int, bool> DtLogitudeGeocellIntMap;

        class CDBDatabaseGeocellMap : public osg::Referenced
        {
        public:
           CDBDatabaseGeocellMap() {};

           std::map<std::string, DtLogitudeGeocellMap> myCdbGeoCellMap;
           std::map<int, DtLogitudeGeocellIntMap> myCdbGeoCellIntMap;
           std::vector<DatasetInfo> myDatasetList;
           osgEarth::GeoExtent myCdbExtent;
        };

        typedef std::map<std::string, osg::ref_ptr<CDBDatabaseGeocellMap> > DtDatabasesGeocellMap;

        // TODO The functions below for the elevation cache are under evaluation and may be removed,
        // if we implement another cache system for indirect rendering

        static bool doesElevationCacheExist(const std::string& shpInstanceFile, const std::string& cachePath);
        static bool getElevationFromCache(const std::string& shpInstanceFile, const std::string& cachePath, std::vector<osg::Vec3d>& modelPoints, const unsigned int expectedFeatureCount );
        static bool saveElevationToCache(const std::string& shpInstanceFile, const std::string& cachePath, const std::vector<osg::Vec3d>& modelPoints);

    private:        
        
        typedef makVrv::LRUCache<std::string, osg::ref_ptr<osg::Image> > ImageCDBsCache;
        ImageCDBsCache myCdbImageryCache;
        Threading::Mutex myCdbImageryCacheMutex;

        typedef makVrv::LRUCache<std::string, osg::ref_ptr<osg::HeightField> > ElevationCDBsCache;
        //ElevationCDBsCache* myCdbElevationCache;
        ElevationCDBsCache myCdbElevationCache;
        Threading::Mutex myCdbElevationCacheMutex;

        size_t myNbCacheSize;

        std::string myCdbRootPath;

        static int theMaxElevationLod;
        
        static int theMaxImageryLod;

        static int theMaxRasterMaterialLod;

        static int theMinT2dLod;

        static int theMaxT2dLod;

        static int theMinLongitude;

        static int theMaxLongitude;

        static int theMinLatitude;

        static int theMaxLatitude;

        mutable Threading::Mutex myMutex;
    };

} } } // namespace mak::osgEarth::CDB