]> git.tdb.fi Git - libs/gl.git/blobdiff - blender/io_mspgl/export_material.py
Move the utility functions in export_material into MaterialExporter
[libs/gl.git] / blender / io_mspgl / export_material.py
index 36dbdc690696346b544e2e367518859df60da693..f01745b98fdf00a1ef7cbc610d14fe03d1c12d38 100644 (file)
@@ -1,87 +1,3 @@
-def create_shadow_method(tech_res, material, resources, detail):
-       from .datafile import Statement
-
-       color_prop = next((p for p in material.properties if p.keyword and "color" in p.keyword), None)
-
-       st = Statement("method", "shadow"+detail)
-       if material.alpha_cutoff>0.0 and color_prop and 'A' in color_prop.tex_channels:
-               st.sub.append(tech_res.create_reference_statement("shader", resources["occluder{}_masked.shader".format(detail)]))
-               ss = Statement("texture", "alpha_map")
-
-               from .export_texture import TextureExporter
-               from .export_texture import SamplerExporter
-               texture_export = TextureExporter()
-               sampler_export = SamplerExporter()
-
-               tex_res = resources[texture_export.get_texture_name(color_prop.texture, color_prop.tex_channels)]
-               ss.sub.append(tech_res.create_reference_statement("texture", tex_res))
-               ss.sub.append(tech_res.create_reference_statement("sampler", resources[sampler_export.get_sampler_name(color_prop.texture)]))
-               st.sub.append(ss)
-
-               ss = Statement("uniforms")
-               ss.sub.append(Statement("uniform", "alpha_cutoff", material.alpha_cutoff))
-               st.sub.append(ss)
-       else:
-               st.sub.append(Statement("shader", "occluder{}.glsl.shader".format(detail)))
-
-       return st;
-
-def create_technique_resource(material, resources):
-       from .datafile import Resource, Statement, Token
-       tech_res = Resource(material.name+".tech", "technique")
-
-       mat_res = resources[material.name+".mat"]
-
-       blend_st = None
-       if material.blend_type=='ALPHA':
-               blend_st = Statement("blend", Token("SRC_ALPHA"), Token("ONE_MINUS_SRC_ALPHA"))
-       elif material.blend_type=='ADDITIVE':
-               blend_st = Statement("blend", Token("ONE"), Token("ONE"))
-       elif material.blend_type=='ADDITIVE_ALPHA':
-               blend_st = Statement("blend", Token("SRC_ALPHA"), Token("ONE"))
-
-       if material.render_mode=='CUSTOM':
-               for m in material.render_methods:
-                       st = Statement("method", m.tag)
-                       if mat_res and m.use_material:
-                               st.sub.append(tech_res.create_reference_statement("material", mat_res))
-
-                       if m.tag=="blended" and blend_st:
-                               st.sub.append(blend_st)
-
-                       shader = m.shader
-                       if shader.endswith(".glsl"):
-                               shader += ".shader"
-                       st.sub.append(Statement("shader", shader))
-
-                       if material.uniforms:
-                               ss = Statement("uniforms")
-                               for u in material.uniforms:
-                                       ss.sub.append(Statement("uniform", u.name, *u.values[:u.size]))
-                               st.sub.append(ss)
-
-                       tech_res.statements.append(st)
-       else:
-               base_method = "blended" if material.blend_type!='NONE' else ""
-               st = Statement("method", base_method)
-               if mat_res:
-                       st.sub.append(tech_res.create_embed_statement("material", mat_res))
-
-               if blend_st:
-                       st.sub.append(blend_st)
-               if material.receive_shadows:
-                       st.sub.append(Statement("receive_shadows", True))
-               if material.image_based_lighting:
-                       st.sub.append(Statement("image_based_lighting", True))
-
-               tech_res.statements.append(st)
-
-               if material.cast_shadows:
-                       tech_res.statements.append(create_shadow_method(tech_res, material, resources, ""));
-                       tech_res.statements.append(create_shadow_method(tech_res, material, resources, "_thsm"));
-
-       return tech_res
-
 class MaterialExporter:
        def export_technique_resources(self, ctx, material, resources):
                from .export_texture import SamplerExporter, TextureExporter
@@ -125,7 +41,96 @@ class MaterialExporter:
                if type(material)!=Material:
                        material = Material(material)
 
-               return create_technique_resource(material, resources)
+               from .datafile import Resource, Statement, Token
+               tech_res = Resource(material.name+".tech", "technique")
+
+               mat_res = resources[material.name+".mat"]
+
+               blend_st = None
+               if material.blend_type=='ALPHA':
+                       blend_st = Statement("blend", Token("SRC_ALPHA"), Token("ONE_MINUS_SRC_ALPHA"))
+               elif material.blend_type=='ADDITIVE':
+                       blend_st = Statement("blend", Token("ONE"), Token("ONE"))
+               elif material.blend_type=='ADDITIVE_ALPHA':
+                       blend_st = Statement("blend", Token("SRC_ALPHA"), Token("ONE"))
+
+               if material.render_mode=='CUSTOM':
+                       for m in material.render_methods:
+                               st = Statement("method", m.tag)
+                               if mat_res and m.use_material:
+                                       st.sub.append(tech_res.create_reference_statement("material", mat_res))
+
+                               if m.tag=="blended" and blend_st:
+                                       st.sub.append(blend_st)
+
+                               shader = m.shader
+                               if shader.endswith(".glsl"):
+                                       shader += ".shader"
+                               st.sub.append(Statement("shader", shader))
+
+                               if material.uniforms:
+                                       ss = Statement("uniforms")
+                                       for u in material.uniforms:
+                                               ss.sub.append(Statement("uniform", u.name, *u.values[:u.size]))
+                                       st.sub.append(ss)
+
+                               if material.face_cull=='BACK':
+                                       st.sub.append(Statement("face_cull", Token("CULL_BACK")))
+
+                               tech_res.statements.append(st)
+               else:
+                       base_method = "blended" if material.blend_type!='NONE' else ""
+                       st = Statement("method", base_method)
+                       if mat_res:
+                               st.sub.append(tech_res.create_embed_statement("material", mat_res))
+
+                       if blend_st:
+                               st.sub.append(blend_st)
+                       if material.receive_shadows:
+                               st.sub.append(Statement("receive_shadows", True))
+                       if material.image_based_lighting:
+                               st.sub.append(Statement("image_based_lighting", True))
+                       if material.face_cull=='BACK':
+                               st.sub.append(Statement("face_cull", Token("CULL_BACK")))
+
+                       tech_res.statements.append(st)
+
+                       if material.cast_shadows:
+                               tech_res.statements.append(self.create_shadow_method(tech_res, material, resources, ""));
+                               tech_res.statements.append(self.create_shadow_method(tech_res, material, resources, "_thsm"));
+
+               return tech_res
+
+       def create_shadow_method(self, tech_res, material, resources, detail):
+               from .datafile import Statement, Token
+
+               color_prop = next((p for p in material.properties if p.keyword and "color" in p.keyword), None)
+
+               st = Statement("method", "shadow"+detail)
+               if material.alpha_cutoff>0.0 and color_prop and 'A' in color_prop.tex_channels:
+                       st.sub.append(tech_res.create_reference_statement("shader", resources["occluder{}_masked.shader".format(detail)]))
+                       ss = Statement("texture", "alpha_map")
+
+                       from .export_texture import TextureExporter
+                       from .export_texture import SamplerExporter
+                       texture_export = TextureExporter()
+                       sampler_export = SamplerExporter()
+
+                       tex_res = resources[texture_export.get_texture_name(color_prop.texture, color_prop.tex_channels)]
+                       ss.sub.append(tech_res.create_reference_statement("texture", tex_res))
+                       ss.sub.append(tech_res.create_reference_statement("sampler", resources[sampler_export.get_sampler_name(color_prop.texture)]))
+                       st.sub.append(ss)
+
+                       ss = Statement("uniforms")
+                       ss.sub.append(Statement("uniform", "alpha_cutoff", material.alpha_cutoff))
+                       st.sub.append(ss)
+               else:
+                       st.sub.append(Statement("shader", "occluder{}.glsl.shader".format(detail)))
+
+               if material.face_cull=='BACK':
+                       st.sub.append(Statement("face_cull", Token("CULL_BACK")))
+
+               return st;
 
        def export_material(self, material, resources):
                from .datafile import Resource, Statement, Token
@@ -173,42 +178,3 @@ class MaterialExporter:
                shader_res.statements.append(st)
 
                return shader_res
-
-
-class MaterialAtlasExporter:
-       def __init__(self):
-               pass
-
-       def export_technique_resources(self, material_atlas, resources):
-               from .datafile import Resource, Statement, Token
-               base_color_name = material_atlas.name+"_base_color.tex"
-               base_color_res = resources.get(base_color_name)
-               if not base_color_res:
-                       base_color_res = Resource(base_color_name, "texture")
-
-                       base_color_res.statements.append(Statement("type", Token("\\2d")))
-                       base_color_res.statements.append(Statement("storage", Token('SRGB_ALPHA'), *material_atlas.size))
-                       base_color_res.statements.append(Statement("raw_data", material_atlas.base_color_data))
-
-                       resources[base_color_name] = base_color_res
-
-               sampler_name = "nearest.samp"
-               sampler_res = resources.get(sampler_name)
-               if not sampler_res:
-                       sampler_res = Resource(sampler_name, "sampler")
-
-                       sampler_res.statements.append(Statement("filter", Token('NEAREST')))
-
-                       resources[sampler_name] = sampler_res
-
-               mat_name = material_atlas.name+".mat"
-               if mat_name not in resources:
-                       mat_res = Resource(mat_name, "material")
-                       mat_res.statements.append(Statement("type", Token('pbr')))
-                       mat_res.statements.append(mat_res.create_reference_statement("base_color_map", base_color_res))
-                       mat_res.statements.append(mat_res.create_reference_statement("sampler", sampler_res))
-
-                       resources[mat_name] = mat_res
-
-       def export_technique(self, material_atlas, resources):
-               return create_technique_resource(material_atlas, resources)