DtIndirectModelsContainer.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2023 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/DtVirtualBaseClassCreator.h>

#include <osg/Node>

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

#include <string>

#define DTINDIRECTMODELSCONTAINER_DEBUG 0

namespace osgDB
{
   class Options;
}


struct MyHashCompare
{
   size_t hash( const std::string& x ) const
   {
      size_t h = 0;
      for( const char* s = x.c_str(); *s; ++s )
         h = (h * 17) ^ *s;
      return h;
   }
   bool equal( const std::string& x, const std::string& y ) const
   {
      return x == y;
   }
};
namespace makVrv
{
   class DtIndirectTextureCollectionPolicy;
   class DtIndirectGeometryRegistrar;
   class DtIndirectTextureCollectionPolicyFactory;
   class DtIndirectLimitTextureStrategy;

   class DtIndirectModelsContainer;

   typedef DtVirtualBaseClassCreator<DtIndirectModelsContainer, const DtIndirectLimitTextureStrategy& > DtIndirectModelsContainerCreator;

   template <>
   inline const char* creatorTypeName<DtIndirectModelsContainer>(void) { return "DtIndirectModelsContainerCreator"; }
   
   class DT_DLL_VRVOSG DtIndirectModelsContainer
   {
   public:
      //friend class so it can be created by its creator
      friend DtIndirectModelsContainerCreator;

      virtual ~DtIndirectModelsContainer();

   protected:
      DtIndirectModelsContainer(DtDe& de, const DtIndirectLimitTextureStrategy& limitTextureStrategy);

   private:

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

   public:

      struct ModelParams
      {
         bool myDaylightControlled = true;
         float myMinRange = -1;
         float myMaxRange = -1;
         bool myCullFaceEnabled = true;
         float myAlphaCutOff = 0.0f;
         float myDisableDecalProjectors = 0.0f;
         bool myAddTopLevelSwitch = false;
         std::string myLightPointGroupName = "";
      };

      virtual bool getModelNode(const std::string& modelName, const std::string& modelDefName
         , const osg::Node*& modelNode, int & modelId, const ModelParams& params, osgDB::Options* options);

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

      virtual const osg::Node* findModelNode(const std::string& modelName) const;

      virtual int findModelId(const std::string& modelName) const;

      virtual bool addModelNodeIfNeeded(osg::ref_ptr<const osg::Node> modelNode, int& modelId, const std::string& modelNameSuffix,
         bool cullFaceEnabled,
         const DtIndirectPolygonOffset& polygonOffset, bool daylightControlled);

      virtual DtIndirectGeometryRegistrar* registrar(void);

   protected:

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

   protected:
      osg::ref_ptr<DtIndirectGeometryRegistrar> myIndirectGeometryRegistrar;

      typedef std::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;
      std::atomic<int> myNextModelId;

      DtDe& myDe;

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