X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=blender%2Fio_mspgl%2Fexport_material.py;h=f51a4a49a88e47c7fbf96055c381be576c7fbb5c;hb=180b20bee11425a776c5ead05afcf6a63945d3b2;hp=356186df17c5b2341e525322cc69f3607ce4513f;hpb=56133280d92c08c1c649a725260a6c4d5afb5e75;p=libs%2Fgl.git diff --git a/blender/io_mspgl/export_material.py b/blender/io_mspgl/export_material.py index 356186df..f51a4a49 100644 --- a/blender/io_mspgl/export_material.py +++ b/blender/io_mspgl/export_material.py @@ -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)