DtGpuArena.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/DtGpuArenaUserData.h>
#include <vrvOsg/DtGpuOperation.h>

#include <vrvCore/DtKeyedData.h>

#include <vrvUtil/DtVirtualBaseClassCreator.h>
#include <vrvUtil/DtSignalConnectionManager.h>

#include <osg/StateAttribute>
#include <osg/OperationThread>

#include <atomic>
#include <deque>
#include <vector>

namespace osg
{
   class StateSet;
   class Camera;
   class RenderInfo;
   class NodeVisitor;
}

namespace makVrv
{
   class DtGpuArena;

   class DtGpuArenaUserData;
   class DtThreadCameraStreamDataManager;
   class DtIndirectGeometry;

   class DtOsgToGlContextTracker;
   class DtIndirectGlTextureObjectManager;
   class DtGlDeleteObjectManager;
   class DtSkyObjectManager;

   class DtDe;

   typedef DtVirtualBaseClassCreator<DtGpuArena> DtGpuArenaCreator;

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

   // This is a top level PreDraw operation set up on the osgViewer.
   // It allows you to setup/run operations that require a context (e.g. compute, transfer, drawing, etc)
   class DT_DLL_VRVOSG DtGpuArena : public osg::Operation
   {
   public:
      friend DtGpuArenaCreator;

      virtual ~DtGpuArena();
   protected:
      DtGpuArena(DtDe& de);

   private:
      DtGpuArena() = delete;
      DtGpuArena& operator=(const DtGpuArena& rhs) = delete;

      DtGpuArena(const DtGpuArena& rhs) = delete;

   public:      
      virtual void operator () (osg::Object* obj) override;
   public:
      virtual void addOneTimeInitOp(DtIndirectGeometry* indirectGeometry);
      virtual void addOneTimeInitOpForNextFrame(DtIndirectGeometry* indirectGeometry);

      virtual void addPersistentOp(DtGpuOperation* op);

      virtual void removePersistentOp(DtGpuOperation* op);

      virtual size_t statsBufferUpdatesRemaining() const;

      virtual size_t statsOneTimeOpsRemaining() const;

   protected:
      virtual void update();

      virtual void slot_timeBudgetUpdated( int milliseconds );

      virtual void processOneTimeInitOperations();

      virtual void waitForGpu();

   protected:
      DtDe& myDe;

      DtThreadCameraStreamDataManager& myTcsDataManager;

      const std::string myName;

      std::vector< osg::observer_ptr<DtIndirectGeometry> > myNextFrameOneTimeInitOps;

      std::mutex myOneTimeInitOpsMutex;
      std::deque< osg::observer_ptr<DtIndirectGeometry> > myOneTimeInitOps;

      std::list<DtGpuOperation*> myPersistentOperations;

      DtSignalConnectionManager myConnections;

      DtOsgToGlContextTracker& myOsgToGlContextTracker;

      DtIndirectGlTextureObjectManager& myIndirectGlTextureObjectManager;

      DtGlDeleteObjectManager& myGlDeleteObjectManager;

      DtGpuArenaUserDataCreator& myGpuArenaUserDataCreator;

      double myTimeBudget;

      unsigned int myLastUpdateFrameNumber;
      size_t myStatsBufferUpdatesRemaining;
      std::atomic<size_t> myStatsOneTimeOpsRemaining;
   };
}