// Copyright Epic Games, Inc. All Rights Reserved.

#include "CoreMinimal.h"
#include "Serialization/ArchiveUObject.h"
#include "UObject/LazyObjectPtr.h"
#include "UObject/SoftObjectPath.h"
#include "UObject/PropertyPortFlags.h"
#include "UObject/UnrealType.h"
#include "Serialization/DuplicatedObject.h"
#include "Serialization/DuplicatedDataReader.h"
#include "UObject/UObjectThreadContext.h"
#include "Internationalization/TextPackageNamespaceUtil.h"

/*----------------------------------------------------------------------------
        FDuplicateDataReader.
----------------------------------------------------------------------------*/
/**
 * Constructor
 *
 * @param	InDuplicatedObjects		map of original object to copy of that object
 * @param	InObjectData			object data to read from
 */
FDuplicateDataReader::FDuplicateDataReader(class FUObjectAnnotationSparse<FDuplicatedObject, false>& InDuplicatedObjects, const FLargeMemoryData& InObjectData, uint32 InPortFlags, UObject* InDestOuter)
    : DuplicatedObjectAnnotation(InDuplicatedObjects), ObjectData(InObjectData), Offset(0)
{
    this->SetIsLoading(true);
    this->SetIsPersistent(true);
    this->ArNoIntraPropertyDelta = true;
    ArPortFlags |= PPF_Duplicate | InPortFlags;

#if USE_STABLE_LOCALIZATION_KEYS
    if (GIsEditor && !(ArPortFlags & (PPF_DuplicateVerbatim | PPF_DuplicateForPIE)))
    {
        SetLocalizationNamespace(TextNamespaceUtil::EnsurePackageNamespace(InDestOuter));
    }
#endif  // USE_STABLE_LOCALIZATION_KEYS
}

void FDuplicateDataReader::SerializeFail()
{
    FUObjectSerializeContext* LoadContext = GetSerializeContext();
    check(LoadContext);
    UE_LOG(LogObj, Fatal, TEXT("FDuplicateDataReader Overread. SerializedObject = %s SerializedProperty = %s"), *GetFullNameSafe(LoadContext->SerializedObject), *GetFullNameSafe(GetSerializedProperty()));
}

FArchive& FDuplicateDataReader::operator<<(FName& N)
{
    FNameEntryId ComparisonIndex;
    FNameEntryId DisplayIndex;
    int32 Number;
    ByteOrderSerialize(&ComparisonIndex, sizeof(ComparisonIndex));
    ByteOrderSerialize(&DisplayIndex, sizeof(DisplayIndex));
    ByteOrderSerialize(&Number, sizeof(Number));
    // copy over the name with a name made from the name index and number
    N = FName(ComparisonIndex, DisplayIndex, Number);
    return *this;
}

FArchive& FDuplicateDataReader::operator<<(UObject*& Object)
{
    UObject* SourceObject = Object;
    Serialize(&SourceObject, sizeof(UObject*));

    FDuplicatedObject ObjectInfo = SourceObject ? DuplicatedObjectAnnotation.GetAnnotation(SourceObject) : FDuplicatedObject();
    if (!ObjectInfo.IsDefault())
    {
        Object = ObjectInfo.DuplicatedObject;
    }
    else
    {
        Object = SourceObject;
    }

    return *this;
}

FArchive& FDuplicateDataReader::operator<<(FLazyObjectPtr& LazyObjectPtr)
{
    FArchive& Ar = *this;
    FUniqueObjectGuid ID;
    Ar << ID;

    if (Ar.GetPortFlags() & PPF_DuplicateForPIE)
    {
        // Remap unique ID if necessary
        ID = ID.FixupForPIE();
    }
    LazyObjectPtr = ID;
    return Ar;
}

FArchive& FDuplicateDataReader::operator<<(FSoftObjectPath& SoftObjectPath)
{
    FArchiveUObject::operator<<(SoftObjectPath);

    // Remap soft reference if necessary
    UObject* SourceObject        = SoftObjectPath.ResolveObject();
    FDuplicatedObject ObjectInfo = SourceObject ? DuplicatedObjectAnnotation.GetAnnotation(SourceObject) : FDuplicatedObject();
    if (!ObjectInfo.IsDefault())
    {
        SoftObjectPath = FSoftObjectPath::GetOrCreateIDForObject(ObjectInfo.DuplicatedObject);
    }

    return *this;
}

// FArchive& FDuplicateDataReader::operator<<(FField*& Field)
//{
//
// }