Redstone1024/Redcraft
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(templates): fix TSharedFromThis not working for indirectly derived

Browse Source
This commit is contained in:
_Redstone_c_ 2023-01-15 18:14:09 +08:00
parent 78979b414a
commit 5c2c1b4a48
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
Redcraft.Utility/Source/Public/Templates/SharedPointer.h
View File

@ -521,15 +521,18 @@ public:
protected:
FORCEINLINE TSharedFromThis() = default;
FORCEINLINE TSharedFromThis(const TSharedFromThis&) = default;
FORCEINLINE TSharedFromThis(TSharedFromThis&&) = default;
FORCEINLINE TSharedFromThis& operator=(const TSharedFromThis&) = default;
FORCEINLINE TSharedFromThis& operator=(TSharedFromThis&&) = default;
FORCEINLINE ~TSharedFromThis() = default;
FORCEINLINE constexpr TSharedFromThis() : WeakThis() { }
FORCEINLINE TSharedFromThis(const TSharedFromThis&) : TSharedFromThis() { }
FORCEINLINE TSharedFromThis& operator=(const TSharedFromThis&) { return *this; }
FORCEINLINE ~TSharedFromThis() = default;
private:
using SharedFromThisType = TSharedFromThis;
// Here it is updated by the private constructor of TSharedRef or TSharedPtr.
mutable TWeakPtr<T> WeakThis;
@ -730,16 +733,18 @@ private:
: Pointer(InPtr), Controller(InController)
{
check(!((Pointer == nullptr) ^ (Controller == nullptr)));
if constexpr (CClass<T>)
if constexpr (CClass<T> && !CVolatile<T> && requires { typename T::SharedFromThisType; })
{
if constexpr (CDerivedFrom<T, TSharedFromThis<T>>)
using SharedFromThisType = T::SharedFromThisType;
if constexpr (CDerivedFrom<T, SharedFromThisType>)
{
if (Pointer != nullptr)
{
const TSharedFromThis<T>& SharedFromThis = *Pointer;
const SharedFromThisType& SharedFromThis = *Pointer;
checkf(!SharedFromThis.DoesSharedInstanceExist(), TEXT("This object is incorrectly managed by multiple TSharedRef or TSharedPtr."));
SharedFromThis.WeakThis = *this;
SharedFromThis.WeakThis = ConstCast<TRemoveCV<T>>(*this);
}
}
}
@ -1206,15 +1211,17 @@ private:
{
check(!((Pointer == nullptr) ^ (Controller == nullptr)));
if constexpr (CClass<T>)
if constexpr (CClass<T> && !CVolatile<T> && requires { typename T::SharedFromThisType; })
{
if constexpr (CDerivedFrom<T, TSharedFromThis<T>>)
using SharedFromThisType = T::SharedFromThisType;
if constexpr (CDerivedFrom<T, SharedFromThisType>)
{
if (Pointer != nullptr)
{
const TSharedFromThis<T>& SharedFromThis = *Pointer;
const SharedFromThisType& SharedFromThis = *Pointer;
checkf(!SharedFromThis.DoesSharedInstanceExist(), TEXT("This object is incorrectly managed by multiple TSharedRef or TSharedPtr."));
SharedFromThis.WeakThis = *this;
SharedFromThis.WeakThis = ConstCast<TRemoveCV<T>>(*this);
}
}
}