from blenderproc.python.modules.loader.LoaderInterface import LoaderInterface
from blenderproc.python.utility.Utility import resolve_path
from blenderproc.python.loader.ObjectLoader import load_obj
[docs]class ObjectLoaderModule(LoaderInterface):
""" Just imports the objects for the given file path
The import will load all materials into cycle nodes.
**Configuration**:
.. list-table::
:widths: 25 100 10
:header-rows: 1
* - Parameter
- Description
- Type
* - path
- The path to the 3D data file to load. Can be either path or paths not both.
- string
* - paths
- A list of paths of 3D data files to load. Can be either path or paths not both.
- list
"""
def __init__(self, config):
LoaderInterface.__init__(self, config)
[docs] def run(self):
if self.config.has_param('path') and self.config.has_param('paths'):
raise Exception("Objectloader can not use path and paths in the same module!")
if self.config.has_param('path'):
file_path = resolve_path(self.config.get_string("path"))
loaded_objects = load_obj(filepath=file_path)
elif self.config.has_param('paths'):
file_paths = self.config.get_list('paths')
loaded_objects = []
# the file paths are mapped here to object names
cache_objects = {}
for file_path in file_paths:
resolved_file_path = resolve_path(file_path)
current_objects = load_obj(filepath=resolved_file_path, cached_objects=cache_objects)
loaded_objects.extend(current_objects)
else:
raise Exception("Loader module needs either a path or paths config value")
if not loaded_objects:
raise Exception("No objects have been loaded here, check the config.")
# Set the add_properties of all imported objects
self._set_properties(loaded_objects)