AutoOp.py

# Copyright 2015 Théo Friberg under GNU GPL 3

if "bpy" in locals():
    import importlib
    importlib.reload(JSONOps)
else:
    from . import JSONOps

import bpy
import os
import glob

class AutomatOperatorFromTexture(bpy.types.Operator):
    """This operator generates automatic materials from textures in Cycles"""

    # Metadata of the operator

    bl_idname = "com.new_automat"
    bl_label = "Automatic Material from Image"
    bl_options = {"UNDO"}

    # Variables used for storing the filepath given by blender's file manager

    filepath: bpy.props.StringProperty(subtype="FILE_PATH")
    filename: bpy.props.StringProperty()
    directory: bpy.props.StringProperty(subtype="FILE_PATH")

    make_seamless: bpy.props.BoolProperty(
        name="Make Seamless",
        description="Make tileable (removes visible borders of the image)",
    )

    def execute(self, context):

        """This is the main runnable method of the operator.

This creates all the node setup."""

        # Create the material

        mat = bpy.data.materials.new(self.filename)

        mat.use_nodes = True
        nodes = mat.node_tree.nodes

        # Empty whatever nodes we allready had.

        for node in nodes.keys():
            nodes.remove(nodes[node])

        nodes_dict = {}

        # Create the main part of the material

        nodes_dict = JSONOps.inflateFile(mat, os.path.dirname(
                                      os.path.realpath(__file__))+os.sep+
                                      "automatic_material.json")



        # We load the images

        image_data = bpy.data.images.load(self.filepath)
        nodes_dict["Color Image"].image = image_data

        # We check if the texture must be made seamless

        if self.make_seamless:
            seamless_vector = JSONOps.inflateFile(mat, os.path.dirname(os.path.realpath(__file__))+os.sep+"seamless_vector.json", -3000, 0)
            links = mat.node_tree.links
            links.new(seamless_vector["Pick Vector"].outputs["Color"], nodes_dict["Color Image"].inputs["Vector"])

        # Below we check potential maps

        modified_fname = self.filename.split(".")

        # Check if we are dealing with maps generated with CrazyBump.
        # If so is the case, the color map is by default suffixed with _COLOR

        known_scheme = False

        if modified_fname[0][-6:] == "_COLOR":

            # We are dealing with CrazyBump and we remove the suffix

            modified_fname[0] = modified_fname[0][:-6]
            known_scheme = True

        other_files = []
        folder = os.path.split(self.filepath)[0]+os.path.sep+"*"
        pattern = folder + ".".join(modified_fname[:-1])+"*."+modified_fname[-1]
        other_files = glob.glob(pattern)

        # We check if we can find a Specularity Map

        specmap = ""

        for file in other_files:
            if "spec" in os.path.split(file)[-1].lower():
                specmap = file
                break

        if len(specmap) > 0:

            spec_map = nodes.new("ShaderNodeTexImage")
            spec_map.location = [nodes_dict["Adjust reflectivity"].location[0],
                                 nodes_dict["Adjust reflectivity"].location[1]+50]
            spec_map.label = "Specularity Map"
            nodes.remove(nodes_dict["Adjust reflectivity"])
            spec_map.image = bpy.data.images.load(specmap)
            links = mat.node_tree.links

            links.new(spec_map.outputs["Color"],
                      nodes_dict["Mix Shaders"].inputs[0])
            if self.make_seamless:
                links.new(seamless_vector["Pick Vector"].outputs["Color"], spec_map.inputs["Vector"])

        # We check if we can find a Normal Map

        normalmap = ""

        for file in other_files:
            if "normal" in os.path.split(file)[-1].lower() or ".".join(os.path.split(file)[1].split(".") [:-1])[-4:] == "_NRM":
                normalmap = file
                break

        if len(normalmap) > 0 and ((not "normal" in self.filename.lower()) or known_scheme):

            normal_map = nodes.new("ShaderNodeTexImage")
            normal_map.location = [nodes_dict["Color Image"].location[0],
                                   nodes_dict["Color Image"].location[1]-240]
            normal_map.label = "Normal Map"
            normal_map.image = bpy.data.images.load(normalmap)
            links = mat.node_tree.links

            normal = nodes.new("ShaderNodeNormalMap")
            normal.location = [nodes_dict["Convert to Bump Map"].location[0],
                               nodes_dict["Convert to Bump Map"].location[1]]
            nodes.remove(nodes_dict["Convert to Bump Map"])
            links.new(normal_map.outputs["Color"],
                      normal.inputs[1])
            links.new(normal.outputs["Normal"],
                      nodes_dict["Diffuse Component"].inputs[2])
            links.new(normal.outputs["Normal"],
                      nodes_dict["Glossy Component"].inputs[2])
            if self.make_seamless:
                links.new(seamless_vector["Pick Vector"].outputs["Color"], normal_map.inputs["Vector"])

        # We check if we can find a Bump Map

        bumpmap = ""

        for file in other_files:
            if "bump" in os.path.split(file.lower())[-1]:
                bumpmap = file
                break

        if len(bumpmap) > 0 and not "bump" in self.filename.lower() and not len(normalmap) > 0:

            bump_map = nodes.new("ShaderNodeTexImage")
            bump_map.location = [nodes_dict["Color Image"].location[0],
                                 nodes_dict["Color Image"].location[1]-240]
            bump_map.label = "Bump Map"
            bump_map.image = bpy.data.images.load(bumpmap)
            links = mat.node_tree.links
            links.new(bump_map.outputs["Color"], nodes_dict["Convert to Bump Map"].inputs[2])
            if self.make_seamless:
                links.new(seamless_vector["Pick Vector"].outputs["Color"], bump_map.inputs["Vector"])

        # Try to add the material to the selected object
        try:
            bpy.context.object.data.materials.append(mat)
        except AttributeError:

            # If there is no object with materials selected,
            # don't add the material to anythinng.

            pass

        # Tell that all went well

        return {"FINISHED"}

    def invoke(self, context, event):

        """This method opens the file browser. After that, the
execute(...) method gets ran, creating the node setup.
It also checks that the render engine is Cycles.  """

        if bpy.context.scene.render.engine == 'CYCLES':
            self.filename = ""
            context.window_manager.fileselect_add(self)
            return {"RUNNING_MODAL"}
        else:
            self.report({'ERROR'}, "Can't generate Cycles material with Blender"
            "internal as active renderer.")
            return {"FINISHED"}