Factory.cpp 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. // SPDX-FileCopyrightText: 2021 Jorrit Rouwe
  2. // SPDX-License-Identifier: MIT
  3. #include <Jolt.h>
  4. #include <Core/Factory.h>
  5. namespace JPH {
  6. Factory Factory::sInstance;
  7. void *Factory::CreateObject(const char *inName)
  8. {
  9. const RTTI *ci = Find(inName);
  10. return ci != nullptr? ci->CreateObject() : nullptr;
  11. }
  12. const RTTI *Factory::Find(const char *inName)
  13. {
  14. ClassNameMap::iterator c = mClassNameMap.find(inName);
  15. return c != mClassNameMap.end()? c->second : nullptr;
  16. }
  17. const RTTI *Factory::Find(uint32 inHash)
  18. {
  19. ClassHashMap::iterator c = mClassHashMap.find(inHash);
  20. return c != mClassHashMap.end()? c->second : nullptr;
  21. }
  22. bool Factory::Register(const RTTI *inRTTI)
  23. {
  24. // Check if we already know the type
  25. if (Find(inRTTI->GetName()) != nullptr)
  26. return true;
  27. // Insert this class by name
  28. mClassNameMap.insert(ClassNameMap::value_type(inRTTI->GetName(), inRTTI));
  29. // Insert this class by hash
  30. if (!mClassHashMap.insert(ClassHashMap::value_type(inRTTI->GetHash(), inRTTI)).second)
  31. {
  32. JPH_ASSERT(false, "Hash collision registering type!");
  33. return false;
  34. }
  35. // Register base classes
  36. for (int i = 0; i < inRTTI->GetBaseClassCount(); ++i)
  37. if (!Register(inRTTI->GetBaseClass(i)))
  38. return false;
  39. // Register attribute classes
  40. for (int i = 0; i < inRTTI->GetAttributeCount(); ++i)
  41. {
  42. const RTTI *rtti = inRTTI->GetAttribute(i)->GetMemberPrimitiveType();
  43. if (rtti != nullptr)
  44. if (!Register(rtti))
  45. return false;
  46. }
  47. return true;
  48. }
  49. vector<const RTTI *> Factory::GetAllClasses()
  50. {
  51. vector<const RTTI *> all_classes;
  52. all_classes.reserve(mClassNameMap.size());
  53. for (ClassNameMap::iterator c = mClassNameMap.begin(); c != mClassNameMap.end(); ++c)
  54. all_classes.push_back(c->second);
  55. return all_classes;
  56. }
  57. } // JPH