Интегрирован модуль импорта моделей из FreeCAD в Blender

2023-02-01 13:43:23 +00:00 · 2023-02-01 13:43:23 +00:00 · 6af0f188a7
commit 6af0f188a7
parent d26698d1a6
25 changed files with 1563 additions and 1 deletions
--- a/cg/blender/import_fcstd/materials.py
+++ b/cg/blender/import_fcstd/materials.py
@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+# Original code by (C) 2019 yorikvanhavre <yorik@uncreated.net>
+# Copyright (C) 2023 Ilia Kurochkin <brothermechanic@gmail.com>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+import logging
+import sys
+import bpy
+from bpy_extras.node_shader_utils import PrincipledBSDFWrapper
+from utils.shininess_to_roughness import shiny_to_rough
+
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.INFO)
+
+
+def set_fem_mat(obj, bobj, fem_mat):
+    if fem_mat.isDerivedFrom("App::MaterialObjectPython"):
+        if fem_mat.References[0][0].Name == obj.Name:
+            fem_mat_name = fem_mat.Material['Name']
+            if 'DiffuseColor' in fem_mat.Material.keys():
+                d_col_str = fem_mat.Material['DiffuseColor']
+                d_col4 = tuple(
+                    map(float, d_col_str[1:-1].split(', ')))
+                d_col = d_col4[:-1]
+            else:
+                d_col = (0.5, 0.5, 0.5)
+            if 'Father' in fem_mat.Material.keys():
+                if fem_mat.Material['Father'] == 'Metal':
+                    me = 1
+                else:
+                    me = 0
+            else:
+                me = 0
+            if 'Shininess' in fem_mat.Material.keys():
+                shiny = float(fem_mat.Material['Shininess'])
+                if shiny == 0:
+                    rg = 0.5
+                else:
+                    rg = shiny_to_rough(shiny)
+            else:
+                rg = 0.5
+            if 'EmissiveColor' in fem_mat.Material.keys():
+                e_col_str = fem_mat.Material['EmissiveColor']
+                e_col4 = tuple(
+                    map(float, e_col_str[1:-1].split(', ')))
+                e_col = e_col4[:-1]
+            else:
+                e_col = (0.0, 0.0, 0.0)
+            if 'Transparency' in fem_mat.Material.keys():
+                tr_str = fem_mat.Material['Transparency']
+                alpha = 1.0 - float(tr_str)
+            else:
+                alpha = 1.0
+
+            logger.debug('Assign %s to object %s', fem_mat_name, obj.Label)
+
+            if fem_mat_name in bpy.data.materials:
+                if len(bobj.material_slots) < 1:
+                    bobj.data.materials.append(bpy.data.materials[fem_mat_name])
+                else:
+                    bobj.material_slots[0].material = bpy.data.materials[fem_mat_name]
+            else:
+                bmat = bpy.data.materials.new(name=fem_mat_name)
+                bmat.use_nodes = True
+                principled = PrincipledBSDFWrapper(bmat, is_readonly=False)
+                principled.base_color = d_col
+                principled.metallic = me
+                principled.roughness = rg
+                principled.emission_color = e_col
+                principled.alpha = alpha
+                bobj.data.materials.append(bmat)