blender/io_mspgl/export_material.py

   1 import os
   2
   3 class MaterialExporter:
   4         def __init__(self):
   5                 self.single_file = True
   6                 self.use_textures = True
   7
   8         def create_texture_exporter(self):
   9                 from .export_texture import TextureExporter
  10                 texture_export = TextureExporter()
  11                 texture_export.inline_data = self.single_file
  12                 return texture_export
  13
  14         def export_technique_resources(self, material, resources):
  15                 texture_export = self.create_texture_exporter()
  16
  17                 mat_name = material.name+".mat"
  18                 if mat_name not in resources:
  19                         resources[mat_name] = self.export_material(material)
  20
  21                 if self.use_textures:
  22                         for s in material.texture_slots:
  23                                 if s and s.texture.type=='IMAGE' and s.texture.image:
  24                                         tex_name = s.texture.name+".tex2d"
  25                                         if tex_name not in resources:
  26                                                 resources[tex_name] = texture_export.export_texture(s.texture)
  27
  28         def export_technique(self, material, *, resources):
  29                 from .datafile import Resource, Statement
  30                 tech_res = Resource(material.name+".tech")
  31
  32                 mat_res = resources[material.name+".mat"]
  33                 textures = {}
  34                 if self.use_textures:
  35                         image_texture_slots = [s for s in material.texture_slots if s and s.texture.type=='IMAGE' and s.texture.image]
  36                         for s in image_texture_slots:
  37                                 if s.use_map_color_diffuse:
  38                                         textures["diffuse_map"] = s.texture
  39                                 elif s.use_map_normal:
  40                                         if s.texture.use_normal_map:
  41                                                 textures["normal_map"] = s.texture
  42                                         else:
  43                                                 textures["displace_map"] = s.texture
  44
  45                 if material.technique:
  46                         if not material.inherit_tech:
  47                                 return tech_res
  48
  49                         if self.single_file:
  50                                 raise Exception("Can't export inherited technique to a single file")
  51
  52                         st = Statement("inherit", material.technique)
  53                         for s, t in textures.items():
  54                                 fn = os.path.basename(t.image.filepath)
  55                                 if t.default_filter and fn:
  56                                         st.sub.append(Statement("texture", s, fn))
  57                                 else:
  58                                         st.sub.append(tech_res.create_reference_statement("texture", s, resources[t.name+".tex2d"]))
  59                         if material.override_material:
  60                                 st.sub.append(tech_res.create_reference_statement("material", "surface", mat_res))
  61                         tech_res.statements.append(st)
  62                 else:
  63                         st = Statement("pass", "")
  64                         if self.single_file:
  65                                 st.sub.append(tech_res.create_embed_statement("material", mat_res))
  66                         else:
  67                                 st.sub.append(tech_res.create_reference_statement("material", mat_res))
  68
  69                         if "diffuse_map" in textures:
  70                                 diffuse_tex = textures["diffuse_map"]
  71                                 tex_res = resources[diffuse_tex.name+".tex2d"]
  72                                 ss = Statement("texunit", 0)
  73                                 fn = os.path.basename(diffuse_tex.image.filepath)
  74                                 if self.single_file:
  75                                         ss.sub.append(tech_res.create_embed_statement("texture2d", tex_res))
  76                                 elif diffuse_tex.default_filter and fn:
  77                                         ss.sub.append(Statement("texture", fn))
  78                                 else:
  79                                         ss.sub.append(tech_res.create_reference_statement("texture", tex_res))
  80                                 st.sub.append(ss)
  81
  82                         tech_res.statements.append(st)
  83
  84                 return tech_res
  85
  86         def export_material(self, material):
  87                 from .datafile import Resource, Statement
  88                 mat_res = Resource(material.name+".mat")
  89                 statements = mat_res.statements
  90
  91                 from .util import get_colormap
  92                 cm = get_colormap(material.srgb_colors)
  93
  94                 if any(s.use_map_color_diffuse for s in material.texture_slots if s):
  95                         statements.append(Statement("diffuse", 1.0, 1.0, 1.0, 1.0))
  96                         amb = cm(material.ambient)
  97                         statements.append(Statement("ambient", amb, amb, amb, 1.0))
  98                 else:
  99                         diff = material.diffuse_color*material.diffuse_intensity
 100                         statements.append(Statement("diffuse", cm(diff.r), cm(diff.g), cm(diff.b), 1.0))
 101                         amb = diff*material.ambient
 102                         statements.append(Statement("ambient", cm(amb.r), cm(amb.g), cm(amb.b), 1.0))
 103                 spec = material.specular_color*material.specular_intensity
 104                 statements.append(Statement("specular", cm(spec.r), cm(spec.g), cm(spec.g), 1.0))
 105                 statements.append(Statement("shininess", material.specular_hardness))
 106
 107                 return mat_res
 108
 109
 110 class MaterialMapExporter:
 111         def __init__(self):
 112                 self.single_file = True
 113
 114         def export_technique_resources(self, material_map, resources):
 115                 from .datafile import Resource, Statement, Token
 116                 diffuse_name = material_map.name+"_diffuse.tex2d"
 117                 if diffuse_name not in resources:
 118                         diffuse_res = Resource(diffuse_name)
 119
 120                         fmt = 'SRGB_ALPHA' if material_map.srgb_colors else 'RGBA'
 121
 122                         diffuse_res.statements.append(Statement("min_filter", Token('NEAREST')))
 123                         diffuse_res.statements.append(Statement("mag_filter", Token('NEAREST')))
 124                         diffuse_res.statements.append(Statement("storage", Token(fmt), *material_map.size))
 125                         diffuse_res.statements.append(Statement("raw_data", material_map.diffuse_data))
 126
 127                         resources[diffuse_name] = diffuse_res
 128
 129                 if "basic_white.mat" not in resources:
 130                         mat_res = Resource("basic_white.mat")
 131                         mat_res.statements.append(Statement("diffuse", 1.0, 1.0, 1.0, 1.0))
 132
 133                         resources["basic_white.mat"] = mat_res
 134
 135         def export_technique(self, material_map, *, resources=None):
 136                 from .datafile import Resource, Statement
 137                 tech_res = Resource(material_map.name+".tech")
 138
 139                 mat_res = resources["basic_white.mat"]
 140                 diffuse_res = resources[material_map.name+"_diffuse.tex2d"]
 141
 142                 if material_map.technique:
 143                         if self.single_file:
 144                                 raise Exception("Can't export inherited technique to a single file")
 145
 146                         st = Statement("inherit", material_map.technique)
 147                         st.sub.append(tech_res.create_reference_statement("texture", "diffuse_map", diffuse_res))
 148                         st.sub.append(tech_res.create_reference_statement("material", "surface", mat_res))
 149                         tech_res.statements.append(st)
 150                 else:
 151                         st = Statement("pass", "")
 152                         if self.single_file:
 153                                 st.sub.append(tech_res.create_embed_statement("material", mat_res))
 154                         else:
 155                                 st.sub.append(tech_res.create_reference_statement("material", mat_res))
 156                         ss = Statement("texunit", 0)
 157                         if self.single_file:
 158                                 ss.sub.append(tech_res.create_embed_statement("texture2d", diffuse_res))
 159                         else:
 160                                 ss.sub.append(tech_res.create_reference_statement("texture", diffuse_res))
 161                         st.sub.append(ss)
 162                         tech_res.statements.append(st)
 163
 164                 return tech_res