Add cast by Tree and refactor

This commit is contained in:
Данил
2025-02-21 20:10:57 +07:00
parent a429424143
commit 5bfafba515
5 changed files with 92 additions and 39 deletions
-6
View File
@@ -1,6 +0,0 @@
#include "RTTI.h"
Classes IRTTI::RTTI_Kind() const
{
return type;
}
-32
View File
@@ -1,32 +0,0 @@
#pragma once
#include "RTTI_Meta.h"
// Interface RTTI
class IRTTI
{
protected:
Classes type;
virtual ~IRTTI() = default;
public:
Classes RTTI_Kind() const;
friend class RTTI;
};
// RTTI metods
class RTTI final
{
RTTI() = default;
public:
template<class Kind>
static bool IsA(Classes type)
{
return Meta<Kind>::TypeStr == type;
}
template<class To, class From>
static To* dyn_cast(From* object)
{
return IsA<To>(static_cast<IRTTI*>(object)->type) ? reinterpret_cast<To*>(object) : nullptr;
}
};
+34
View File
@@ -0,0 +1,34 @@
#include "RTTI.h"
void IRTTI::TreeArray::push_back(const Classes type)
{
size++;
Classes* NewTree = new Classes[size];
for (size_t i = 0; i < size - 1; ++i)
NewTree[i] = Tree[i];
NewTree[size - 1] = type;
}
IRTTI::TreeArray::TreeArray()
{
Tree = new Classes;
size = 1;
}
void IRTTI::SetType(const Classes type)
{
this->type = type;
//Tree.push_back(type);
}
IRTTI::IRTTI()
{
SetType(Classes::IRTTI);
}
Classes IRTTI::RTTI_Kind() const
{
return type;
}
+56
View File
@@ -0,0 +1,56 @@
#pragma once
#include "RTTI_Meta.h"
// Interface RTTI
GENERATE_META(IRTTI)
class IRTTI
{
struct TreeArray
{
size_t size;
Classes* Tree;
void push_back(const Classes type);
TreeArray();
};
TreeArray Tree;
Classes type;
protected:
void SetType(Classes type);
IRTTI();
~IRTTI() = default;
public:
Classes RTTI_Kind() const;
friend class RTTI;
};
// RTTI metods
class RTTI final
{
RTTI() = default;
public:
template<class Kind>
static bool IsA(Classes type)
{
return Meta<Kind>::Type == type;
}
template<class To, class From>
static To* dyn_cast(From* object)
{
IRTTI* InterfaceOfObject = static_cast<IRTTI*>(object);
if (IsA<To>(InterfaceOfObject->type))
return reinterpret_cast<To*>(object);
const size_t coutType = InterfaceOfObject->Tree.size;
for (size_t i = 0; i < coutType; ++i)
{
if (InterfaceOfObject->Tree.Tree[i] == Meta<To>::Type)
return reinterpret_cast<To*>(object);
}
return nullptr;
}
};
+2 -1
View File
@@ -4,7 +4,7 @@
#define GENERATE_META(Class) \ #define GENERATE_META(Class) \
class Class; \ class Class; \
template<> struct Meta <Class> { \ template<> struct Meta <Class> { \
static constexpr Classes TypeStr = Classes::##Class; \ static constexpr Classes Type = Classes::##Class; \
}; };
// Predeclarate struct for meta information // Predeclarate struct for meta information
@@ -13,4 +13,5 @@ template<class T> struct Meta;
// Enum type classes // Enum type classes
enum class Classes enum class Classes
{ {
IRTTI
}; };