def _setup_material(self):
"""Setup object material"""
available_materials = {
'metal': abr_nodes.material_metal_tool_cap,
'plastic': abr_nodes.material_3Dprinted_plastic
}
material = self.config.scenario_setup.object_material.lower()
if material not in available_materials:
raise ValueError(f'Requested material "{material}" is not supported')
# make sure cap is selected
blnd.select_object(self.obj.name)
# remove any material that's currently assigned to the object and then
# setup the metal for the cap
blnd.remove_material_nodes(self.obj)
blnd.clear_orphaned_materials()
# add default material and setup nodes (without specifying empty, to get
# it created automatically)
self.obj_mat = blnd.add_default_material(self.obj)
available_materials[material].setup_material(self.obj_mat)
def reset(self):
blnd.clear_all_objects()
blnd.clear_orphaned_materials()
def main():
"""First tear down any material assigned with the object, then create everything from scratch"""
remove_material_nodes()
clear_orphaned_materials()
mat = add_default_material()
setup_material_metal_tool_cap(mat)