DtIndirectModelsContainer.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtOsgThreadingUtils.h>
#include <vrvOsg/DtIndirectPolygonOffset.h>

#include <vrvIndirectRendering/DtModelInfo.h>

#include <vrvUtil/DtVirtualBaseClass.h>

#include <boost/unordered_map.hpp>
#include <tbb/concurrent_unordered_map.h>

#include <osg/Node>

namespace osgDB
{
   class Options;
}

namespace makVrv
{
   class DtIndirectTextureCollectionPolicy;
   class DtIndirectGeometryRegistrar;
   class DtIndirectTextureCollectionPolicyFactory;

#define DTINDIRECTMODELSCONTAINER_DEBUG 0

   class DtIndirectModelsContainerCreator;
   class DT_DLL_VRVOSG DtIndirectModelsContainer
   {
   public:
      //friend class so it can be created by its creator
      friend class DtIndirectModelsContainerCreator;

      virtual ~DtIndirectModelsContainer();

   protected:
      DtIndirectModelsContainer(DtDe& de, DtIndirectTextureCollectionPolicyFactory& indirectTextureCollectionPolicyFactory);

   private:

      DtIndirectModelsContainer() = delete;
      DtIndirectModelsContainer(const DtIndirectModelsContainer &other) = delete;
      DtIndirectModelsContainer &operator=(const DtIndirectModelsContainer &other) = delete;

   public:

      virtual bool getModelNode(const std::string& modelName, const osg::Node*& modelNode, int & modelId,
         bool daylightControlled, float minRange = -1, float maxRange = -1, osgDB::Options *options = 0,
         bool cullFaceEnabled = true);

      virtual bool hasModelNode(const std::string& modelName) const;

      virtual bool addModelNodeIfNeeded(const osg::Node* modelNode, int& modelId, const std::string& modelNameSuffix,
         bool cullFaceEnabled = true,
         const DtIndirectPolygonOffset& polygonOffset = DtIndirectPolygonOffset(0.0f,0.0f), bool daylightControlled = false);

      virtual void registerModelWithRegistrar(const osg::Node* modelNode, const std::string& modelName, const DtModelInfo& modelInfo,
         bool cullFaceEnabled = true,
         const DtIndirectPolygonOffset& polygonOffset = DtIndirectPolygonOffset(0.0f, 0.0f), bool daylightControlled = false);

      virtual DtIndirectGeometryRegistrar* registrar(void);
   
   protected:

      bool checkGeometriesOk(const osg::Node* modelNode, const std::string& modelName);

   protected:
      DtIndirectTextureCollectionPolicy* myTextureCollectionPolicy;
      osg::ref_ptr<DtIndirectGeometryRegistrar> myIndirectGeometryRegistrar;

      typedef boost::unordered_map<std::string, osg::ref_ptr< const osg::Node > > MapMarkerNameToModelNode;
      MapMarkerNameToModelNode myMapMarkerNameToModelNode;
      Threading::Mutex myMapMarkerNameToModelNodeMutex;

      typedef tbb::concurrent_unordered_map<std::string, DtModelInfo> ModelsToModelInfo;
      ModelsToModelInfo myModelsToModelInfo;
      tbb::atomic<int> myNextModelId;

      DtDe& myDe;

#if DTINDIRECTMODELSCONTAINER_DEBUG
      static tbb::atomic<int> myRefCount;
#endif
   };

   class DT_DLL_VRVOSG DtIndirectModelsContainerCreator : public DtVirtualBaseClass
   {
   public:
      DtIndirectModelsContainerCreator(DtDe &de);

      virtual ~DtIndirectModelsContainerCreator();

   private:

      DtIndirectModelsContainerCreator();
      DtIndirectModelsContainerCreator(const DtIndirectModelsContainerCreator &other);
      DtIndirectModelsContainerCreator &operator=(const DtIndirectModelsContainerCreator &other);


   public:
      virtual DtIndirectModelsContainer* create() const;

      static DtIndirectModelsContainerCreator& instance(DtDe& de);

      static const std::string& theClassName();

   protected:
      DtDe& myDe;

      DtIndirectTextureCollectionPolicyFactory& myIndirectTextureCollectionPolicyFactory;
   };
}