DtSpeedTreeCore.h

Go to the documentation of this file.

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


#pragma once

#include <vrvSt/vrvSt.h>

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

#include <vrvSt/DtSpeedTreeTcsUserData.h>

#include <unordered_set>

namespace makOsgSpeedTree
{
   class Core;
   struct SForestRenderInfo;
}

namespace osg
{
   class Camera;
}

namespace makVrv
{
   class DtThreadCameraStreamDataManager;

   class DT_DLL_VRVST DtSpeedTreeCore : public DtVirtualBaseClass
   {
   
   public:
      typedef std::unordered_set<int> ProgramBlocks;

   public:
      DtSpeedTreeCore(DtDe& de);

      virtual ~DtSpeedTreeCore();

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

   public:
      static DtSpeedTreeCore& instance(DtDe& de);

      static void registerInstance(DtDe& de, DtSpeedTreeCore* anInstance);

   public:

      virtual makOsgSpeedTree::Core* speedTreeCore(void)
      {
         return mySpeedTreeCore;
      }
      
      virtual void slot_displayToBeDestroyed(const std::string& displayName, DtDe* de);

      virtual void set3DLodDistances(float fHighDetail3dDistance,
         float fLowDetail3dDistance);

      virtual void setBillboardStartEndDistances(float fBillboardStartDistance,
         float fBillboardEndDistance);

      virtual void setTreesEnabled(bool enabled);
      virtual void enableBillboardRendering(bool enable);
      virtual void enable3DTreeRendering(bool enable);

      virtual void setAmbientScalar(float val);
      virtual void setDiffuseScalar(float val);
      virtual void setSpecularScalar(float val);
      virtual void setTransmissionScalar(float val);
      virtual void setAlphaScalar(float val);

      virtual void setVertexUserFunctionFileName(
         const std::string& vertexUserFunctionFilename);

      virtual void setMultiViewVertexUserFunctionFileName(
          const std::string& vertexUserFunctionFilename);
      
      virtual void setFragmentUserFunctionFileName(
         const std::string& fragmentUserFunctionFilename);

      virtual const std::string& vertexUserFunctionFileName() const;

      virtual const std::string& multiViewVertexUserFunctionFileName() const;

      virtual const std::string& fragmentUserFunctionFileName() const;

      virtual void setFreezeEnabled(bool enable);
      virtual bool freezeEnabled() const;

      virtual void setProfilingActive(bool active);

      //Get the program id's that have been bound already.
      // This changed from being in the forest because it needs to be cleared for all
      // forests if one forest uploads new functions to the core
      virtual ProgramBlocks& programBlocks();

      virtual DtSpeedTreeTcsUserData* getOrCreateUserData(const osg::Camera* osgCamera);

      virtual DtSpeedTreeTcsUserData* findUserData(const osg::Camera* osgCamera);

      virtual const DtSpeedTreeTcsUserData* findUserData(const osg::Camera* osgCamera) const;

      virtual void clearAllUserDataUniformLocationMaps();

      virtual void incrementNumForestsRenderedThisFrame(void);

      virtual void resetNumForestsRenderedThisFrame(void);

      virtual int numForestsRenderedThisFrame(void) const;
      
   protected:
      virtual void connectSignals(void);
      virtual void disconnectSignals(void);

      virtual void slot_preTick(void);
   protected:
      static void initRenderInfo(
         makOsgSpeedTree::SForestRenderInfo& sForestRenderInfo);

   protected:
      DtDe& myDe;

      makOsgSpeedTree::Core* mySpeedTreeCore;
      DtSignalConnectionManager myConnections;

      std::string myVertexUserFunctionFilename;
      std::string myMultiViewVertexUserFunctionFileName;
      std::string myFragmentUserFunctionFilename;
      bool myFreezeEnabled;

      ProgramBlocks myProgramBlocks;

      DtThreadCameraStreamDataManager& myTcsDataManager;
      DtSpeedTreeTcsUserDataCreator& myUserDataCreator;

      std::atomic<int> myNumForestsRenderedThisFrame;
   }; 
}