DtIndirectGeometryRegistrar.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvOsg/DtOsgThreadingUtils.h>
#include <vrvOsg/DtIndirectGeometryAggregator.h>
#include <vrvOsg/DtIndirectEffectTexturesTemplate.h>
#include <vrvOsg/DtIndirectGeometryStateManager.h>
#include <vrvOsg/DtIndirectSwitchManager.h>
#include <vrvOsg/DtIndirectLodManager.h>
#include <vrvOsg/DtIndirectModelRegistrationInfo.h>

#include <vrvUtil/DtVirtualBaseClassCreator.h>

#include <vrvUtil/signalslib.h>

#include <osg/Node>
#include <osg/Geometry>
#include <osg/Texture>
#include <osg/Referenced>

#include <tbb/concurrent_queue.h>
#include <atomic>
#include <tbb/task.h>

#include <string>
#include <vector>
#include <unordered_set>

#include <shared_mutex>

#define INDIRECTGEOMETRYREGISTRAR_DEBUG 0

namespace osgUtil{
   class CullVisitor;
};

namespace osgEarth {
   class VirtualProgram;
};

namespace makVrv
{
   class DtGlBuffer;

   class DtIndirectGeometryRegistrar;

   class DtIndirectTextureCollectionPolicy;
   class DtIndirectModelInfoContainer;

   class DtOsgShaderManager;

   class DtIndirectModelInfo;

   class DtOsgRenderer;


   class DtIndirectGeometryRegistrar;

   typedef DtVirtualBaseClassCreator<DtIndirectGeometryRegistrar, const std::string&, const DtIndirectLimitTextureStrategy& > DtIndirectGeometryRegistrarCreator;

   template <>
   inline const char* creatorTypeName<DtIndirectGeometryRegistrar>(void) { return "DtIndirectGeometryRegistrarCreator"; }




   class  DT_DLL_VRVOSG DtIndirectGeometryRegistrar : public osg::Referenced
   {
   protected:
      friend DtIndirectGeometryRegistrarCreator;

   protected:
      DtIndirectGeometryRegistrar(DtDe& de
         , const std::string& textureCollectionPolicyName
         , const DtIndirectLimitTextureStrategy& limitTextureStrategy
         );

      virtual ~DtIndirectGeometryRegistrar();

   private:
       DtIndirectGeometryRegistrar() = delete;

      DtIndirectGeometryRegistrar(const DtIndirectGeometryRegistrar& other) = delete;

      DtIndirectGeometryRegistrar(const DtIndirectGeometryRegistrar& lhs, const osg::CopyOp& copyop) = delete;

      DtIndirectGeometryRegistrar& operator=(const DtIndirectGeometryRegistrar& other) = delete;
   public:
      virtual void setName(const std::string& name);

      // ! Thread safe method - can be called in any paging thread or the main render thread
      // ! This needs to be 'queued' up because the actual registration can happen when rebuild rebuildIfNeeded
      // ! is called from somewhere in the main thread (e.g. via the drawImplementation of DtIndirectGeometry)
      // \param modelId: The model id
      // \param modelFileName is used only for debugging messages for the aggregator
      virtual void queueUpRegisterModel(const DtIndirectModelRegistrationInfo& info);

      virtual void processRegisterModelQueue();

      virtual void processRegisterModelQueueAsync();

      virtual bool beingProcessed();
      
      virtual bool hasTasksEnqueued();

      virtual unsigned int updateBuildNumber();

      virtual unsigned int numModelsInQueue() const;

      virtual int numModels(void) const;

      virtual int modelId(int index) const;

      virtual int numGeometriesForModelID(int modelId) const;

      virtual bool isModelUploaded(int modelId) const;

      virtual const DtIndirectModelInfo* findModelInfoForModelId(int modelId) const;

      virtual const DtIndirectModelInfoContainer* modelInfoContainer() const;

      virtual DtIndirectModelInfoContainer* modelInfoContainer();

      virtual const DtIndirectTextureCollectionPolicy& textureCollectionPolicy(void) const;

      virtual const DtIndirectEffectTexturesTemplate& effectTexturesState(void) const;

      virtual DtIndirectGeometryStateManager& stateManager(void) const;

      virtual GLenum indicesType(void) const;

      virtual void releaseGLObjects(osg::State* state) const;

      virtual bool registeredModelsDirty() const;

      virtual void setPagerHandlesModelLoading(bool value);
      virtual bool pagerHandlesModelLoading();

      virtual int numTexCoordArrays(void) const;
      
      virtual std::shared_timed_mutex& addingModelsMutex();

      virtual void findModelInfosThatUseThisImageFileForItsTextures(std::set<DtIndirectModelInfo*>& modelInfos, const std::string& absoluteFileName) const;
   protected:
      virtual void registerModel(const DtIndirectModelRegistrationInfo& info);

   protected:

      DtIndirectModelInfoContainer* myModelInfoContainer;

      DtIndirectEffectTexturesTemplate myEffectTexturesState;

      bool myPagerHandlesModelLoading;
     
      typedef tbb::concurrent_queue<DtIndirectModelRegistrationInfo> QueueRegisterModelOps;
      QueueRegisterModelOps myQueueRegisterModelOps;

      DtIndirectTextureCollectionPolicy* myTextureCollectionPolicy;

      mutable bool myRegisteredModelsDirty;
      unsigned int myCurrentBuildNumber;

      DtIndirectGeometryAggregator* myAggregator;
      std::atomic<bool> myBeingProcessed;
      std::atomic<int> myNumberOfTasksEnqueued;
      std::shared_timed_mutex myAddingModelsMutex;

      mutable DtIndirectGeometryStateManager myMutableStateContainer;

      DtDe& myDe;

      DtOsgShaderManager& myShaderManager;

      std::unordered_map< std::string, std::set<int> > myImagesToModelsThatAreUsingIt;
   };

}