diff --git a/io_import_images_as_planes.py b/io_import_images_as_planes.py
index 6a4797bf444e4d8117c5f44688851436f573a526..222c76729c5d08f2edf347645d166db6e5d612d0 100644
--- a/io_import_images_as_planes.py
+++ b/io_import_images_as_planes.py
@@ -19,7 +19,7 @@
bl_info = {
"name": "Import Images as Planes",
"author": "Florian Meyer (tstscr)",
- "version": (1, 1),
+ "version": (1, 5),
"blender": (2, 6, 3),
"location": "File > Import > Images as Planes",
"description": "Imports images and creates planes with the appropriate "
@@ -70,10 +70,9 @@ EXTENSIONS = [ext for ext_ls in EXT_LIST.values() for ext in ext_ls]
# -----------------------------------------------------------------------------
-# Functions
+# misc
def set_image_options(self, image):
image.use_premultiply = self.use_premultiply
-
if self.relative:
image.filepath = bpy.path.relpath(image.filepath)
@@ -88,8 +87,48 @@ def is_image_fn_single(fn, ext_key):
return ext in EXT_LIST[ext_key]
-def create_image_textures(self, image):
+def align_planes(self, planes):
+ gap = self.align_offset
+ offset = 0
+ for i, plane in enumerate(planes):
+ offset += (plane.dimensions.x / 2.0) + gap
+ if i == 0:
+ continue
+ move_local = mathutils.Vector((offset, 0.0, 0.0))
+ move_world = plane.location + move_local * plane.matrix_world.inverted()
+ plane.location += move_world
+ offset += (plane.dimensions.x / 2.0)
+
+
+def generate_paths(self):
+ directory, fn = os.path.split(self.filepath)
+
+ if fn and not self.all_in_directory:
+ # test for extension
+ if not is_image_fn_any(fn):
+ return [], directory
+
+ return [self.filepath], directory
+ if not fn or self.all_in_directory:
+ imagepaths = []
+ files_in_directory = os.listdir(directory)
+ # clean files from nonimages
+ files_in_directory = [fn for fn in files_in_directory
+ if is_image_fn_any(fn)]
+ # clean from unwanted extensions
+ if self.extension != "*":
+ files_in_directory = [fn for fn in files_in_directory
+ if is_image_fn_single(fn, self.extension)]
+ # create paths
+ for fn in files_in_directory:
+ imagepaths.append(os.path.join(directory, fn))
+ #print(imagepaths)
+ return imagepaths, directory
+
+# -----------------------------------------------------------------------------
+# Blender
+def create_image_textures(self, image):
fn_full = os.path.normpath(bpy.path.abspath(image.filepath))
# look for texture with importsettings
@@ -152,7 +191,14 @@ def create_material_for_texture(self, texture):
def create_image_plane(self, context, material):
- img = material.texture_slots[0].texture.image
+ engine = context.scene.render.engine
+ if engine == 'BLENDER_RENDER':
+ img = material.texture_slots[0].texture.image
+ if engine == 'CYCLES':
+ nodes = material.node_tree.nodes
+ img_node = [node for node in nodes if node.type == 'TEX_IMAGE'][0]
+ img = img_node.image
+
px, py = img.size
# can't load data
@@ -187,51 +233,123 @@ def create_image_plane(self, context, material):
return plane
-def generate_paths(self):
- directory, fn = os.path.split(self.filepath)
-
- if fn and not self.all_in_directory:
- # test for extension
- if not is_image_fn_any(fn):
- return [], directory
-
- return [self.filepath], directory
-
- if not fn or self.all_in_directory:
- imagepaths = []
- files_in_directory = os.listdir(directory)
- # clean files from nonimages
- files_in_directory = [fn for fn in files_in_directory
- if is_image_fn_any(fn)]
- # clean from unwanted extensions
- if self.extension != "*":
- files_in_directory = [fn for fn in files_in_directory
- if is_image_fn_single(fn, self.extension)]
- # create paths
- for fn in files_in_directory:
- imagepaths.append(os.path.join(directory, fn))
-
- #print(imagepaths)
- return imagepaths, directory
-
-
-def align_planes(self, planes):
- gap = self.align_offset
- offset = 0
- for i, plane in enumerate(planes):
- offset += (plane.dimensions.x / 2.0) + gap
- if i == 0:
- continue
- move_local = mathutils.Vector((offset, 0.0, 0.0))
- move_world = plane.location + move_local * plane.matrix_world.inverted()
- plane.location += move_world
- offset += (plane.dimensions.x / 2.0)
+# -----------------------------------------------------------------------------
+# Cycles
+def get_input_links(node, nodes, links):
+ input_links = []
+ for link in links:
+ if link.to_node == node:
+ input_links.append(link)
+
+ sorted_links = []
+ while input_links:
+ for input in node.inputs:
+ for link in input_links:
+ if link.to_socket == input:
+ sorted_links.append(link)
+ input_links.remove(link)
+ return sorted_links
+
+def get_input_nodes(node, nodes, links):
+ input_nodes = []
+ input_links = get_input_links(node, nodes, links)
+ for link in input_links:
+ input_nodes.append(link.from_node)
+ return input_nodes
+
+def auto_align_nodes(node_tree):
+ print('\nAligning Nodes')
+ x_gap = 200
+ y_gap = 100
+ nodes = node_tree.nodes
+ links = node_tree.links
+ to_node = [node for node in nodes if node.type == 'OUTPUT_MATERIAL'][0]
+
+ def align(to_node, nodes, links):
+ from_nodes = get_input_nodes(to_node, nodes, links)
+
+ for i, node in enumerate(from_nodes):
+ node.location.x = to_node.location.x - x_gap
+ node.location.y = to_node.location.y
+ node.location.y -= i * y_gap
+ node.location.y += (len(from_nodes)-1) * y_gap / (len(from_nodes))
+ align(node, nodes, links)
+
+ align(to_node, nodes, links)
+
+def clean_node_tree(node_tree):
+ nodes = node_tree.nodes
+ for node in nodes:
+ if not node.type == 'OUTPUT_MATERIAL':
+ nodes.remove(node)
+ return node_tree.nodes[0]
+
+def create_cycles_material(self, image):
+ name_compat = bpy.path.display_name_from_filepath(image.filepath)
+ material = None
+ for mat in bpy.data.materials:
+ if mat.name == name_compat and self.overwrite_node_tree:
+ material = mat
+ if not material:
+ material = bpy.data.materials.new(name=name_compat)
+
+ material.use_nodes = True
+ node_tree = material.node_tree
+ out_node = clean_node_tree(node_tree)
+
+ if self.shader == 'BSDF_DIFFUSE':
+ bsdf_diffuse = node_tree.nodes.new('BSDF_DIFFUSE')
+ tex_image = node_tree.nodes.new('TEX_IMAGE')
+ tex_image.image = image
+ tex_image.show_texture = True
+ node_tree.links.new(out_node.inputs[0], bsdf_diffuse.outputs[0])
+ node_tree.links.new(bsdf_diffuse.inputs[0], tex_image.outputs[0])
+
+ if self.shader == 'EMISSION':
+ emission = node_tree.nodes.new('EMISSION')
+ tex_image = node_tree.nodes.new('TEX_IMAGE')
+ tex_image.image = image
+ tex_image.show_texture = True
+ node_tree.links.new(out_node.inputs[0], emission.outputs[0])
+ node_tree.links.new(emission.inputs[0], tex_image.outputs[0])
+
+ if self.shader == 'BSDF_DIFFUSE + BSDF TRANSPARENT':
+ bsdf_diffuse = node_tree.nodes.new('BSDF_DIFFUSE')
+ bsdf_transparent = node_tree.nodes.new('BSDF_TRANSPARENT')
+ mix_shader = node_tree.nodes.new('MIX_SHADER')
+ tex_image = node_tree.nodes.new('TEX_IMAGE')
+ tex_image.image = image
+ tex_image.show_texture = True
+ node_tree.links.new(out_node.inputs[0], mix_shader.outputs[0])
+ node_tree.links.new(mix_shader.inputs[0], tex_image.outputs[1])
+ node_tree.links.new(mix_shader.inputs[1], bsdf_diffuse.outputs[0])
+ node_tree.links.new(mix_shader.inputs[2], bsdf_transparent.outputs[0])
+ node_tree.links.new(bsdf_diffuse.inputs[0], tex_image.outputs[0])
+
+
+ if self.shader == 'EMISSION + BSDF TRANSPARENT':
+ emission = node_tree.nodes.new('EMISSION')
+ bsdf_transparent = node_tree.nodes.new('BSDF_TRANSPARENT')
+ mix_shader = node_tree.nodes.new('MIX_SHADER')
+ tex_image = node_tree.nodes.new('TEX_IMAGE')
+ tex_image.image = image
+ tex_image.show_texture = True
+ node_tree.links.new(out_node.inputs[0], mix_shader.outputs[0])
+ node_tree.links.new(mix_shader.inputs[0], tex_image.outputs[1])
+ node_tree.links.new(mix_shader.inputs[1], emission.outputs[0])
+ node_tree.links.new(mix_shader.inputs[2], bsdf_transparent.outputs[0])
+ node_tree.links.new(emission.inputs[0], tex_image.outputs[0])
+
+
+ auto_align_nodes(node_tree)
+ return material
# -----------------------------------------------------------------------------
# Main
def import_images(self, context):
+ engine = context.scene.render.engine
import_list, directory = generate_paths(self)
images = []
textures = []
@@ -241,16 +359,24 @@ def import_images(self, context):
for path in import_list:
images.append(load_image(path, directory))
- for image in images:
- set_image_options(self, image)
- textures.append(create_image_textures(self, image))
-
- for texture in textures:
- materials.append(create_material_for_texture(self, texture))
+ if engine == 'BLENDER_RENDER':
+ for image in images:
+ set_image_options(self, image)
+ textures.append(create_image_textures(self, image))
+
+ for texture in textures:
+ materials.append(create_material_for_texture(self, texture))
+
+ for material in materials:
+ planes.append(create_image_plane(self, context, material))
+
+ if engine == 'CYCLES':
+ for image in images:
+ materials.append(create_cycles_material(self, image))
+
+ for material in materials:
+ planes.append(create_image_plane(self, context, material))
- for material in materials:
- plane = create_image_plane(self, context, material)
- planes.append(plane)
context.scene.update()
if self.align:
@@ -316,7 +442,7 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
('psd', 'PSD (.psd)', 'Photoshop Document')),
)
use_dimension = BoolProperty(name="Use image dimensions",
- description="Use the images pixels to derive planes size",
+ description="Use the images pixels to derive planes size in Blender Units",
default=False,
)
factor = IntProperty(name="Pixels/BU",
@@ -355,6 +481,29 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
default=False,
)
+ shader = EnumProperty(
+ name="Shader",
+ description="Shader",
+ items=(
+ ('BSDF_DIFFUSE',
+ 'Diffuse',
+ 'Diffuse Shader'),
+ ('EMISSION',
+ 'Emission',
+ 'Emission Shader'),
+ ('BSDF_DIFFUSE + BSDF TRANSPARENT',
+ 'Diffuse + Transparent',
+ 'Diffuse + Transparent Mix'),
+ ('EMISSION + BSDF TRANSPARENT',
+ 'Emission + Transparent',
+ 'Emission + Transparent Mix')),
+ )
+ overwrite_node_tree = BoolProperty(
+ name="overwrite Material",
+ description="overwrite existing Material with new nodetree (based on material name)",
+ default=True,
+ )
+
# -------------
# Image Options
use_premultiply = BoolProperty(name="Premultiply",
@@ -368,6 +517,7 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
)
def draw(self, context):
+ engine = context.scene.render.engine
layout = self.layout
box = layout.box()
@@ -381,14 +531,21 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
row.active = bpy.data.is_saved
row.prop(self, "relative")
- box = layout.box()
- box.label("Material mappings:", icon='MATERIAL')
- box.prop(self, "use_shadeless")
- box.prop(self, "use_transparency")
- box.prop(self, "use_premultiply")
- box.prop(self, "transparency_method", expand=True)
- box.prop(self, "use_transparent_shadows")
-
+ if engine == 'BLENDER_RENDER':
+ box = layout.box()
+ box.label("Material Settings: (Blender)", icon='MATERIAL')
+ box.prop(self, "use_shadeless")
+ box.prop(self, "use_transparency")
+ box.prop(self, "use_premultiply")
+ box.prop(self, "transparency_method", expand=True)
+ box.prop(self, "use_transparent_shadows")
+
+ if engine == 'CYCLES':
+ box = layout.box()
+ box.label("Material Settings: (Cycles)", icon='MATERIAL')
+ box.prop(self, 'shader', expand = True)
+ box.prop(self, 'overwrite_node_tree')
+
box = layout.box()
box.label("Plane dimensions:", icon='ARROW_LEFTRIGHT')
box.prop(self, "use_dimension")
@@ -434,4 +591,4 @@ def unregister():
bpy.types.INFO_MT_file_import.remove(import_images_button)
if __name__ == '__main__':
- register()
+ register()
\ No newline at end of file