77 lines
3.7 KiB
C++
77 lines
3.7 KiB
C++
|
// Copyright Epic Games, Inc. All Rights Reserved.
|
||
|
|
|
||
|
|
/*==================================================================================================
|
||
|
|
FixupNeedsLoadForEditorGameCommandlet.cpp: Fixes outdated NeedsLoadForEditorGame flags on exports
|
||
|
|
====================================================================================================*/
|
||
|
|
|
||
|
|
#include "Commandlets/FixupNeedsLoadForEditorGameCommandlet.h"
|
||
|
|
#include "UObject/Class.h"
|
||
|
|
#include "UObject/Package.h"
|
||
|
|
#include "UObject/LinkerLoad.h"
|
||
|
|
|
||
|
|
int32 UFixupNeedsLoadForEditorGameCommandlet::InitializeResaveParameters(const TArray<FString>& Tokens, TArray<FString>& MapPathNames)
|
||
|
|
{
|
||
|
|
int32 Result = Super::InitializeResaveParameters(Tokens, MapPathNames);
|
||
|
|
// We need ResaveClasses to be specified, otherwise we won't know what to update
|
||
|
|
if (Result == 0 && !ResaveClasses.Num())
|
||
|
|
{
|
||
|
|
UE_LOG(LogContentCommandlet, Error, TEXT("FixupNeedsLoadForEditorGame commandlet requires at least one resave class name. Use -RESAVECLASS=ClassA,ClassB,ClassC to specify resave classes."));
|
||
|
|
Result = 1;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
for (FName& ClassName: ResaveClasses)
|
||
|
|
{
|
||
|
|
if (!ResaveClassNeedsLoadForEditorGameValues.Contains(ClassName))
|
||
|
|
{
|
||
|
|
UClass* ResaveClass = FindObject<UClass>(ANY_PACKAGE, *ClassName.ToString());
|
||
|
|
if (ResaveClass)
|
||
|
|
{
|
||
|
|
UObject* DefaultObject = ResaveClass->GetDefaultObject();
|
||
|
|
ResaveClassNeedsLoadForEditorGameValues.Add(ClassName, DefaultObject->NeedsLoadForEditorGame());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if (Verbosity != UResavePackagesCommandlet::ONLY_ERRORS)
|
||
|
|
{
|
||
|
|
UE_LOG(LogContentCommandlet, Warning, TEXT("Resave Class \"%s\" could not be found. Make sure the class name is valid and that it's a native class."), *ClassName.ToString());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (ResaveClassNeedsLoadForEditorGameValues.Num() == 0)
|
||
|
|
{
|
||
|
|
UE_LOG(LogContentCommandlet, Error, TEXT("Got %d classes to resave but none of the exist."), ResaveClasses.Num());
|
||
|
|
Result = 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return Result;
|
||
|
|
}
|
||
|
|
|
||
|
|
void UFixupNeedsLoadForEditorGameCommandlet::PerformPreloadOperations(FLinkerLoad* PackageLinker, bool& bSavePackage)
|
||
|
|
{
|
||
|
|
Super::PerformPreloadOperations(PackageLinker, bSavePackage);
|
||
|
|
if (bSavePackage)
|
||
|
|
{
|
||
|
|
// The package contains a class we want to check but we don't want to save it unless
|
||
|
|
// any of the exports has an outdated (compared to a CDO) NeedsLoadForEditorGame flag.
|
||
|
|
bSavePackage = false;
|
||
|
|
for (int32 ExportIndex = 0; ExportIndex < PackageLinker->ExportMap.Num(); ExportIndex++)
|
||
|
|
{
|
||
|
|
FName ExportClassName = PackageLinker->GetExportClassName(ExportIndex);
|
||
|
|
bool* bNeedsLoadForEditorGameValuePtr = ResaveClassNeedsLoadForEditorGameValues.Find(ExportClassName);
|
||
|
|
if (bNeedsLoadForEditorGameValuePtr)
|
||
|
|
{
|
||
|
|
FObjectExport& Export = PackageLinker->ExportMap[ExportIndex];
|
||
|
|
// The condition below may seem confusing because bNotAlwaysLoadedForEditorGame = !bNeedsLoadForEditorGame
|
||
|
|
// At the moment we only update assets that had bNotAlwaysLoadedForEditorGame set to true but NeedsLoadForEditorGame returns true
|
||
|
|
// and either bNotForClient or bNotForServer is true.
|
||
|
|
if (Export.bNotAlwaysLoadedForEditorGame &&
|
||
|
|
Export.bNotAlwaysLoadedForEditorGame == *bNeedsLoadForEditorGameValuePtr &&
|
||
|
|
(Export.bNotForClient || Export.bNotForServer))
|
||
|
|
{
|
||
|
|
bSavePackage = true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|