DtIndirectPerInstanceDataManager.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/DtIndirectManager.h>

#include <vrvUtil/DtVirtualBaseClassCreator.h>

#include <vrvMath/DtVector4.h>
#include <vrvMath/DtReusableMemoryVector.h>

#include <unordered_map>

namespace osg
{
   class State;
}

namespace makVrv
{
   class DtInstanceContainer;
   class DtDe;
   class DtGlBuffer;

   #define CPU_SIDE_COMPILATION 1
   #include <../data/shaders/indirect/PerInstanceData.h>
   #undef CPU_SIDE_COMPILATION

   class DtIndirectPerInstanceDataManager;

   typedef DtVirtualBaseClassCreator<DtIndirectPerInstanceDataManager, DtInstanceContainer&> DtIndirectPerInstanceDataManagerCreator;

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

   struct PerInstanceDataInvalidator
   {
   public:
      void operator()(PerInstanceData& info);
   };

   typedef DtReusableMemoryVector<PerInstanceData, PerInstanceDataInvalidator> DtReusablePerInstanceData;

   class DT_DLL_VRVOSG DtIndirectPerInstanceDataManager : public DtIndirectManager
   {
   public:
      friend DtIndirectPerInstanceDataManagerCreator;

      virtual ~DtIndirectPerInstanceDataManager();
   protected:
      DtIndirectPerInstanceDataManager(DtDe& de, DtInstanceContainer& container);

   private:
      DtIndirectPerInstanceDataManager() = delete;
      DtIndirectPerInstanceDataManager(const DtIndirectPerInstanceDataManager &other) = delete;
      DtIndirectPerInstanceDataManager &operator=(const DtIndirectPerInstanceDataManager &other) = delete; 
   public:
      virtual void setColor(int instanceId, const Vector4_32& color);
      virtual Vector4_32 color(int instanceId) const;

      virtual void setOtherData0(int instanceId, const Vector4_32& data);
      virtual Vector4_32 otherData0(int instanceId) const;

      virtual void setVisibility(int instanceId, bool visible);
      virtual bool visibility(int instanceId) const;

      virtual void setLightPointMaxDistance(int instanceId, float distance);
      virtual float lightPointMaxDistance(int instanceId) const;

      virtual void setLightPointGroupId(int instanceId, int lightPointGroupId);
      virtual int lightPointGroupId(int instanceId) const;

      virtual void addInstance(int perGeomInstanceIndex, int numPerGeomInstances, int instanceId, int lightPointGroupId, bool visible);

      virtual void removeInstance(int instanceId);

      virtual void processPerInstanceData(void);

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

      virtual const DtGlBuffer* gpuPerInstanceDataBuffer(void) const;

   protected:

      DtGlBuffer* myGpuPerInstanceDataBuffer;

      DtReusablePerInstanceData myVectorPerInstanceData;

      mutable bool myMutablePerInstanceDataDirty;

      typedef std::unordered_map<int, int> MapInstanceIdToOffsetInInstanceDataBuffer;
      MapInstanceIdToOffsetInInstanceDataBuffer myMapInstanceIdToOffsetInInstanceDataBuffer;
   };

}