Merge branch 'main' of https://gitea-web.iepose.cn/yang/SpaceStudio

SpaceStudioClient/Plugins/NeoKinectUnreal/Source/NeoKinectUnreal/NeoKinectUnreal.Build.cs

@@ -19,6 +19,8 @@ public class NeoKinectUnreal : ModuleRules
 		{
 			"Core",
 			"CoreUObject",
+            "OpenCV",
+			"OpenCVHelper"
 			// ... add other public dependencies that you statically link with here ...
 		});
 
@@ -34,6 +36,7 @@ public class NeoKinectUnreal : ModuleRules
 
 		string neoKinectPath = Path.Combine(thirdPartyPath, "NeoKinect");
 		string kinectPath = Path.Combine(thirdPartyPath, "Kinect");
+        //string openCVPath = Path.Combine(thirdPartyPath, "OpenCV");
 
         PublicSystemLibraryPaths.Add(Path.Combine(neoKinectPath, "Lib"));
 		PublicIncludePaths.Add(Path.Combine(neoKinectPath, "Inc"));
@@ -64,5 +67,6 @@ public class NeoKinectUnreal : ModuleRules
 		const string ProjectDLLDir = "$(ProjectDir)/Binaries/Win64";
 		RuntimeDependencies.Add(Path.Combine(ProjectDLLDir, KinectFaceDllName));
 		RuntimeDependencies.Add(Path.Combine(ProjectDLLDir, "NuiDatabase/..."), StagedFileType.NonUFS);
+
     }
 }

SpaceStudioClient/Plugins/NeoKinectUnreal/Source/NeoKinectUnreal/Private/NeoKinectManager.cpp

@@ -12,7 +12,9 @@
 #include "Engine/TextureRenderTarget2D.h"
 #include "HAL/PlatformProcess.h"
 #include "UObject/Package.h"
-
+#include "PreOpenCVHeaders.h"
+#include "opencv2/opencv.hpp"
+#include "PostOpenCVHeaders.h"
 using namespace NeoKinect;
 
 //~ STRUCTS METHODS
@@ -74,7 +76,6 @@ FFrameTextureData UNeoKinectManager::FramesData[static_cast<int32>(EKinectFrame:
 UNeoKinectManager* UNeoKinectManager::EventsDispatcher = nullptr;
 FTSTicker::FDelegateHandle UNeoKinectManager::TickerHandle;
 
-
 #pragma endregion Statics
 
 //~ INITIALIZATION / UNINITIALIZATION
@@ -238,6 +239,7 @@ void UNeoKinectManager::InitStatics()
 		fd.UFrameType = static_cast<EKinectFrame>(i);
 		fd.BufferSize = fd.Width * fd.Height * fd.PixelSize;
 	}
+
 }
 #pragma endregion Initialization_Uninitialization
 
@@ -1139,6 +1141,28 @@ void UNeoKinectManager::UpdateFrames()
 			if (Buffer && FrameData.IsTextureValid())
 			{
 				ThreadBuffer.Lock();
+				{
+					if (FrameData.KFrameType == KinectFrameType::BodyIndex)
+					{
+						cv::Mat binaryImage(FrameData.Height, FrameData.Width, CV_8UC4);
+						memcpy(binaryImage.data, Buffer, FrameData.Height * FrameData.Width * FrameData.PixelSize);
+						cv::Mat processedImage;
+						cv::resize(binaryImage, processedImage, cv::Size(FrameData.Width / 2, FrameData.Height / 2), 0, 0, cv::INTER_CUBIC);
+						cv::Mat kernel = cv::getStructuringElement(cv::MORPH_ELLIPSE, cv::Size(5, 5));
+						cv::morphologyEx(processedImage, processedImage, cv::MORPH_OPEN, kernel); // 去噪
+						cv::morphologyEx(processedImage, processedImage, cv::MORPH_CLOSE, kernel); // 填充
+						cv::GaussianBlur(processedImage, processedImage, cv::Size(5, 5), 1.0);
+						cv::resize(processedImage, processedImage, cv::Size(FrameData.Width, FrameData.Height), 0, 0, cv::INTER_CUBIC);
+
+						RHIUpdateTexture2D(
+							FrameData.Texture->GetRenderTargetResource()->GetRenderTargetTexture(),
+							0,
+							FUpdateTextureRegion2D(0, 0, 0, 0, FrameData.Width, FrameData.Height),
+							FrameData.Width * FrameData.PixelSize,
+							processedImage.data
+						);
+					}
+					else
 					{
 						RHIUpdateTexture2D(
 							FrameData.Texture->GetRenderTargetResource()->GetRenderTargetTexture(),
@@ -1148,6 +1172,7 @@ void UNeoKinectManager::UpdateFrames()
 							Buffer
 						);
 					}
+				}
 				ThreadBuffer.Unlock();
 			}
 		}

SpaceStudioClient/Plugins/NeoKinectUnreal/Source/NeoKinectUnreal/Public/NeoKinectManager.h

@@ -7,7 +7,6 @@
 #include "NeoKinectEnums.h"
 #include "PixelFormat.h"
 #include "Containers/Ticker.h"
-
 #include "NeoKinectManager.generated.h"
 
 class FKinectThread;