+import os
+
+def linear_to_srgb(l):
+ if l<0.0031308:
+ return 12.92*l
+ else:
+ return 1.055*(l**(1/2.4))-0.055
+
+def get_colormap(srgb):
+ if srgb:
+ return linear_to_srgb
+ else:
+ return lambda x: x
+
+def image_name(i):
+ fp = i.filepath
+ if fp:
+ return os.path.split(fp)[1]
+ else:
+ return i.name
+
+
+class MaterialExporter:
+ def __init__(self):
+ self.textures = 'REF'
+
+ def export_technique_resources(self, material, resources):
+ from .export_texture import TextureExporter
+ texture_export = TextureExporter()
+
+ mat_name = material.name+".mat"
+ if mat_name not in resources:
+ resources[mat_name] = self.export_material(material)
+
+ if self.textures=='INLINE':
+ for s in material.texture_slots:
+ if s and s.texture.type=='IMAGE':
+ tex_name = s.texture.name+".tex2d"
+ if tex_name not in resources:
+ resources[tex_name] = texture_export.export_texture(s.texture)
+
+ def export_technique(self, material, *, resources):
+ from .datafile import Resource, Statement
+ tech_res = Resource(material.name+".tech")
+
+ mat_res = resources[material.name+".mat"]
+ image_texture_slots = [s for s in material.texture_slots if s and s.texture.type=='IMAGE']
+
+ if material.technique:
+ if not obj.inherit_tech:
+ return []
+
+ st = Statement("inherit", material.technique)
+ for slot in image_texture_slots:
+ name = image_name(slot.texture.image)
+ if slot.use_map_color_diffuse:
+ st.sub.append(Statement("texture", "diffuse_map", name))
+ elif slot.use_map_normal:
+ st.sub.append(Statement("texture", "normal_map", name))
+ 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", "")
+ st.sub.append(tech_res.create_embed_statement("material", mat_res))
+
+ if self.textures!='NONE':
+ diffuse_tex = None
+ for slot in image_texture_slots:
+ if slot.use_map_color_diffuse:
+ diffuse_tex = slot.texture
+ break
+
+ if diffuse_tex:
+ ss = Statement("texunit", 0)
+ if self.textures=='INLINE':
+ tex_res = resources[slot.texture.name+".tex2d"]
+ ss.sub.append(tech_res.create_embed_statement("texture2d", tex_res))
+ elif tex.image:
+ ss.sub.append(Statement("texture", image_name(tex.image)))
+ st.sub.append(ss)
+
+ tech_res.statements.append(st)
+
+ return tech_res
+
+ def export_material(self, material):
+ from .datafile import Resource, Statement
+ mat_res = Resource(material.name+".mat")
+ statements = mat_res.statements
+
+ cm = get_colormap(material.srgb_colors)
+ if 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*material.diffuse_intensity
+ statements.append(Statement("diffuse", cm(diff.r), cm(diff.g), cm(diff.b), 1.0))
+ amb = diff*material.ambient
+ statements.append(Statement("ambient", cm(amb.r), cm(amb.g), cm(amb.b), 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", material.specular_hardness))
+
+ return mat_res