workshop rework

Content/Python/ue_workshop/materials.py

@@ -0,0 +1,34 @@
+"""Small opaque PBR materials, generated without external assets."""
+import unreal
+
+from .paths import GENERATED, slug
+
+
+def color(name, rgb=(0.15, 0.5, 0.8), roughness=0.55, metallic=0.0):
+    path = f"{GENERATED}/Materials/M_{slug(name)}"
+    material = unreal.load_asset(path)
+    if material is None:
+        material = unreal.AssetToolsHelpers.get_asset_tools().create_asset(
+            "M_" + name, GENERATED + "/Materials", unreal.Material, unreal.MaterialFactoryNew())
+    if not isinstance(material, unreal.Material):
+        raise TypeError(f"{path} nie jest materialem.")
+    lib = unreal.MaterialEditingLibrary
+    lib.delete_all_material_expressions(material)
+    base = lib.create_material_expression(material, unreal.MaterialExpressionConstant3Vector, -300, 0)
+    base.set_editor_property("constant", unreal.LinearColor(*rgb, 1.0))
+    lib.connect_material_property(base, "", unreal.MaterialProperty.MP_BASE_COLOR)
+    for offset, value, prop in [(160, roughness, unreal.MaterialProperty.MP_ROUGHNESS),
+                                (280, metallic, unreal.MaterialProperty.MP_METALLIC)]:
+        node = lib.create_material_expression(material, unreal.MaterialExpressionConstant, -300, offset)
+        node.set_editor_property("r", float(value))
+        lib.connect_material_property(node, "", prop)
+    lib.recompile_material(material)
+    unreal.EditorAssetLibrary.save_loaded_asset(material)
+    return material
+
+
+def assign(actor, material, slot=0):
+    component = actor.get_component_by_class(unreal.StaticMeshComponent)
+    if component is None:
+        raise TypeError("Material mozna przypisac do aktora ze StaticMeshComponent.")
+    component.set_material(slot, material)