Andron666
137 lines
5.1 KiB
C++
137 lines
5.1 KiB
C++
/*
|
|
* Copyright (c) <2017> Side Effects Software Inc.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*
|
|
* Produced by:
|
|
* Mykola Konyk
|
|
* Side Effects Software Inc
|
|
* 123 Front Street West, Suite 1401
|
|
* Toronto, Ontario
|
|
* Canada M5J 2M2
|
|
* 416-504-9876
|
|
*
|
|
*/
|
|
|
|
#include "HoudiniAttributeDataComponent.h"
|
|
|
|
#include "HoudiniApi.h"
|
|
#include "HoudiniEngineRuntimePrivatePCH.h"
|
|
#include "HoudiniEngineUtils.h"
|
|
#include "HoudiniEngine.h"
|
|
#include "HoudiniPluginSerializationVersion.h"
|
|
|
|
UHoudiniAttributeDataComponent::UHoudiniAttributeDataComponent( const FObjectInitializer & ObjectInitializer )
|
|
: Super( ObjectInitializer )
|
|
{
|
|
}
|
|
|
|
UHoudiniAttributeDataComponent::~UHoudiniAttributeDataComponent()
|
|
{}
|
|
|
|
void
|
|
UHoudiniAttributeDataComponent::SetAttributeData( FHoudiniPointAttributeData&& InVertexAttributeData )
|
|
{
|
|
VertexAttributeData.Add( InVertexAttributeData );
|
|
}
|
|
|
|
FHoudiniPointAttributeData*
|
|
UHoudiniAttributeDataComponent::GetAttributeData( class UStaticMeshComponent* StaticMeshComponent )
|
|
{
|
|
for ( FHoudiniPointAttributeData& AttributeData : VertexAttributeData )
|
|
{
|
|
if ( AttributeData.Component.IsValid() && AttributeData.Component.Get() == StaticMeshComponent )
|
|
{
|
|
return &AttributeData;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
void
|
|
UHoudiniAttributeDataComponent::Serialize( FArchive & Ar )
|
|
{
|
|
Super::Serialize( Ar );
|
|
|
|
Ar.UsingCustomVersion( FHoudiniCustomSerializationVersion::GUID );
|
|
|
|
int32 Version = VER_HOUDINI_PLUGIN_SERIALIZATION_AUTOMATIC_VERSION;
|
|
Ar << Version;
|
|
}
|
|
|
|
bool
|
|
UHoudiniAttributeDataComponent::Upload( HAPI_NodeId GeoNodeId, class UStaticMeshComponent* StaticMeshComponent ) const
|
|
{
|
|
for ( const FHoudiniPointAttributeData& AttributeData : VertexAttributeData )
|
|
{
|
|
if ( AttributeData.Component.IsValid() && AttributeData.Component.Get() == StaticMeshComponent )
|
|
{
|
|
HAPI_AttributeInfo AttributeInfoPoint;
|
|
FHoudiniApi:: AttributeInfo_Init(&AttributeInfoPoint);
|
|
//FMemory::Memzero< HAPI_AttributeInfo >( AttributeInfoPoint );
|
|
AttributeInfoPoint.count = AttributeData.Count;
|
|
AttributeInfoPoint.tupleSize = AttributeData.TupleSize;
|
|
AttributeInfoPoint.exists = true;
|
|
AttributeInfoPoint.owner = HAPI_ATTROWNER_POINT;
|
|
AttributeInfoPoint.originalOwner = HAPI_ATTROWNER_INVALID;
|
|
|
|
const char* AttrNameRaw = TCHAR_TO_ANSI( *AttributeData.AttrName );
|
|
|
|
switch ( AttributeData.DataType )
|
|
{
|
|
case EHoudiniVertexAttributeDataType::VADT_Bool:
|
|
case EHoudiniVertexAttributeDataType::VADT_Int32:
|
|
{
|
|
AttributeInfoPoint.storage = HAPI_STORAGETYPE_INT;
|
|
|
|
HOUDINI_CHECK_ERROR_RETURN( FHoudiniApi::AddAttribute(
|
|
FHoudiniEngine::Get().GetSession(), GeoNodeId,
|
|
0, AttrNameRaw, &AttributeInfoPoint ), false );
|
|
|
|
HOUDINI_CHECK_ERROR_RETURN( FHoudiniApi::SetAttributeIntData(
|
|
FHoudiniEngine::Get().GetSession(),
|
|
GeoNodeId, 0, AttrNameRaw, &AttributeInfoPoint,
|
|
AttributeData.IntData.GetData(), 0, AttributeInfoPoint.count ), false );
|
|
|
|
break;
|
|
}
|
|
case EHoudiniVertexAttributeDataType::VADT_Float:
|
|
{
|
|
AttributeInfoPoint.storage = HAPI_STORAGETYPE_FLOAT;
|
|
|
|
HOUDINI_CHECK_ERROR_RETURN( FHoudiniApi::AddAttribute(
|
|
FHoudiniEngine::Get().GetSession(), GeoNodeId,
|
|
0, AttrNameRaw, &AttributeInfoPoint ), false );
|
|
|
|
HOUDINI_CHECK_ERROR_RETURN( FHoudiniApi::SetAttributeFloatData(
|
|
FHoudiniEngine::Get().GetSession(),
|
|
GeoNodeId, 0, AttrNameRaw, &AttributeInfoPoint,
|
|
AttributeData.FloatData.GetData(), 0, AttributeInfoPoint.count ), false );
|
|
|
|
break;
|
|
}
|
|
default:
|
|
checkNoEntry();
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
return true;
|
|
}
|