three.js/utils/exporters/blender/addons/io_three/exporter/material.py

from .. import constants, logger
from . import base_classes, utilities, api


class Material(base_classes.BaseNode):
    def __init__(self, node, parent):
        logger.debug('Material().__init__(%s)', node)
        base_classes.BaseNode.__init__(self, node, parent, 
            constants.MATERIAL)
        
        self.__common_attributes()
        if self[constants.TYPE] == constants.THREE_PHONG:
            self.__phong_attributes()

        textures = self.parent.options.get(constants.MAPS)
        if textures:
            self.__update_maps()

    def __common_attributes(self):
        logger.debug('Material().__common_attributes()')
        dispatch = {
            constants.PHONG: constants.THREE_PHONG,
            constants.LAMBERT: constants.THREE_LAMBERT,
            constants.BASIC: constants.THREE_BASIC
        }
        shader_type = api.material.type(self.node)
        self[constants.TYPE] = dispatch[shader_type]

        ambient = api.material.ambient_color(self.node)
        self[constants.AMBIENT] = utilities.rgb2int(ambient)

        diffuse = api.material.diffuse_color(self.node)
        self[constants.COLOR] = utilities.rgb2int(diffuse)
    
        emissive = api.material.emissive_color(self.node)
        self[constants.EMISSIVE] = utilities.rgb2int(emissive)

        vertex_color = api.material.use_vertex_colors(self.node)
        self[constants.VERTEX_COLORS] = vertex_color

        self[constants.BLENDING] = api.material.blending(self.node)

        self[constants.DEPTH_TEST] = api.material.depth_test(self.node)

        self[constants.DEPTH_WRITE] = api.material.depth_write(self.node)

    def __phong_attributes(self):
        logger.debug('Material().__phong_attributes()')
        specular = api.material.specular_color(self.node)
        self[constants.SPECULAR] = utilities.rgb2int(specular)
        self[constants.SHININESS] = api.material.specular_coef(self.node)

    def __update_maps(self):
        logger.debug('Material().__update_maps()')

        mapping = (
            (api.material.diffuse_map, constants.MAP),
            (api.material.specular_map, constants.SPECULAR_MAP),
            (api.material.light_map, constants.LIGHT_MAP)
        )

        for func,key in mapping:
            map_node = func(self.node)
            if map_node:
                logger.info('Found map node %s for %s', map_node, key)
                tex_inst = self.scene.texture(map_node.name)
                self[key] = tex_inst[constants.UUID] 

        if self[constants.TYPE] ==  constants.THREE_PHONG:
            mapping = (
                (api.material.bump_map, constants.BUMP_MAP,
                 constants.BUMP_SCALE, api.material.bump_scale),
                (api.material.normal_map, constants.NORMAL_MAP,
                 constants.NORMAL_SCALE, api.material.normal_scale)
            )

            for func, map_key, scale_key, scale_func in mapping:
                map_node = func(self.node)
                if not map_node: continue
                logger.info('Found map node %s for %s', map_node, map_key)
                tex_inst = self.scene.texture(map_node.name)
                self[map_key] = tex_inst[constants.UUID] 
                self[scale_key] = scale_func(self.node)