【转载】Blender 程序化分配材质

中文译文链接：Blender程序化分配材质 — BimAnt翻译整理
英文原文链接：Assign a material to an object in Blender using Python

本文介绍如何使用 Python 创建新材质、添加着色器、创建新对象 并 将材质分配给该对象。

首先，创建一种新材质（定义一个创建新材质的函数）。该函数将 id 字符串作为新材质的名称：

import bpy

def newMaterial(id):

    mat = bpy.data.materials.get(id)

    if mat is None:
        mat = bpy.data.materials.new(name=id)

    mat.use_nodes = True

    if mat.node_tree:
        mat.node_tree.links.clear()
        mat.node_tree.nodes.clear()

    return mat

然后为材质添加着色器。输入着色器的类型（即漫反射、发光、光滑(泽)）和 rgb 颜色：

def newShader(id, type, r, g, b):

    mat = newMaterial(id)

    nodes = mat.node_tree.nodes
    links = mat.node_tree.links
    # 材质输出
    output = nodes.new(type='ShaderNodeOutputMaterial')
    
    # 根据所选类型新建对应着色器节点
    if type == "diffuse":
        shader = nodes.new(type='ShaderNodeBsdfDiffuse')
        nodes["Diffuse BSDF"].inputs[0].default_value = (r, g, b, 1)

    elif type == "emission":
        shader = nodes.new(type='ShaderNodeEmission')
        nodes["Emission"].inputs[0].default_value = (r, g, b, 1)
        nodes["Emission"].inputs[1].default_value = 1

    elif type == "glossy":
        shader = nodes.new(type='ShaderNodeBsdfGlossy')
        nodes["Glossy BSDF"].inputs[0].default_value = (r, g, b, 1)
        nodes["Glossy BSDF"].inputs[1].default_value = 0
    
    # 将新建的着色器节点链接到输出
    links.new(shader.outputs[0], output.inputs[0])

    return mat

然后定义绘制函数，其中除了会创建材质，还会创建对象，并为该激活对象分配材质，最后调用该绘制函数：

def drawObject():

    mat = newShader("Shader1", "diffuse", 1, 1, 1)
    bpy.ops.mesh.primitive_cube_add(size=2, align='WORLD', location=(0, 0, 0))
    bpy.context.active_object.data.materials.append(mat)

drawObject()