You've already forked UnrealEngineUWP
mirror of
https://github.com/izzy2lost/UnrealEngineUWP.git
synced 2026-03-26 18:15:20 -07:00
4c1bb11c29
This represents UE4/Main @ 14525125 + cherrypicked fixes #skipundocheck [CL 14551026 by Marc Audy in ue5-main branch]
121 lines
3.1 KiB
C++
121 lines
3.1 KiB
C++
// Copyright Epic Games, Inc. All Rights Reserved.
|
|
|
|
#include "DatasmithDirectLink.h"
|
|
|
|
#include "DatasmithExporterManager.h"
|
|
#include "IDatasmithSceneElements.h"
|
|
#include "DirectLink/DatasmithDirectLinkTools.h"
|
|
|
|
#include "DirectLinkEndpoint.h"
|
|
#include "DirectLinkMisc.h"
|
|
|
|
#include "Containers/Ticker.h"
|
|
#include "MeshDescription.h"
|
|
#include "Misc/CommandLine.h"
|
|
#include "Modules/ModuleInterface.h"
|
|
#include "Modules/ModuleManager.h"
|
|
|
|
|
|
DEFINE_LOG_CATEGORY(LogDatasmithDirectLinkExporterAPI);
|
|
|
|
class FDatasmithDirectLinkImpl
|
|
{
|
|
public:
|
|
|
|
static int32 ValidateCommunicationSetup()
|
|
{
|
|
return
|
|
(FModuleManager::Get().LoadModule(TEXT("Messaging")) ? 0 : 1) +
|
|
(FModuleManager::Get().LoadModule(TEXT("UdpMessaging")) ? 0 : 2) +
|
|
(FModuleManager::Get().LoadModule(TEXT("Networking")) ? 0 : 3) +
|
|
(FParse::Param(FCommandLine::Get(), TEXT("Messaging")) ? 0 : 4);
|
|
}
|
|
|
|
FDatasmithDirectLinkImpl()
|
|
: Endpoint(MakeShared<DirectLink::FEndpoint, ESPMode::ThreadSafe>(TEXT("DatasmithExporter")))
|
|
{
|
|
Endpoint->SetVerbose();
|
|
check(ValidateCommunicationSetup() == 0);
|
|
// #ue_directlink_integration app specific endpoint name, and source name.
|
|
}
|
|
|
|
bool InitializeForScene(TSharedRef<IDatasmithScene>& Scene)
|
|
{
|
|
UE_LOG(LogDatasmithDirectLinkExporterAPI, Log, TEXT("InitializeForScene"));
|
|
|
|
Endpoint->RemoveSource(Source);
|
|
|
|
// Use the scene's label to name the source
|
|
const TCHAR* SourceName = FCString::Strlen(Scene->GetLabel()) == 0 ? TEXT("unnamed") : Scene->GetLabel();
|
|
|
|
Source = Endpoint->AddSource(SourceName, DirectLink::EVisibility::Public);
|
|
bool bSnapshotNow = false;
|
|
Endpoint->SetSourceRoot(Source, &*Scene, bSnapshotNow);
|
|
CurrentScene = Scene;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool UpdateScene(TSharedRef<IDatasmithScene>& Scene)
|
|
{
|
|
UE_LOG(LogDatasmithDirectLinkExporterAPI, Log, TEXT("UpdateScene"));
|
|
if (CurrentScene != Scene)
|
|
{
|
|
InitializeForScene(Scene);
|
|
}
|
|
|
|
Endpoint->SnapshotSource(Source);
|
|
DumpDatasmithScene(Scene, TEXT("send"));
|
|
return true;
|
|
}
|
|
|
|
TSharedRef<DirectLink::FEndpoint, ESPMode::ThreadSafe> GetEnpoint() const
|
|
{
|
|
return Endpoint;
|
|
}
|
|
|
|
private:
|
|
TSharedRef<DirectLink::FEndpoint, ESPMode::ThreadSafe> Endpoint;
|
|
DirectLink::FSourceHandle Source;
|
|
TSharedPtr<IDatasmithScene> CurrentScene;
|
|
};
|
|
|
|
|
|
TUniquePtr<FDatasmithDirectLinkImpl> DirectLinkImpl;
|
|
|
|
int32 FDatasmithDirectLink::ValidateCommunicationSetup()
|
|
{ return FDatasmithDirectLinkImpl::ValidateCommunicationSetup(); }
|
|
|
|
bool FDatasmithDirectLink::Shutdown()
|
|
{
|
|
DirectLinkImpl.Reset();
|
|
FDatasmithExporterManager::Shutdown();
|
|
return true;
|
|
}
|
|
|
|
FDatasmithDirectLink::FDatasmithDirectLink()
|
|
{
|
|
if (!DirectLinkImpl)
|
|
{
|
|
DirectLinkImpl = MakeUnique<FDatasmithDirectLinkImpl>();
|
|
}
|
|
}
|
|
|
|
FDatasmithDirectLink::~FDatasmithDirectLink() = default;
|
|
|
|
bool FDatasmithDirectLink::InitializeForScene(TSharedRef<IDatasmithScene>& Scene)
|
|
{ return DirectLinkImpl->InitializeForScene(Scene); }
|
|
|
|
bool FDatasmithDirectLink::UpdateScene(TSharedRef<IDatasmithScene>& Scene)
|
|
{ return DirectLinkImpl->UpdateScene(Scene); }
|
|
|
|
TSharedRef<DirectLink::FEndpoint, ESPMode::ThreadSafe> FDatasmithDirectLink::GetEnpoint()
|
|
{
|
|
if (!DirectLinkImpl)
|
|
{
|
|
DirectLinkImpl = MakeUnique<FDatasmithDirectLinkImpl>();
|
|
}
|
|
|
|
return DirectLinkImpl->GetEnpoint();
|
|
}
|