godotengine
/
godot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Flake8 compliance (except for unused variables)

This commit is contained in:
Nobody Really 2016-07-09 12:25:21 +02:00
parent 52fdc65a5d
commit c973f3a1da
3 changed files with 1086 additions and 693 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

245
tools/export/blender25/godot_export_manager.py
View File

@ -18,24 +18,27 @@
# Script copyright (c) Andreas Esau
import bpy
from bpy.props import (StringProperty, BoolProperty, EnumProperty,
FloatProperty, IntProperty, CollectionProperty)
import os
from bpy.app.handlers import persistent
from mathutils import Matrix
bl_info = {
"name": "Godot Export Manager",
"author": "Andreas Esau",
"version": (1, 0),
"blender": (2, 7, 0),
"location": "Scene Properties > Godot Export Manager",
"description": "Godot Export Manager uses the Better Collada Exporter to manage Export Groups and automatically export the objects groups to Collada Files.",
"description": "Godot Export Manager uses the Better Collada Exporter"
"to manage Export Groups and automatically export the objects groups"
"to Collada Files.",
"warning": "",
"wiki_url": ("http://www.godotengine.org"),
"tracker_url": "",
"category": "Import-Export"}
import bpy
from bpy.props import StringProperty, BoolProperty, EnumProperty, FloatProperty, FloatVectorProperty, IntProperty, CollectionProperty, PointerProperty
import os
from bpy.app.handlers import persistent
from mathutils import Vector, Matrix
class godot_export_manager(bpy.types.Panel):
bl_label = "Godot Export Manager"
@ -45,8 +48,8 @@ class godot_export_manager(bpy.types.Panel):
bpy.types.Scene.godot_export_on_save = BoolProperty(default=False)
### draw function for all ui elements
def draw(self, context):
""" Draw function for all ui elements """
layout = self.layout
split = self.layout.split()
scene = bpy.data.scenes[0]
@ -59,9 +62,13 @@ class godot_export_manager(bpy.types.Panel):
row = layout.row()
col = row.column(align=True)
op = col.operator("scene.godot_add_objects_to_group", text="Add selected objects to Group", icon="COPYDOWN")
op = col.operator("scene.godot_add_objects_to_group",
text="Add selected objects to Group",
icon="COPYDOWN")
op = col.operator("scene.godot_delete_objects_from_group", text="Delete selected objects from Group", icon="PASTEDOWN")
op = col.operator("scene.godot_delete_objects_from_group",
text="Delete selected objects from Group",
icon="PASTEDOWN")
row = layout.row()
col = row.column()
@ -70,11 +77,15 @@ class godot_export_manager(bpy.types.Panel):
row = layout.row()
col = row.column()
col.template_list("UI_List_Godot", "dummy", scene, "godot_export_groups", scene, "godot_export_groups_index", rows=1, maxrows=10, type='DEFAULT')
col.template_list("UI_List_Godot", "dummy", scene,
"godot_export_groups", scene,
"godot_export_groups_index", rows=1, maxrows=10,
type='DEFAULT')
col = row.column(align=True)
col.operator("scene.godot_add_export_group", text="", icon="ZOOMIN")
col.operator("scene.godot_delete_export_group", text="", icon="ZOOMOUT")
col.operator("scene.godot_delete_export_group", text="",
icon="ZOOMOUT")
col.operator("scene.godot_export_all_groups", text="", icon="EXPORT")
if len(scene.godot_export_groups) > 0:
@ -113,9 +124,10 @@ class godot_export_manager(bpy.types.Panel):
col.prop(group, "use_metadata")
### Custom template_list look
class UI_List_Godot(bpy.types.UIList):
def draw_item(self, context, layout, data, item, icon, active_data, active_propname, index):
""" Custom template_list look """
def draw_item(self, context, layout, data, item, icon, active_data,
active_propname, index):
ob = data
slot = item
col = layout.row(align=True)
@ -123,11 +135,14 @@ class UI_List_Godot(bpy.types.UIList):
col.label(text=item.name, icon="GROUP")
col.prop(item, "active", text="")
op = col.operator("scene.godot_select_group_objects", text="", emboss=False, icon="RESTRICT_SELECT_OFF")
op = col.operator("scene.godot_select_group_objects",
text="", emboss=False, icon="RESTRICT_SELECT_OFF")
op.idx = index
op = col.operator("scene.godot_export_group", text="", emboss=False, icon="EXPORT")
op = col.operator("scene.godot_export_group", text="", emboss=False,
icon="EXPORT")
op.idx = index
class add_objects_to_group(bpy.types.Operator):
bl_idname = "scene.godot_add_objects_to_group"
bl_label = "Add Objects to Group"
@ -141,20 +156,24 @@ class add_objects_to_group(bpy.types.Operator):
objects_str = ""
if len(scene.godot_export_groups) > 0:
for i, object in enumerate(context.selected_objects):
if object.name not in scene.godot_export_groups[scene.godot_export_groups_index].nodes:
node = scene.godot_export_groups[scene.godot_export_groups_index].nodes.add()
if object.name not in scene.godot_export_groups[
scene.godot_export_groups_index].nodes:
node = scene.godot_export_groups[
scene.godot_export_groups_index].nodes.add()
node.name = object.name
if i == 0:
objects_str += object.name
else:
objects_str += ", "+object.name
self.report({'INFO'}, objects_str + " added to group." )
self.report({'INFO'}, objects_str + " added to group.")
if self.undo:
bpy.ops.ed.undo_push(message="Objects added to group")
else:
self.report({'WARNING'}, "Create a group first." )
return{'FINISHED'}
self.report({'WARNING'}, "Create a group first.")
return {'FINISHED'}
class del_objects_from_group(bpy.types.Operator):
bl_idname = "scene.godot_delete_objects_from_group"
@ -172,21 +191,25 @@ class del_objects_from_group(bpy.types.Operator):
objects_str = ""
j = 0
for i, node in enumerate(scene.godot_export_groups[scene.godot_export_groups_index].nodes):
for i, node in enumerate(scene.godot_export_groups[
scene.godot_export_groups_index].nodes):
if node.name in selected_objects:
scene.godot_export_groups[scene.godot_export_groups_index].nodes.remove(i)
scene.godot_export_groups[
scene.godot_export_groups_index].nodes.remove(i)
if j == 0:
objects_str += object.name
else:
objects_str += ", "+object.name
j+=1
j += 1
self.report({'INFO'}, objects_str + " deleted from group." )
self.report({'INFO'}, objects_str + " deleted from group.")
bpy.ops.ed.undo_push(message="Objects deleted from group")
else:
self.report({'WARNING'}, "There is no group to delete from." )
return{'FINISHED'}
self.report({'WARNING'}, "There is no group to delete from.")
return {'FINISHED'}
class select_group_objects(bpy.types.Operator):
bl_idname = "scene.godot_select_group_objects"
@ -203,7 +226,8 @@ class select_group_objects(bpy.types.Operator):
if node.name in bpy.data.objects:
bpy.data.objects[node.name].select = True
context.scene.objects.active = bpy.data.objects[node.name]
return{'FINISHED'}
return {'FINISHED'}
class export_groups_autosave(bpy.types.Operator):
@ -217,9 +241,10 @@ class export_groups_autosave(bpy.types.Operator):
for i in range(len(scene.godot_export_groups)):
if scene.godot_export_groups[i].active:
bpy.ops.scene.godot_export_group(idx=i)
self.report({'INFO'}, "All Groups exported." )
self.report({'INFO'}, "All Groups exported.")
bpy.ops.ed.undo_push(message="Export all Groups")
return{'FINISHED'}
return {'FINISHED'}
class export_all_groups(bpy.types.Operator):
@ -233,46 +258,49 @@ class export_all_groups(bpy.types.Operator):
for i in range(0, len(scene.godot_export_groups)):
bpy.ops.scene.godot_export_group(idx=i, export_all=True)
self.report({'INFO'}, "All Groups exported." )
return{'FINISHED'}
self.report({'INFO'}, "All Groups exported.")
return {'FINISHED'}
class export_group(bpy.types.Operator):
bl_idname = "scene.godot_export_group"
bl_label = "Export Group"
bl_description = "Exports the active group to destination folder as Collada file."
bl_description = "Exports the active group to destination folder"\
" as Collada file."
idx = IntProperty(default=0)
export_all = BoolProperty(default=False)
def copy_object_recursive(self, ob, parent, single_user = True):
def copy_object_recursive(self, ob, parent, single_user=True):
new_ob = bpy.data.objects[ob.name].copy()
if single_user or ob.type=="ARMATURE":
if single_user or ob.type == "ARMATURE":
new_mesh_data = new_ob.data.copy()
new_ob.data = new_mesh_data
bpy.context.scene.objects.link(new_ob)
if ob != parent:
new_ob.parent = parent
new_ob.parent = parent
else:
new_ob.parent = None
for child in ob.children:
self.copy_object_recursive(child, new_ob, single_user)
new_ob.select = True
return new_ob
def delete_object(self, ob):
if ob != None:
if ob is not None:
for child in ob.children:
self.delete_object(child)
bpy.context.scene.objects.unlink(ob)
bpy.data.objects.remove(ob)
def convert_group_to_node(self, group):
if group.dupli_group != None:
if group.dupli_group is not None:
for object in group.dupli_group.objects:
if object.parent == None:
if object.parent is None:
object = self.copy_object_recursive(object, object, True)
matrix = Matrix(object.matrix_local)
object.matrix_local = Matrix()
@ -297,15 +325,16 @@ class export_group(bpy.types.Operator):
bpy.ops.ed.undo_push(message="Clear not existent Group Nodes.")
path = group[self.idx].export_path
if (path.find("//")==0 or path.find("\\\\")==0):
#if relative, convert to absolute
if (path.find("//") == 0 or path.find("\\\\") == 0):
# If relative, convert to absolute
path = bpy.path.abspath(path)
path = path.replace("\\", "/")
### if path exists and group export name is set the group will be exported
if os.path.exists(path) and group[self.idx].export_name != "":
# If path exists and group export name is set the group will be
# exported
if os.path.exists(path) and group[self.idx].export_name != "":
context.scene.layers = [True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True, True]
context.scene.layers = [True] * 20
if group[self.idx].export_name.endswith(".dae"):
path = os.path.join(path, group[self.idx].export_name)
@ -319,17 +348,19 @@ class export_group(bpy.types.Operator):
object.select = False
context.scene.objects.active = None
### make particle duplicates, parent and select them
# Make particle duplicates, parent and select them
nodes_to_be_added = []
if group[self.idx].use_include_particle_duplicates:
for i, object in enumerate(group[self.idx].nodes):
if bpy.data.objects[object.name].type != "EMPTY":
context.scene.objects.active = bpy.data.objects[object.name]
context.scene.objects.active = bpy.data.objects[
object.name]
bpy.data.objects[object.name].select = True
bpy.ops.object.duplicates_make_real()
for object in context.selected_objects:
nodes_to_be_added.append(object)
bpy.ops.object.parent_set(type="OBJECT", keep_transform=False)
bpy.ops.object.parent_set(type="OBJECT",
keep_transform=False)
for object in context.selected_objects:
object.select = False
@ -338,17 +369,35 @@ class export_group(bpy.types.Operator):
for object in nodes_to_be_added:
object.select = True
### select all other nodes from the group
# Select all other nodes from the group
for i, object in enumerate(group[self.idx].nodes):
if bpy.data.objects[object.name].type == "EMPTY":
self.convert_group_to_node(bpy.data.objects[object.name])
else:
bpy.data.objects[object.name].select = True
bpy.ops.object.transform_apply(location=group[self.idx].apply_loc, rotation=group[self.idx].apply_rot, scale=group[self.idx].apply_scale)
bpy.ops.export_scene.dae(check_existing=True, filepath=path, filter_glob="*.dae", object_types=group[self.idx].object_types, use_export_selected=group[self.idx].use_export_selected, use_mesh_modifiers=group[self.idx].use_mesh_modifiers, use_tangent_arrays=group[self.idx].use_tangent_arrays, use_triangles=group[self.idx].use_triangles, use_copy_images=group[self.idx].use_copy_images, use_active_layers=group[self.idx].use_active_layers, use_anim=group[self.idx].use_anim, use_anim_action_all=group[self.idx].use_anim_action_all, use_anim_skip_noexp=group[self.idx].use_anim_skip_noexp, use_anim_optimize=group[self.idx].use_anim_optimize, anim_optimize_precision=group[self.idx].anim_optimize_precision, use_metadata=group[self.idx].use_metadata)
bpy.ops.object.transform_apply(location=group[self.idx].apply_loc,
rotation=group[self.idx].apply_rot,
scale=group[self.idx].apply_scale)
bpy.ops.export_scene.dae(
check_existing=True, filepath=path, filter_glob="*.dae",
object_types=group[self.idx].object_types,
use_export_selected=group[self.idx].use_export_selected,
use_mesh_modifiers=group[self.idx].use_mesh_modifiers,
use_tangent_arrays=group[self.idx].use_tangent_arrays,
use_triangles=group[self.idx].use_triangles,
use_copy_images=group[self.idx].use_copy_images,
use_active_layers=group[self.idx].use_active_layers,
use_anim=group[self.idx].use_anim,
use_anim_action_all=group[self.idx].use_anim_action_all,
use_anim_skip_noexp=group[self.idx].use_anim_skip_noexp,
use_anim_optimize=group[self.idx].use_anim_optimize,
anim_optimize_precision=group[
self.idx].anim_optimize_precision,
use_metadata=group[self.idx].use_metadata)
self.report({'INFO'}, '"'+group[self.idx].name+'"' + " Group exported." )
self.report({'INFO'},
'"' + group[self.idx].name + '" Group exported.')
msg = "Export Group "+group[self.idx].name
bpy.ops.ed.undo_push(message="")
@ -356,14 +405,16 @@ class export_group(bpy.types.Operator):
bpy.ops.ed.undo_push(message=msg)
else:
self.report({'INFO'}, "Define Export Name and Export Path." )
self.report({'INFO'}, "Define Export Name and Export Path.")
return{'FINISHED'}
class add_export_group(bpy.types.Operator):
bl_idname = "scene.godot_add_export_group"
bl_label = "Adds a new export Group"
bl_description = "Creates a new Export Group with the selected Objects assigned to it."
bl_description = "Creates a new Export Group with the selected"\
" Objects assigned to it."
def execute(self, context):
scene = context.scene
@ -375,6 +426,7 @@ class add_export_group(bpy.types.Operator):
node.name = object.name
scene.godot_export_groups_index = len(scene.godot_export_groups)-1
bpy.ops.ed.undo_push(message="Create New Export Group")
return{'FINISHED'}
@ -385,6 +437,7 @@ class del_export_group(bpy.types.Operator):
def invoke(self, context, event):
wm = context.window_manager
return wm.invoke_confirm(self, event)
def execute(self, context):
@ -394,7 +447,8 @@ class del_export_group(bpy.types.Operator):
if scene.godot_export_groups_index > 0:
scene.godot_export_groups_index -= 1
bpy.ops.ed.undo_push(message="Delete Export Group")
return{'FINISHED'}
return {'FINISHED'}
class godot_node_list(bpy.types.PropertyGroup):
@ -408,28 +462,74 @@ class godot_export_groups(bpy.types.PropertyGroup):
export_path = StringProperty(subtype="DIR_PATH")
active = BoolProperty(default=True, description="Export Group")
object_types = EnumProperty(name="Object Types", options={'ENUM_FLAG'}, items=(('EMPTY', "Empty", ""), ('CAMERA', "Camera", ""), ('LAMP', "Lamp", ""), ('ARMATURE', "Armature", ""), ('MESH', "Mesh", ""), ('CURVE', "Curve", ""), ), default={'EMPTY', 'CAMERA', 'LAMP', 'ARMATURE', 'MESH', 'CURVE'})
object_types = EnumProperty(name="Object Types", options={'ENUM_FLAG'},
items=(('EMPTY', "Empty", ""),
('CAMERA', "Camera", ""),
('LAMP', "Lamp", ""),
('ARMATURE', "Armature", ""),
('MESH', "Mesh", ""),
('CURVE', "Curve", ""), ),
default={'EMPTY', 'CAMERA', 'LAMP',
'ARMATURE', 'MESH', 'CURVE'})
apply_scale = BoolProperty(name="Apply Scale", description="Apply Scale before export.", default=False)
apply_rot = BoolProperty(name="Apply Rotation", description="Apply Rotation before export.", default=False)
apply_loc = BoolProperty(name="Apply Location", description="Apply Location before export.", default=False)
apply_scale = BoolProperty(name="Apply Scale",
description="Apply Scale before export.",
default=False)
apply_rot = BoolProperty(name="Apply Rotation",
description="Apply Rotation before export.",
default=False)
apply_loc = BoolProperty(name="Apply Location",
description="Apply Location before export.",
default=False)
use_export_selected = BoolProperty(name="Selected Objects", description="Export only selected objects (and visible in active layers if that applies).", default=True)
use_mesh_modifiers = BoolProperty(name="Apply Modifiers", description="Apply modifiers to mesh objects (on a copy!).", default=True)
use_tangent_arrays = BoolProperty(name="Tangent Arrays", description="Export Tangent and Binormal arrays (for normalmapping).", default=False)
use_triangles = BoolProperty(name="Triangulate", description="Export Triangles instead of Polygons.", default=False)
use_export_selected = BoolProperty(name="Selected Objects",
description="Export only selected"
"objects (and visible in active layers "
"if that applies).", default=True)
use_mesh_modifiers = BoolProperty(name="Apply Modifiers",
description="Apply modifiers to mesh"
" objects (on a copy!).", default=True)
use_tangent_arrays = BoolProperty(name="Tangent Arrays",
description="Export Tangent and Binormal"
" arrays (for normalmapping).",
default=False)
use_triangles = BoolProperty(name="Triangulate",
description="Export Triangles instead of"
" Polygons.", default=False)
use_copy_images = BoolProperty(name="Copy Images", description="Copy Images (create images/ subfolder)", default=False)
use_active_layers = BoolProperty(name="Active Layers", description="Export only objects on the active layers.", default=True)
use_anim = BoolProperty(name="Export Animation", description="Export keyframe animation", default=False)
use_anim_action_all = BoolProperty(name="All Actions", description=("Export all actions for the first armature found in separate DAE files"), default=False)
use_anim_skip_noexp = BoolProperty(name="Skip (-noexp) Actions", description="Skip exporting of actions whose name end in (-noexp). Useful to skip control animations.", default=True)
use_anim_optimize = BoolProperty(name="Optimize Keyframes", description="Remove double keyframes", default=True)
use_copy_images = BoolProperty(name="Copy Images",
description="Copy Images (create images/ "
"subfolder)", default=False)
use_active_layers = BoolProperty(name="Active Layers",
description="Export only objects on the"
" active layers.", default=True)
use_anim = BoolProperty(name="Export Animation",
description="Export keyframe animation",
default=False)
use_anim_action_all = BoolProperty(name="All Actions",
description=("Export all actions for "
"the first armature found"
" in separate DAE files"),
default=False)
use_anim_skip_noexp = BoolProperty(name="Skip (-noexp) Actions",
description="Skip exporting of"
" actions whose name end in (-noexp)."
" Useful to skip control animations.",
default=True)
use_anim_optimize = BoolProperty(name="Optimize Keyframes",
description="Remove double keyframes",
default=True)
anim_optimize_precision = FloatProperty(name="Precision", description=("Tolerence for comparing double keyframes (higher for greater accuracy)"), min=1, max=16, soft_min=1, soft_max=16, default=6.0)
anim_optimize_precision = FloatProperty(
name="Precision", description=("Tolerence for comparing double "
"keyframes (higher for greater "
"accuracy)"), min=1, max=16,
soft_min=1, soft_max=16, default=6.0)
use_metadata = BoolProperty(name="Use Metadata", default=True, options={'HIDDEN'})
use_include_particle_duplicates = BoolProperty(name="Include Particle Duplicates", default=True)
use_metadata = BoolProperty(name="Use Metadata", default=True,
options={'HIDDEN'})
use_include_particle_duplicates = BoolProperty(
name="Include Particle Duplicates", default=True)
def register():
@ -446,7 +546,8 @@ def register():
bpy.utils.register_class(select_group_objects)
bpy.utils.register_class(UI_List_Godot)
bpy.types.Scene.godot_export_groups = CollectionProperty(type=godot_export_groups)
bpy.types.Scene.godot_export_groups = CollectionProperty(
type=godot_export_groups)
bpy.types.Scene.godot_export_groups_index = IntProperty(default=0, min=0)

33
tools/export/blender25/io_scene_dae/__init__.py
View File

@ -18,35 +18,30 @@
# <pep8-80 compliant>
import bpy
from bpy.props import StringProperty, BoolProperty, FloatProperty, EnumProperty
from bpy_extras.io_utils import ExportHelper
bl_info = {
"name": "Better Collada Exporter",
"author": "Juan Linietsky",
"blender": (2, 5, 8),
"api": 38691,
"location": "File > Import-Export",
"description": ("Export DAE Scenes, This plugin actually works better! otherwise contact me."),
"description": ("Export DAE Scenes, This plugin actually works better! "
"otherwise contact me."),
"warning": "",
"wiki_url": ("http://www.godotengine.org"),
"tracker_url": "",
"support": 'OFFICIAL',
"category": "Import-Export"}
if "bpy" in locals():
import imp
if "export_dae" in locals():
imp.reload(export_dae)
import bpy
from bpy.props import StringProperty, BoolProperty, FloatProperty, EnumProperty
from bpy_extras.io_utils import (ExportHelper,
path_reference_mode,
axis_conversion,
)
class ExportDAE(bpy.types.Operator, ExportHelper):
'''Selection to DAE'''
bl_idname = "export_scene.dae"
@ -74,7 +69,8 @@ class ExportDAE(bpy.types.Operator, ExportHelper):
use_export_selected = BoolProperty(
name="Selected Objects",
description="Export only selected objects (and visible in active layers if that applies).",
description="Export only selected objects (and visible in active "
"layers if that applies).",
default=False,
)
use_mesh_modifiers = BoolProperty(
@ -84,7 +80,8 @@ class ExportDAE(bpy.types.Operator, ExportHelper):
)
use_tangent_arrays = BoolProperty(
name="Tangent Arrays",
description="Export Tangent and Binormal arrays (for normalmapping).",
description="Export Tangent and Binormal arrays "
"(for normalmapping).",
default=False,
)
use_triangles = BoolProperty(
@ -92,6 +89,7 @@ class ExportDAE(bpy.types.Operator, ExportHelper):
description="Export Triangles instead of Polygons.",
default=False,
)
use_copy_images = BoolProperty(
name="Copy Images",
description="Copy Images (create images/ subfolder)",
@ -109,12 +107,14 @@ class ExportDAE(bpy.types.Operator, ExportHelper):
)
use_anim_action_all = BoolProperty(
name="All Actions",
description=("Export all actions for the first armature found in separate DAE files"),
description=("Export all actions for the first armature found "
"in separate DAE files"),
default=False,
)
use_anim_skip_noexp = BoolProperty(
name="Skip (-noexp) Actions",
description="Skip exporting of actions whose name end in (-noexp). Useful to skip control animations.",
description="Skip exporting of actions whose name end in (-noexp)."
" Useful to skip control animations.",
default=True,
)
use_anim_optimize = BoolProperty(
@ -140,7 +140,8 @@ class ExportDAE(bpy.types.Operator, ExportHelper):
@property
def check_extension(self):
return True # return self.batch_mode == 'OFF'
# return self.batch_mode == 'OFF'
return True
def check(self, context):
return True

1501
tools/export/blender25/io_scene_dae/export_dae.py
View File

File diff suppressed because it is too large Load Diff