1 files changed, 68 insertions, 0 deletions
diff --git a/io_scene_gltf2/blender/exp/gltf2_blender_gather_materials_sheen.py b/io_scene_gltf2/blender/exp/gltf2_blender_gather_materials_sheen.py
new file mode 100644
index 00000000..03625ecb
--- /dev/null
+++ b/io_scene_gltf2/blender/exp/gltf2_blender_gather_materials_sheen.py
@@ -0,0 +1,68 @@
+# SPDX-License-Identifier: Apache-2.0
+# Copyright 2018-2022 The glTF-Blender-IO authors.
+
+import bpy
+from io_scene_gltf2.io.com.gltf2_io_extensions import Extension
+from io_scene_gltf2.blender.exp import gltf2_blender_get
+from io_scene_gltf2.blender.exp import gltf2_blender_gather_texture_info
+
+
+def export_sheen(blender_material, export_settings):
+    sheen_extension = {}
+
+    sheenColor_socket = gltf2_blender_get.get_socket(blender_material, "sheenColor")
+    sheenRoughness_socket = gltf2_blender_get.get_socket(blender_material, "sheenRoughness")
+
+    if sheenColor_socket is None or sheenRoughness_socket is None:
+        return None, None
+
+    sheenColor_non_linked = isinstance(sheenColor_socket, bpy.types.NodeSocket) and not sheenColor_socket.is_linked
+    sheenRoughness_non_linked = isinstance(sheenRoughness_socket, bpy.types.NodeSocket) and not sheenRoughness_socket.is_linked
+
+
+    use_actives_uvmaps = []
+
+    if sheenColor_non_linked is True:
+        color = sheenColor_socket.default_value[:3]
+        if color != (0.0, 0.0, 0.0):
+            sheen_extension['sheenColorFactor'] = color
+    else:
+        # Factor
+        fac = gltf2_blender_get.get_factor_from_socket(sheenColor_socket, kind='RGB')
+        if fac is not None and fac != [0.0, 0.0, 0.0]:
+            sheen_extension['sheenColorFactor'] = fac
+        
+        # Texture
+        if gltf2_blender_get.has_image_node_from_socket(sheenColor_socket):
+            original_sheenColor_texture, original_sheenColor_use_active_uvmap, _ = gltf2_blender_gather_texture_info.gather_texture_info(
+                sheenColor_socket,
+                (sheenColor_socket,),
+                export_settings,
+            )
+            sheen_extension['sheenColorTexture'] = original_sheenColor_texture
+            if original_sheenColor_use_active_uvmap:
+                use_actives_uvmaps.append("sheenColorTexture")
+
+
+    if sheenRoughness_non_linked is True:
+        fac = sheenRoughness_socket.default_value
+        if fac != 0.0:
+            sheen_extension['sheenRoughnessFactor'] = fac
+    else:
+        # Factor
+        fac = gltf2_blender_get.get_factor_from_socket(sheenRoughness_socket, kind='VALUE')
+        if fac is not None and fac != 0.0:
+            sheen_extension['sheenRoughnessFactor'] = fac
+        
+        # Texture
+        if gltf2_blender_get.has_image_node_from_socket(sheenRoughness_socket):
+            original_sheenRoughness_texture, original_sheenRoughness_use_active_uvmap, _ = gltf2_blender_gather_texture_info.gather_texture_info(
+                sheenRoughness_socket,
+                (sheenRoughness_socket,),
+                export_settings,
+            )
+            sheen_extension['sheenRoughnessTexture'] = original_sheenRoughness_texture
+            if original_sheenRoughness_use_active_uvmap:
+                use_actives_uvmaps.append("sheenRoughnessTexture")
+    
+    return Extension('KHR_materials_sheen', sheen_extension, False), use_actives_uvmaps