Source Code Generator

Public/SourceCodeGenerator.h

@@ -0,0 +1,101 @@
+#pragma once
+
+#include "CoreMinimal.h"
+#include "IScriptGeneratorPluginInterface.h"
+
+class FSourceCodeGeneratorModule : public IScriptGeneratorPluginInterface
+{
+private:
+	static constexpr auto LargeInitialBufferSizes = 256 * 1024;
+
+	static constexpr auto SmallInitialBufferSizes = 1 * 1024;
+
+	typedef TStringBuilder<LargeInitialBufferSizes> FBigStringBuilder;
+
+	typedef TStringBuilder<SmallInitialBufferSizes> FSmallStringBuilder;
+
+public:
+	/** IModuleInterface implementation */
+	virtual void StartupModule() override;
+
+	virtual void ShutdownModule() override;
+
+public:
+	/** IScriptGeneratorPluginInterface implementation */
+	virtual FString GetGeneratedCodeModuleName() const override;
+
+	virtual bool SupportsTarget(const FString& TargetName) const override;
+
+	virtual bool ShouldExportClassesForModule(const FString& ModuleName, EBuildModuleType::Type ModuleType,
+	                                          const FString& ModuleGeneratedIncludeDirectory) const override;
+
+	virtual void Initialize(const FString& RootLocalPath, const FString& RootBuildPath,
+	                        const FString& OutputDirectory, const FString& IncludeBase) override;
+
+	virtual void ExportClass(UClass* Class, const FString& SourceHeaderFilename,
+	                         const FString& GeneratedHeaderFilename, bool bHasChanged) override;
+
+	virtual void FinishExport() override;
+
+	virtual FString GetGeneratorName() const override;
+
+private:
+	static bool CanExportClass(const UClass* Class);
+
+	static bool CanExportFunction(const UFunction* Function);
+
+	static bool CanExportProperty(const FProperty* Property);
+
+	static void ExportFunction(FBigStringBuilder& StringBuilder, const UClass* Class, const UFunction* Function);
+
+	static void ExportProperty(FBigStringBuilder& StringBuilder, const UClass* Class, const FProperty* Property);
+
+	static bool IsClassTypeSupported(const UClass* Class);
+
+	static bool IsPropertyTypeSupported(const FProperty* Property);
+
+	static FString GetReturnPropertySignature(const FProperty* Property);
+
+	static FString GetParamPropertySignature(const FProperty* Property);
+
+	static FString GetFunctionSignature(const UClass* Class, const UFunction* Function);
+
+	static FString GetFunctionParamName(const UFunction* Function);
+
+	static FString GetFunctionDefaultValue(const UFunction* Function);
+
+	static bool FindDefaultValue(const TMap<FName, FString>* MetaMap, const FProperty* Param, FString& String);
+
+	static void GetPropertySignature(const FProperty* Property, FString& String);
+
+	static void GetDependencyClasses(const FProperty* Property, TSet<UClass*>& DependencyClasses);
+
+	static void SaveIfChanged(const FString& FileName, const FString& String);
+
+	FString GetHeaderFile(UClass* Class);
+
+	static FString GenerateInclude(const FString& FileName);
+
+	FString GetInclude(UClass* Class);
+
+	static void GetPlugins(const FString& InPathName, TMap<FString, FString>& Plugins);
+
+	static void GetModules(const FString& InPathName, TMap<FString, FString>& Modules);
+
+	static void GetModules(const FString& InPathName, TSet<FString>& Modules);
+
+private:
+	FString OutputPath;
+
+	TArray<FString> ExportModules;
+
+	TSet<FString> Project;
+
+	TMap<FString, FString> HeaderPath;
+
+	TArray<UClass*> ExportClasses;
+
+	const FString BindingSuffix = FString(TEXT(".binding.inl"));
+
+	const FString HeaderSuffix = FString(TEXT(".header.inl"));
+};