]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/export_material.py
Rework Blender material and texture export
[libs/gl.git] / blender / io_mspgl / export_material.py
index 356186df17c5b2341e525322cc69f3607ce4513f..f51a4a49a88e47c7fbf96055c381be576c7fbb5c 100644 (file)
@@ -1,5 +1,24 @@
 import os
 
+def create_technique_resource(material, resources, single_file):
+       from .datafile import Resource, Statement
+       tech_res = Resource(material.name+".tech")
+
+       mat_res = resources[material.name+".mat"]
+
+       st = Statement("pass", "")
+       if single_file:
+               st.sub.append(tech_res.create_embed_statement("material", mat_res))
+       else:
+               st.sub.append(tech_res.create_reference_statement("material", mat_res))
+
+       if material.render_mode=='CUSTOM':
+               st.sub.append(Statement("shader", material.shader))
+
+       tech_res.statements.append(st)
+
+       return tech_res
+
 class MaterialExporter:
        def __init__(self):
                self.single_file = True
@@ -14,97 +33,53 @@ class MaterialExporter:
        def export_technique_resources(self, material, resources):
                texture_export = self.create_texture_exporter()
 
-               mat_name = material.name+".mat"
-               if mat_name not in resources:
-                       resources[mat_name] = self.export_material(material)
+               from .material import Material
+               material = Material(material)
 
-               if False and self.use_textures:
-                       for s in material.texture_slots:
-                               if s and s.texture.type=='IMAGE' and s.texture.image:
-                                       tex_name = s.texture.name+".tex2d"
+               if self.use_textures:
+                       for p in material.properties:
+                               if p.texture:
+                                       tex_name = p.texture.image.name+".tex2d"
                                        if tex_name not in resources:
-                                               resources[tex_name] = texture_export.export_texture(s.texture)
+                                               resources[tex_name] = texture_export.export_texture(p.texture, p.tex_usage)
+
+               mat_name = material.name+".mat"
+               if mat_name not in resources:
+                       resources[mat_name] = self.export_material(material, resources=resources)
 
        def export_technique(self, material, *, resources):
-               from .datafile import Resource, Statement
-               tech_res = Resource(material.name+".tech")
-
-               mat_res = resources[material.name+".mat"]
-               textures = {}
-               if False and self.use_textures:
-                       image_texture_slots = [s for s in material.texture_slots if s and s.texture.type=='IMAGE' and s.texture.image]
-                       for s in image_texture_slots:
-                               if s.use_map_color_diffuse:
-                                       textures["diffuse_map"] = s.texture
-                               elif s.use_map_normal:
-                                       if s.texture.use_normal_map:
-                                               textures["normal_map"] = s.texture
-                                       else:
-                                               textures["displace_map"] = s.texture
-
-               if material.technique:
-                       if not material.inherit_tech:
-                               return tech_res
+               return create_technique_resource(material, resources, self.single_file)
 
-                       if self.single_file:
-                               raise Exception("Can't export inherited technique to a single file")
-
-                       st = Statement("inherit", material.technique)
-                       for s, t in textures.items():
-                               fn = os.path.basename(t.image.filepath)
-                               if t.default_filter and fn:
-                                       st.sub.append(Statement("texture", s, fn))
-                               else:
-                                       st.sub.append(tech_res.create_reference_statement("texture", s, resources[t.name+".tex2d"]))
-                       if material.override_material:
-                               st.sub.append(tech_res.create_reference_statement("material", "surface", mat_res))
-                       tech_res.statements.append(st)
-               else:
-                       st = Statement("pass", "")
-                       if self.single_file:
-                               st.sub.append(tech_res.create_embed_statement("material", mat_res))
-                       else:
-                               st.sub.append(tech_res.create_reference_statement("material", mat_res))
-
-                       if "diffuse_map" in textures:
-                               diffuse_tex = textures["diffuse_map"]
-                               tex_res = resources[diffuse_tex.name+".tex2d"]
-                               ss = Statement("texunit", 0)
-                               fn = os.path.basename(diffuse_tex.image.filepath)
-                               if self.single_file:
-                                       ss.sub.append(tech_res.create_embed_statement("texture2d", tex_res))
-                               elif diffuse_tex.default_filter and fn:
-                                       ss.sub.append(Statement("texture", fn))
-                               else:
-                                       ss.sub.append(tech_res.create_reference_statement("texture", tex_res))
-                               st.sub.append(ss)
-
-                       tech_res.statements.append(st)
-
-               return tech_res
-
-       def export_material(self, material):
+       def export_material(self, material, *, resources):
                from .datafile import Resource, Statement
                mat_res = Resource(material.name+".mat")
-               statements = mat_res.statements
 
-               from .util import get_colormap
-               cm = get_colormap(material.srgb_colors)
-
-               if False and any(s.use_map_color_diffuse for s in material.texture_slots if s):
-                       statements.append(Statement("diffuse", 1.0, 1.0, 1.0, 1.0))
-                       amb = cm(material.ambient)
-                       statements.append(Statement("ambient", amb, amb, amb, 1.0))
-               else:
-                       diff = material.diffuse_color
-                       statements.append(Statement("diffuse", cm(diff[0]), cm(diff[1]), cm(diff[2]), 1.0))
-                       statements.append(Statement("ambient", cm(diff[0]), cm(diff[1]), cm(diff[2]), 1.0))
-               spec = material.specular_color*material.specular_intensity
-               statements.append(Statement("specular", cm(spec.r), cm(spec.g), cm(spec.g), 1.0))
-               statements.append(Statement("shininess", min(2/material.roughness**2-2, 250)))
+               st = Statement("pbr")
+               st.sub.append(self.create_property_statement(mat_res, material.base_color, "base_color", resources))
+               st.sub.append(self.create_property_statement(mat_res, material.metalness, "metalness", resources))
+               st.sub.append(self.create_property_statement(mat_res, material.roughness, "roughness", resources))
+               st.sub.append(self.create_property_statement(mat_res, material.normal, "normal", resources, tex_only=True))
+               st.sub.append(self.create_property_statement(mat_res, material.emission, "emission", resources))
+               mat_res.statements.append(st)
 
                return mat_res
 
+       def create_property_statement(self, mat_res, prop, keyword, resources, *, tex_only=False):
+               from .datafile import Statement
+               if self.use_textures and prop.texture:
+                       tex_res = resources[prop.texture.image.name+".tex2d"]
+                       fn = os.path.basename(prop.texture.image.filepath)
+                       if self.single_file:
+                               raise Exception("Can't export textures to a single file")
+                       elif prop.texture.default_filter and fn:
+                               return Statement(keyword+"_map", fn)
+                       else:
+                               return mat_res.create_reference_statement(keyword+"_map", tex_res)
+               elif type(prop.value)==tuple:
+                       return Statement(keyword, *prop.value)
+               else:
+                       return Statement(keyword, prop.value)
+
 
 class MaterialMapExporter:
        def __init__(self):
@@ -112,52 +87,25 @@ class MaterialMapExporter:
 
        def export_technique_resources(self, material_map, resources):
                from .datafile import Resource, Statement, Token
-               diffuse_name = material_map.name+"_diffuse.tex2d"
-               if diffuse_name not in resources:
-                       diffuse_res = Resource(diffuse_name)
-
-                       fmt = 'SRGB_ALPHA' if material_map.srgb_colors else 'RGBA'
+               base_color_name = material_map.name+"_base_color.tex2d"
+               if base_color_name not in resources:
+                       base_color_res = Resource(base_color_name)
 
-                       diffuse_res.statements.append(Statement("min_filter", Token('NEAREST')))
-                       diffuse_res.statements.append(Statement("mag_filter", Token('NEAREST')))
-                       diffuse_res.statements.append(Statement("storage", Token(fmt), *material_map.size))
-                       diffuse_res.statements.append(Statement("raw_data", material_map.diffuse_data))
+                       base_color_res.statements.append(Statement("min_filter", Token('NEAREST')))
+                       base_color_res.statements.append(Statement("mag_filter", Token('NEAREST')))
+                       base_color_res.statements.append(Statement("storage", Token('SRGB_ALPHA'), *material_map.size))
+                       base_color_res.statements.append(Statement("raw_data", material_map.base_color_data))
 
-                       resources[diffuse_name] = diffuse_res
-
-               if "basic_white.mat" not in resources:
-                       mat_res = Resource("basic_white.mat")
-                       mat_res.statements.append(Statement("diffuse", 1.0, 1.0, 1.0, 1.0))
-
-                       resources["basic_white.mat"] = mat_res
-
-       def export_technique(self, material_map, *, resources=None):
-               from .datafile import Resource, Statement
-               tech_res = Resource(material_map.name+".tech")
+                       resources[base_color_name] = base_color_res
 
-               mat_res = resources["basic_white.mat"]
-               diffuse_res = resources[material_map.name+"_diffuse.tex2d"]
-
-               if material_map.technique:
-                       if self.single_file:
-                               raise Exception("Can't export inherited technique to a single file")
+               mat_name = material_map.name+".mat"
+               if mat_name not in resources:
+                       mat_res = Resource(mat_name)
+                       st = Statement("pbr")
+                       st.sub.append(mat_res.create_reference_statement("base_color_map", base_color_res))
+                       mat_res.statements.append(st)
 
-                       st = Statement("inherit", material_map.technique)
-                       st.sub.append(tech_res.create_reference_statement("texture", "diffuse_map", diffuse_res))
-                       st.sub.append(tech_res.create_reference_statement("material", "surface", mat_res))
-                       tech_res.statements.append(st)
-               else:
-                       st = Statement("pass", "")
-                       if self.single_file:
-                               st.sub.append(tech_res.create_embed_statement("material", mat_res))
-                       else:
-                               st.sub.append(tech_res.create_reference_statement("material", mat_res))
-                       ss = Statement("texunit", 0)
-                       if self.single_file:
-                               ss.sub.append(tech_res.create_embed_statement("texture2d", diffuse_res))
-                       else:
-                               ss.sub.append(tech_res.create_reference_statement("texture", diffuse_res))
-                       st.sub.append(ss)
-                       tech_res.statements.append(st)
+                       resources[mat_name] = mat_res
 
-               return tech_res
+       def export_technique(self, material_map, *, resources):
+               return create_technique_resource(material_map, resources, self.single_file)