You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
867 lines
29 KiB
867 lines
29 KiB
// Copyright Epic Games, Inc. All Rights Reserved. |
|
|
|
using System; |
|
using System.Collections.Concurrent; |
|
using EpicGames.Core; |
|
using EpicGames.UHT.Types; |
|
using EpicGames.UHT.Utils; |
|
using System.Collections.Generic; |
|
using System.IO; |
|
using System.Linq; |
|
using System.Text; |
|
using System.Threading.Tasks; |
|
using EpicGames.UHT.Tables; |
|
using UnrealBuildTool; |
|
|
|
namespace SourceCodeGeneratorUbtPlugin |
|
{ |
|
[UnrealHeaderTool] |
|
internal class SourceCodeGenerator |
|
{ |
|
[UhtExporter(Name = "UnrealCSharp", Description = "UnrealCSharp Source Code Generator", |
|
Options = UhtExporterOptions.Default, ModuleName = "UnrealCSharp")] |
|
private static void SourceCodeGeneratorExporter(IUhtExportFactory factory) |
|
{ |
|
var SettingFilePath = Path.Combine(new string(factory.Session.ProjectDirectory), |
|
"Config", |
|
"DefaultUnrealCSharpEditorSetting.ini"); |
|
|
|
if (File.Exists(SettingFilePath)) |
|
{ |
|
var SettingConfigFile = new ConfigFile(new FileReference(SettingFilePath)); |
|
|
|
if (SettingConfigFile.TryGetSection("/Script/UnrealCSharpCore.UnrealCSharpEditorSetting", |
|
out var SettingConfigSection)) |
|
{ |
|
var SettingConfigHierarchySection = new ConfigHierarchySection( |
|
new List<ConfigFileSection> { SettingConfigSection }); |
|
|
|
if (SettingConfigHierarchySection.TryGetValue("bEnableExport", |
|
out var Value)) |
|
{ |
|
if (bool.Parse(Value.ToLower())) |
|
{ |
|
new SourceCodeGenerator(factory).Generate(); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
private readonly IUhtExportFactory Factory; |
|
|
|
private UhtSession Session => Factory.Session; |
|
|
|
private ConcurrentQueue<UhtClass> ExportClasses = new(); |
|
|
|
private HashSet<string> Project = new(); |
|
|
|
private Dictionary<string, string> HeaderPath = new(); |
|
|
|
private const string BindingSuffix = ".binding.inl"; |
|
|
|
private const string HeaderSuffix = ".header.inl"; |
|
|
|
private readonly HashSet<string> ClassNameBlacklist = new(); |
|
|
|
public SourceCodeGenerator(IUhtExportFactory factory) |
|
{ |
|
Factory = factory; |
|
} |
|
|
|
/// <summary> |
|
/// Export all the classes in all the packages |
|
/// </summary> |
|
private void Generate() |
|
{ |
|
if (Session is { EngineDirectory: not null, ProjectDirectory: not null }) |
|
{ |
|
GetModules(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Source/")), Project); |
|
|
|
GetModules(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Source/")), HeaderPath); |
|
|
|
GetModules(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Plugins/")), HeaderPath); |
|
|
|
GetPlugins(Path.GetFullPath(Path.Combine(Session.ProjectDirectory, "Plugins/")), HeaderPath); |
|
|
|
GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Developer/")), HeaderPath); |
|
|
|
GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Editor/")), HeaderPath); |
|
|
|
GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Programs/")), HeaderPath); |
|
|
|
GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Source/Runtime/")), HeaderPath); |
|
|
|
GetModules(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Plugins/")), HeaderPath); |
|
|
|
GetPlugins(Path.GetFullPath(Path.Combine(Session.EngineDirectory, "Plugins/")), HeaderPath); |
|
} |
|
|
|
var SettingFilePath = Path.Combine(new string(Session.ProjectDirectory), |
|
"Config", |
|
"DefaultUnrealCSharpEditorSetting.ini"); |
|
|
|
var ExportModule = new List<string>(); |
|
|
|
if (File.Exists(SettingFilePath)) |
|
{ |
|
var SettingConfigFile = new ConfigFile(new FileReference(SettingFilePath)); |
|
|
|
if (SettingConfigFile.TryGetSection("/Script/UnrealCSharpCore.UnrealCSharpEditorSetting", |
|
out var SettingConfigSection)) |
|
{ |
|
var SettingConfigHierarchySection = new ConfigHierarchySection( |
|
new List<ConfigFileSection> { SettingConfigSection }); |
|
|
|
if (SettingConfigHierarchySection.TryGetValues("ExportModule", |
|
out var Values)) |
|
{ |
|
foreach (var value in Values) |
|
{ |
|
ExportModule.Add(value); |
|
} |
|
} |
|
|
|
if (SettingConfigHierarchySection.TryGetValues("ClassBlacklist", out var blacklistValues)) |
|
{ |
|
foreach (var className in blacklistValues) |
|
{ |
|
if (!string.IsNullOrEmpty(className)) |
|
{ |
|
ClassNameBlacklist.Add(className); |
|
Console.WriteLine($"[BlackList] {className} will be skipped."); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
List<Task?> tasks = new(); |
|
|
|
foreach (var module in Session.Modules) |
|
{ |
|
if (module.Module.ModuleType != UHTModuleType.EngineRuntime && |
|
module.Module.ModuleType != UHTModuleType.GameRuntime) |
|
{ |
|
continue; |
|
} |
|
|
|
if (ExportModule.All(x => |
|
string.Compare(x, module.Module.Name, StringComparison.OrdinalIgnoreCase) != 0)) |
|
{ |
|
continue; |
|
} |
|
|
|
QueueClassExports(module.ScriptPackage, tasks); |
|
} |
|
|
|
var waitTasks = tasks.Where(Task => Task != null).Cast<Task>().ToArray(); |
|
|
|
if (waitTasks.Length > 0) |
|
{ |
|
Task.WaitAll(waitTasks); |
|
} |
|
|
|
Finish(); |
|
} |
|
|
|
/// <summary> |
|
/// Collect the classes to be exported for the given package and type |
|
/// </summary> |
|
/// <param name="type">Type to test for exporting</param> |
|
/// <param name="tasks">Collection of queued tasks</param> |
|
private void QueueClassExports(UhtType type, List<Task?> tasks) |
|
{ |
|
if (type is UhtClass classObj) |
|
{ |
|
if (CanExportClass(classObj)) |
|
{ |
|
tasks.Add(Factory.CreateTask(_ => { ExportClass(classObj); })); |
|
} |
|
} |
|
|
|
foreach (var child in type.Children) |
|
{ |
|
QueueClassExports(child, tasks); |
|
} |
|
} |
|
|
|
/// <summary> |
|
/// Test to see if the given class should be exported |
|
/// </summary> |
|
/// <param name="classObj">Class to test</param> |
|
/// <returns>True if the class should be exported, false if not</returns> |
|
protected virtual bool CanExportClass(UhtClass classObj) |
|
{ |
|
if (ClassNameBlacklist.Contains(classObj.SourceName)) |
|
{ |
|
return false; |
|
} |
|
|
|
return !classObj.ClassFlags.HasAnyFlags(EClassFlags.Interface) && |
|
IsClassTypeSupported(classObj); |
|
} |
|
|
|
/// <summary> |
|
/// Test to see if the given function should be exported |
|
/// </summary> |
|
/// <param name="function">Function to test</param> |
|
/// <returns>True if the function should be exported</returns> |
|
protected virtual bool CanExportFunction(UhtFunction function) |
|
{ |
|
if (function.Outer is UhtClass classObj) |
|
{ |
|
if (!classObj.ClassFlags.HasAnyFlags(EClassFlags.RequiredAPI) && |
|
!function.FunctionExportFlags.HasAnyFlags(UhtFunctionExportFlags.RequiredAPI)) |
|
{ |
|
return false; |
|
} |
|
} |
|
else |
|
{ |
|
return false; |
|
} |
|
|
|
if (function.FunctionType != UhtFunctionType.Function) |
|
{ |
|
return false; |
|
} |
|
|
|
if (!function.FunctionFlags.HasAnyFlags(EFunctionFlags.Public)) |
|
{ |
|
return false; |
|
} |
|
|
|
if (function.Deprecated) |
|
{ |
|
return false; |
|
} |
|
|
|
if (function.FunctionFlags.HasAnyFlags(EFunctionFlags.EditorOnly)) |
|
{ |
|
return false; |
|
} |
|
|
|
if (function.FunctionExportFlags.HasAnyFlags(UhtFunctionExportFlags.CustomThunk)) |
|
{ |
|
return false; |
|
} |
|
|
|
if (function is { HasReturnProperty: true, ReturnProperty: UhtMapProperty }) |
|
{ |
|
return false; |
|
} |
|
|
|
foreach (var child in function.Children) |
|
{ |
|
if (child is UhtProperty property && !IsPropertyTypeSupported(property)) |
|
{ |
|
return false; |
|
} |
|
} |
|
|
|
return true; |
|
} |
|
|
|
/// <summary> |
|
/// Test to see if the given property should be exported |
|
/// </summary> |
|
/// <param name="property">Property to test</param> |
|
/// <returns>True if the property should be exported</returns> |
|
protected virtual bool CanExportProperty(UhtProperty property) |
|
{ |
|
return !property.Deprecated && |
|
!property.PropertyFlags.HasAnyFlags(EPropertyFlags.EditorOnly) && |
|
property.PropertyFlags.HasAnyFlags(EPropertyFlags.NativeAccessSpecifierPublic) && |
|
IsPropertyTypeSupported(property); |
|
} |
|
|
|
private static bool IsClassTypeSupported(UhtClass classObj) |
|
{ |
|
return !classObj.Deprecated && |
|
(classObj == classObj.Session.UObject || |
|
classObj == classObj.Session.UClass || |
|
classObj.ClassFlags.HasAnyFlags(EClassFlags.RequiredAPI | EClassFlags.MinimalAPI)); |
|
} |
|
|
|
private static bool IsPropertyTypeSupported(UhtProperty property) |
|
{ |
|
if (property is UhtObjectProperty objectProperty) |
|
{ |
|
if (!IsClassTypeSupported(objectProperty.Class)) |
|
{ |
|
return false; |
|
} |
|
} |
|
|
|
if (property is UhtInterfaceProperty interfaceProperty) |
|
{ |
|
if (interfaceProperty.InterfaceClass == interfaceProperty.Session.IInterface) |
|
{ |
|
return false; |
|
} |
|
} |
|
|
|
if (property is UhtDelegateProperty or UhtMulticastDelegateProperty or UhtFieldPathProperty) |
|
{ |
|
return false; |
|
} |
|
|
|
if (property is UhtStructProperty { ScriptStruct.HasNoOpConstructor: false }) |
|
{ |
|
return false; |
|
} |
|
|
|
if (property is UhtArrayProperty arrayProperty) |
|
{ |
|
return IsPropertyTypeSupported(arrayProperty.ValueProperty); |
|
} |
|
|
|
if (property is UhtSetProperty setProperty) |
|
{ |
|
return IsPropertyTypeSupported(setProperty.ValueProperty); |
|
} |
|
|
|
if (property is UhtMapProperty mapProperty) |
|
{ |
|
return IsPropertyTypeSupported(mapProperty.KeyProperty) && |
|
IsPropertyTypeSupported(mapProperty.ValueProperty); |
|
} |
|
|
|
return property is { IsBitfield: false, IsStaticArray: false }; |
|
} |
|
|
|
/// <summary> |
|
/// Export the given class |
|
/// </summary> |
|
/// <param name="classObj">Class to export</param> |
|
private void ExportClass(UhtClass classObj) |
|
{ |
|
using BorrowStringBuilder borrower = new(StringBuilderCache.Big); |
|
|
|
if (ExportClass(borrower.StringBuilder, classObj)) |
|
{ |
|
var fileName = Factory.MakePath(classObj.EngineName, BindingSuffix); |
|
|
|
SaveIfChanged(fileName, borrower.StringBuilder); |
|
|
|
ExportClasses.Enqueue(classObj); |
|
} |
|
} |
|
|
|
private void Finish() |
|
{ |
|
var packages = new Dictionary<UhtPackage, List<string>>(); |
|
|
|
foreach (var ExportClass in ExportClasses) |
|
{ |
|
if (!packages.ContainsKey(ExportClass.Package)) |
|
{ |
|
packages.Add(ExportClass.Package, new List<string>()); |
|
} |
|
|
|
packages[ExportClass.Package].AddUnique(ExportClass.EngineName); |
|
} |
|
|
|
foreach (var package in packages) |
|
{ |
|
package.Value.Sort(); |
|
|
|
var borrow = new BorrowStringBuilder(StringBuilderCache.Big); |
|
|
|
var builder = borrow.StringBuilder; |
|
|
|
builder.Append("#pragma once\r\n\r\n"); |
|
|
|
foreach (var value in package.Value) |
|
{ |
|
builder.Append(GenerateInclude($"{value}{BindingSuffix}")); |
|
} |
|
|
|
var fileName = Factory.MakePath(package.Key.Module, HeaderSuffix); |
|
|
|
SaveIfChanged(fileName, builder); |
|
} |
|
} |
|
|
|
private bool ExportClass(StringBuilder builder, UhtClass classObj) |
|
{ |
|
builder.Append("#pragma once\r\n\r\n" + |
|
"PRAGMA_DISABLE_DEPRECATION_WARNINGS\r\n\r\n"); |
|
|
|
var DependencyClasses = new HashSet<UhtClass> { classObj }; |
|
|
|
using BorrowStringBuilder bodyBorrower = new(StringBuilderCache.Big); |
|
|
|
var bodyBuilder = bodyBorrower.StringBuilder; |
|
|
|
bodyBuilder.AppendFormat("namespace\r\n" + |
|
"{{\r\n" + |
|
"\tstruct FRegister{0}\r\n" + |
|
"\t{{\r\n" + |
|
"\t\tFRegister{1}()\r\n" + |
|
"\t\t{{\r\n" + |
|
"\t\t\tTBindingClassBuilder<{2}>(NAMESPACE_BINDING)\r\n", |
|
classObj.EngineName, |
|
classObj.EngineName, |
|
classObj.SourceName); |
|
|
|
var bExportProperty = false; |
|
|
|
foreach (var child in classObj.Children) |
|
{ |
|
if (child is UhtProperty property && CanExportProperty(property)) |
|
{ |
|
GetDependencyClasses(property, DependencyClasses); |
|
|
|
ExportProperty(bodyBuilder, classObj, property); |
|
|
|
bExportProperty = true; |
|
} |
|
} |
|
|
|
var bExportFunction = false; |
|
|
|
foreach (var function in classObj.Functions.Reverse()) |
|
{ |
|
if (CanExportFunction(function)) |
|
{ |
|
foreach (var child in function.Children) |
|
{ |
|
if (child is UhtProperty property) |
|
{ |
|
GetDependencyClasses(property, DependencyClasses); |
|
} |
|
} |
|
|
|
ExportFunction(bodyBuilder, classObj, function); |
|
|
|
bExportFunction = true; |
|
} |
|
} |
|
|
|
foreach (var DependencyClass in DependencyClasses) |
|
{ |
|
builder.Append(GetInclude(DependencyClass)); |
|
} |
|
|
|
builder.Append("\r\n"); |
|
|
|
bodyBuilder.Remove(bodyBuilder.Length - 2, 2); |
|
|
|
bodyBuilder.AppendFormat(";\r\n" + |
|
"\t\t}}\r\n" + |
|
"\t}};\r\n" + |
|
"\r\n" + |
|
"\t[[maybe_unused]] FRegister{0} Register{1};\r\n" + |
|
"}}\r\n\r\n", |
|
classObj.EngineName, |
|
classObj.EngineName); |
|
|
|
builder.Append(bodyBuilder); |
|
|
|
builder.Append("PRAGMA_ENABLE_DEPRECATION_WARNINGS\r\n"); |
|
|
|
return bExportProperty || bExportFunction; |
|
} |
|
|
|
private static void GetDependencyClasses(UhtProperty property, HashSet<UhtClass> DependencyClasses) |
|
{ |
|
if (property is UhtClassProperty classProperty) |
|
{ |
|
if (classProperty.MetaClass != null) |
|
{ |
|
DependencyClasses.Add(classProperty.MetaClass); |
|
} |
|
} |
|
else if (property is UhtObjectProperty objectProperty) |
|
{ |
|
DependencyClasses.Add(objectProperty.Class); |
|
} |
|
else if (property is UhtSoftClassProperty softClassProperty) |
|
{ |
|
if (softClassProperty.MetaClass != null) |
|
{ |
|
DependencyClasses.Add(softClassProperty.MetaClass); |
|
} |
|
} |
|
else if (property is UhtSoftObjectProperty softObjectProperty) |
|
{ |
|
DependencyClasses.Add(softObjectProperty.Class); |
|
} |
|
else if (property is UhtArrayProperty arrayProperty) |
|
{ |
|
GetDependencyClasses(arrayProperty.ValueProperty, DependencyClasses); |
|
} |
|
else if (property is UhtSetProperty setProperty) |
|
{ |
|
GetDependencyClasses(setProperty.ValueProperty, DependencyClasses); |
|
} |
|
else if (property is UhtMapProperty mapProperty) |
|
{ |
|
GetDependencyClasses(mapProperty.KeyProperty, DependencyClasses); |
|
|
|
GetDependencyClasses(mapProperty.ValueProperty, DependencyClasses); |
|
} |
|
} |
|
|
|
private static void ExportFunction(StringBuilder builder, UhtClass classObj, UhtFunction function) |
|
{ |
|
builder.Append( |
|
$"\t\t\t\t.Function(\"{function.SourceName}\", BINDING_OVERLOAD({GetFunctionSignature(classObj, function)}, &{classObj.SourceName}::{function.SourceName}"); |
|
|
|
if (function.HasParameters) |
|
{ |
|
builder.Append($", {GetFunctionParamName(function)}"); |
|
} |
|
|
|
if (!function.FunctionFlags.HasAnyFlags(EFunctionFlags.Static)) |
|
{ |
|
builder.Append(", EFunctionInteract::Virtual"); |
|
} |
|
|
|
if (function.HasParameters) |
|
{ |
|
builder.Append(GetFunctionDefaultValue(function)); |
|
} |
|
|
|
builder.Append("))\r\n"); |
|
} |
|
|
|
private static void ExportProperty(StringBuilder builder, UhtClass classObj, UhtProperty property) |
|
{ |
|
builder.Append( |
|
$"\t\t\t\t.Property(\"{property.SourceName}\", BINDING_PROPERTY(&{classObj.SourceName}::{property.SourceName}))\r\n"); |
|
} |
|
|
|
private static string GetParamPropertySignature(UhtProperty? property) |
|
{ |
|
if (property == null) |
|
{ |
|
return "void"; |
|
} |
|
|
|
var builder = new StringBuilder(); |
|
|
|
var caps = property.PropertyCaps; |
|
|
|
var isInterfaceProp = property is UhtInterfaceProperty; |
|
|
|
var passCppArgsByRef = caps.HasAnyFlags(UhtPropertyCaps.PassCppArgsByRef); |
|
|
|
var isConstParam = property.PropertyFlags.HasAnyFlags(EPropertyFlags.ConstParm) || |
|
(isInterfaceProp && !property.PropertyFlags.HasAnyFlags(EPropertyFlags.OutParm)); |
|
|
|
var isConstArgsByRef = property.ArrayDimensions == null && passCppArgsByRef && |
|
!property.PropertyFlags.HasAnyFlags( |
|
EPropertyFlags.ConstParm | EPropertyFlags.OutParm); |
|
|
|
var isOnConstClass = false; |
|
|
|
if (property is UhtObjectProperty objectProperty) |
|
{ |
|
isOnConstClass = objectProperty.Class.ClassFlags.HasAnyFlags(EClassFlags.Const) || |
|
objectProperty.MetaData.TryGetValue(UhtNames.NativeConst, out _); |
|
} |
|
|
|
var shouldHaveRef = |
|
property.PropertyFlags.HasAnyFlags(EPropertyFlags.OutParm | EPropertyFlags.ReferenceParm); |
|
|
|
var constAtTheBeginning = isOnConstClass || isConstArgsByRef || (isConstParam && !shouldHaveRef); |
|
|
|
if (constAtTheBeginning) |
|
{ |
|
builder.Append("const "); |
|
} |
|
|
|
if (property is UhtEnumProperty enumProperty) |
|
{ |
|
builder.Append(enumProperty.Enum.CppType); |
|
} |
|
else |
|
{ |
|
property.AppendText(builder, UhtPropertyTextType.ExportMember); |
|
} |
|
|
|
var fromConstClass = false; |
|
|
|
if (property.Outer is UhtClass outerClass) |
|
{ |
|
fromConstClass = outerClass.ClassFlags.HasAnyFlags(EClassFlags.Const); |
|
} |
|
|
|
var constAtTheEnd = (fromConstClass || (isConstParam && shouldHaveRef)) && !constAtTheBeginning; |
|
|
|
if (constAtTheEnd) |
|
{ |
|
builder.Append(" const"); |
|
} |
|
|
|
shouldHaveRef = passCppArgsByRef; |
|
|
|
if (shouldHaveRef || |
|
property.ArrayDimensions == null && |
|
(passCppArgsByRef || |
|
property.PropertyFlags.HasAnyFlags(EPropertyFlags.OutParm | EPropertyFlags.ReferenceParm))) |
|
{ |
|
if (property is UhtArrayProperty or UhtStrProperty or UhtTextProperty or UhtInterfaceProperty |
|
or UhtSoftObjectProperty) |
|
{ |
|
if (property.RefQualifier != UhtPropertyRefQualifier.None) |
|
{ |
|
builder.Append('&'); |
|
} |
|
} |
|
else |
|
{ |
|
builder.Append('&'); |
|
} |
|
} |
|
|
|
return builder.ToString(); |
|
} |
|
|
|
private static string GetReturnPropertySignature(UhtProperty? property) |
|
{ |
|
if (property == null) |
|
{ |
|
return "void"; |
|
} |
|
|
|
var builder = new StringBuilder(); |
|
|
|
if (property.PropertyFlags.HasAnyFlags(EPropertyFlags.ConstParm) || |
|
property.RefQualifier == UhtPropertyRefQualifier.ConstRef) |
|
{ |
|
builder.Append("const "); |
|
} |
|
|
|
if (property is UhtEnumProperty enumProperty) |
|
{ |
|
builder.Append(enumProperty.Enum.CppType); |
|
} |
|
else |
|
{ |
|
property.AppendText(builder, UhtPropertyTextType.ExportMember); |
|
} |
|
|
|
if (property.RefQualifier == UhtPropertyRefQualifier.ConstRef) |
|
{ |
|
builder.Append('&'); |
|
} |
|
|
|
return builder.ToString(); |
|
} |
|
|
|
private static string GetFunctionSignature(UhtClass classObj, UhtFunction function) |
|
{ |
|
var builder = new StringBuilder(); |
|
|
|
builder.Append(GetReturnPropertySignature(function.ReturnProperty)); |
|
|
|
if (function.FunctionFlags.HasAnyFlags(EFunctionFlags.Static)) |
|
{ |
|
builder.Append("(*)"); |
|
} |
|
else |
|
{ |
|
builder.Append($"({classObj.SourceName}::*)"); |
|
} |
|
|
|
builder.Append('('); |
|
|
|
foreach (var child in function.Children) |
|
{ |
|
if (child is UhtProperty property) |
|
{ |
|
if (!property.PropertyFlags.HasAnyFlags(EPropertyFlags.ReturnParm)) |
|
{ |
|
builder.Append(GetParamPropertySignature(property)); |
|
|
|
builder.Append(", "); |
|
} |
|
} |
|
} |
|
|
|
if (function.HasParameters) |
|
{ |
|
builder.Remove(builder.Length - 2, 2); |
|
} |
|
|
|
builder.Append(')'); |
|
|
|
if (function.FunctionFlags.HasAnyFlags(EFunctionFlags.Const)) |
|
{ |
|
builder.Append("const"); |
|
} |
|
|
|
return builder.ToString(); |
|
} |
|
|
|
private static string GetFunctionParamName(UhtFunction function) |
|
{ |
|
var builder = new StringBuilder(); |
|
|
|
builder.Append("TArray<FString>{"); |
|
|
|
foreach (var child in function.Children) |
|
{ |
|
if (child is UhtProperty property) |
|
{ |
|
if (!property.PropertyFlags.HasAnyFlags(EPropertyFlags.ReturnParm)) |
|
{ |
|
builder.Append($"\"{property.SourceName}\", "); |
|
} |
|
} |
|
} |
|
|
|
builder.Remove(builder.Length - 2, 2); |
|
|
|
builder.Append('}'); |
|
|
|
return builder.ToString(); |
|
} |
|
|
|
private static string GetFunctionDefaultValue(UhtFunction function) |
|
{ |
|
if (!function.FunctionFlags.HasAnyFlags(EFunctionFlags.BlueprintCallable)) |
|
{ |
|
return ""; |
|
} |
|
|
|
var builder = new StringBuilder(); |
|
|
|
foreach (var child in function.Children) |
|
{ |
|
if (child is UhtProperty property) |
|
{ |
|
if (!property.PropertyFlags.HasAnyFlags(EPropertyFlags.ReturnParm)) |
|
{ |
|
if (property.DefaultValueTokens != null) |
|
{ |
|
builder.Append(", "); |
|
|
|
var DefaultValue = new StringBuilder(); |
|
|
|
foreach (var ValueToken in property.DefaultValueTokens) |
|
{ |
|
DefaultValue.Append(ValueToken.Value); |
|
} |
|
|
|
if (DefaultValue.ToString() == "NULL") |
|
{ |
|
builder.Append("nullptr"); |
|
} |
|
else |
|
{ |
|
builder.Append(DefaultValue); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
|
|
return builder.ToString(); |
|
} |
|
|
|
private string GetHeaderFile(UhtClass classObj) |
|
{ |
|
return Path.Combine(HeaderPath[classObj.Package.Module.ShortName], classObj.HeaderFile.FilePath); |
|
} |
|
|
|
private static string GenerateInclude(string file) |
|
{ |
|
return $"#include \"{file}\"\r\n"; |
|
} |
|
|
|
private string GetInclude(UhtClass classObj) |
|
{ |
|
return GenerateInclude(GetHeaderFile(classObj)); |
|
} |
|
|
|
private void SaveIfChanged(string file, StringBuilder builder) |
|
{ |
|
if (File.Exists(file) && File.ReadAllText(file) == builder.ToString()) |
|
{ |
|
return; |
|
} |
|
|
|
Factory.CommitOutput(file, builder); |
|
} |
|
|
|
private void GetPlugins(string InPathName, Dictionary<string, string> Plugins) |
|
{ |
|
var Suffix = "*.uplugin"; |
|
|
|
var DirectoryInfo = new DirectoryInfo(InPathName); |
|
|
|
foreach (var Item in DirectoryInfo.GetFiles(Suffix)) |
|
{ |
|
if (Item.DirectoryName != null) |
|
{ |
|
Plugins[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName; |
|
} |
|
} |
|
|
|
foreach (var Directories in DirectoryInfo.GetDirectories()) |
|
{ |
|
foreach (var Item in Directories.GetFiles(Suffix, SearchOption.AllDirectories)) |
|
{ |
|
if (Item.DirectoryName != null) |
|
{ |
|
Plugins[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName; |
|
} |
|
} |
|
} |
|
} |
|
|
|
private void GetModules(string InPathName, Dictionary<string, string> Modules) |
|
{ |
|
var Suffix = "*.Build.cs"; |
|
|
|
var DirectoryInfo = new DirectoryInfo(InPathName); |
|
|
|
foreach (var Item in DirectoryInfo.GetFiles(Suffix)) |
|
{ |
|
if (Item.DirectoryName != null) |
|
{ |
|
Modules[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName; |
|
} |
|
} |
|
|
|
foreach (var Directories in DirectoryInfo.GetDirectories()) |
|
{ |
|
foreach (var Item in Directories.GetFiles(Suffix, SearchOption.AllDirectories)) |
|
{ |
|
if (Item.DirectoryName != null) |
|
{ |
|
Modules[Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)] = Item.DirectoryName; |
|
} |
|
} |
|
} |
|
} |
|
|
|
private void GetModules(string InPathName, HashSet<string> Modules) |
|
{ |
|
var Suffix = "*.Build.cs"; |
|
|
|
var DirectoryInfo = new DirectoryInfo(InPathName); |
|
|
|
foreach (var Item in DirectoryInfo.GetFiles(Suffix)) |
|
{ |
|
Modules.Add(Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)); |
|
} |
|
|
|
foreach (var Directories in DirectoryInfo.GetDirectories()) |
|
{ |
|
foreach (var Item in Directories.GetFiles(Suffix, SearchOption.AllDirectories)) |
|
{ |
|
if (Item.DirectoryName != null) |
|
{ |
|
Modules.Add(Item.Name.Remove(Item.Name.Length - Suffix.Length + 1)); |
|
} |
|
} |
|
} |
|
} |
|
} |
|
} |