Andron666
317 lines
10 KiB
C++
317 lines
10 KiB
C++
/*
|
|
* Copyright (c) <2021> Side Effects Software Inc.
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. The name of Side Effects Software may not be used to endorse or
|
|
* promote products derived from this software without specific prior
|
|
* written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY SIDE EFFECTS SOFTWARE "AS IS" AND ANY EXPRESS
|
|
* OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN
|
|
* NO EVENT SHALL SIDE EFFECTS SOFTWARE BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
|
|
* OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
|
|
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
|
|
* EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "HoudiniPublicAPIProcessHDANode.h"
|
|
|
|
#include "HoudiniPublicAPI.h"
|
|
#include "HoudiniPublicAPIBlueprintLib.h"
|
|
#include "HoudiniPublicAPIAssetWrapper.h"
|
|
#include "HoudiniPublicAPIInputTypes.h"
|
|
|
|
#include "HoudiniEngineRuntimePrivatePCH.h"
|
|
|
|
|
|
UHoudiniPublicAPIProcessHDANode::UHoudiniPublicAPIProcessHDANode(const FObjectInitializer& ObjectInitializer)
|
|
: Super(ObjectInitializer)
|
|
{
|
|
if ( HasAnyFlags(RF_ClassDefaultObject) == false )
|
|
{
|
|
AddToRoot();
|
|
}
|
|
|
|
AssetWrapper = nullptr;
|
|
bCookSuccess = false;
|
|
bBakeSuccess = false;
|
|
|
|
HoudiniAsset = nullptr;
|
|
InstantiateAt = FTransform::Identity;
|
|
WorldContextObject = nullptr;
|
|
SpawnInLevelOverride = nullptr;
|
|
bEnableAutoCook = true;
|
|
bEnableAutoBake = false;
|
|
BakeDirectoryPath = FString();
|
|
BakeMethod = EHoudiniEngineBakeOption::ToActor;
|
|
bRemoveOutputAfterBake = false;
|
|
bRecenterBakedActors = false;
|
|
bReplacePreviousBake = false;
|
|
bDeleteInstantiatedAssetOnCompletionOrFailure = false;
|
|
}
|
|
|
|
UHoudiniPublicAPIProcessHDANode*
|
|
UHoudiniPublicAPIProcessHDANode::ProcessHDA(
|
|
UHoudiniAsset* InHoudiniAsset,
|
|
const FTransform& InInstantiateAt,
|
|
const TMap<FName, FHoudiniParameterTuple>& InParameters,
|
|
const TMap<int32, UHoudiniPublicAPIInput*>& InNodeInputs,
|
|
const TMap<FName, UHoudiniPublicAPIInput*>& InParameterInputs,
|
|
UObject* InWorldContextObject,
|
|
ULevel* InSpawnInLevelOverride,
|
|
const bool bInEnableAutoCook,
|
|
const bool bInEnableAutoBake,
|
|
const FString& InBakeDirectoryPath,
|
|
const EHoudiniEngineBakeOption InBakeMethod,
|
|
const bool bInRemoveOutputAfterBake,
|
|
const bool bInRecenterBakedActors,
|
|
const bool bInReplacePreviousBake,
|
|
const bool bInDeleteInstantiatedAssetOnCompletionOrFailure)
|
|
{
|
|
UHoudiniPublicAPIProcessHDANode* Node = NewObject<UHoudiniPublicAPIProcessHDANode>();
|
|
|
|
Node->HoudiniAsset = InHoudiniAsset;
|
|
Node->InstantiateAt = InInstantiateAt;
|
|
Node->Parameters = InParameters;
|
|
Node->NodeInputs = InNodeInputs;
|
|
Node->ParameterInputs = InParameterInputs;
|
|
Node->WorldContextObject = InWorldContextObject;
|
|
Node->SpawnInLevelOverride = InSpawnInLevelOverride;
|
|
Node->bEnableAutoCook = bInEnableAutoCook;
|
|
Node->bEnableAutoBake = bInEnableAutoBake;
|
|
Node->BakeDirectoryPath = InBakeDirectoryPath;
|
|
Node->BakeMethod = InBakeMethod;
|
|
Node->bRemoveOutputAfterBake = bInRemoveOutputAfterBake;
|
|
Node->bRecenterBakedActors = bInRecenterBakedActors;
|
|
Node->bReplacePreviousBake = bInReplacePreviousBake;
|
|
Node->bDeleteInstantiatedAssetOnCompletionOrFailure = bInDeleteInstantiatedAssetOnCompletionOrFailure;
|
|
|
|
return Node;
|
|
}
|
|
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::Activate()
|
|
{
|
|
UHoudiniPublicAPI* API = UHoudiniPublicAPIBlueprintLib::GetAPI();
|
|
if (!IsValid(API))
|
|
{
|
|
HandleFailure();
|
|
return;
|
|
}
|
|
|
|
AssetWrapper = UHoudiniPublicAPIAssetWrapper::CreateEmptyWrapper(API);
|
|
if (!IsValid(AssetWrapper))
|
|
{
|
|
HandleFailure();
|
|
return;
|
|
}
|
|
|
|
AssetWrapper->GetOnPreInstantiationDelegate().AddDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePreInstantiation);
|
|
AssetWrapper->GetOnPostInstantiationDelegate().AddDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostInstantiation);
|
|
AssetWrapper->GetOnPostCookDelegate().AddDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostAutoCook);
|
|
AssetWrapper->GetOnPreProcessStateExitedDelegate().AddDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePreProcess);
|
|
AssetWrapper->GetOnPostProcessingDelegate().AddDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostProcessing);
|
|
AssetWrapper->GetOnPostBakeDelegate().AddDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostAutoBake);
|
|
|
|
if (!API->InstantiateAssetWithExistingWrapper(
|
|
AssetWrapper,
|
|
HoudiniAsset,
|
|
InstantiateAt,
|
|
WorldContextObject,
|
|
SpawnInLevelOverride,
|
|
bEnableAutoCook,
|
|
bEnableAutoBake,
|
|
BakeDirectoryPath,
|
|
BakeMethod,
|
|
bRemoveOutputAfterBake,
|
|
bRecenterBakedActors,
|
|
bReplacePreviousBake))
|
|
{
|
|
HandleFailure();
|
|
return;
|
|
}
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::UnbindDelegates()
|
|
{
|
|
AssetWrapper->GetOnPreInstantiationDelegate().RemoveDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePreInstantiation);
|
|
AssetWrapper->GetOnPostInstantiationDelegate().RemoveDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostInstantiation);
|
|
AssetWrapper->GetOnPostCookDelegate().RemoveDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostAutoCook);
|
|
AssetWrapper->GetOnPreProcessStateExitedDelegate().RemoveDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePreProcess);
|
|
AssetWrapper->GetOnPostProcessingDelegate().RemoveDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostProcessing);
|
|
AssetWrapper->GetOnPostBakeDelegate().RemoveDynamic(this, &UHoudiniPublicAPIProcessHDANode::HandlePostAutoBake);
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandleFailure()
|
|
{
|
|
if (Failed.IsBound())
|
|
Failed.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
|
|
UnbindDelegates();
|
|
|
|
RemoveFromRoot();
|
|
|
|
if (bDeleteInstantiatedAssetOnCompletionOrFailure && IsValid(AssetWrapper))
|
|
AssetWrapper->DeleteInstantiatedAsset();
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandleComplete()
|
|
{
|
|
if (Completed.IsBound())
|
|
Completed.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
|
|
UnbindDelegates();
|
|
|
|
RemoveFromRoot();
|
|
|
|
if (bDeleteInstantiatedAssetOnCompletionOrFailure && IsValid(AssetWrapper))
|
|
AssetWrapper->DeleteInstantiatedAsset();
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandlePreInstantiation(UHoudiniPublicAPIAssetWrapper* InAssetWrapper)
|
|
{
|
|
if (InAssetWrapper != AssetWrapper)
|
|
{
|
|
HOUDINI_LOG_WARNING(
|
|
TEXT("[UHoudiniPublicAPIProcessHDANode] Received delegate event from unexpected asset wrapper (%s vs %s)!"),
|
|
IsValid(AssetWrapper) ? *(AssetWrapper->GetName()) : TEXT(""),
|
|
IsValid(InAssetWrapper) ? *(InAssetWrapper->GetName()) : TEXT(""));
|
|
return;
|
|
}
|
|
|
|
// Set any parameters specified when the node was created
|
|
if (Parameters.Num() > 0 && IsValid(AssetWrapper))
|
|
{
|
|
AssetWrapper->SetParameterTuples(Parameters);
|
|
}
|
|
|
|
if (PreInstantiation.IsBound())
|
|
PreInstantiation.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandlePostInstantiation(UHoudiniPublicAPIAssetWrapper* InAssetWrapper)
|
|
{
|
|
if (InAssetWrapper != AssetWrapper)
|
|
{
|
|
HOUDINI_LOG_WARNING(
|
|
TEXT("[UHoudiniPublicAPIProcessHDANode] Received delegate event from unexpected asset wrapper (%s vs %s)!"),
|
|
IsValid(AssetWrapper) ? *(AssetWrapper->GetName()) : TEXT(""),
|
|
IsValid(InAssetWrapper) ? *(InAssetWrapper->GetName()) : TEXT(""));
|
|
return;
|
|
}
|
|
|
|
// Set any inputs specified when the node was created
|
|
if (IsValid(AssetWrapper))
|
|
{
|
|
if (NodeInputs.Num() > 0)
|
|
{
|
|
AssetWrapper->SetInputsAtIndices(NodeInputs);
|
|
}
|
|
if (ParameterInputs.Num() > 0)
|
|
{
|
|
AssetWrapper->SetInputParameters(ParameterInputs);
|
|
}
|
|
|
|
// // Set any parameters specified when the node was created
|
|
// if (Parameters.Num() > 0)
|
|
// {
|
|
// AssetWrapper->SetParameterTuples(Parameters);
|
|
// }
|
|
}
|
|
|
|
if (PostInstantiation.IsBound())
|
|
PostInstantiation.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
|
|
if (!bEnableAutoCook)
|
|
HandleComplete();
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandlePostAutoCook(UHoudiniPublicAPIAssetWrapper* InAssetWrapper, const bool bInCookSuccess)
|
|
{
|
|
if (InAssetWrapper != AssetWrapper)
|
|
{
|
|
HOUDINI_LOG_WARNING(
|
|
TEXT("[UHoudiniPublicAPIProcessHDANode] Received delegate event from unexpected asset wrapper (%s vs %s)!"),
|
|
IsValid(AssetWrapper) ? *(AssetWrapper->GetName()) : TEXT(""),
|
|
IsValid(InAssetWrapper) ? *(InAssetWrapper->GetName()) : TEXT(""));
|
|
return;
|
|
}
|
|
|
|
bCookSuccess = bInCookSuccess;
|
|
|
|
if (PostAutoCook.IsBound())
|
|
PostAutoCook.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandlePreProcess(UHoudiniPublicAPIAssetWrapper* InAssetWrapper)
|
|
{
|
|
if (InAssetWrapper != AssetWrapper)
|
|
{
|
|
HOUDINI_LOG_WARNING(
|
|
TEXT("[UHoudiniPublicAPIProcessHDANode] Received delegate event from unexpected asset wrapper (%s vs %s)!"),
|
|
IsValid(AssetWrapper) ? *(AssetWrapper->GetName()) : TEXT(""),
|
|
IsValid(InAssetWrapper) ? *(InAssetWrapper->GetName()) : TEXT(""));
|
|
return;
|
|
}
|
|
|
|
if (PreProcess.IsBound())
|
|
PreProcess.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandlePostProcessing(UHoudiniPublicAPIAssetWrapper* InAssetWrapper)
|
|
{
|
|
if (InAssetWrapper != AssetWrapper)
|
|
{
|
|
HOUDINI_LOG_WARNING(
|
|
TEXT("[UHoudiniPublicAPIProcessHDANode] Received delegate event from unexpected asset wrapper (%s vs %s)!"),
|
|
IsValid(AssetWrapper) ? *(AssetWrapper->GetName()) : TEXT(""),
|
|
IsValid(InAssetWrapper) ? *(InAssetWrapper->GetName()) : TEXT(""));
|
|
return;
|
|
}
|
|
|
|
if (PostProcessing.IsBound())
|
|
PostProcessing.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
|
|
if (!bEnableAutoBake)
|
|
HandleComplete();
|
|
}
|
|
|
|
void
|
|
UHoudiniPublicAPIProcessHDANode::HandlePostAutoBake(UHoudiniPublicAPIAssetWrapper* InAssetWrapper, const bool bInBakeSuccess)
|
|
{
|
|
if (InAssetWrapper != AssetWrapper)
|
|
{
|
|
HOUDINI_LOG_WARNING(
|
|
TEXT("[UHoudiniPublicAPIProcessHDANode] Received delegate event from unexpected asset wrapper (%s vs %s)!"),
|
|
IsValid(AssetWrapper) ? *(AssetWrapper->GetName()) : TEXT(""),
|
|
IsValid(InAssetWrapper) ? *(InAssetWrapper->GetName()) : TEXT(""));
|
|
return;
|
|
}
|
|
|
|
bBakeSuccess = bInBakeSuccess;
|
|
|
|
if (PostAutoBake.IsBound())
|
|
PostAutoBake.Broadcast(AssetWrapper, bCookSuccess, bBakeSuccess);
|
|
|
|
HandleComplete();
|
|
}
|