DtEffectControllerShaderCache.h

Go to the documentation of this file.

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


#pragma once

#include <vrvOsg/vrvOsg.h>

#include <string>

#include <tbb/spin_rw_mutex.h>
#include <map>

namespace makVrv
{
   class DtOsgFileCache;
   class DtDe;

   class DT_DLL_VRVOSG DtEffectControllerShaderCache
   {
   public:
      DtEffectControllerShaderCache(DtDe& de, DtOsgFileCache& fileCache);

      virtual ~DtEffectControllerShaderCache();

   private:
      
      DtEffectControllerShaderCache();
      DtEffectControllerShaderCache(const DtEffectControllerShaderCache &other);
      DtEffectControllerShaderCache &operator=(const DtEffectControllerShaderCache &other);

   public:
      virtual const std::string& vertShaderSource(const std::string& vertShaderFile, int textureUnit) const;

      virtual const std::string& fragShaderSource(const std::string& fragShaderFile, bool forTexArray, const std::string& effectUnitName) const;
   protected:
      virtual std::string generateVertShaderSource(const std::string& vertShaderFile, int textureUnit) const;

      virtual std::string generateFragShaderSource(const std::string& fragShaderFile, bool forTexArray, const std::string& effectUnitName) const;
   protected:
      const DtOsgFileCache& myFileCache;

      typedef std::pair<std::string, int> VertShaderKey;
      typedef std::map<VertShaderKey, std::string> MapVertShaders;

      mutable MapVertShaders myMapVertShaders;

      typedef std::pair<std::string, bool> FragShaderKey;
      typedef std::map<FragShaderKey, std::string> MapFragShaders;
      mutable MapFragShaders myMapFragShaders;

      mutable  tbb::spin_rw_mutex myFragMapMutex;
      mutable  tbb::spin_rw_mutex myVertMapMutex;

      DtDe& myDe;
   };
}