ArchiveReplaceOrClearExternalReferences.h

Go to the documentation of this file.

// Copyright Epic Games, Inc. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Templates/Casts.h"
#include "UObject/Package.h"
#include "Serialization/ArchiveReplaceObjectRef.h"
 
/*----------------------------------------------------------------------------
FArchiveReplaceOrClearExternalReferences.
----------------------------------------------------------------------------*/
template< class T >
class FArchiveReplaceOrClearExternalReferences : public FArchiveReplaceObjectRef<T>
{
    typedef FArchiveReplaceObjectRef<T> TSuper;
public:
    FArchiveReplaceOrClearExternalReferences
        ( UObject* InSearchObject
        , const TMap<T*, T*>& InReplacementMap
        , UPackage* InDestPackage
        , EArchiveReplaceObjectFlags Flags = EArchiveReplaceObjectFlags::None)
        : TSuper(InSearchObject, InReplacementMap, EArchiveReplaceObjectFlags::DelayStart | Flags)
        , DestPackage(InDestPackage)
    {
        if (!(Flags & EArchiveReplaceObjectFlags::DelayStart))
        {
            this->SerializeSearchObject();
        }
    }
 
    UE_DEPRECATED(5.0, "Use version that passes Flags instead of just bDelayStart")
    FArchiveReplaceOrClearExternalReferences
        ( UObject* InSearchObject
        , const TMap<T*, T*>& InReplacementMap
        , UPackage* InDestPackage
        , bool bDelayStart)
        : TSuper(InSearchObject, InReplacementMap, EArchiveReplaceObjectFlags::DelayStart)
        , DestPackage(InDestPackage)
    {
        if (!bDelayStart)
        {
            this->SerializeSearchObject();
        }
    }
 
    FArchive& operator<<(UObject*& Obj)
    {
        UObject* Resolved = Obj;
        TSuper::operator<<(Resolved);
        // if Resolved is a private object in another package just clear the reference:
        if (Resolved)
        {
            UPackage* ObjPackage = Resolved->GetPackage();
            if (ObjPackage)
            {
                CA_SUPPRESS(6011);
                if (ObjPackage != Obj && 
                    DestPackage != ObjPackage && 
                    Obj->GetOutermostObject() != TSuper::GetSearchObject()->GetOutermostObject() &&
                    !Obj->HasAnyFlags(RF_Public))
                {
                    Resolved = nullptr;
                }
            }
        }
        Obj = Resolved;
        return *this;
    }
 
protected:
    UPackage* DestPackage;
};