Andron666
142 lines
5.1 KiB
Python
142 lines
5.1 KiB
Python
# 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.
|
|
|
|
""" An example script that instantiates an HDA that contains a TOP network.
|
|
After the HDA itself has cooked (in on_post_process) we iterate over all
|
|
TOP networks in the HDA and print their paths. Auto-bake is then enabled for
|
|
PDG and the TOP networks are cooked.
|
|
|
|
"""
|
|
|
|
import os
|
|
|
|
import unreal
|
|
|
|
_g_wrapper = None
|
|
|
|
|
|
def get_test_hda_path():
|
|
return '/HoudiniEngine/Examples/hda/pdg_pighead_grid_2_0.pdg_pighead_grid_2_0'
|
|
|
|
|
|
def get_test_hda():
|
|
return unreal.load_object(None, get_test_hda_path())
|
|
|
|
|
|
def get_hda_directory():
|
|
""" Attempt to get the directory containing the .hda files. This is
|
|
/HoudiniEngine/Examples/hda. In newer versions of UE we can use
|
|
``unreal.SystemLibrary.get_system_path(asset)``, in older versions
|
|
we manually construct the path to the plugin.
|
|
|
|
Returns:
|
|
(str): the path to the example hda directory or ``None``.
|
|
|
|
"""
|
|
if hasattr(unreal.SystemLibrary, 'get_system_path'):
|
|
return os.path.dirname(os.path.normpath(
|
|
unreal.SystemLibrary.get_system_path(get_test_hda())))
|
|
else:
|
|
plugin_dir = os.path.join(
|
|
os.path.normpath(unreal.Paths.project_plugins_dir()),
|
|
'Runtime', 'HoudiniEngine', 'Content', 'Examples', 'hda'
|
|
)
|
|
if not os.path.exists(plugin_dir):
|
|
plugin_dir = os.path.join(
|
|
os.path.normpath(unreal.Paths.engine_plugins_dir()),
|
|
'Runtime', 'HoudiniEngine', 'Content', 'Examples', 'hda'
|
|
)
|
|
if not os.path.exists(plugin_dir):
|
|
return None
|
|
return plugin_dir
|
|
|
|
|
|
def delete_instantiated_asset():
|
|
global _g_wrapper
|
|
if _g_wrapper:
|
|
result = _g_wrapper.delete_instantiated_asset()
|
|
_g_wrapper = None
|
|
return result
|
|
else:
|
|
return False
|
|
|
|
|
|
def on_pre_instantiation(in_wrapper):
|
|
print('on_pre_instantiation')
|
|
|
|
# Set the hda_directory parameter to the directory that contains the
|
|
# example .hda files
|
|
hda_directory = get_hda_directory()
|
|
print('Setting "hda_directory" to {0}'.format(hda_directory))
|
|
in_wrapper.set_string_parameter_value('hda_directory', hda_directory)
|
|
|
|
# Cook the HDA (not PDG yet)
|
|
in_wrapper.recook()
|
|
|
|
|
|
def on_post_bake(in_wrapper, success):
|
|
# in_wrapper.on_post_bake_delegate.remove_callable(on_post_bake)
|
|
print('bake complete ... {}'.format('success' if success else 'failed'))
|
|
|
|
|
|
def on_post_process(in_wrapper):
|
|
print('on_post_process')
|
|
|
|
# in_wrapper.on_post_processing_delegate.remove_callable(on_post_process)
|
|
|
|
# Iterate over all PDG/TOP networks and nodes and log them
|
|
print('TOP networks:')
|
|
for network_path in in_wrapper.get_pdgtop_network_paths():
|
|
print('\t{}'.format(network_path))
|
|
for node_path in in_wrapper.get_pdgtop_node_paths(network_path):
|
|
print('\t\t{}'.format(node_path))
|
|
|
|
# Enable PDG auto-bake (auto bake TOP nodes after they are cooked)
|
|
in_wrapper.set_pdg_auto_bake_enabled(True)
|
|
# Bind to PDG post bake delegate (called after all baking is complete)
|
|
in_wrapper.on_post_pdg_bake_delegate.add_callable(on_post_bake)
|
|
# Cook the specified TOP node
|
|
in_wrapper.pdg_cook_node('topnet1', 'HE_OUT_PIGHEAD_GRID')
|
|
|
|
|
|
def run():
|
|
# get the API singleton
|
|
api = unreal.HoudiniPublicAPIBlueprintLib.get_api()
|
|
|
|
global _g_wrapper
|
|
|
|
# instantiate an asset, disabling auto-cook of the asset
|
|
_g_wrapper = api.instantiate_asset(get_test_hda(), unreal.Transform(), enable_auto_cook=False)
|
|
|
|
# Bind to the on pre instantiation delegate (before the first cook) and
|
|
# set parameters
|
|
_g_wrapper.on_pre_instantiation_delegate.add_callable(on_pre_instantiation)
|
|
# Bind to the on post processing delegate (after a cook and after all
|
|
# outputs have been generated in Unreal)
|
|
_g_wrapper.on_post_processing_delegate.add_callable(on_post_process)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run()
|