DtVirtualProgramCache.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2021 VT-MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <vrvCore/DtVirtualBaseClassCreator.h>

#include <osg/ref_ptr>

#include <string>
#include <map>

namespace osg
{
   class StateSet;
}
namespace osgEarth
{
   class VirtualProgram;
}

namespace makVrv
{
   class DtDe;

   class DtVirtualProgramCache;

   typedef DtVirtualBaseClassCreator<DtVirtualProgramCache> DtVirtualProgramCacheCreator;

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

   class DT_DLL_VRVOSG DtVirtualProgramCache
   {
   public:
      friend DtVirtualProgramCacheCreator;
      
      virtual ~DtVirtualProgramCache();

   protected:
      DtVirtualProgramCache(DtDe& de);

   private:
      DtVirtualProgramCache() = delete;

      DtVirtualProgramCache(const DtVirtualProgramCache& rhs) = delete;

      DtVirtualProgramCache &operator=(const DtVirtualProgramCache& rhs) = delete;

   public:
      virtual bool findAndSetOn(const std::string& strHash, osg::StateSet* stateSet);

      virtual void cache(const std::string& strHash, osgEarth::VirtualProgram* virtualProgram);

      virtual void clear();

      virtual void addRef(void);

      virtual void removeRef(void);

      virtual int refCount(void) const;
      
   protected:
      typedef osg::ref_ptr<osgEarth::VirtualProgram> VirtualProgramRef;
      typedef std::map<std::string, VirtualProgramRef> StringVpMap;

      StringVpMap myMapHashesToVirtualPrograms;

      DtDe& myDe;

      tbb::atomic<int> myRefCount;
      mutable tbb::mutex myMapMutex;
   };

}