initial commit

Author: AntiAnti

Resources/Icon128.png

@@ -0,0 +1,26 @@
+{
+	"FileVersion": 3,
+	"Version": 1,
+	"VersionName": "1.0",
+	"FriendlyName": "SimplePyTorch",
+	"Description": "Simple wrapper to launch TorchScript models in UE4",
+	"Category": "Other",
+	"CreatedBy": "YuriNK",
+	"CreatedByURL": "https://vrmocapstudio.com",
+	"DocsURL": "",
+	"MarketplaceURL": "",
+	"SupportURL": "",
+	"EngineVersion": "4.26.0",
+	"CanContainContent": true,
+	"Installed": true,
+	"Modules": [
+		{
+			"Name": "SimplePyTorch",
+			"Type": "Runtime",
+			"LoadingPhase": "Default",
+			"WhitelistPlatforms": [
+				"Win64"
+			]
+		}
+	]
+}

SimplePyTorch.uplugin

@@ -0,0 +1,95 @@
+// VR IK Body Plugin
+// (c) Yuri N Kalinin, 2021, ykasczc@gmail.com. All right reserved.
+
+#include "SimplePyTorch.h"
+#include "HAL/PlatformProcess.h"
+#include "Misc/Paths.h"
+#include "Interfaces/IPluginManager.h"
+
+#define LOCTEXT_NAMESPACE "FSimplePyTorchModule"
+
+void FSimplePyTorchModule::StartupModule()
+{
+	// Init DLL from a Path
+	FString FilePath;
+	const FString szBinaries = TEXT("Binaries");
+	const FString szPlatform = TEXT("Win64");
+	
+#if WITH_EDITOR
+	auto ThisPlugin = IPluginManager::Get().FindPlugin(TEXT("SimplePyTorch"));
+	if (ThisPlugin.IsValid())
+	{
+		FilePath = FPaths::ConvertRelativePathToFull(ThisPlugin->GetBaseDir());
+	}
+#else
+	FilePath = FPaths::ConvertRelativePathToFull(FPaths::ProjectDir());
+#endif	
+	FilePath = FilePath / szBinaries / szPlatform / TEXT("torchscript_wrapper.dll");
+	WrapperDllHandle = NULL;
+	bDllLoaded = false;
+
+#if PLATFORM_WINDOWS
+	if (FPaths::FileExists(FilePath))
+	{
+		WrapperDllHandle = FPlatformProcess::GetDllHandle(*FilePath);
+
+		if (WrapperDllHandle != NULL)
+		{
+			FuncTSW_LoadScriptModel = (__TSW_LoadScriptModel)FPlatformProcess::GetDllExport(WrapperDllHandle, TEXT("TSW_LoadScriptModel"));
+			if (FuncTSW_LoadScriptModel == NULL)
+			{
+				UE_LOG(LogTemp, Error, TEXT("SimplePyTorchModule: can't load function __TSW_LoadScriptModel"));
+				return;
+			}
+			FuncTSW_CheckModel = (__TSW_CheckModel)FPlatformProcess::GetDllExport(WrapperDllHandle, TEXT("TSW_CheckModel"));
+			if (FuncTSW_CheckModel == NULL)
+			{
+				UE_LOG(LogTemp, Error, TEXT("SimplePyTorchModule: can't load function __TSW_CheckModel"));
+				return;
+			}
+			FuncTSW_Forward1d = (__TSW_Forward1d)FPlatformProcess::GetDllExport(WrapperDllHandle, TEXT("TSW_Forward1d"));
+			if (FuncTSW_Forward1d == NULL)
+			{
+				UE_LOG(LogTemp, Error, TEXT("SimplePyTorchModule: can't load function __TSW_Forward1d"));
+				return;
+			}
+			FuncTSW_ForwardTensor = (__TSW_ForwardTensor)FPlatformProcess::GetDllExport(WrapperDllHandle, TEXT("TSW_ForwardTensor"));
+			if (FuncTSW_ForwardTensor == NULL)
+			{
+				UE_LOG(LogTemp, Error, TEXT("SimplePyTorchModule: can't load function __TSW_ForwardTensor"));
+				return;
+			}
+			FuncTSW_ForwardPass_Def = (__TSW_ForwardPass_Def)FPlatformProcess::GetDllExport(WrapperDllHandle, TEXT("TSW_ForwardPass_Def"));
+			if (FuncTSW_ForwardPass_Def == NULL)
+			{
+				UE_LOG(LogTemp, Error, TEXT("SimplePyTorchModule: can't load function __TSW_ForwardPass_Def"));
+				return;
+			}
+			FuncTSW_Execute_Def = (__TSW_Execute_Def)FPlatformProcess::GetDllExport(WrapperDllHandle, TEXT("TSW_Execute_Def"));
+			if (FuncTSW_Execute_Def == NULL)
+			{
+				UE_LOG(LogTemp, Error, TEXT("SimplePyTorchModule: can't load function __TSW_Execute_Def"));
+				return;
+			}
+			
+			bDllLoaded = true;
+		}
+	}
+	else
+	{
+		UE_LOG(LogTemp, Error, TEXT("FSimplePyTorchModule. Can't find dll: %s"), *FilePath);
+	}
+#endif
+}
+
+void FSimplePyTorchModule::ShutdownModule()
+{
+	if (WrapperDllHandle != NULL)
+	{
+		FPlatformProcess::FreeDllHandle(WrapperDllHandle);
+	}
+}
+
+#undef LOCTEXT_NAMESPACE
+	
+IMPLEMENT_MODULE(FSimplePyTorchModule, SimplePyTorch)