DtKeyedData.h

Go to the documentation of this file.

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


#pragma once

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

#include <tbb/concurrent_hash_map.h>

#include <thread>

namespace makVrv
{
   class DtDe;

   class DtKeyedDataUserData;
   class DtTcsUserDataCreator;

   class DT_DLL_VRVCORE DtKeyedData
   {
   public:
      DtKeyedData(DtDe& de, const std::string& name);
      virtual ~DtKeyedData();
   private:
      DtKeyedData() = delete;
      DtKeyedData(const DtKeyedData& rhs) = delete;
      DtKeyedData& operator=(const DtKeyedData& rhs) = delete;

   public:

      struct UserDataCreationResult
      {
         DtKeyedDataUserData* myTcsUserData = nullptr;
         bool myCreated = false;
      };
   public:
      template<class ClassToCreate, class ...ConstructorArgs>
      UserDataCreationResult getOrCreateUserData(const void* userPointer, DtVirtualBaseClassCreator<ClassToCreate, ConstructorArgs...>* creator
         , ConstructorArgs&&... constructorArgs);

      virtual DtKeyedDataUserData* findUserData(const void* userPointer) const;

      virtual void deleteUserData(const void* userPointer);

   protected:
      DtDe& myDe;

      typedef tbb::concurrent_hash_map< const void*, DtKeyedDataUserData*> UserDataHashTable;
      UserDataHashTable myUserDataHashTable;

      const std::string myName;

      static std::atomic<int> theCount;
   };

   template<class ClassToCreate, class ...ConstructorArgs>
   DtKeyedData::UserDataCreationResult DtKeyedData::getOrCreateUserData(const void* userPointer, DtVirtualBaseClassCreator<ClassToCreate, ConstructorArgs...>* creator
      , ConstructorArgs&&... constructorArgs)
   {
      UserDataCreationResult result;
      UserDataHashTable::accessor accessor;
      const bool inserted = myUserDataHashTable.insert(accessor, userPointer);
      if (inserted)
      {
         auto& entry = accessor->second;
         entry = creator->create(std::forward<ConstructorArgs>(constructorArgs)...);
      }

      DtKeyedDataUserData* tcsUserData = accessor->second;
      if (tcsUserData == nullptr)
      {
         myDe.DT_LOG_WARN << "tcsUserData == nullptr" << std::endl << std::endl;
      }
      return { tcsUserData, inserted };
   }

   class DT_DLL_VRVCORE DtKeyedDataUserData
   {
   public:
      DtKeyedDataUserData(DtDe& de, const std::string& name);
      virtual ~DtKeyedDataUserData();
   private:
      DtKeyedDataUserData() = delete;
      DtKeyedDataUserData(const DtKeyedDataUserData& rhs) = delete;
      DtKeyedDataUserData& operator=(const DtKeyedDataUserData& rhs) = delete;
   protected:
      DtDe& myDe;

      const std::string myName;

      static std::atomic<int> theCount;
   };

}