diff options
Diffstat (limited to 'io_scene_gltf2/blender/exp/gltf2_blender_gather_materials.py')
-rwxr-xr-x | io_scene_gltf2/blender/exp/gltf2_blender_gather_materials.py | 154 |
1 files changed, 154 insertions, 0 deletions
diff --git a/io_scene_gltf2/blender/exp/gltf2_blender_gather_materials.py b/io_scene_gltf2/blender/exp/gltf2_blender_gather_materials.py new file mode 100755 index 00000000..427f07ce --- /dev/null +++ b/io_scene_gltf2/blender/exp/gltf2_blender_gather_materials.py @@ -0,0 +1,154 @@ +# Copyright 2018 The glTF-Blender-IO authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +import bpy + +from io_scene_gltf2.blender.exp.gltf2_blender_gather_cache import cached +from io_scene_gltf2.io.com import gltf2_io +from io_scene_gltf2.blender.exp import gltf2_blender_gather_texture_info +from io_scene_gltf2.blender.exp import gltf2_blender_gather_material_normal_texture_info_class +from io_scene_gltf2.blender.exp import gltf2_blender_gather_material_occlusion_texture_info_class + +from io_scene_gltf2.blender.exp import gltf2_blender_gather_materials_pbr_metallic_roughness +from io_scene_gltf2.blender.exp import gltf2_blender_generate_extras +from io_scene_gltf2.blender.exp import gltf2_blender_get + + +@cached +def gather_material(blender_material, export_settings): + """ + Gather the material used by the blender primitive. + + :param blender_material: the blender material used in the glTF primitive + :param export_settings: + :return: a glTF material + """ + if not __filter_material(blender_material, export_settings): + return None + + material = gltf2_io.Material( + alpha_cutoff=__gather_alpha_cutoff(blender_material, export_settings), + alpha_mode=__gather_alpha_mode(blender_material, export_settings), + double_sided=__gather_double_sided(blender_material, export_settings), + emissive_factor=__gather_emissive_factor(blender_material, export_settings), + emissive_texture=__gather_emissive_texture(blender_material, export_settings), + extensions=__gather_extensions(blender_material, export_settings), + extras=__gather_extras(blender_material, export_settings), + name=__gather_name(blender_material, export_settings), + normal_texture=__gather_normal_texture(blender_material, export_settings), + occlusion_texture=__gather_occlusion_texture(blender_material, export_settings), + pbr_metallic_roughness=__gather_pbr_metallic_roughness(blender_material, export_settings) + ) + + return material + # material = blender_primitive['material'] + # + # if get_material_requires_texcoords(glTF, material) and not export_settings['gltf_texcoords']: + # material = -1 + # + # if get_material_requires_normals(glTF, material) and not export_settings['gltf_normals']: + # material = -1 + # + # # Meshes/primitives without material are allowed. + # if material >= 0: + # primitive.material = material + # else: + # print_console('WARNING', 'Material ' + internal_primitive[ + # 'material'] + ' not found. Please assign glTF 2.0 material or enable Blinn-Phong material in export.') + + +def __filter_material(blender_material, export_settings): + # if not blender_material.use_nodes: + # return False + # if not blender_material.node_tree: + # return False + return True + + +def __gather_alpha_cutoff(blender_material, export_settings): + if blender_material.blend_method == 'CLIP': + return blender_material.alpha_threshold + return None + + +def __gather_alpha_mode(blender_material, export_settings): + if blender_material.blend_method == 'CLIP': + return 'MASK' + elif blender_material.blend_method == 'BLEND': + return 'BLEND' + return None + + +def __gather_double_sided(blender_material, export_settings): + return None + + +def __gather_emissive_factor(blender_material, export_settings): + emissive_socket = gltf2_blender_get.get_socket_or_texture_slot(blender_material, "Emissive") + if emissive_socket is None: + emissive_socket = gltf2_blender_get.get_socket_or_texture_slot_old(blender_material, "EmissiveFactor") + if isinstance(emissive_socket, bpy.types.NodeSocket) and not emissive_socket.is_linked: + return list(emissive_socket.default_value)[0:3] + return None + + +def __gather_emissive_texture(blender_material, export_settings): + emissive = gltf2_blender_get.get_socket_or_texture_slot(blender_material, "Emissive") + if emissive is None: + emissive = gltf2_blender_get.get_socket_or_texture_slot_old(blender_material, "Emissive") + return gltf2_blender_gather_texture_info.gather_texture_info((emissive,), export_settings) + + +def __gather_extensions(blender_material, export_settings): + extensions = {} + + + # TODO specular glossiness extension + + return extensions if extensions else None + + +def __gather_extras(blender_material, export_settings): + if export_settings['gltf_extras']: + return gltf2_blender_generate_extras.generate_extras(blender_material) + return None + + +def __gather_name(blender_material, export_settings): + return blender_material.name + + +def __gather_normal_texture(blender_material, export_settings): + normal = gltf2_blender_get.get_socket_or_texture_slot(blender_material, "Normal") + if normal is None: + normal = gltf2_blender_get.get_socket_or_texture_slot_old(blender_material, "Normal") + return gltf2_blender_gather_material_normal_texture_info_class.gather_material_normal_texture_info_class( + (normal,), + export_settings) + + +def __gather_occlusion_texture(blender_material, export_settings): + occlusion = gltf2_blender_get.get_socket_or_texture_slot(blender_material, "Occlusion") + if occlusion is None: + occlusion = gltf2_blender_get.get_socket_or_texture_slot_old(blender_material, "Occlusion") + return gltf2_blender_gather_material_occlusion_texture_info_class.gather_material_occlusion_texture_info_class( + (occlusion,), + export_settings) + + +def __gather_pbr_metallic_roughness(blender_material, export_settings): + return gltf2_blender_gather_materials_pbr_metallic_roughness.gather_material_pbr_metallic_roughness( + blender_material, + export_settings) + |