Skip to content
Snippets Groups Projects
Commit 030cbd62 authored by Florian Meyer's avatar Florian Meyer
Browse files

added preliminary support for cycles materials

parent 6daaf5e6
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
io_import_images_as_planes.py
+ 220
− 63
View file @ 030cbd62
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
bl_info = { bl_info = {
"name": "Import Images as Planes", "name": "Import Images as Planes",
"author": "Florian Meyer (tstscr)", "author": "Florian Meyer (tstscr)",
"version": (1, 1), "version": (1, 5),
"blender": (2, 6, 3), "blender": (2, 6, 3),
"location": "File > Import > Images as Planes", "location": "File > Import > Images as Planes",
"description": "Imports images and creates planes with the appropriate " "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] ...@@ -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): def set_image_options(self, image):
image.use_premultiply = self.use_premultiply image.use_premultiply = self.use_premultiply
if self.relative: if self.relative:
image.filepath = bpy.path.relpath(image.filepath) image.filepath = bpy.path.relpath(image.filepath)
...@@ -88,8 +87,48 @@ def is_image_fn_single(fn, ext_key): ...@@ -88,8 +87,48 @@ def is_image_fn_single(fn, ext_key):
return ext in EXT_LIST[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)) fn_full = os.path.normpath(bpy.path.abspath(image.filepath))
# look for texture with importsettings # look for texture with importsettings
...@@ -152,7 +191,14 @@ def create_material_for_texture(self, texture): ...@@ -152,7 +191,14 @@ def create_material_for_texture(self, texture):
def create_image_plane(self, context, material): 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 px, py = img.size
# can't load data # can't load data
...@@ -187,51 +233,123 @@ def create_image_plane(self, context, material): ...@@ -187,51 +233,123 @@ def create_image_plane(self, context, material):
return plane return plane
def generate_paths(self): # -----------------------------------------------------------------------------
directory, fn = os.path.split(self.filepath) # Cycles
def get_input_links(node, nodes, links):
if fn and not self.all_in_directory: input_links = []
# test for extension for link in links:
if not is_image_fn_any(fn): if link.to_node == node:
return [], directory input_links.append(link)
return [self.filepath], directory sorted_links = []
while input_links:
if not fn or self.all_in_directory: for input in node.inputs:
imagepaths = [] for link in input_links:
files_in_directory = os.listdir(directory) if link.to_socket == input:
# clean files from nonimages sorted_links.append(link)
files_in_directory = [fn for fn in files_in_directory input_links.remove(link)
if is_image_fn_any(fn)] return sorted_links
# clean from unwanted extensions
if self.extension != "*": def get_input_nodes(node, nodes, links):
files_in_directory = [fn for fn in files_in_directory input_nodes = []
if is_image_fn_single(fn, self.extension)] input_links = get_input_links(node, nodes, links)
# create paths for link in input_links:
for fn in files_in_directory: input_nodes.append(link.from_node)
imagepaths.append(os.path.join(directory, fn)) return input_nodes
#print(imagepaths) def auto_align_nodes(node_tree):
return imagepaths, directory print('\nAligning Nodes')
x_gap = 200
y_gap = 100
def align_planes(self, planes): nodes = node_tree.nodes
gap = self.align_offset links = node_tree.links
offset = 0 to_node = [node for node in nodes if node.type == 'OUTPUT_MATERIAL'][0]
for i, plane in enumerate(planes):
offset += (plane.dimensions.x / 2.0) + gap def align(to_node, nodes, links):
if i == 0: from_nodes = get_input_nodes(to_node, nodes, links)
continue
move_local = mathutils.Vector((offset, 0.0, 0.0)) for i, node in enumerate(from_nodes):
move_world = plane.location + move_local * plane.matrix_world.inverted() node.location.x = to_node.location.x - x_gap
plane.location += move_world node.location.y = to_node.location.y
offset += (plane.dimensions.x / 2.0) 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 # Main
def import_images(self, context): def import_images(self, context):
engine = context.scene.render.engine
import_list, directory = generate_paths(self) import_list, directory = generate_paths(self)
images = [] images = []
textures = [] textures = []
...@@ -241,16 +359,24 @@ def import_images(self, context): ...@@ -241,16 +359,24 @@ def import_images(self, context):
for path in import_list: for path in import_list:
images.append(load_image(path, directory)) images.append(load_image(path, directory))
for image in images: if engine == 'BLENDER_RENDER':
set_image_options(self, image) for image in images:
textures.append(create_image_textures(self, image)) 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 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() context.scene.update()
if self.align: if self.align:
...@@ -316,7 +442,7 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper): ...@@ -316,7 +442,7 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
('psd', 'PSD (.psd)', 'Photoshop Document')), ('psd', 'PSD (.psd)', 'Photoshop Document')),
) )
use_dimension = BoolProperty(name="Use image dimensions", 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, default=False,
) )
factor = IntProperty(name="Pixels/BU", factor = IntProperty(name="Pixels/BU",
...@@ -355,6 +481,29 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper): ...@@ -355,6 +481,29 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
default=False, 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 # Image Options
use_premultiply = BoolProperty(name="Premultiply", use_premultiply = BoolProperty(name="Premultiply",
...@@ -368,6 +517,7 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper): ...@@ -368,6 +517,7 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
) )
def draw(self, context): def draw(self, context):
engine = context.scene.render.engine
layout = self.layout layout = self.layout
box = layout.box() box = layout.box()
...@@ -381,14 +531,21 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper): ...@@ -381,14 +531,21 @@ class IMPORT_OT_image_to_plane(Operator, ImportHelper, AddObjectHelper):
row.active = bpy.data.is_saved row.active = bpy.data.is_saved
row.prop(self, "relative") row.prop(self, "relative")
box = layout.box() if engine == 'BLENDER_RENDER':
box.label("Material mappings:", icon='MATERIAL') box = layout.box()
box.prop(self, "use_shadeless") box.label("Material Settings: (Blender)", icon='MATERIAL')
box.prop(self, "use_transparency") box.prop(self, "use_shadeless")
box.prop(self, "use_premultiply") box.prop(self, "use_transparency")
box.prop(self, "transparency_method", expand=True) box.prop(self, "use_premultiply")
box.prop(self, "use_transparent_shadows") 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 = layout.box()
box.label("Plane dimensions:", icon='ARROW_LEFTRIGHT') box.label("Plane dimensions:", icon='ARROW_LEFTRIGHT')
box.prop(self, "use_dimension") box.prop(self, "use_dimension")
...@@ -434,4 +591,4 @@ def unregister(): ...@@ -434,4 +591,4 @@ def unregister():
bpy.types.INFO_MT_file_import.remove(import_images_button) bpy.types.INFO_MT_file_import.remove(import_images_button)
if __name__ == '__main__': if __name__ == '__main__':
register() register()
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment