blender/io_mspgl/export_material.py

   1 import os
   2
   3 def create_technique_resource(material, resources, single_file):
   4         from .datafile import Resource, Statement
   5         tech_res = Resource(material.name+".tech")
   6
   7         mat_res = resources[material.name+".mat"]
   8
   9         st = Statement("pass", "")
  10         if single_file:
  11                 st.sub.append(tech_res.create_embed_statement("material", mat_res))
  12         else:
  13                 st.sub.append(tech_res.create_reference_statement("material", mat_res))
  14
  15         if material.render_mode=='CUSTOM':
  16                 st.sub.append(Statement("shader", material.shader))
  17
  18         tech_res.statements.append(st)
  19
  20         return tech_res
  21
  22 class MaterialExporter:
  23         def __init__(self):
  24                 self.single_file = True
  25                 self.use_textures = True
  26
  27         def create_texture_exporter(self):
  28                 from .export_texture import TextureExporter
  29                 texture_export = TextureExporter()
  30                 texture_export.inline_data = self.single_file
  31                 return texture_export
  32
  33         def export_technique_resources(self, material, resources):
  34                 texture_export = self.create_texture_exporter()
  35
  36                 from .material import Material
  37                 material = Material(material)
  38
  39                 if self.use_textures:
  40                         for p in material.properties:
  41                                 if p.texture:
  42                                         tex_name = p.texture.image.name+".tex2d"
  43                                         if tex_name not in resources:
  44                                                 resources[tex_name] = texture_export.export_texture(p.texture, p.tex_usage)
  45
  46                 mat_name = material.name+".mat"
  47                 if mat_name not in resources:
  48                         resources[mat_name] = self.export_material(material, resources=resources)
  49
  50         def export_technique(self, material, *, resources):
  51                 return create_technique_resource(material, resources, self.single_file)
  52
  53         def export_material(self, material, *, resources):
  54                 from .datafile import Resource, Statement
  55                 mat_res = Resource(material.name+".mat")
  56
  57                 st = Statement("pbr")
  58                 st.sub.append(self.create_property_statement(mat_res, material.base_color, "base_color", resources))
  59                 st.sub.append(self.create_property_statement(mat_res, material.metalness, "metalness", resources))
  60                 st.sub.append(self.create_property_statement(mat_res, material.roughness, "roughness", resources))
  61                 st.sub.append(self.create_property_statement(mat_res, material.normal, "normal", resources, tex_only=True))
  62                 st.sub.append(self.create_property_statement(mat_res, material.emission, "emission", resources))
  63                 mat_res.statements.append(st)
  64
  65                 return mat_res
  66
  67         def create_property_statement(self, mat_res, prop, keyword, resources, *, tex_only=False):
  68                 from .datafile import Statement
  69                 if self.use_textures and prop.texture:
  70                         tex_res = resources[prop.texture.image.name+".tex2d"]
  71                         fn = os.path.basename(prop.texture.image.filepath)
  72                         if self.single_file:
  73                                 raise Exception("Can't export textures to a single file")
  74                         elif prop.texture.default_filter and fn:
  75                                 return Statement(keyword+"_map", fn)
  76                         else:
  77                                 return mat_res.create_reference_statement(keyword+"_map", tex_res)
  78                 elif type(prop.value)==tuple:
  79                         return Statement(keyword, *prop.value)
  80                 else:
  81                         return Statement(keyword, prop.value)
  82
  83
  84 class MaterialMapExporter:
  85         def __init__(self):
  86                 self.single_file = True
  87
  88         def export_technique_resources(self, material_map, resources):
  89                 from .datafile import Resource, Statement, Token
  90                 base_color_name = material_map.name+"_base_color.tex2d"
  91                 if base_color_name not in resources:
  92                         base_color_res = Resource(base_color_name)
  93
  94                         base_color_res.statements.append(Statement("min_filter", Token('NEAREST')))
  95                         base_color_res.statements.append(Statement("mag_filter", Token('NEAREST')))
  96                         base_color_res.statements.append(Statement("storage", Token('SRGB_ALPHA'), *material_map.size))
  97                         base_color_res.statements.append(Statement("raw_data", material_map.base_color_data))
  98
  99                         resources[base_color_name] = base_color_res
 100
 101                 mat_name = material_map.name+".mat"
 102                 if mat_name not in resources:
 103                         mat_res = Resource(mat_name)
 104                         st = Statement("pbr")
 105                         st.sub.append(mat_res.create_reference_statement("base_color_map", base_color_res))
 106                         mat_res.statements.append(st)
 107
 108                         resources[mat_name] = mat_res
 109
 110         def export_technique(self, material_map, *, resources):
 111                 return create_technique_resource(material_map, resources, self.single_file)