SWBF-msh-Blender-IO/addons/io_scene_swbf_msh/msh_material_gather.py

""" Gathers the Blender materials and returns them as a dictionary of
    strings and Material objects. """

import bpy
from typing import Dict
from .msh_material import *

from .msh_material_utilities import _RENDERTYPES_MAPPING

import os

def gather_materials() -> Dict[str, Material]:
    """ Gathers the Blender materials and returns them as
        a dictionary of strings and Material objects. """

    materials: Dict[str, Material] = {}

    for blender_material in bpy.data.materials:
        materials[blender_material.name] = read_material(blender_material)

    return materials

def read_material(blender_material: bpy.types.Material) -> Material:
    """ Reads a the swbf_msh properties from a Blender material and
        returns a Material object. """

    result = Material()

    if blender_material.swbf_msh_mat is None:
        return result

    props = blender_material.swbf_msh_mat

    result.specular_color = props.specular_color.copy()
    result.rendertype = _read_material_props_rendertype(props)
    result.flags = _read_material_props_flags(props)
    result.data = _read_material_props_data(props)

    if "UNSUPPORTED" not in props.rendertype:
        result.texture0 = os.path.basename(props.diffuse_map)
        result.texture1 = _read_normal_map_or_distortion_map_texture(props)
        result.texture2 = _read_detail_texture(props)
        result.texture3 = _read_envmap_texture(props)
    
    else:
        result.texture0 = os.path.basename(props.texture_0)
        result.texture1 = os.path.basename(props.texture_1)
        result.texture2 = os.path.basename(props.texture_2)
        result.texture3 = os.path.basename(props.texture_3)

    return result


def _read_material_props_rendertype(props) -> Rendertype:
    if "UNSUPPORTED" in props.rendertype:
        return Rendertype(props.rendertype_value)
    else:
        return _RENDERTYPES_MAPPING[props.rendertype]

def _read_material_props_flags(props) -> MaterialFlags:
    if "REFRACTION" in props.rendertype:
        return MaterialFlags.BLENDED_TRANSPARENCY

    flags = MaterialFlags.NONE

    if props.blended_transparency:
        flags |= MaterialFlags.BLENDED_TRANSPARENCY
    if props.additive_transparency:
        flags |= MaterialFlags.ADDITIVE_TRANSPARENCY
    if props.hardedged_transparency:
        flags |= MaterialFlags.HARDEDGED_TRANSPARENCY
    if props.unlit:
        flags |= MaterialFlags.UNLIT
    if props.glow:
        flags |= MaterialFlags.GLOW
    if props.perpixel:
        flags |= MaterialFlags.PERPIXEL
    if props.specular:
        flags |= MaterialFlags.SPECULAR
    if props.doublesided:
        flags |= MaterialFlags.DOUBLESIDED

    return flags

def _read_material_props_data(props) -> Tuple[int, int]:
    if "UNSUPPORTED" in props.rendertype:
        return (props.data_value_0, props.data_value_1)
    if "SCROLLING" in props.rendertype:
        return (props.scroll_speed_u, props.scroll_speed_v)
    if "BLINK" in props.rendertype:
        return (props.blink_min_brightness, props.blink_speed)
    if "NORMALMAPPED_TILED" in props.rendertype:
        return (props.normal_map_tiling_u, props.normal_map_tiling_v)
    if "REFRACTION" in props.rendertype:
        return (0, 0)
    if "ANIMATED" in props.rendertype:
        return (int(str(props.animation_length).split("_")[1]), props.animation_speed)

    return (props.detail_map_tiling_u, props.detail_map_tiling_v)



def _read_normal_map_or_distortion_map_texture(props) -> str:
    if "REFRACTION" in props.rendertype:
        return os.path.basename(props.distortion_map)
    if "NORMALMAPPED" in props.rendertype:
        return os.path.basename(props.normal_map)

    return ""

def _read_detail_texture(props) -> str:
    if "REFRACTION" in props.rendertype:
        return ""

    return os.path.basename(props.detail_map)

def _read_envmap_texture(props) -> str:
    if "ENVMAPPED" not in props.rendertype:
        return ""

    return os.path.basename(props.environment_map)