diff options
| -rw-r--r-- | btrace/__init__.py | 123 | ||||
| -rw-r--r-- | btrace/bTrace.py | 2610 | ||||
| -rw-r--r-- | btrace/bTrace_props.py | 951 |
3 files changed, 2162 insertions, 1522 deletions
diff --git a/btrace/__init__.py b/btrace/__init__.py index 40ff4df1..c1f62512 100644 --- a/btrace/__init__.py +++ b/btrace/__init__.py @@ -1,43 +1,96 @@ -#BEGIN GPL LICENSE BLOCK +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### -#This program is free software; you can redistribute it and/or -#modify it under the terms of the GNU General Public License -#as published by the Free Software Foundation; either version 2 -#of the License, or (at your option) any later version. - -#This program is distributed in the hope that it will be useful, -#but WITHOUT ANY WARRANTY; without even the implied warranty of -#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -#GNU General Public License for more details. - -#You should have received a copy of the GNU General Public License -#along with this program; if not, write to the Free Software Foundation, -#Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. - -#END GPL LICENCE BLOCK bl_info = { "name": "Btrace", "author": "liero, crazycourier, Atom, Meta-Androcto, MacKracken", - "version": (1, 1, ), - "blender": (2, 68, 0), + "version": (1, 2, 1), + "blender": (2, 78, 0), "location": "View3D > Tools", "description": "Tools for converting/animating objects/particles into curves", "warning": "", "wiki_url": "https://wiki.blender.org/index.php/Extensions:2.6/Py/Scripts/Curve/Btrace", - "tracker_url": "https://developer.blender.org/maniphest/task/edit/form/2/", "category": "Add Curve"} +if "bpy" in locals(): + import importlib + importlib.reload(bTrace_props) + importlib.reload(bTrace) +else: + from . import bTrace_props + from . import bTrace import bpy -from .bTrace import * -import selection_utils -from bpy.props import FloatProperty, EnumProperty, IntProperty, BoolProperty, FloatVectorProperty +from bpy.types import AddonPreferences +from .bTrace_props import ( + TracerProperties, + addTracerObjectPanel, + ) +from .bTrace import ( + OBJECT_OT_convertcurve, + OBJECT_OT_objecttrace, + OBJECT_OT_objectconnect, + OBJECT_OT_writing, + OBJECT_OT_particletrace, + OBJECT_OT_traceallparticles, + OBJECT_OT_curvegrow, + OBJECT_OT_reset, + OBJECT_OT_fcnoise, + OBJECT_OT_meshfollow, + OBJECT_OT_materialChango, + OBJECT_OT_clearColorblender, + ) +from bpy.props import ( + EnumProperty, + PointerProperty, + ) + + +# Add-on Preferences +class btrace_preferences(AddonPreferences): + bl_idname = __name__ + + expand_enum = EnumProperty( + name="UI Options", + items=[ + ('list', "Drop down list", + "Show all the items as dropdown list in the Tools Region"), + ('col', "Enable Expanded UI Panel", + "Show all the items expanded in the Tools Region in a column"), + ('row', "Icons only in a row", + "Show all the items as icons expanded in a row in the Tools Region") + ], + description="", + default='list' + ) + + def draw(self, context): + layout = self.layout + layout.label("UI Options:") -### Define Classes to register -classes = [ + row = layout.row(align=True) + row.prop(self, "expand_enum", text="UI Options", expand=True) + + +# Define Classes to register +classes = ( TracerProperties, - TracerPropertiesMenu, addTracerObjectPanel, OBJECT_OT_convertcurve, OBJECT_OT_objecttrace, @@ -50,18 +103,22 @@ classes = [ OBJECT_OT_fcnoise, OBJECT_OT_meshfollow, OBJECT_OT_materialChango, - OBJECT_OT_clearColorblender - ] + OBJECT_OT_clearColorblender, + btrace_preferences, + ) + def register(): - for c in classes: - bpy.utils.register_class(c) - bpy.types.WindowManager.curve_tracer = bpy.props.PointerProperty(type=TracerProperties) - bpy.types.WindowManager.btrace_menu = bpy.props.PointerProperty(type=TracerPropertiesMenu, update=deselect_others) + for cls in classes: + bpy.utils.register_class(cls) + bpy.types.WindowManager.curve_tracer = PointerProperty(type=TracerProperties) + def unregister(): - for c in classes: - bpy.utils.unregister_class(c) + for cls in classes: + bpy.utils.unregister_class(cls) del bpy.types.WindowManager.curve_tracer + + if __name__ == "__main__": register() diff --git a/btrace/bTrace.py b/btrace/bTrace.py index 9ac2cfcd..82d93f9d 100644 --- a/btrace/bTrace.py +++ b/btrace/bTrace.py @@ -1,750 +1,206 @@ -#BEGIN GPL LICENSE BLOCK +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### + +# TO DO LIST # +# Add more options to curve radius/modulation plus cyclic/connect curve option -#This program is free software; you can redistribute it and/or -#modify it under the terms of the GNU General Public License -#as published by the Free Software Foundation; either version 2 -#of the License, or (at your option) any later version. - -#This program is distributed in the hope that it will be useful, -#but WITHOUT ANY WARRANTY; without even the implied warranty of -#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -#GNU General Public License for more details. +import bpy +import selection_utils +from bpy.types import Operator +from random import ( + choice as rand_choice, + random as rand_random, + randint as rand_randint, + uniform as rand_uniform, + ) -#You should have received a copy of the GNU General Public License -#along with this program; if not, write to the Free Software Foundation, -#Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. -#END GPL LICENCE BLOCK +def error_handlers(self, op_name, error, reports="ERROR", func=False): + if self and reports: + self.report({'WARNING'}, reports + " (See Console for more info)") -bl_info = { - "name": "Btrace", - "author": "liero, crazycourier, Atom, Meta-Androcto, MacKracken", - "version": (1, 1, ), - "blender": (2, 71, 0), - "location": "View3D > Toolshelf > Addons Tab", - "description": "Tools for converting/animating objects/particles into curves", - "warning": "Still under development, bug reports appreciated", - "wiki_url": "", - "tracker_url": "https://developer.blender.org/T29563", - "category": "Add Curve"} + is_func = "Function" if func else "Operator" + print("\n[Btrace]\n{}: {}\nError: {}\n".format(op_name, is_func, error)) -#### TO DO LIST #### -### [ ] Add more options to curve radius/modulation plus cyclic/connect curve option +# Object Trace +# creates a curve with a modulated radius connecting points of a mesh -import bpy -import selection_utils -from bpy.props import FloatProperty, EnumProperty, IntProperty, BoolProperty, FloatVectorProperty - - -def deselect_others(ob, context): - """For tool menu select, deselects others if one selected""" - selected = addTracerObjectPanel.selected - ob[selected] = False - keys = [key for key in ob.keys() if ob[key]] # all the True keys - if len(keys) <= 0: - ob[selected] = True # reselect - return None - for key in keys: - addTracerObjectPanel.selected = key - ob[key] = True - - -# Class for properties panel -class TracerPropertiesMenu(bpy.types.PropertyGroup): - """Toolbar show/hide booleans for tool options""" - tool_objectTrace = BoolProperty(name="Object Trace", default=False, description="Trace selected mesh object with a curve", update=deselect_others) - tool_objectsConnect = BoolProperty(name="Objects Connect", default=False, description="Connect objects with a curve controlled by hooks", update=deselect_others) - tool_particleTrace = BoolProperty(name="Particle Trace", default=False, description="Trace particle path with a curve", update=deselect_others) - tool_meshFollow = BoolProperty(name="Mesh Follow", default=False, description="Follow selection items on animated mesh object", update=deselect_others) - tool_particleConnect = BoolProperty(name="Particle Connect", default=False, description="Connect particles with a curves and animated over particle lifetime", update=deselect_others) - tool_growCurve = BoolProperty(name="Grow Curve", default=False, description="Animate curve bevel over time by keyframing points radius", update=deselect_others) - tool_handwrite = BoolProperty(name="Handwriting", default=False, description="Create and Animate curve using the grease pencil", update=deselect_others) - tool_fcurve = BoolProperty(name="F-Curve Noise", default=False, description="Add F-Curve noise to selected objects", update=deselect_others) - tool_colorblender = BoolProperty(name="Color Blender", default=False, description="Add F-Curve noise to selected objects", update=deselect_others) - - -# Class to define properties -class TracerProperties(bpy.types.PropertyGroup): - """Options for tools""" - curve_spline = EnumProperty(name="Spline", items=(("POLY", "Poly", "Use Poly spline type"), ("NURBS", "Nurbs", "Use Nurbs spline type"), ("BEZIER", "Bezier", "Use Bezier spline type")), description="Choose which type of spline to use when curve is created", default="BEZIER") - curve_handle = EnumProperty(name="Handle", items=(("ALIGNED", "Aligned", "Use Aligned Handle Type"), ("AUTOMATIC", "Automatic", "Use Auto Handle Type"), ("FREE_ALIGN", "Free Align", "Use Free Handle Type"), ("VECTOR", "Vector", "Use Vector Handle Type")), description="Choose which type of handle to use when curve is created", default="VECTOR") - curve_resolution = IntProperty(name="Bevel Resolution", min=1, max=32, default=4, description="Adjust the Bevel resolution") - curve_depth = FloatProperty(name="Bevel Depth", min=0.0, max=100.0, default=0.1, description="Adjust the Bevel depth") - curve_u = IntProperty(name="Resolution U", min=0, max=64, default=12, description="Adjust the Surface resolution") - curve_join = BoolProperty(name="Join Curves", default=False, description="Join all the curves after they have been created") - curve_smooth = BoolProperty(name="Smooth", default=True, description="Render curve smooth") - # Option to Duplicate Mesh - object_duplicate = BoolProperty(name="Apply to Copy", default=False, description="Apply curve to a copy of object") - # Distort Mesh options - distort_modscale = IntProperty(name="Modulation Scale", min=0, max=50, default=2, description="Add a scale to modulate the curve at random points, set to 0 to disable") - distort_noise = FloatProperty(name="Mesh Noise", min=0.0, max=50.0, default=0.00, description="Adjust noise added to mesh before adding curve") - # Particle Options - particle_step = IntProperty(name="Step Size", min=1, max=50, default=5, description="Sample one every this number of frames") - particle_auto = BoolProperty(name='Auto Frame Range', default=True, description='Calculate Frame Range from particles life') - particle_f_start = IntProperty(name='Start Frame', min=1, max=5000, default=1, description='Start frame') - particle_f_end = IntProperty(name='End Frame', min=1, max=5000, default=250, description='End frame') - # F-Curve Modifier Properties - fcnoise_rot = BoolProperty(name="Rotation", default=False, description="Affect Rotation") - fcnoise_loc = BoolProperty(name="Location", default=True, description="Affect Location") - fcnoise_scale = BoolProperty(name="Scale", default=False, description="Affect Scale") - fcnoise_amp = IntProperty(name="Amp", min=1, max=500, default=5, description="Adjust the amplitude") - fcnoise_timescale = FloatProperty(name="Time Scale", min=1, max=500, default=50, description="Adjust the time scale") - fcnoise_key = BoolProperty(name="Add Keyframe", default=True, description="Keyframe is needed for tool, this adds a LocRotScale keyframe") - show_curve_settings = BoolProperty(name="Curve Settings", default=False, description="Change the curve settings for the created curve") - material_settings = BoolProperty(name="Material Settings", default=False, description="Change the material settings for the created curve") - particle_settings = BoolProperty(name="Particle Settings", default=False, description="Show the settings for the created curve") - animation_settings = BoolProperty(name="Animation Settings", default=False, description="Show the settings for the Animations") - distort_curve = BoolProperty(name="Add Distortion", default=False, description="Set options to distort the final curve") - connect_noise = BoolProperty(name="F-Curve Noise", default=False, description="Adds F-Curve Noise Modifier to selected objects") - settings_objectTrace = BoolProperty(name="Object Trace Settings", default=False, description="Trace selected mesh object with a curve") - settings_objectsConnect = BoolProperty(name="Objects Connect Settings", default=False, description="Connect objects with a curve controlled by hooks") - settings_objectTrace = BoolProperty(name="Object Trace Settings", default=False, description="Trace selected mesh object with a curve") - respect_order = BoolProperty(name="Order", default=False, description="Remember order objects were selected") - settings_particleTrace = BoolProperty(name="Particle Trace Settings", default=False, description="Trace particle path with a curve") - settings_particleConnect = BoolProperty(name="Particle Connect Settings", default=False, description="Connect particles with a curves and animated over particle lifetime") - settings_growCurve = BoolProperty(name="Grow Curve Settings", default=False, description="Animate curve bevel over time by keyframing points radius") - settings_fcurve = BoolProperty(name="F-Curve Settings", default=False, description="F-Curve Settings") - settings_toggle = BoolProperty(name="Settings", default=False, description="Toggle Settings") - # Animation Options - anim_auto = BoolProperty(name='Auto Frame Range', default=True, description='Automatically calculate Frame Range') - anim_f_start = IntProperty(name='Start', min=1, max=2500, default=1, description='Start frame / Hidden object') - anim_length = IntProperty(name='Duration', min=1, soft_max=1000, max=2500, default=100, description='Animation Length') - anim_f_fade = IntProperty(name='Fade After', min=0, soft_max=250, max=2500, default=10, description='Fade after this frames / Zero means no fade') - anim_delay = IntProperty(name='Grow', min=0, max=50, default=5, description='Frames it takes a point to grow') - anim_tails = BoolProperty(name='Tails on endpoints', default=True, description='Set radius to zero for open splines endpoints') - anim_keepr = BoolProperty(name='Keep Radius', default=True, description='Try to keep radius data from original curve') - animate = BoolProperty(name="Animate Result", default=False, description='Animate the final curve objects') - # Convert to Curve options - convert_conti = BoolProperty(name='Continuous', default=True, description='Create a continuous curve using verts from mesh') - convert_everyedge = BoolProperty(name='Every Edge', default=False, description='Create a curve from all verts in a mesh') - convert_edgetype = EnumProperty(name="Edge Type for Curves", - items=(("CONTI", "Continuous", "Create a continuous curve using verts from mesh"), ("EDGEALL", "All Edges", "Create a curve from every edge in a mesh")), - description="Choose which type of spline to use when curve is created", default="CONTI") - convert_joinbefore = BoolProperty(name="Join objects before convert", default=False, description='Join all selected mesh to one object before converting to mesh') - # Mesh Follow Options - fol_edge_select = BoolProperty(name='Edge', default=False, description='Grow from edges') - fol_vert_select = BoolProperty(name='Vertex', default=False, description='Grow from verts') - fol_face_select = BoolProperty(name='Face', default=True, description='Grow from faces') - fol_mesh_type = EnumProperty(name='Mesh type', default='VERTS', description='Mesh feature to draw cruves from', items=( - ("VERTS", "Verts", "Draw from Verts"), ("EDGES", "Edges", "Draw from Edges"), ("FACES", "Faces", "Draw from Faces"), ("OBJECT", "Object", "Draw from Object origin"))) - fol_start_frame = IntProperty(name="Start Frame", min=1, max=2500, default=1, description="Start frame for range to trace") - fol_end_frame = IntProperty(name="End Frame", min=1, max=2500, default=250, description="End frame for range to trace") - fol_perc_verts = FloatProperty(name="Reduce selection by", min=0.001, max=1.000, default=0.5, description="percentage of total verts to trace") - fol_sel_option = EnumProperty(name="Selection type", description="Choose which objects to follow", default="RANDOM", items=( - ("RANDOM", "Random", "Follow Random items"), ("CUSTOM", "Custom Select", "Follow selected items"), ("ALL", "All", "Follow all items"))) - trace_mat_color = FloatVectorProperty(name="Material Color", description="Choose material color", min=0, max=1, default=(0.0,0.3,0.6), subtype="COLOR") - trace_mat_random = BoolProperty(name="Random Color", default=False, description='Make the material colors random') - - # Material custom Properties properties - mat_simple_adv_toggle = EnumProperty(name="Material Options", items=(("SIMPLE", "Simple", "Show Simple Material Options"), ("ADVANCED", "Advanced", "Show Advanced Material Options")), description="Choose which Material Options to show", default="SIMPLE") - mat_run_color_blender = BoolProperty(name="Run Color Blender", default=False, description="Generate colors from a color scheme") - mmColors = bpy.props.EnumProperty( - items=(("RANDOM", "Random", "Use random colors"), - ("CUSTOM", "Custom", "Use custom colors"), - ("BW", "Black/White", "Use Black and White"), - ("BRIGHT", "Bright Colors", "Use Bright colors"), - ("EARTH", "Earth", "Use Earth colors"), - ("GREENBLUE", "Green to Blue", "Use Green to Blue colors")), - description="Choose which type of colors the materials uses", - default="BRIGHT", - name="Define a color palette") - # Custom property for how many keyframes to skip - mmSkip = bpy.props.IntProperty(name="frames", min=1, max=500, default=20, description="Number of frames between each keyframes") - # Custom property to enable/disable random order for the - mmBoolRandom = bpy.props.BoolProperty(name="Random Order", default=False, description="Randomize the order of the colors") - # Custom Color properties - mmColor1 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.8, 0.8, 0.8), description="Custom Color 1", subtype="COLOR") - mmColor2 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.8, 0.8, 0.3), description="Custom Color 2", subtype="COLOR") - mmColor3 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.8, 0.5, 0.6), description="Custom Color 3", subtype="COLOR") - mmColor4 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.2, 0.8, 0.289), description="Custom Color 4", subtype="COLOR") - mmColor5 = bpy.props.FloatVectorProperty(min=0, max=1, default=(1.0, 0.348, 0.8), description="Custom Color 5", subtype="COLOR") - mmColor6 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.4, 0.67, 0.8), description="Custom Color 6", subtype="COLOR") - mmColor7 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.66, 0.88, 0.8), description="Custom Color 7", subtype="COLOR") - mmColor8 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.8, 0.38, 0.22), description="Custom Color 8", subtype="COLOR") - # BW Color properties - bwColor1 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.0,0.0,0.0), description="Black/White Color 1", subtype="COLOR") - bwColor2 = bpy.props.FloatVectorProperty(min=0, max=1, default=(1.0,1.0,1.0), description="Black/White Color 2", subtype="COLOR") - # Bright Color properties - brightColor1 = bpy.props.FloatVectorProperty(min=0, max=1, default=(1.0, 0.0, 0.75), description="Bright Color 1", subtype="COLOR") - brightColor2 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.0,1.0,1.0), description="Bright Color 2", subtype="COLOR") - brightColor3 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.0,1.0,0.0), description="Bright Color 3", subtype="COLOR") - brightColor4 = bpy.props.FloatVectorProperty(min=0, max=1, default=(1.0,1.0,0.0), description="Bright Color 4", subtype="COLOR") - # Earth Color Properties - earthColor1 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.068, 0.019, 0.014), description="Earth Color 1", subtype="COLOR") - earthColor2 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.089, 0.060, 0.047), description="Earth Color 2", subtype="COLOR") - earthColor3 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.188, 0.168, 0.066), description="Earth Color 3", subtype="COLOR") - earthColor4 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.445, 0.296, 0.065), description="Earth Color 4", subtype="COLOR") - earthColor5 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.745, 0.332, 0.065), description="Earth Color 5", subtype="COLOR") - # Green to Blue Color properties - greenblueColor1 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.296, 0.445, 0.074), description="Green/Blue Color 1", subtype="COLOR") - greenblueColor2 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.651, 1.0, 0.223), description="Green/Blue Color 2", subtype="COLOR") - greenblueColor3 = bpy.props.FloatVectorProperty(min=0, max=1, default=(0.037, 0.047, 0.084), description="Green/Blue Color 3", subtype="COLOR") - - -############################ -## Draw Brush panel in Toolbar -############################ -class addTracerObjectPanel(bpy.types.Panel): - bl_label = "Btrace: Panel" - bl_space_type = 'VIEW_3D' - bl_region_type = 'TOOLS' - bl_context = 'objectmode' - selected = "tool_objectTrace" - bl_category = 'Create' - bl_options = {'DEFAULT_CLOSED'} - - - def draw(self, context): - layout = self.layout - Btrace = bpy.context.window_manager.curve_tracer - btracemenu = props = bpy.context.window_manager.btrace_menu - obj = bpy.context.object - - - ############################ - ## Color Blender Panel options - ############################ - def color_blender(): - """Buttons for Color Blender""" - row = box.row() - row.label("Color palette") - row.prop(Btrace, 'mmColors', text="") - # Show Custom Colors if selected - if Btrace.mmColors == 'CUSTOM': - row = box.row(align=True) - row.prop(Btrace, 'mmColor1', text="") - row.prop(Btrace, 'mmColor2', text="") - row.prop(Btrace, 'mmColor3', text="") - row.prop(Btrace, 'mmColor4', text="") - row.prop(Btrace, 'mmColor5', text="") - row.prop(Btrace, 'mmColor6', text="") - row.prop(Btrace, 'mmColor7', text="") - row.prop(Btrace, 'mmColor8', text="") - # Show Earth Colors - elif Btrace.mmColors == 'BW': - row = box.row(align=True) - row.prop(Btrace, 'bwColor1', text="") - row.prop(Btrace, 'bwColor2', text="") - # Show Earth Colors - elif Btrace.mmColors == 'BRIGHT': - row = box.row(align=True) - row.prop(Btrace, 'brightColor1', text="") - row.prop(Btrace, 'brightColor2', text="") - row.prop(Btrace, 'brightColor3', text="") - row.prop(Btrace, 'brightColor4', text="") - # Show Earth Colors - elif Btrace.mmColors == 'EARTH': - row = box.row(align=True) - row.prop(Btrace, 'earthColor1', text="") - row.prop(Btrace, 'earthColor2', text="") - row.prop(Btrace, 'earthColor3', text="") - row.prop(Btrace, 'earthColor4', text="") - row.prop(Btrace, 'earthColor5', text="") - # Show Earth Colors - elif Btrace.mmColors == 'GREENBLUE': - row = box.row(align=True) - row.prop(Btrace, 'greenblueColor1', text="") - row.prop(Btrace, 'greenblueColor2', text="") - row.prop(Btrace, 'greenblueColor3', text="") - elif Btrace.mmColors == 'RANDOM': - row = box.row() - - ############################ - ## Curve Panel options - ############################ - def curve_settings(): - """Button for curve options""" - row = self.layout.row() - row = box.row(align=True) - - row.prop(Btrace, 'show_curve_settings', icon='CURVE_BEZCURVE', text="Curve Settings") - row.prop(Btrace, 'material_settings', icon='MATERIAL_DATA', text="Material Settings") - if Btrace.material_settings: - row = box.row() - row.label(text="Material Settings", icon='COLOR') - row = box.row() - row.prop(Btrace, "trace_mat_random") - if not Btrace.trace_mat_random: - row = box.row() - row.prop(Btrace, "trace_mat_color", text="") - else: - row.prop(Btrace, "mat_run_color_blender") - if Btrace.mat_run_color_blender: - row = box.row() - row.operator("object.colorblenderclear", text="Reset Material Keyframes", icon="KEY_DEHLT") - row.prop(Btrace, 'mmSkip', text="Keyframe every") - - color_blender() - row = box.row() - if Btrace.show_curve_settings: - #if or btracemenu.tool_handwrite: - if len(bpy.context.selected_objects) > 0 and obj.type == 'CURVE': # selected curve options - col = box.column(align=True) - col.label(text="Edit Curves for", icon='CURVE_BEZCURVE') - col.label(text="Selected Curve Bevel Options") - row = col.row(align=True) - row.prop(obj.data, 'bevel_depth', text="Depth") - row.prop(obj.data, 'bevel_resolution', text="Resolution") - row = col.row(align=True) - row.prop(obj.data, 'resolution_u') - else: # For new curve - box.label(text="New Curve Settings", icon='CURVE_BEZCURVE') - box.prop(Btrace, "curve_spline") - box.prop(Btrace, "curve_handle") - box.label(text="Bevel Options") - col = box.column(align=True) - row = col.row(align=True) - row.prop(Btrace, "curve_depth", text="Depth") - row.prop(Btrace, "curve_resolution", text="Resolution") - row = col.row(align=True) - row.prop(Btrace, "curve_u") - - ############################ - ## Grow Animation Panel options - ############################ - def add_grow(): - """Button for grow animation option""" - row = box.row() - row.label(text="Animate Final Curve") - row = box.row() - row.prop(Btrace, "animate", text="Add Grow Curve Animation", icon="META_BALL") - row.label("") - if Btrace.animate: - box.label(text='Frame Animation Settings:', icon="META_BALL") - col = box.column(align=True) - col.prop(Btrace, 'anim_auto') - if not Btrace.anim_auto: - row = col.row(align=True) - row.prop(Btrace, 'anim_f_start') - row.prop(Btrace, 'anim_length') - row = col.row(align=True) - row.prop(Btrace, 'anim_delay') - row.prop(Btrace, 'anim_f_fade') - - box.label(text='Additional Settings') - row = box.row() - row.prop(Btrace, 'anim_tails') - row.prop(Btrace, 'anim_keepr') - - ################################################################## - ## Start Btrace Panel - ################################################################## - col = self.layout.column(align=True) - #col.label(text="Trace Tools") - row = col.row() - row.prop(btracemenu, "tool_objectTrace", text="Ojbect Trace", icon="FORCE_MAGNETIC") - row.prop(btracemenu, "tool_objectsConnect", text="Object Connect", icon="OUTLINER_OB_EMPTY") - row = col.row() - row.prop(btracemenu, "tool_meshFollow", text="Mesh Follow", icon="DRIVER") - row.prop(btracemenu, "tool_handwrite", text="Handwriting", icon='BRUSH_DATA') - row = col.row() - row.prop(btracemenu, "tool_particleTrace", icon="PARTICLES", text="Particle Trace") - row.prop(btracemenu, "tool_particleConnect", icon="MOD_PARTICLES", text="Particle Connect") - row = layout.row() - col = layout.column(align=True) - row = col.row() - row.prop(btracemenu, "tool_growCurve", icon="META_BALL", text="Grow Animation") - row.prop(btracemenu, "tool_fcurve", text="Fcurve Noise", icon='RNDCURVE') - row = col.row() - row.prop(btracemenu, "tool_colorblender", text="Color Blender", icon="COLOR") - row.label(text="") - row = layout.row() - - ########################## - ## Start Object Tools - ########################## - sel = bpy.context.selected_objects - ############################ - ### Object Trace - ############################ - if btracemenu.tool_objectTrace: - row = layout.row() - row.label(text=" Trace Tool:", icon="FORCE_CURVE") - box = self.layout.box() - row = box.row() - row.label(text="Object Trace", icon="FORCE_MAGNETIC") - row.operator("object.btobjecttrace", text="Run!", icon="PLAY") - row = box.row() - row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') - myselected = "Selected %d" % len(bpy.context.selected_objects) - row.label(text=myselected) - if Btrace.settings_toggle: - row = box.row() - row.label(text='Edge Draw Method') - row = box.row(align=True) - row.prop(Btrace, 'convert_edgetype') - box.prop(Btrace, "object_duplicate") - if len(sel) > 1: - box.prop(Btrace, 'convert_joinbefore') - else: - Btrace.convert_joinbefore = False - row = box.row() - row.prop(Btrace, "distort_curve") - if Btrace.distort_curve: - col = box.column(align=True) - col.prop(Btrace, "distort_modscale") - col.prop(Btrace, "distort_noise") - row = box.row() - curve_settings() # Show Curve/material settings - add_grow() # Grow settings here - - ############################ - ### Objects Connect - ############################ - if btracemenu.tool_objectsConnect: - row = layout.row() - row.label(text=" Trace Tool:", icon="FORCE_CURVE") - box = self.layout.box() - row = box.row() - row.label(text="Objects Connect", icon="OUTLINER_OB_EMPTY") - row.operator("object.btobjectsconnect", text="Run!", icon="PLAY") - row = box.row() - row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') - row.label(text="") - if Btrace.settings_toggle: - row = box.row() - row.prop(Btrace, "respect_order", text="Selection Options") - if Btrace.respect_order: - box.operator("object.select_order", text="Click to start order selection", icon='UV_SYNC_SELECT') - row = box.row() - row.prop(Btrace, "connect_noise", text="Add F-Curve Noise") - if Btrace.connect_noise: - row = box.row() - row.label(text="F-Curve Noise", icon='RNDCURVE') - row = box.row(align=True) - row.prop(Btrace, "fcnoise_rot") - row.prop(Btrace, "fcnoise_loc") - row.prop(Btrace, "fcnoise_scale") - col = box.column(align=True) - col.prop(Btrace, "fcnoise_amp") - col.prop(Btrace, "fcnoise_timescale") - box.prop(Btrace, "fcnoise_key") - curve_settings() # Show Curve/material settings - add_grow() # Grow settings here - - ############################ - ### Mesh Follow - ############################ - if btracemenu.tool_meshFollow: - row = layout.row() - row.label(text=" Trace Tool:", icon="FORCE_CURVE") - box = self.layout.box() - row = box.row() - row.label(text="Mesh Follow", icon="DRIVER") - row.operator("object.btmeshfollow", text="Run!", icon="PLAY") - row = box.row() - if Btrace.fol_mesh_type == 'OBJECT': - a, b = "Trace Object", "SNAP_VOLUME" - if Btrace.fol_mesh_type == 'VERTS': - a, b = "Trace Verts", "SNAP_VERTEX" - if Btrace.fol_mesh_type == 'EDGES': - a, b = "Trace Edges", "SNAP_EDGE" - if Btrace.fol_mesh_type == 'FACES': - a, b = "Trace Faces", "SNAP_FACE" - row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') - row.label(text=a, icon=b) - if Btrace.settings_toggle: - col = box.column(align=True) - row = col.row(align=True) - row.prop(Btrace, "fol_mesh_type", expand=True) - row = col.row(align=True) - if Btrace.fol_mesh_type != 'OBJECT': - row.prop(Btrace, "fol_sel_option", expand=True) - row = box.row() - if Btrace.fol_sel_option == 'RANDOM': - row.label("Random Select of Total") - row.prop(Btrace, "fol_perc_verts", text="%") - if Btrace.fol_sel_option == 'CUSTOM': - row.label("Choose selection in Edit Mode") - if Btrace.fol_sel_option == 'ALL': - row.label("Select All items") - col = box.column(align=True) - col.label("Time Options", icon="TIME") - col.prop(Btrace, "particle_step") - row = col.row(align=True) - row.prop(Btrace, "fol_start_frame") - row.prop(Btrace, "fol_end_frame") - curve_settings() # Show Curve/material settings - add_grow() # Grow settings here - - ############################ - ### Handwriting Tools - ############################ - if btracemenu.tool_handwrite: - row = layout.row() - row.label(text=" Trace Tool:", icon="FORCE_CURVE") - box = self.layout.box() - row = box.row() - row.label(text='Handwriting', icon='BRUSH_DATA') - row.operator("curve.btwriting", text="Run!", icon='PLAY') - row = box.row() - row = box.row() - row.label(text='Grease Pencil Writing Tools') - col = box.column(align=True) - row = col.row(align=True) - row.operator("gpencil.draw", text="Draw", icon='BRUSH_DATA').mode = 'DRAW' - row.operator("gpencil.draw", text="Poly", icon='VPAINT_HLT').mode = 'DRAW_POLY' - row = col.row(align=True) - row.operator("gpencil.draw", text="Line", icon='ZOOMOUT').mode = 'DRAW_STRAIGHT' - row.operator("gpencil.draw", text="Erase", icon='TPAINT_HLT').mode = 'ERASER' - row = box.row() - row.operator("gpencil.data_unlink", text="Delete Grease Pencil Layer", icon="CANCEL") - row = box.row() - curve_settings() # Show Curve/material settings - add_grow() # Grow settings here - - ############################ - ### Particle Trace - ############################ - if btracemenu.tool_particleTrace: - row = layout.row() - row.label(text=" Trace Tool:", icon="FORCE_CURVE") - box = self.layout.box() - row = box.row() - row.label(text="Particle Trace", icon="PARTICLES") - row.operator("particles.particletrace", text="Run!", icon="PLAY") - row = box.row() - row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') - row.label(text="") - if Btrace.settings_toggle: - box.prop(Btrace, "particle_step") - row = box.row() - row.prop(Btrace, "curve_join") - curve_settings() # Show Curve/material settings - add_grow() # Grow settings here - - ############################ - ### Connect Particles - ############################ - if btracemenu.tool_particleConnect: - row = layout.row() - row.label(text=" Trace Tool:", icon="FORCE_CURVE") - box = self.layout.box() - row = box.row() - row.label(text='Particle Connect', icon='MOD_PARTICLES') - row.operator("particles.connect", icon="PLAY", text='Run!') - row = box.row() - row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') - row.label(text="") - if Btrace.settings_toggle: - box.prop(Btrace, "particle_step") - row = box.row() - row.prop(Btrace, 'particle_auto') - if not Btrace.particle_auto: - row = box.row(align=True) - row.prop(Btrace, 'particle_f_start') - row.prop(Btrace, 'particle_f_end') - curve_settings() # Show Curve/material settings - add_grow() # Grow settings here - - ####################### - #### Grow Animation #### - ####################### - if btracemenu.tool_growCurve: - row = layout.row() - row.label(text=" Curve Tool:", icon="OUTLINER_OB_CURVE") - box = self.layout.box() - row = box.row() - row.label(text="Grow Curve", icon="META_BALL") - row.operator('curve.btgrow', text='Run!', icon='PLAY') - row = box.row() - row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') - row.operator('object.btreset', icon='KEY_DEHLT') - if Btrace.settings_toggle: - box.label(text='Frame Animation Settings:') - col = box.column(align=True) - col.prop(Btrace, 'anim_auto') - if not Btrace.anim_auto: - row = col.row(align=True) - row.prop(Btrace, 'anim_f_start') - row.prop(Btrace, 'anim_length') - row = col.row(align=True) - row.prop(Btrace, 'anim_delay') - row.prop(Btrace, 'anim_f_fade') - - box.label(text='Additional Settings') - row = box.row() - row.prop(Btrace, 'anim_tails') - row.prop(Btrace, 'anim_keepr') - - ####################### - #### F-Curve Noise Curve #### - ####################### - if btracemenu.tool_fcurve: - row = layout.row() - row.label(text=" Curve Tool:", icon="OUTLINER_OB_CURVE") - box = self.layout.box() - row = box.row() - row.label(text="F-Curve Noise", icon='RNDCURVE') - row.operator("object.btfcnoise", icon='PLAY', text="Run!") - row = box.row() - row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') - row.operator('object.btreset', icon='KEY_DEHLT') - if Btrace.settings_toggle: - row = box.row(align=True) - row.prop(Btrace, "fcnoise_rot") - row.prop(Btrace, "fcnoise_loc") - row.prop(Btrace, "fcnoise_scale") - col = box.column(align=True) - col.prop(Btrace, "fcnoise_amp") - col.prop(Btrace, "fcnoise_timescale") - box.prop(Btrace, "fcnoise_key") - - ####################### - #### Color Blender #### - ####################### - if btracemenu.tool_colorblender: - row = layout.row() - row.label(text=" Curve/Object Tool:", icon="OUTLINER_OB_CURVE") - box = self.layout.box() - row = box.row() - row.label(text="Color Blender", icon="COLOR") - row.operator("object.colorblender", icon='PLAY', text="Run!") - row = box.row() - row.operator("object.colorblenderclear", text="Reset Keyframes", icon="KEY_DEHLT") - row.prop(Btrace, 'mmSkip', text="Keyframe every") - color_blender() - -###### END PANEL ############## -############################### - - -################## ################## ################## ############ -## Object Trace -## creates a curve with a modulated radius connecting points of a mesh -################## ################## ################## ############ - -class OBJECT_OT_objecttrace(bpy.types.Operator): +class OBJECT_OT_objecttrace(Operator): bl_idname = "object.btobjecttrace" bl_label = "Btrace: Object Trace" - bl_description = "Trace selected mesh object with a curve with the option to animate" + bl_description = ("Trace selected mesh object with a curve with the option to animate\n" + "The Active Object has to be of a Mesh or Font type") bl_options = {'REGISTER', 'UNDO'} - @classmethod def poll(cls, context): - return (context.object and context.object.type in {'MESH', 'FONT'}) + return (context.object and + context.object.type in {'MESH', 'FONT'}) def invoke(self, context, event): - import bpy - - # Run through each selected object and convert to to a curved object - brushObj = bpy.context.selected_objects - Btrace = bpy.context.window_manager.curve_tracer - # Duplicate Mesh - if Btrace.object_duplicate: - bpy.ops.object.duplicate_move() - brushObj = bpy.context.selected_objects - # Join Mesh - if Btrace.convert_joinbefore: - if len(brushObj) > 1: # Only run if multiple objects selected - bpy.ops.object.join() - brushObj = bpy.context.selected_objects - - for i in brushObj: - bpy.context.scene.objects.active = i - if i and i.type != 'CURVE': - bpy.ops.object.btconvertcurve() - addtracemat(bpy.context.object.data) - if Btrace.animate: - bpy.ops.curve.btgrow() - return {'FINISHED'} - - -################## ################## ################## ############ -## Objects Connect -## connect selected objects with a curve + hooks to each node -## possible handle types: 'FREE' 'AUTO' 'VECTOR' 'ALIGNED' -################## ################## ################## ############ - -class OBJECT_OT_objectconnect(bpy.types.Operator): + try: + # Run through each selected object and convert to to a curved object + brushObj = context.selected_objects + Btrace = context.window_manager.curve_tracer + check_materials = True + # Duplicate Mesh + if Btrace.object_duplicate: + bpy.ops.object.duplicate_move() + brushObj = context.selected_objects + # Join Mesh + if Btrace.convert_joinbefore: + if len(brushObj) > 1: # Only run if multiple objects selected + bpy.ops.object.join() + brushObj = context.selected_objects + + for i in brushObj: + context.scene.objects.active = i + if i and i.type != 'CURVE': + bpy.ops.object.btconvertcurve() + # Materials + trace_mats = addtracemat(bpy.context.object.data) + if not trace_mats and check_materials is True: + check_materials = False + if Btrace.animate: + bpy.ops.curve.btgrow() + + if check_materials is False: + self.report({'WARNING'}, "Some Materials could not be added") + + return {'FINISHED'} + + except Exception as e: + error_handlers(self, "object.btobjecttrace", e, + "Object Trace could not be completed") + + return {'CANCELLED'} + + +# Objects Connect +# connect selected objects with a curve + hooks to each node +# possible handle types: 'FREE' 'AUTO' 'VECTOR' 'ALIGNED' + +class OBJECT_OT_objectconnect(Operator): bl_idname = "object.btobjectsconnect" bl_label = "Btrace: Objects Connect" - bl_description = "Connect selected objects with a curve and add hooks to each node" + bl_description = ("Connect selected objects with a curve and add hooks to each node\n" + "Needs at least two objects selected") bl_options = {'REGISTER', 'UNDO'} @classmethod def poll(cls, context): - return len(bpy.context.selected_objects) > 1 + return len(context.selected_objects) > 1 def invoke(self, context, event): - import bpy, selection_utils - list = [] - Btrace = bpy.context.window_manager.curve_tracer - curve_handle = Btrace.curve_handle - if curve_handle == 'AUTOMATIC': # hackish because of naming conflict in api - curve_handle = 'AUTO' - # Check if Btrace group exists, if not create - bgroup = bpy.data.groups.keys() - if 'Btrace' not in bgroup: - bpy.ops.group.create(name="Btrace") - # check if noise - if Btrace.connect_noise: - bpy.ops.object.btfcnoise() - # check if respect order is checked, create list of objects - if Btrace.respect_order == True: - selobnames = selection_utils.selected - obnames = [] - for ob in selobnames: - obnames.append(bpy.data.objects[ob]) - else: - obnames = bpy.context.selected_objects # No selection order - - for a in obnames: - list.append(a) - a.select = False - - # trace the origins - tracer = bpy.data.curves.new('tracer', 'CURVE') - tracer.dimensions = '3D' - spline = tracer.splines.new('BEZIER') - spline.bezier_points.add(len(list) - 1) - curve = bpy.data.objects.new('curve', tracer) - bpy.context.scene.objects.link(curve) - - # render ready curve - tracer.resolution_u = Btrace.curve_u - tracer.bevel_resolution = Btrace.curve_resolution # Set bevel resolution from Panel options - tracer.fill_mode = 'FULL' - tracer.bevel_depth = Btrace.curve_depth # Set bevel depth from Panel options - - # move nodes to objects - for i in range(len(list)): - p = spline.bezier_points[i] - p.co = list[i].location - p.handle_right_type = curve_handle - p.handle_left_type = curve_handle - - bpy.context.scene.objects.active = curve - bpy.ops.object.mode_set(mode='OBJECT') - - # place hooks - for i in range(len(list)): - list[i].select = True - curve.data.splines[0].bezier_points[i].select_control_point = True - bpy.ops.object.mode_set(mode='EDIT') - bpy.ops.object.hook_add_selob() + try: + lists = [] + Btrace = context.window_manager.curve_tracer + curve_handle = Btrace.curve_handle + if curve_handle == 'AUTOMATIC': # hackish because of naming conflict in api + curve_handle = 'AUTO' + # Check if Btrace group exists, if not create + bgroup = bpy.data.groups.keys() + if 'Btrace' not in bgroup: + bpy.ops.group.create(name="Btrace") + # check if noise + if Btrace.connect_noise: + bpy.ops.object.btfcnoise() + # check if respect order is checked, create list of objects + if Btrace.respect_order is True: + selobnames = selection_utils.selected + obnames = [] + for ob in selobnames: + obnames.append(bpy.data.objects[ob]) + else: + obnames = bpy.context.selected_objects # No selection order + + for a in obnames: + lists.append(a) + a.select = False + + # trace the origins + tracer = bpy.data.curves.new('tracer', 'CURVE') + tracer.dimensions = '3D' + spline = tracer.splines.new('BEZIER') + spline.bezier_points.add(len(lists) - 1) + curve = bpy.data.objects.new('curve', tracer) + bpy.context.scene.objects.link(curve) + + # render ready curve + tracer.resolution_u = Btrace.curve_u + # Set bevel resolution from Panel options + tracer.bevel_resolution = Btrace.curve_resolution + tracer.fill_mode = 'FULL' + # Set bevel depth from Panel options + tracer.bevel_depth = Btrace.curve_depth + + # move nodes to objects + for i in range(len(lists)): + p = spline.bezier_points[i] + p.co = lists[i].location + p.handle_right_type = curve_handle + p.handle_left_type = curve_handle + + bpy.context.scene.objects.active = curve bpy.ops.object.mode_set(mode='OBJECT') - curve.data.splines[0].bezier_points[i].select_control_point = False - list[i].select = False - - bpy.ops.object.select_all(action='DESELECT') - curve.select = True # selected curve after it's created - addtracemat(bpy.context.object.data) # Add material - if Btrace.animate: # Add Curve Grow it? - bpy.ops.curve.btgrow() - bpy.ops.object.group_link(group="Btrace") # add to Btrace group - if Btrace.animate: - bpy.ops.curve.btgrow() # Add grow curve - return {'FINISHED'} - - -################## ################## ################## ############ -## Particle Trace -## creates a curve from each particle of a system -################## ################## ################## ############ -def curvetracer(curvename, splinename): + + # place hooks + for i in range(len(lists)): + lists[i].select = True + curve.data.splines[0].bezier_points[i].select_control_point = True + bpy.ops.object.mode_set(mode='EDIT') + bpy.ops.object.hook_add_selob() + bpy.ops.object.mode_set(mode='OBJECT') + curve.data.splines[0].bezier_points[i].select_control_point = False + lists[i].select = False + + bpy.ops.object.select_all(action='DESELECT') + curve.select = True # selected curve after it's created + # Materials + check_materials = True + trace_mats = addtracemat(bpy.context.object.data) + if not trace_mats and check_materials is True: + check_materials = False + + if Btrace.animate: # Add Curve Grow it? + bpy.ops.curve.btgrow() + + bpy.ops.object.group_link(group="Btrace") # add to Btrace group + if Btrace.animate: + bpy.ops.curve.btgrow() # Add grow curve + + if check_materials is False: + self.report({'WARNING'}, "Some Materials could not be added") + + return {'FINISHED'} + + except Exception as e: + error_handlers(self, "object.btobjectsconnect", e, + "Objects Connect could not be completed") + + return {'CANCELLED'} + + +# Particle Trace +# creates a curve from each particle of a system + +def curvetracer(curvename, splinename): Btrace = bpy.context.window_manager.curve_tracer tracer = bpy.data.curves.new(splinename, 'CURVE') tracer.dimensions = '3D' @@ -757,762 +213,940 @@ def curvetracer(curvename, splinename): tracer.bevel_resolution = Btrace.curve_resolution tracer.bevel_depth = Btrace.curve_depth tracer.resolution_u = Btrace.curve_u + return tracer, curve -class OBJECT_OT_particletrace(bpy.types.Operator): +class OBJECT_OT_particletrace(Operator): bl_idname = "particles.particletrace" bl_label = "Btrace: Particle Trace" - bl_description = "Creates a curve from each particle of a system. Keeping particle amount under 250 will make this run faster" + bl_description = ("Creates a curve from each particle of a system.\n" + "Keeping particle amount under 250 will make this run faster") bl_options = {'REGISTER', 'UNDO'} @classmethod def poll(cls, context): - return (bpy.context.object and bpy.context.object.particle_systems) + return (context.object is not None and + context.object.particle_systems) def execute(self, context): - Btrace = bpy.context.window_manager.curve_tracer - particle_step = Btrace.particle_step # step size in frames - obj = bpy.context.object - ps = obj.particle_systems.active - curvelist = [] - curve_handle = Btrace.curve_handle - if curve_handle == 'AUTOMATIC': # hackish naming conflict - curve_handle = 'AUTO' - if curve_handle == 'FREE_ALIGN': # hackish naming conflict - curve_handle = 'FREE' - - # Check if Btrace group exists, if not create - bgroup = bpy.data.groups.keys() - if 'Btrace' not in bgroup: - bpy.ops.group.create(name="Btrace") - - if Btrace.curve_join: - tracer = curvetracer('Tracer', 'Splines') - for x in ps.particles: - if not Btrace.curve_join: - tracer = curvetracer('Tracer.000', 'Spline.000') - spline = tracer[0].splines.new('BEZIER') - spline.bezier_points.add((x.lifetime - 1) // particle_step) # add point to spline based on step size - for t in list(range(int(x.lifetime))): - bpy.context.scene.frame_set(t + x.birth_time) - if not t % particle_step: - p = spline.bezier_points[t // particle_step] - p.co = x.location - p.handle_right_type = curve_handle - p.handle_left_type = curve_handle - particlecurve = tracer[1] - curvelist.append(particlecurve) - # add to group - bpy.ops.object.select_all(action='DESELECT') - for curveobject in curvelist: - curveobject.select = True - bpy.context.scene.objects.active = curveobject - bpy.ops.object.group_link(group="Btrace") - addtracemat(curveobject.data) # Add material - - if Btrace.animate: - bpy.ops.curve.btgrow() # Add grow curve - - return {'FINISHED'} - - -########################################################################### -## Particle Connect -## connect all particles in active system with a continuous animated curve -########################################################################### - -class OBJECT_OT_traceallparticles(bpy.types.Operator): - bl_idname = 'particles.connect' - bl_label = 'Connect Particles' - bl_description = 'Create a continuous animated curve from particles in active system' + try: + Btrace = context.window_manager.curve_tracer + particle_step = Btrace.particle_step # step size in frames + obj = bpy.context.object + ps = obj.particle_systems.active + curvelist = [] + curve_handle = Btrace.curve_handle + check_materials = True + + if curve_handle == 'AUTOMATIC': # hackish naming conflict + curve_handle = 'AUTO' + if curve_handle == 'FREE_ALIGN': # hackish naming conflict + curve_handle = 'FREE' + + # Check if Btrace group exists, if not create + bgroup = bpy.data.groups.keys() + if 'Btrace' not in bgroup: + bpy.ops.group.create(name="Btrace") + + if Btrace.curve_join: + tracer = curvetracer('Tracer', 'Splines') + for x in ps.particles: + if not Btrace.curve_join: + tracer = curvetracer('Tracer.000', 'Spline.000') + spline = tracer[0].splines.new('BEZIER') + # add point to spline based on step size + spline.bezier_points.add((x.lifetime - 1) // particle_step) + for t in list(range(int(x.lifetime))): + bpy.context.scene.frame_set(t + x.birth_time) + if not t % particle_step: + p = spline.bezier_points[t // particle_step] + p.co = x.location + p.handle_right_type = curve_handle + p.handle_left_type = curve_handle + particlecurve = tracer[1] + curvelist.append(particlecurve) + # add to group + bpy.ops.object.select_all(action='DESELECT') + for curveobject in curvelist: + curveobject.select = True + bpy.context.scene.objects.active = curveobject + bpy.ops.object.group_link(group="Btrace") + # Materials + trace_mats = addtracemat(curveobject.data) + if not trace_mats and check_materials is True: + check_materials = False + + if Btrace.animate: + bpy.ops.curve.btgrow() # Add grow curve + + if check_materials is False: + self.report({'WARNING'}, "Some Materials could not be added") + + return {'FINISHED'} + + except Exception as e: + error_handlers(self, "particles.particletrace", e, + "Particle Trace could not be completed") + + return {'CANCELLED'} + + +# Particle Connect +# connect all particles in active system with a continuous animated curve + +class OBJECT_OT_traceallparticles(Operator): + bl_idname = "particles.connect" + bl_label = "Connect Particles" + bl_description = ("Create a continuous animated curve from particles in active system\n" + "Needs an Object with a particle system attached") bl_options = {'REGISTER', 'UNDO'} @classmethod def poll(cls, context): - return (bpy.context.object and bpy.context.object.particle_systems) + return (context.object is not None and + context.object.particle_systems) def execute(self, context): + try: + obj = context.object + ps = obj.particle_systems.active + setting = ps.settings + + # Grids distribution not supported + if setting.distribution == 'GRID': + self.report({'INFO'}, + "Grid distribution mode for particles not supported") + return{'CANCELLED'} + + Btrace = context.window_manager.curve_tracer + # Get frame start + particle_f_start = Btrace.particle_f_start + # Get frame end + particle_f_end = Btrace.particle_f_end + curve_handle = Btrace.curve_handle + # hackish because of naming conflict in api + if curve_handle == 'AUTOMATIC': + curve_handle = 'AUTO' + if curve_handle == 'FREE_ALIGN': + curve_handle = 'FREE' + + # define what kind of object to create + tracer = bpy.data.curves.new('Splines', 'CURVE') + # Create new object with settings listed above + curve = bpy.data.objects.new('Tracer', tracer) + # Link newly created object to the scene + bpy.context.scene.objects.link(curve) + # add a new Bezier point in the new curve + spline = tracer.splines.new('BEZIER') + spline.bezier_points.add(setting.count - 1) - obj = bpy.context.object - ps = obj.particle_systems.active - set = ps.settings + tracer.dimensions = '3D' + tracer.resolution_u = Btrace.curve_u + tracer.bevel_resolution = Btrace.curve_resolution + tracer.fill_mode = 'FULL' + tracer.bevel_depth = Btrace.curve_depth + + if Btrace.particle_auto: + f_start = int(setting.frame_start) + f_end = int(setting.frame_end + setting.lifetime) + else: + if particle_f_end <= particle_f_start: + particle_f_end = particle_f_start + 1 + f_start = particle_f_start + f_end = particle_f_end + + for bFrames in range(f_start, f_end): + bpy.context.scene.frame_set(bFrames) + if not (bFrames - f_start) % Btrace.particle_step: + for bFrames in range(setting.count): + if ps.particles[bFrames].alive_state != 'UNBORN': + e = bFrames + bp = spline.bezier_points[bFrames] + pt = ps.particles[e] + bp.co = pt.location + bp.handle_right_type = curve_handle + bp.handle_left_type = curve_handle + bp.keyframe_insert('co') + bp.keyframe_insert('handle_left') + bp.keyframe_insert('handle_right') + # Select new curve + bpy.ops.object.select_all(action='DESELECT') + curve.select = True + bpy.context.scene.objects.active = curve + + # Materials + trace_mats = addtracemat(curve.data) + if not trace_mats: + self.report({'WARNING'}, "Some Materials could not be added") + + if Btrace.animate: + bpy.ops.curve.btgrow() - # Grids distribution not supported - if set.distribution == 'GRID': - self.report({'INFO'},"Grid distribution mode for particles not supported.") return{'FINISHED'} - Btrace = bpy.context.window_manager.curve_tracer - particle_f_start = Btrace.particle_f_start # Get frame start - particle_f_end = Btrace.particle_f_end # Get frame end - curve_handle = Btrace.curve_handle - if curve_handle == 'AUTOMATIC': # hackish because of naming conflict in api - curve_handle = 'AUTO' - if curve_handle == 'FREE_ALIGN': - curve_handle = 'FREE' - tracer = bpy.data.curves.new('Splines','CURVE') # define what kind of object to create - curve = bpy.data.objects.new('Tracer',tracer) # Create new object with settings listed above - bpy.context.scene.objects.link(curve) # Link newly created object to the scene - spline = tracer.splines.new('BEZIER') # add a new Bezier point in the new curve - spline.bezier_points.add(set.count-1) - - tracer.dimensions = '3D' - tracer.resolution_u = Btrace.curve_u - tracer.bevel_resolution = Btrace.curve_resolution - tracer.fill_mode = 'FULL' - tracer.bevel_depth = Btrace.curve_depth - - if Btrace.particle_auto: - f_start = int(set.frame_start) - f_end = int(set.frame_end + set.lifetime) - else: - if particle_f_end <= particle_f_start: - particle_f_end = particle_f_start + 1 - f_start = particle_f_start - f_end = particle_f_end - - for bFrames in range(f_start, f_end): - bpy.context.scene.frame_set(bFrames) - if not (bFrames-f_start) % Btrace.particle_step: - for bFrames in range(set.count): - if ps.particles[bFrames].alive_state != 'UNBORN': - e = bFrames - bp = spline.bezier_points[bFrames] - pt = ps.particles[e] - bp.co = pt.location - #bp.handle_left = pt.location - #bp.handle_right = pt.location - bp.handle_right_type = curve_handle - bp.handle_left_type = curve_handle - bp.keyframe_insert('co') - bp.keyframe_insert('handle_left') - bp.keyframe_insert('handle_right') - # Select new curve - bpy.ops.object.select_all(action='DESELECT') - curve.select = True - bpy.context.scene.objects.active = curve - addtracemat(curve.data) #Add material - if Btrace.animate: - bpy.ops.curve.btgrow() - return{'FINISHED'} - -################## ################## ################## ############ -## Writing Tool -## Writes a curve by animating its point's radii -## -################## ################## ################## ############ -class OBJECT_OT_writing(bpy.types.Operator): - bl_idname = 'curve.btwriting' - bl_label = 'Write' - bl_description = 'Use Grease Pencil to write and convert to curves' + except Exception as e: + error_handlers(self, "particles.connect", e, + "Connect Particles could not be completed") + + return {'CANCELLED'} + + +# Writing Tool +# Writes a curve by animating its point's radii + +class OBJECT_OT_writing(Operator): + bl_idname = "curve.btwriting" + bl_label = "Write" + bl_description = ("Use Grease Pencil to write and convert to curves\n" + "Needs an existing Grease Pencil layer attached to the Scene") bl_options = {'REGISTER', 'UNDO'} - # @classmethod ### Removed so panel still draws if nothing is selected - # def poll(cls, context): - # return (context.scene.grease_pencil or context.object.grease_pencil != None) + @classmethod + def poll(cls, context): + gp = context.scene.grease_pencil + return gp and gp.layers def execute(self, context): - Btrace = bpy.context.window_manager.curve_tracer - obj = bpy.context.object - gactive = bpy.context.active_object # set selected object before convert - bpy.ops.gpencil.convert(type='CURVE') - gactiveCurve = bpy.context.active_object # get curve after convert - # render ready curve - gactiveCurve.data.resolution_u = Btrace.curve_u - gactiveCurve.data.bevel_resolution = Btrace.curve_resolution # Set bevel resolution from Panel options - gactiveCurve.data.fill_mode = 'FULL' - gactiveCurve.data.bevel_depth = Btrace.curve_depth # Set bevel depth from Panel options - - writeObj = bpy.context.selected_objects - if Btrace.animate: - for i in writeObj: - bpy.context.scene.objects.active = i - bpy.ops.curve.btgrow() - addtracemat(bpy.context.object.data) #Add material - else: - for i in writeObj: - bpy.context.scene.objects.active = i - addtracemat(bpy.context.object.data) #Add material - - # Delete grease pencil strokes - bpy.context.scene.objects.active = gactive - bpy.ops.gpencil.data_unlink() - bpy.context.scene.objects.active = gactiveCurve - # Smooth object - bpy.ops.object.shade_smooth() - # Return to first frame - bpy.context.scene.frame_set(Btrace.anim_f_start) - - return{'FINISHED'} - -################## ################## ################## ############ -## Create Curve -## Convert mesh to curve using either Continuous, All Edges, or Sharp Edges -## Option to create noise -################## ################## ################## ############ - -class OBJECT_OT_convertcurve(bpy.types.Operator): + try: + # first check if the Grease Pencil is attached to the Scene + tool_settings = context.scene.tool_settings + source_data = tool_settings.grease_pencil_source + if source_data in {"OBJECT"}: + self.report({'WARNING'}, + "Operation Cancelled. " + "The Grease Pencil data-block is attached to an Object") + return {"CANCELLED"} + + Btrace = context.window_manager.curve_tracer + # this is hacky - store objects in the scene for comparison later + store_objects = [ob for ob in context.scene.objects] + + gactive = context.active_object + # checking if there are any strokes the easy way + if not bpy.ops.gpencil.convert.poll(): + self.report({'WARNING'}, + "Operation Cancelled. " + "Are there any Grease Pencil Strokes ?") + return {'CANCELLED'} + + bpy.ops.gpencil.convert(type='CURVE') + # get curve after convert (compare the scenes to get the difference) + scene_obj = context.scene.objects + check_materials = True + + for obj in scene_obj: + if obj not in store_objects and obj.type == "CURVE": + gactiveCurve = obj + break + + # render ready curve + gactiveCurve.data.resolution_u = Btrace.curve_u + # Set bevel resolution from Panel options + gactiveCurve.data.bevel_resolution = Btrace.curve_resolution + gactiveCurve.data.fill_mode = 'FULL' + # Set bevel depth from Panel options + gactiveCurve.data.bevel_depth = Btrace.curve_depth + + writeObj = context.selected_objects + if Btrace.animate: + for i in writeObj: + context.scene.objects.active = i + bpy.ops.curve.btgrow() + # Materials + trace_mats = addtracemat(bpy.context.object.data) + if not trace_mats and check_materials is True: + check_materials = False + else: + for i in writeObj: + context.scene.objects.active = i + # Materials + trace_mats = addtracemat(bpy.context.object.data) + if not trace_mats and check_materials is True: + check_materials = False + + # Delete grease pencil strokes + context.scene.objects.active = gactive + bpy.ops.gpencil.data_unlink() + context.scene.objects.active = gactiveCurve + # Smooth object + bpy.ops.object.shade_smooth() + # Return to first frame + bpy.context.scene.frame_set(Btrace.anim_f_start) + + if check_materials is False: + self.report({'WARNING'}, "Some Materials could not be added") + + return{'FINISHED'} + + except Exception as e: + error_handlers(self, "curve.btwriting", e, + "Grease Pencil conversion could not be completed") + + return {'CANCELLED'} + + +# Create Curve +# Convert mesh to curve using either Continuous, All Edges, or Sharp Edges +# Option to create noise + +class OBJECT_OT_convertcurve(Operator): bl_idname = "object.btconvertcurve" bl_label = "Btrace: Create Curve" - bl_description = "Convert mesh to curve using either Continuous, All Edges, or Sharp Edges" + bl_description = ("Convert Mesh to Curve using either Continuous, " + "All Edges or Sharp Edges\n" + "Active Object has to be of a Mesh or Font type") bl_options = {'REGISTER', 'UNDO'} - def execute(self, context): - import bpy, random, mathutils - from mathutils import Vector - - Btrace = bpy.context.window_manager.curve_tracer - traceobjects = bpy.context.selected_objects # create a list with all the selected objects + @classmethod + def poll(cls, context): + return (context.object is not None and + context.object.type in {"MESH", "FONT"}) - obj = bpy.context.object + def execute(self, context): + try: + Btrace = context.window_manager.curve_tracer + obj = context.object - ### Convert Font - if obj.type == 'FONT': - bpy.ops.object.mode_set(mode='OBJECT') - bpy.ops.object.convert(target='CURVE') # Convert edges to curve - bpy.context.object.data.dimensions = '3D' + # Convert Font + if obj.type == 'FONT': + bpy.ops.object.mode_set(mode='OBJECT') + bpy.ops.object.convert(target='CURVE') # Convert edges to curve + bpy.context.object.data.dimensions = '3D' - # make a continuous edge through all vertices - if obj.type == 'MESH': - # Add noise to mesh - if Btrace.distort_curve: - for v in obj.data.vertices: - for u in range(3): - v.co[u] += Btrace.distort_noise * (random.uniform(-1,1)) + # make a continuous edge through all vertices + if obj.type == 'MESH': + # Add noise to mesh + if Btrace.distort_curve: + for v in obj.data.vertices: + for u in range(3): + v.co[u] += Btrace.distort_noise * (rand_uniform(-1, 1)) - if Btrace.convert_edgetype == 'CONTI': - ## Start Continuous edge - bpy.ops.object.mode_set(mode='EDIT') - bpy.ops.mesh.select_all(action='SELECT') - bpy.ops.mesh.delete(type='EDGE_FACE') - bpy.ops.mesh.select_all(action='DESELECT') - verts = bpy.context.object.data.vertices - bpy.ops.object.mode_set(mode='OBJECT') - li = [] - p1 = random.randint(0,len(verts)-1) - - for v in verts: - li.append(v.index) - li.remove(p1) - for z in range(len(li)): - x = [] - for px in li: - d = verts[p1].co - verts[px].co # find distance from first vert - x.append(d.length) - p2 = li[x.index(min(x))] # find the shortest distance list index - verts[p1].select = verts[p2].select = True + if Btrace.convert_edgetype == 'CONTI': + # Start Continuous edge bpy.ops.object.mode_set(mode='EDIT') - bpy.context.tool_settings.mesh_select_mode = [True, False, False] - bpy.ops.mesh.edge_face_add() + bpy.ops.mesh.select_all(action='SELECT') + bpy.ops.mesh.delete(type='EDGE_FACE') bpy.ops.mesh.select_all(action='DESELECT') + verts = bpy.context.object.data.vertices bpy.ops.object.mode_set(mode='OBJECT') - # verts[p1].select = verts[p2].select = False #Doesn't work after Bmesh merge - li.remove(p2) # remove item from list. - p1 = p2 - # Convert edges to curve - bpy.ops.object.mode_set(mode='OBJECT') - bpy.ops.object.convert(target='CURVE') + li = [] + p1 = rand_randint(0, len(verts) - 1) + + for v in verts: + li.append(v.index) + li.remove(p1) + for z in range(len(li)): + x = [] + for px in li: + d = verts[p1].co - verts[px].co # find distance from first vert + x.append(d.length) + p2 = li[x.index(min(x))] # find the shortest distance list index + verts[p1].select = verts[p2].select = True + bpy.ops.object.mode_set(mode='EDIT') + bpy.context.tool_settings.mesh_select_mode = [True, False, False] + bpy.ops.mesh.edge_face_add() + bpy.ops.mesh.select_all(action='DESELECT') + bpy.ops.object.mode_set(mode='OBJECT') + li.remove(p2) # remove item from list. + p1 = p2 + # Convert edges to curve + bpy.ops.object.mode_set(mode='OBJECT') + bpy.ops.object.convert(target='CURVE') - if Btrace.convert_edgetype == 'EDGEALL': - ## Start All edges - bpy.ops.object.mode_set(mode='EDIT') - bpy.ops.mesh.select_all(action='SELECT') - bpy.ops.mesh.delete(type='ONLY_FACE') - bpy.ops.object.mode_set() - bpy.ops.object.convert(target='CURVE') - for sp in obj.data.splines: - sp.type = Btrace.curve_spline - - obj = bpy.context.object - # Set spline type to custom property in panel - bpy.ops.object.editmode_toggle() - bpy.ops.curve.spline_type_set(type=Btrace.curve_spline) - # Set handle type to custom property in panel - bpy.ops.curve.handle_type_set(type=Btrace.curve_handle) - bpy.ops.object.editmode_toggle() - obj.data.fill_mode = 'FULL' - # Set resolution to custom property in panel - obj.data.bevel_resolution = Btrace.curve_resolution - obj.data.resolution_u = Btrace.curve_u - # Set depth to custom property in panel - obj.data.bevel_depth = Btrace.curve_depth - # Smooth object - bpy.ops.object.shade_smooth() - # Modulate curve radius and add distortion - if Btrace.distort_curve: - scale = Btrace.distort_modscale - if scale == 0: - return {'FINISHED'} - for u in obj.data.splines: - for v in u.bezier_points: - v.radius = scale*round(random.random(),3) - return {'FINISHED'} - - -################## ################## ################## ############ -## Mesh Follow, trace vertex or faces -## Create curve at center of selection item, extruded along animation -## Needs to be animated mesh!!! -################## ################## ################## ############ - -class OBJECT_OT_meshfollow(bpy.types.Operator): + if Btrace.convert_edgetype == 'EDGEALL': + # Start All edges + bpy.ops.object.mode_set(mode='EDIT') + bpy.ops.mesh.select_all(action='SELECT') + bpy.ops.mesh.delete(type='ONLY_FACE') + bpy.ops.object.mode_set() + bpy.ops.object.convert(target='CURVE') + for sp in obj.data.splines: + sp.type = Btrace.curve_spline + + obj = context.object + # Set spline type to custom property in panel + bpy.ops.object.editmode_toggle() + bpy.ops.curve.spline_type_set(type=Btrace.curve_spline) + # Set handle type to custom property in panel + bpy.ops.curve.handle_type_set(type=Btrace.curve_handle) + bpy.ops.object.editmode_toggle() + obj.data.fill_mode = 'FULL' + # Set resolution to custom property in panel + obj.data.bevel_resolution = Btrace.curve_resolution + obj.data.resolution_u = Btrace.curve_u + # Set depth to custom property in panel + obj.data.bevel_depth = Btrace.curve_depth + # Smooth object + bpy.ops.object.shade_smooth() + # Modulate curve radius and add distortion + if Btrace.distort_curve: + scale = Btrace.distort_modscale + if scale == 0: + return {'FINISHED'} + for u in obj.data.splines: + for v in u.bezier_points: + v.radius = scale * round(rand_random(), 3) + + return {'FINISHED'} + + except Exception as e: + error_handlers(self, "object.btconvertcurve", e, + "Conversion could not be completed") + + return {'CANCELLED'} + + +# Mesh Follow, trace vertex or faces +# Create curve at center of selection item, extruded along animation +# Needs to be an animated mesh!!! + +class OBJECT_OT_meshfollow(Operator): bl_idname = "object.btmeshfollow" bl_label = "Btrace: Vertex Trace" bl_description = "Trace Vertex or Face on an animated mesh" bl_options = {'REGISTER', 'UNDO'} - @classmethod def poll(cls, context): return (context.object and context.object.type in {'MESH'}) def execute(self, context): - import bpy, random - from mathutils import Vector - - Btrace = bpy.context.window_manager.curve_tracer - distort_curve = Btrace.distort_curve # modulate the resulting curve - stepsize = Btrace.particle_step - traceobjects = bpy.context.selected_objects # create a list with all the selected objects - - obj = bpy.context.object - scn = bpy.context.scene - meshdata = obj.data - cursor = bpy.context.scene.cursor_location.copy() # Store the location to restore at end of script - drawmethod = Btrace.fol_mesh_type # Draw from Edges, Verts, or Faces - if drawmethod == 'VERTS': - meshobjs = obj.data.vertices - if drawmethod == 'FACES': - meshobjs = obj.data.polygons # untested - if drawmethod == 'EDGES': - meshobjs = obj.data.edges # untested - - # Frame properties - start_frame, end_frame = Btrace.fol_start_frame, Btrace.fol_end_frame - if start_frame > end_frame: # Make sure the math works - startframe = end_frame - 5 # if start past end, goto (end - 5) - frames = int((end_frame - start_frame) / stepsize) - - def getsel_option(): # Get selection objects. - sel = [] - seloption, fol_mesh_type = Btrace.fol_sel_option, Btrace.fol_mesh_type # options = 'random', 'custom', 'all' - if fol_mesh_type == 'OBJECT': - pass - else: - if seloption == 'CUSTOM': - for i in meshobjs: - if i.select == True: + try: + Btrace = context.window_manager.curve_tracer + stepsize = Btrace.particle_step + + obj = context.object + scn = context.scene + drawmethod = Btrace.fol_mesh_type # Draw from Edges, Verts, or Faces + + if drawmethod == 'VERTS': + meshobjs = obj.data.vertices + if drawmethod == 'FACES': + meshobjs = obj.data.polygons # untested + if drawmethod == 'EDGES': + meshobjs = obj.data.edges # untested + + # Frame properties + start_frame, end_frame = Btrace.fol_start_frame, Btrace.fol_end_frame + if start_frame > end_frame: # Make sure the math works + start_frame = end_frame - 5 # if start past end, goto (end - 5) + frames = int((end_frame - start_frame) / stepsize) + + def getsel_option(): # Get selection objects + sel = [] + # options are 'random', 'custom', 'all' + seloption, fol_mesh_type = Btrace.fol_sel_option, Btrace.fol_mesh_type + if fol_mesh_type == 'OBJECT': + pass + else: + if seloption == 'CUSTOM': + for i in meshobjs: + if i.select is True: + sel.append(i.index) + if seloption == 'RANDOM': + for i in list(meshobjs): + sel.append(i.index) + finalsel = int(len(sel) * Btrace.fol_perc_verts) + remove = len(sel) - finalsel + for i in range(remove): + sel.pop(rand_randint(0, len(sel) - 1)) + if seloption == 'ALL': + for i in list(meshobjs): sel.append(i.index) - if seloption == 'RANDOM': - for i in list(meshobjs): - sel.append(i.index) - finalsel = int(len(sel) * Btrace.fol_perc_verts) - remove = len(sel) - finalsel - for i in range(remove): - sel.pop(random.randint(0, len(sel) - 1)) - if seloption == 'ALL': - for i in list(meshobjs): - sel.append(i.index) - - return sel - - def get_coord(objindex): - obj_co = [] # list of vector coordinates to use - frame_x = start_frame - for i in range(frames): # create frame numbers list - scn.frame_set(frame_x) - if drawmethod != 'OBJECT': - followed_item = meshobjs[objindex] - if drawmethod == 'VERTS': - g_co = obj.matrix_local * followed_item.co # find Vert vector - - if drawmethod == 'FACES': - g_co = obj.matrix_local * followed_item.normal # find Face vector - - if drawmethod == 'EDGES': - v1 = followed_item.vertices[0] - v2 = followed_item.vertices[1] - co1 = bpy.context.object.data.vertices[v1] - co2 = bpy.context.object.data.vertices[v2] - localcenter = co1.co.lerp(co2.co, 0.5) - g_co = obj.matrix_world * localcenter - - if drawmethod == 'OBJECT': - g_co = objindex.location.copy() - - obj_co.append(g_co) - frame_x = frame_x + stepsize - - scn.frame_set(start_frame) - return obj_co - - def make_curve(co_list): - Btrace = bpy.context.window_manager.curve_tracer - tracer = bpy.data.curves.new('tracer','CURVE') - tracer.dimensions = '3D' - spline = tracer.splines.new('BEZIER') - spline.bezier_points.add(len(co_list)- 1) - curve = bpy.data.objects.new('curve',tracer) - scn.objects.link(curve) - curvelist.append(curve) - # render ready curve - tracer.resolution_u = Btrace.curve_u - tracer.bevel_resolution = Btrace.curve_resolution # Set bevel resolution from Panel options - tracer.fill_mode = 'FULL' - tracer.bevel_depth = Btrace.curve_depth # Set bevel depth from Panel options - curve_handle = Btrace.curve_handle - if curve_handle == 'AUTOMATIC': # hackish AUTOMATIC doesn't work here - curve_handle = 'AUTO' - - # move bezier points to objects - for i in range(len(co_list)): - p = spline.bezier_points[i] - p.co = co_list[i] - p.handle_right_type = curve_handle - p.handle_left_type = curve_handle - return curve - - # Run methods - # Check if Btrace group exists, if not create - bgroup = bpy.data.groups.keys() - if 'Btrace' not in bgroup: - bpy.ops.group.create(name="Btrace") - - Btrace = bpy.context.window_manager.curve_tracer - sel = getsel_option() # Get selection - curvelist = [] # list to use for grow curve - - if Btrace.fol_mesh_type == 'OBJECT': - vector_list = get_coord(obj) - curvelist.append(make_curve(vector_list)) - else: - for i in sel: - vector_list = get_coord(i) - curvelist.append(make_curve(vector_list)) - # Select new curves and add to group - bpy.ops.object.select_all(action='DESELECT') - for curveobject in curvelist: - if curveobject.type == 'CURVE': - curveobject.select = True - bpy.context.scene.objects.active = curveobject - bpy.ops.object.group_link(group="Btrace") - addtracemat(curveobject.data) - curveobject.select = False - if Btrace.animate: # Add grow curve - for curveobject in curvelist: - curveobject.select = True - bpy.ops.curve.btgrow() - for curveobject in curvelist: - curveobject.select = False + return sel + + def get_coord(objindex): + obj_co = [] # list of vector coordinates to use + frame_x = start_frame + for i in range(frames): # create frame numbers list + scn.frame_set(frame_x) + if drawmethod != 'OBJECT': + followed_item = meshobjs[objindex] + if drawmethod == 'VERTS': + # find Vert vector + g_co = obj.matrix_local * followed_item.co + + if drawmethod == 'FACES': + # find Face vector + g_co = obj.matrix_local * followed_item.normal + + if drawmethod == 'EDGES': + v1 = followed_item.vertices[0] + v2 = followed_item.vertices[1] + co1 = bpy.context.object.data.vertices[v1] + co2 = bpy.context.object.data.vertices[v2] + localcenter = co1.co.lerp(co2.co, 0.5) + g_co = obj.matrix_world * localcenter + + if drawmethod == 'OBJECT': + g_co = objindex.location.copy() + + obj_co.append(g_co) + frame_x = frame_x + stepsize + + scn.frame_set(start_frame) + return obj_co + + def make_curve(co_list): + Btrace = bpy.context.window_manager.curve_tracer + tracer = bpy.data.curves.new('tracer', 'CURVE') + tracer.dimensions = '3D' + spline = tracer.splines.new('BEZIER') + spline.bezier_points.add(len(co_list) - 1) + curve = bpy.data.objects.new('curve', tracer) + scn.objects.link(curve) + curvelist.append(curve) + # render ready curve + tracer.resolution_u = Btrace.curve_u + # Set bevel resolution from Panel options + tracer.bevel_resolution = Btrace.curve_resolution + tracer.fill_mode = 'FULL' + # Set bevel depth from Panel options + tracer.bevel_depth = Btrace.curve_depth + curve_handle = Btrace.curve_handle + + # hackish AUTOMATIC doesn't work here + if curve_handle == 'AUTOMATIC': + curve_handle = 'AUTO' + + # move bezier points to objects + for i in range(len(co_list)): + p = spline.bezier_points[i] + p.co = co_list[i] + p.handle_right_type = curve_handle + p.handle_left_type = curve_handle + return curve - obj.select = False # Deselect original object - return {'FINISHED'} + # Run methods + # Check if Btrace group exists, if not create + bgroup = bpy.data.groups.keys() + if 'Btrace' not in bgroup: + bpy.ops.group.create(name="Btrace") -################################################################### -#### Add Tracer Material -################################################################### + Btrace = bpy.context.window_manager.curve_tracer + sel = getsel_option() # Get selection + curvelist = [] # list to use for grow curve + check_materials = True -def addtracemat(matobj): - matslots = bpy.context.object.data.materials.items() # Check if a material exists, skip if it does - if len(matslots) < 1: # Make sure there is only one material slot - engine = bpy.context.scene.render.engine - Btrace = bpy.context.window_manager.curve_tracer - if not Btrace.mat_run_color_blender: # Check if color blender is to be run - if Btrace.trace_mat_random: # Create Random color for each item - # Use random color from chosen palette, assign color lists for each palette - import random - brightColors = [Btrace.brightColor1, Btrace.brightColor2, Btrace.brightColor3, Btrace.brightColor4] - bwColors = [Btrace.bwColor1, Btrace.bwColor2] - customColors = [Btrace.mmColor1, Btrace.mmColor2, Btrace.mmColor3, Btrace.mmColor4, Btrace.mmColor5, Btrace.mmColor6, Btrace.mmColor7, Btrace.mmColor8] - earthColors = [Btrace.earthColor1, Btrace.earthColor2, Btrace.earthColor3, Btrace.earthColor4, Btrace.earthColor5] - greenblueColors = [Btrace.greenblueColor1, Btrace.greenblueColor2, Btrace.greenblueColor3] - if Btrace.mmColors == 'BRIGHT': - #print(random.randint(0, len(brightColors) - 1)) - mat_color = brightColors[random.randint(0, len(brightColors) - 1)] - if Btrace.mmColors == 'BW': - mat_color = bwColors[random.randint(0, len(bwColors) - 1)] - if Btrace.mmColors == 'CUSTOM': - mat_color = customColors[random.randint(0, len(customColors) - 1)] - if Btrace.mmColors == 'EARTH': - mat_color = earthColors[random.randint(0, len(earthColors) - 1)] - if Btrace.mmColors == 'GREENBLUE': - mat_color = greenblueColors[random.randint(0, len(greenblueColors) - 1)] - if Btrace.mmColors == 'RANDOM': - mat_color = (random.random(), random.random(), random.random()) - else: # Choose Single color - mat_color = Btrace.trace_mat_color - # mat_color = Btrace.trace_mat_color - TraceMat = bpy.data.materials.new('TraceMat') - # add cycles or BI render material options - if engine == 'CYCLES': - TraceMat.use_nodes = True - Diffuse_BSDF = TraceMat.node_tree.nodes['Diffuse BSDF'] - r, g, b = mat_color[0], mat_color[1], mat_color[2] - Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] - TraceMat.diffuse_color = mat_color + if Btrace.fol_mesh_type == 'OBJECT': + vector_list = get_coord(obj) + curvelist.append(make_curve(vector_list)) else: - TraceMat.diffuse_color = mat_color - TraceMat.specular_intensity = 0.5 - # Add material to object - matobj.materials.append(bpy.data.materials.get(TraceMat.name)) - - else: # Run color blender operator - bpy.ops.object.colorblender() + for i in sel: + vector_list = get_coord(i) + curvelist.append(make_curve(vector_list)) + # Select new curves and add to group + bpy.ops.object.select_all(action='DESELECT') + for curveobject in curvelist: + if curveobject.type == 'CURVE': + curveobject.select = True + context.scene.objects.active = curveobject + bpy.ops.object.group_link(group="Btrace") + # Materials + trace_mats = addtracemat(curveobject.data) + if not trace_mats and check_materials is True: + check_materials = False + + curveobject.select = False + + if Btrace.animate: # Add grow curve + for curveobject in curvelist: + curveobject.select = True + bpy.ops.curve.btgrow() + for curveobject in curvelist: + curveobject.select = False - return {'FINISHED'} + obj.select = False # Deselect original object -################################################################### -#### Add Color Blender Material -################################################################### + if check_materials is False: + self.report({'WARNING'}, "Some Materials could not be added") -# This is the magical material changer! -class OBJECT_OT_materialChango(bpy.types.Operator): - bl_idname = 'object.colorblender' - bl_label = 'Color Blender' - bl_options = {'REGISTER', 'UNDO'} + return {'FINISHED'} - def execute(self, context): + except Exception as e: + error_handlers(self, "object.btmeshfollow", e, + "Vertex Trace could not be completed") - import bpy, random - Btrace = bpy.context.window_manager.curve_tracer # properties panel - colorObjects = bpy.context.selected_objects - - # Set color lists - brightColors = [Btrace.brightColor1, Btrace.brightColor2, Btrace.brightColor3, Btrace.brightColor4] - bwColors = [Btrace.bwColor1, Btrace.bwColor2] - customColors = [Btrace.mmColor1, Btrace.mmColor2, Btrace.mmColor3, Btrace.mmColor4, Btrace.mmColor5, Btrace.mmColor6, Btrace.mmColor7, Btrace.mmColor8] - earthColors = [Btrace.earthColor1, Btrace.earthColor2, Btrace.earthColor3, Btrace.earthColor4, Btrace.earthColor5] - greenblueColors = [Btrace.greenblueColor1, Btrace.greenblueColor2, Btrace.greenblueColor3] - - colorList = Btrace.mmColors - engine = bpy.context.scene.render.engine - - # Go through each selected object and run the operator - for i in colorObjects: - theObj = i - # Check to see if object has materials - checkMaterials = len(theObj.data.materials) - if engine == 'CYCLES': - materialName = "colorblendMaterial" - madMat = bpy.data.materials.new(materialName) - madMat.use_nodes = True - if checkMaterials == 0: - theObj.data.materials.append(madMat) - else: - theObj.material_slots[0].material = madMat - else: # This is internal - if checkMaterials == 0: - # Add a material - materialName = "colorblendMaterial" - madMat = bpy.data.materials.new(materialName) - theObj.data.materials.append(madMat) - else: - pass # pass since we have what we need - # assign the first material of the object to "mat" - madMat = theObj.data.materials[0] + return {'CANCELLED'} - # Numbers of frames to skip between keyframes - skip = Btrace.mmSkip +# Add Tracer Material +def addtracemat(matobj): + try: + # Check if a material exists, skip if it does + matslots = bpy.context.object.data.materials.items() + if len(matslots) < 1: # Make sure there is only one material slot + engine = bpy.context.scene.render.engine + Btrace = bpy.context.window_manager.curve_tracer - # Random material function - def colorblenderRandom(): - randomRGB = (random.random(), random.random(), random.random()) - if engine == 'CYCLES': - Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] - mat_color = randomRGB - r, g, b = mat_color[0], mat_color[1], mat_color[2] - Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] - madMat.diffuse_color = mat_color + # Check if color blender is to be run + if not Btrace.mat_run_color_blender: + # Create Random color for each item + if Btrace.trace_mat_random: + # Use random color from chosen palette, + # assign color lists for each palette + brightColors = [ + Btrace.brightColor1, Btrace.brightColor2, + Btrace.brightColor3, Btrace.brightColor4 + ] + bwColors = [ + Btrace.bwColor1, Btrace.bwColor2 + ] + customColors = [ + Btrace.mmColor1, Btrace.mmColor2, Btrace.mmColor3, + Btrace.mmColor4, Btrace.mmColor5, Btrace.mmColor6, + Btrace.mmColor7, Btrace.mmColor8 + ] + earthColors = [ + Btrace.earthColor1, Btrace.earthColor2, + Btrace.earthColor3, Btrace.earthColor4, + Btrace.earthColor5 + ] + greenblueColors = [ + Btrace.greenblueColor1, Btrace.greenblueColor2, + Btrace.greenblueColor3 + ] + if Btrace.mmColors == 'BRIGHT': + mat_color = brightColors[ + rand_randint(0, len(brightColors) - 1) + ] + if Btrace.mmColors == 'BW': + mat_color = bwColors[ + rand_randint(0, len(bwColors) - 1) + ] + if Btrace.mmColors == 'CUSTOM': + mat_color = customColors[ + rand_randint(0, len(customColors) - 1) + ] + if Btrace.mmColors == 'EARTH': + mat_color = earthColors[ + rand_randint(0, len(earthColors) - 1) + ] + if Btrace.mmColors == 'GREENBLUE': + mat_color = greenblueColors[ + rand_randint(0, len(greenblueColors) - 1) + ] + if Btrace.mmColors == 'RANDOM': + mat_color = (rand_random(), rand_random(), rand_random()) else: - madMat.diffuse_color = randomRGB + # Choose Single color + mat_color = Btrace.trace_mat_color - def colorblenderCustom(): + TraceMat = bpy.data.materials.new('TraceMat') + # add cycles or BI render material options if engine == 'CYCLES': - Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] - mat_color = random.choice(customColors) + TraceMat.use_nodes = True + Diffuse_BSDF = TraceMat.node_tree.nodes['Diffuse BSDF'] r, g, b = mat_color[0], mat_color[1], mat_color[2] Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] - madMat.diffuse_color = mat_color + TraceMat.diffuse_color = mat_color else: - madMat.diffuse_color = random.choice(customColors) + TraceMat.diffuse_color = mat_color + TraceMat.specular_intensity = 0.5 + # Add material to object + matobj.materials.append(bpy.data.materials.get(TraceMat.name)) - # Black and white color - def colorblenderBW(): - if engine == 'CYCLES': - Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] - mat_color = random.choice(bwColors) - r, g, b = mat_color[0], mat_color[1], mat_color[2] - Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] - madMat.diffuse_color = mat_color - else: - madMat.diffuse_color = random.choice(bwColors) + else: + # Run color blender operator + bpy.ops.object.colorblender() - # Bright colors - def colorblenderBright(): - if engine == 'CYCLES': - Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] - mat_color = random.choice(brightColors) - r, g, b = mat_color[0], mat_color[1], mat_color[2] - Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] - madMat.diffuse_color = mat_color - else: - madMat.diffuse_color = random.choice(brightColors) + return True - # Earth Tones - def colorblenderEarth(): - if engine == 'CYCLES': - Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] - mat_color = random.choice(earthColors) - r, g, b = mat_color[0], mat_color[1], mat_color[2] - Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] - madMat.diffuse_color = mat_color - else: - madMat.diffuse_color = random.choice(earthColors) + except Exception as e: + error_handlers(False, "addtracemat", e, "Function error", True) - # Green to Blue Tones - def colorblenderGreenBlue(): - if engine == 'CYCLES': - Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] - mat_color = random.choice(greenblueColors) - r, g, b = mat_color[0], mat_color[1], mat_color[2] - Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] - madMat.diffuse_color = mat_color - else: - madMat.diffuse_color = random.choice(greenblueColors) - - # define frame start/end variables - scn = bpy.context.scene - start = scn.frame_start - end = scn.frame_end - # Go to each frame in iteration and add material - while start<=(end+(skip-1)): - - bpy.context.scene.frame_set(frame=start) - - # Check what colors setting is checked and run the appropriate function - if Btrace.mmColors=='RANDOM': - colorblenderRandom() - elif Btrace.mmColors=='CUSTOM': - colorblenderCustom() - elif Btrace.mmColors=='BW': - colorblenderBW() - elif Btrace.mmColors=='BRIGHT': - colorblenderBright() - elif Btrace.mmColors=='EARTH': - colorblenderEarth() - elif Btrace.mmColors=='GREENBLUE': - colorblenderGreenBlue() - else: - pass + return False + + +# Add Color Blender Material +# This is the magical material changer! +class OBJECT_OT_materialChango(Operator): + bl_idname = "object.colorblender" + bl_label = "Color Blender" + bl_options = {'REGISTER', 'UNDO'} - # Add keyframe to material + def execute(self, context): + try: + # properties panel + Btrace = context.window_manager.curve_tracer + colorObjects = context.selected_objects + + # Set color lists + brightColors = [ + Btrace.brightColor1, Btrace.brightColor2, + Btrace.brightColor3, Btrace.brightColor4 + ] + bwColors = [Btrace.bwColor1, Btrace.bwColor2] + customColors = [ + Btrace.mmColor1, Btrace.mmColor2, Btrace.mmColor3, + Btrace.mmColor4, Btrace.mmColor5, Btrace.mmColor6, + Btrace.mmColor7, Btrace.mmColor8 + ] + earthColors = [ + Btrace.earthColor1, Btrace.earthColor2, Btrace.earthColor3, + Btrace.earthColor4, Btrace.earthColor5 + ] + greenblueColors = [ + Btrace.greenblueColor1, Btrace.greenblueColor2, + Btrace.greenblueColor3 + ] + engine = context.scene.render.engine + + # Go through each selected object and run the operator + for i in colorObjects: + theObj = i + # Check to see if object has materials + checkMaterials = len(theObj.data.materials) if engine == 'CYCLES': - madMat.node_tree.nodes['Diffuse BSDF'].inputs[0].keyframe_insert('default_value') - madMat.keyframe_insert('diffuse_color') # not sure if this is need, it's viewport color only - else: - madMat.keyframe_insert('diffuse_color') + materialName = "colorblendMaterial" + madMat = bpy.data.materials.new(materialName) + madMat.use_nodes = True + if checkMaterials == 0: + theObj.data.materials.append(madMat) + else: + theObj.material_slots[0].material = madMat + else: # This is internal + if checkMaterials == 0: + # Add a material + materialName = "colorblendMaterial" + madMat = bpy.data.materials.new(materialName) + theObj.data.materials.append(madMat) + else: + pass # pass since we have what we need + # assign the first material of the object to "mat" + madMat = theObj.data.materials[0] + + # Numbers of frames to skip between keyframes + skip = Btrace.mmSkip + + # Random material function + def colorblenderRandom(): + randomRGB = (rand_random(), rand_random(), rand_random()) + if engine == 'CYCLES': + Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] + mat_color = randomRGB + r, g, b = mat_color[0], mat_color[1], mat_color[2] + Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] + madMat.diffuse_color = mat_color + else: + madMat.diffuse_color = randomRGB + + def colorblenderCustom(): + if engine == 'CYCLES': + Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] + mat_color = rand_choice(customColors) + r, g, b = mat_color[0], mat_color[1], mat_color[2] + Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] + madMat.diffuse_color = mat_color + else: + madMat.diffuse_color = rand_choice(customColors) + + # Black and white color + def colorblenderBW(): + if engine == 'CYCLES': + Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] + mat_color = rand_choice(bwColors) + r, g, b = mat_color[0], mat_color[1], mat_color[2] + Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] + madMat.diffuse_color = mat_color + else: + madMat.diffuse_color = rand_choice(bwColors) + + # Bright colors + def colorblenderBright(): + if engine == 'CYCLES': + Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] + mat_color = rand_choice(brightColors) + r, g, b = mat_color[0], mat_color[1], mat_color[2] + Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] + madMat.diffuse_color = mat_color + else: + madMat.diffuse_color = rand_choice(brightColors) + + # Earth Tones + def colorblenderEarth(): + if engine == 'CYCLES': + Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] + mat_color = rand_choice(earthColors) + r, g, b = mat_color[0], mat_color[1], mat_color[2] + Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] + madMat.diffuse_color = mat_color + else: + madMat.diffuse_color = rand_choice(earthColors) + + # Green to Blue Tones + def colorblenderGreenBlue(): + if engine == 'CYCLES': + Diffuse_BSDF = madMat.node_tree.nodes['Diffuse BSDF'] + mat_color = rand_choice(greenblueColors) + r, g, b = mat_color[0], mat_color[1], mat_color[2] + Diffuse_BSDF.inputs[0].default_value = [r, g, b, 1] + madMat.diffuse_color = mat_color + else: + madMat.diffuse_color = rand_choice(greenblueColors) + + # define frame start/end variables + scn = context.scene + start = scn.frame_start + end = scn.frame_end + + # Go to each frame in iteration and add material + while start <= (end + (skip - 1)): + bpy.context.scene.frame_set(frame=start) + + # Check what colors setting is checked and run the appropriate function + if Btrace.mmColors == 'RANDOM': + colorblenderRandom() + elif Btrace.mmColors == 'CUSTOM': + colorblenderCustom() + elif Btrace.mmColors == 'BW': + colorblenderBW() + elif Btrace.mmColors == 'BRIGHT': + colorblenderBright() + elif Btrace.mmColors == 'EARTH': + colorblenderEarth() + elif Btrace.mmColors == 'GREENBLUE': + colorblenderGreenBlue() + else: + pass + + # Add keyframe to material + if engine == 'CYCLES': + madMat.node_tree.nodes[ + 'Diffuse BSDF'].inputs[0].keyframe_insert('default_value') + # not sure if this is need, it's viewport color only + madMat.keyframe_insert('diffuse_color') + else: + madMat.keyframe_insert('diffuse_color') + + # Increase frame number + start += skip + + return{'FINISHED'} + + except Exception as e: + error_handlers(self, "object.colorblender", e, + "Color Blender could not be completed") + + return {'CANCELLED'} - # Increase frame number - start += skip - return{'FINISHED'} -###### This clears the keyframes ###### -class OBJECT_OT_clearColorblender(bpy.types.Operator): - bl_idname = 'object.colorblenderclear' - bl_label = 'Clear colorblendness' +# This clears the keyframes +class OBJECT_OT_clearColorblender(Operator): + bl_idname = "object.colorblenderclear" + bl_label = "Clear colorblendness" + bl_description = "Clear the color keyframes" bl_options = {'REGISTER', 'UNDO'} def invoke(self, context, event): + try: + colorObjects = context.selected_objects + engine = context.scene.render.engine - import bpy, random - mcolorblend = context.window_manager.colorblender_operator # properties panel - colorObjects = bpy.context.selected_objects - - # Go through each selected object and run the operator - for i in colorObjects: - theObj = i - # assign the first material of the object to "mat" - matCl = theObj.data.materials[0] + # Go through each selected object and run the operator + for i in colorObjects: + theObj = i + # assign the first material of the object to "mat" + matCl = theObj.data.materials[0] + + # define frame start/end variables + scn = context.scene + start = scn.frame_start + end = scn.frame_end + + # Remove all keyframes from diffuse_color, super sloppy + while start <= (end + 100): + context.scene.frame_set(frame=start) + try: + if engine == 'CYCLES': + matCl.node_tree.nodes[ + 'Diffuse BSDF'].inputs[0].keyframe_delete('default_value') + elif engine == 'BLENDER_RENDER': + matCl.keyframe_delete('diffuse_color') + except: + pass + start += 1 - # define frame start/end variables - scn = bpy.context.scene - start = scn.frame_start - end = scn.frame_end + return{'FINISHED'} - # Remove all keyframes from diffuse_color, super sloppy need to find better way - while start <= (end * 2): - bpy.context.scene.frame_set(frame=start) - matCl.keyframe_delete('diffuse_color') - start += 1 + except Exception as e: + error_handlers(self, "object.colorblenderclear", e, + "Reset Keyframes could not be completed") - return{'FINISHED'} + return {'CANCELLED'} -################## ################## ################## ############ -## F-Curve Noise -## will add noise modifiers to each selected object f-curves -## change type to: 'rotation' | 'location' | 'scale' | '' to effect all -## first record a keyframe for this to work (to generate the f-curves) -################## ################## ################## ############ +# F-Curve Noise +# will add noise modifiers to each selected object f-curves +# change type to: 'rotation' | 'location' | 'scale' | '' to effect all +# first record a keyframe for this to work (to generate the f-curves) -class OBJECT_OT_fcnoise(bpy.types.Operator): +class OBJECT_OT_fcnoise(Operator): bl_idname = "object.btfcnoise" bl_label = "Btrace: F-curve Noise" bl_options = {'REGISTER', 'UNDO'} def execute(self, context): - import bpy, random - - Btrace = bpy.context.window_manager.curve_tracer - amp = Btrace.fcnoise_amp - timescale = Btrace.fcnoise_timescale - addkeyframe = Btrace.fcnoise_key - - # This sets properties for Loc, Rot and Scale if they're checked in the Tools window - noise_rot = 'rotation' - noise_loc = 'location' - noise_scale = 'scale' - if not Btrace.fcnoise_rot: - noise_rot = 'none' - if not Btrace.fcnoise_loc: - noise_loc = 'none' - if not Btrace.fcnoise_scale: - noise_scale = 'none' - - type = noise_loc, noise_rot, noise_scale # Add settings from panel for type of keyframes - amplitude = amp - time_scale = timescale - - for i in bpy.context.selected_objects: - # Add keyframes, this is messy and should only add keyframes for what is checked - if addkeyframe == True: - bpy.ops.anim.keyframe_insert(type="LocRotScale") - for obj in bpy.context.selected_objects: - if obj.animation_data: - for c in obj.animation_data.action.fcurves: - if c.data_path.startswith(type): - # clean modifiers - for m in c.modifiers : - c.modifiers.remove(m) - # add noide modifiers - n = c.modifiers.new('NOISE') - n.strength = amplitude - n.scale = time_scale - n.phase = random.randint(0,999) - return {'FINISHED'} - -################## ################## ################## ############ -## Curve Grow Animation -## Animate curve radius over length of time -################## ################## ################## ############ -class OBJECT_OT_curvegrow(bpy.types.Operator): - bl_idname = 'curve.btgrow' - bl_label = 'Run Script' - bl_description = 'Keyframe points radius' + try: + Btrace = context.window_manager.curve_tracer + amp = Btrace.fcnoise_amp + timescale = Btrace.fcnoise_timescale + addkeyframe = Btrace.fcnoise_key + + # This sets properties for Loc, Rot and Scale + # if they're checked in the Tools window + noise_rot = 'rotation' + noise_loc = 'location' + noise_scale = 'scale' + if not Btrace.fcnoise_rot: + noise_rot = 'none' + if not Btrace.fcnoise_loc: + noise_loc = 'none' + if not Btrace.fcnoise_scale: + noise_scale = 'none' + + # Add settings from panel for type of keyframes + types = noise_loc, noise_rot, noise_scale + amplitude = amp + time_scale = timescale + + for i in context.selected_objects: + # Add keyframes, this is messy and should only + # add keyframes for what is checked + if addkeyframe is True: + bpy.ops.anim.keyframe_insert(type="LocRotScale") + for obj in context.selected_objects: + if obj.animation_data: + for c in obj.animation_data.action.fcurves: + if c.data_path.startswith(types): + # clean modifiers + for m in c.modifiers: + c.modifiers.remove(m) + # add noide modifiers + n = c.modifiers.new('NOISE') + n.strength = amplitude + n.scale = time_scale + n.phase = rand_randint(0, 999) + + return {'FINISHED'} + + except Exception as e: + error_handlers(self, "object.btfcnoise", e, + "F-curve Noise could not be completed") + + return {'CANCELLED'} + + +# Curve Grow Animation +# Animate curve radius over length of time + +class OBJECT_OT_curvegrow(Operator): + bl_idname = "curve.btgrow" + bl_label = "Run Script" + bl_description = "Keyframe points radius" bl_options = {'REGISTER', 'UNDO'} @classmethod @@ -1520,109 +1154,107 @@ class OBJECT_OT_curvegrow(bpy.types.Operator): return (context.object and context.object.type in {'CURVE'}) def execute(self, context): - Btrace = bpy.context.window_manager.curve_tracer - anim_f_start, anim_length, anim_auto = Btrace.anim_f_start, Btrace.anim_length, Btrace.anim_auto - curve_resolution, curve_depth = Btrace.curve_resolution, Btrace.curve_depth - # make the curve visible - objs = bpy.context.selected_objects - for i in objs: # Execute on multiple selected objects - bpy.context.scene.objects.active = i - obj = bpy.context.active_object - try: - obj.data.fill_mode = 'FULL' - except: - obj.data.dimensions = '3D' - obj.data.fill_mode = 'FULL' - if not obj.data.bevel_resolution: - obj.data.bevel_resolution = curve_resolution - if not obj.data.bevel_depth: - obj.data.bevel_depth = curve_depth - if anim_auto: - anim_f_start = bpy.context.scene.frame_start - anim_length = bpy.context.scene.frame_end - # get points data and beautify - actual, total = anim_f_start, 0 - for sp in obj.data.splines: - total += len(sp.points) + len(sp.bezier_points) - step = anim_length / total - for sp in obj.data.splines: - sp.radius_interpolation = 'BSPLINE' - po = [p for p in sp.points] + [p for p in sp.bezier_points] - if not Btrace.anim_keepr: - for p in po: - p.radius = 1 - if Btrace.anim_tails and not sp.use_cyclic_u: - po[0].radius = po[-1].radius = 0 - po[1].radius = po[-2].radius = .65 - ra = [p.radius for p in po] - - # record the keyframes - for i in range(len(po)): - po[i].radius = 0 - po[i].keyframe_insert('radius', frame=actual) - actual += step - po[i].radius = ra[i] - po[i].keyframe_insert('radius', frame=(actual + Btrace.anim_delay)) - - if Btrace.anim_f_fade: - po[i].radius = ra[i] - po[i].keyframe_insert('radius', frame=(actual + Btrace.anim_f_fade - step)) + try: + # not so nice with the nested try blocks, however the inside one + # is used as a switch statement + Btrace = context.window_manager.curve_tracer + anim_f_start, anim_length, anim_auto = Btrace.anim_f_start, \ + Btrace.anim_length, \ + Btrace.anim_auto + curve_resolution, curve_depth = Btrace.curve_resolution, \ + Btrace.curve_depth + # make the curve visible + objs = context.selected_objects + # Execute on multiple selected objects + for i in objs: + context.scene.objects.active = i + obj = context.active_object + try: + obj.data.fill_mode = 'FULL' + except: + obj.data.dimensions = '3D' + obj.data.fill_mode = 'FULL' + if not obj.data.bevel_resolution: + obj.data.bevel_resolution = curve_resolution + if not obj.data.bevel_depth: + obj.data.bevel_depth = curve_depth + if anim_auto: + anim_f_start = bpy.context.scene.frame_start + anim_length = bpy.context.scene.frame_end + # get points data and beautify + actual, total = anim_f_start, 0 + for sp in obj.data.splines: + total += len(sp.points) + len(sp.bezier_points) + step = anim_length / total + for sp in obj.data.splines: + sp.radius_interpolation = 'BSPLINE' + po = [p for p in sp.points] + [p for p in sp.bezier_points] + if not Btrace.anim_keepr: + for p in po: + p.radius = 1 + if Btrace.anim_tails and not sp.use_cyclic_u: + po[0].radius = po[-1].radius = 0 + po[1].radius = po[-2].radius = .65 + ra = [p.radius for p in po] + + # record the keyframes + for i in range(len(po)): po[i].radius = 0 - po[i].keyframe_insert('radius', frame=(actual + Btrace.anim_delay + Btrace.anim_f_fade)) + po[i].keyframe_insert('radius', frame=actual) + actual += step + po[i].radius = ra[i] + po[i].keyframe_insert( + 'radius', + frame=(actual + Btrace.anim_delay) + ) + + if Btrace.anim_f_fade: + po[i].radius = ra[i] + po[i].keyframe_insert( + 'radius', + frame=(actual + Btrace.anim_f_fade - step) + ) + po[i].radius = 0 + po[i].keyframe_insert( + 'radius', + frame=(actual + Btrace.anim_delay + Btrace.anim_f_fade) + ) + + bpy.context.scene.frame_set(Btrace.anim_f_start) - bpy.context.scene.frame_set(Btrace.anim_f_start) - return{'FINISHED'} - -################## ################## ################## ############ -## Remove animation and curve radius data -################## ################## ################## ############ -class OBJECT_OT_reset(bpy.types.Operator): - bl_idname = 'object.btreset' - bl_label = 'Clear animation' - bl_description = 'Remove animation / curve radius data' + return{'FINISHED'} + + except Exception as e: + error_handlers(self, "curve.btgrow", e, + "Grow curve could not be completed") + + return {'CANCELLED'} + + +# Remove animation and curve radius data +class OBJECT_OT_reset(Operator): + bl_idname = "object.btreset" + bl_label = "Clear animation" + bl_description = "Remove animation / curve radius data" bl_options = {'REGISTER', 'UNDO'} def execute(self, context): - objs = bpy.context.selected_objects - for i in objs: # Execute on multiple selected objects - bpy.context.scene.objects.active = i - obj = bpy.context.active_object - obj.animation_data_clear() - if obj.type == 'CURVE': - for sp in obj.data.splines: - po = [p for p in sp.points] + [p for p in sp.bezier_points] - for p in po: - p.radius = 1 - return{'FINISHED'} - -### Define Classes to register -classes = [ - TracerProperties, - TracerPropertiesMenu, - addTracerObjectPanel, - OBJECT_OT_convertcurve, - OBJECT_OT_objecttrace, - OBJECT_OT_objectconnect, - OBJECT_OT_writing, - OBJECT_OT_particletrace, - OBJECT_OT_traceallparticles, - OBJECT_OT_curvegrow, - OBJECT_OT_reset, - OBJECT_OT_fcnoise, - OBJECT_OT_meshfollow, - OBJECT_OT_materialChango, - OBJECT_OT_clearColorblender - ] - -def register(): - for c in classes: - bpy.utils.register_class(c) - bpy.types.WindowManager.curve_tracer = bpy.props.PointerProperty(type=TracerProperties) - bpy.types.WindowManager.btrace_menu = bpy.props.PointerProperty(type=TracerPropertiesMenu, update=deselect_others) - -def unregister(): - for c in classes: - bpy.utils.unregister_class(c) - del bpy.types.WindowManager.curve_tracer -if __name__ == "__main__": - register() + try: + objs = context.selected_objects + for i in objs: # Execute on multiple selected objects + context.scene.objects.active = i + obj = context.active_object + obj.animation_data_clear() + if obj.type == 'CURVE': + for sp in obj.data.splines: + po = [p for p in sp.points] + [p for p in sp.bezier_points] + for p in po: + p.radius = 1 + + return{'FINISHED'} + + except Exception as e: + error_handlers(self, "object.btreset", e, + "Clear animation could not be completed") + + return {'CANCELLED'} diff --git a/btrace/bTrace_props.py b/btrace/bTrace_props.py new file mode 100644 index 00000000..bff3b88f --- /dev/null +++ b/btrace/bTrace_props.py @@ -0,0 +1,951 @@ +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### + +import bpy +from bpy.types import ( + Panel, + PropertyGroup, + ) +from bpy.props import ( + FloatProperty, + EnumProperty, + IntProperty, + BoolProperty, + FloatVectorProperty, + ) + + +# Class to define properties +class TracerProperties(PropertyGroup): + """Options for tools""" + curve_spline = EnumProperty( + name="Spline", + items=(("POLY", "Poly", "Use Poly spline type"), + ("NURBS", "Nurbs", "Use Nurbs spline type"), + ("BEZIER", "Bezier", "Use Bezier spline type")), + description="Choose which type of spline to use when curve is created", + default="BEZIER" + ) + curve_handle = EnumProperty( + name="Handle", + items=(("ALIGNED", "Aligned", "Use Aligned Handle Type"), + ("AUTOMATIC", "Automatic", "Use Auto Handle Type"), + ("FREE_ALIGN", "Free Align", "Use Free Handle Type"), + ("VECTOR", "Vector", "Use Vector Handle Type")), + description="Choose which type of handle to use when curve is created", + default="VECTOR" + ) + curve_resolution = IntProperty( + name="Bevel Resolution", + min=1, max=32, + default=4, + description="Adjust the Bevel resolution" + ) + curve_depth = FloatProperty( + name="Bevel Depth", + min=0.0, max=100.0, + default=0.1, + description="Adjust the Bevel depth" + ) + curve_u = IntProperty( + name="Resolution U", + min=0, max=64, + default=12, + description="Adjust the Surface resolution" + ) + curve_join = BoolProperty( + name="Join Curves", + default=False, + description="Join all the curves after they have been created" + ) + curve_smooth = BoolProperty( + name="Smooth", + default=True, + description="Render curve smooth" + ) + # Option to Duplicate Mesh + object_duplicate = BoolProperty( + name="Apply to Copy", + default=False, + description="Apply curve to a copy of object" + ) + # Distort Mesh options + distort_modscale = IntProperty( + name="Modulation Scale", + min=0, max=50, + default=2, + description="Add a scale to modulate the curve at random points, set to 0 to disable" + ) + distort_noise = FloatProperty( + name="Mesh Noise", + min=0.0, max=50.0, + default=0.00, + description="Adjust noise added to mesh before adding curve" + ) + # Particle Options + particle_step = IntProperty( + name="Step Size", + min=1, max=50, + default=5, + description="Sample one every this number of frames" + ) + particle_auto = BoolProperty( + name="Auto Frame Range", + default=True, + description="Calculate Frame Range from particles life" + ) + particle_f_start = IntProperty( + name='Start Frame', + min=1, max=5000, + default=1, + description='Start frame' + ) + particle_f_end = IntProperty( + name="End Frame", + min=1, max=5000, + default=250, + description="End frame" + ) + # F-Curve Modifier Properties + fcnoise_rot = BoolProperty( + name="Rotation", + default=False, + description="Affect Rotation" + ) + fcnoise_loc = BoolProperty( + name="Location", + default=True, + description="Affect Location" + ) + fcnoise_scale = BoolProperty( + name="Scale", + default=False, + description="Affect Scale" + ) + fcnoise_amp = IntProperty( + name="Amp", + min=1, max=500, + default=5, + description="Adjust the amplitude" + ) + fcnoise_timescale = FloatProperty( + name="Time Scale", + min=1, max=500, + default=50, + description="Adjust the time scale" + ) + fcnoise_key = BoolProperty( + name="Add Keyframe", + default=True, + description="Keyframe is needed for tool, this adds a LocRotScale keyframe" + ) + show_curve_settings = BoolProperty( + name="Curve Settings", + default=False, + description="Change the curve settings for the created curve" + ) + material_settings = BoolProperty( + name="Material Settings", + default=False, + description="Change the material settings for the created curve" + ) + particle_settings = BoolProperty( + name="Particle Settings", + default=False, + description="Show the settings for the created curve" + ) + animation_settings = BoolProperty( + name="Animation Settings", + default=False, + description="Show the settings for the Animations" + ) + distort_curve = BoolProperty( + name="Add Distortion", + default=False, + description="Set options to distort the final curve" + ) + connect_noise = BoolProperty( + name="F-Curve Noise", + default=False, + description="Adds F-Curve Noise Modifier to selected objects" + ) + settings_objectTrace = BoolProperty( + name="Object Trace Settings", + default=False, + description="Trace selected mesh object with a curve" + ) + settings_objectsConnect = BoolProperty( + name="Objects Connect Settings", + default=False, + description="Connect objects with a curve controlled by hooks" + ) + settings_objectTrace = BoolProperty( + name="Object Trace Settings", + default=False, + description="Trace selected mesh object with a curve" + ) + respect_order = BoolProperty( + name="Order", + default=False, + description="Remember order objects were selected" + ) + settings_particleTrace = BoolProperty( + name="Particle Trace Settings", + default=False, + description="Trace particle path with a curve" + ) + settings_particleConnect = BoolProperty( + name="Particle Connect Settings", + default=False, + description="Connect particles with a curves and animated over particle lifetime" + ) + settings_growCurve = BoolProperty( + name="Grow Curve Settings", + default=False, + description="Animate curve bevel over time by keyframing points radius" + ) + settings_fcurve = BoolProperty( + name="F-Curve Settings", + default=False, + description="F-Curve Settings" + ) + settings_toggle = BoolProperty( + name="Settings", + default=False, + description="Toggle Settings" + ) + # Animation Options + anim_auto = BoolProperty( + name="Auto Frame Range", + default=True, + description="Automatically calculate Frame Range" + ) + anim_f_start = IntProperty( + name="Start", + min=1, max=2500, + default=1, + description="Start frame / Hidden object" + ) + anim_length = IntProperty( + name="Duration", + min=1, + soft_max=1000, max=2500, + default=100, + description="Animation Length" + ) + anim_f_fade = IntProperty( + name="Fade After", + min=0, + soft_max=250, max=2500, + default=10, + description="Fade after this frames / Zero means no fade" + ) + anim_delay = IntProperty( + name="Grow", + min=0, max=50, + default=5, + description="Frames it takes a point to grow" + ) + anim_tails = BoolProperty( + name='Tails on endpoints', + default=True, + description='Set radius to zero for open splines endpoints' + ) + anim_keepr = BoolProperty( + name="Keep Radius", + default=True, + description="Try to keep radius data from original curve" + ) + animate = BoolProperty( + name="Animate Result", + default=False, + description="Animate the final curve objects" + ) + # Convert to Curve options + convert_conti = BoolProperty( + name="Continuous", + default=True, + description="Create a continuous curve using verts from mesh" + ) + convert_everyedge = BoolProperty( + name="Every Edge", + default=False, + description="Create a curve from all verts in a mesh" + ) + convert_edgetype = EnumProperty( + name="Edge Type for Curves", + items=(("CONTI", "Continuous", "Create a continuous curve using verts from mesh"), + ("EDGEALL", "All Edges", "Create a curve from every edge in a mesh")), + description="Choose which type of spline to use when curve is created", + default="CONTI" + ) + convert_joinbefore = BoolProperty( + name="Join objects before convert", + default=False, + description="Join all selected mesh to one object before converting to mesh" + ) + # Mesh Follow Options + fol_edge_select = BoolProperty( + name="Edge", + default=False, + description="Grow from edges" + ) + fol_vert_select = BoolProperty( + name="Vertex", + default=False, + description="Grow from verts" + ) + fol_face_select = BoolProperty( + name="Face", + default=True, + description="Grow from faces" + ) + fol_mesh_type = EnumProperty( + name="Mesh type", + default="VERTS", + description="Mesh feature to draw cruves from", + items=(("VERTS", "Verts", "Draw from Verts"), + ("EDGES", "Edges", "Draw from Edges"), + ("FACES", "Faces", "Draw from Faces"), + ("OBJECT", "Object", "Draw from Object origin")) + ) + fol_start_frame = IntProperty( + name="Start Frame", + min=1, max=2500, + default=1, + description="Start frame for range to trace" + ) + fol_end_frame = IntProperty( + name="End Frame", + min=1, max=2500, + default=250, + description="End frame for range to trace" + ) + fol_perc_verts = FloatProperty( + name="Reduce selection by", + min=0.001, max=1.000, + default=0.5, + description="percentage of total verts to trace" + ) + fol_sel_option = EnumProperty( + name="Selection type", + description="Choose which objects to follow", + default="RANDOM", + items=(("RANDOM", "Random", "Follow Random items"), + ("CUSTOM", "Custom Select", "Follow selected items"), + ("ALL", "All", "Follow all items")) + ) + trace_mat_color = FloatVectorProperty( + name="Material Color", + description="Choose material color", + min=0, max=1, + default=(0.0, 0.3, 0.6), + subtype="COLOR" + ) + trace_mat_random = BoolProperty( + name="Random Color", + default=False, + description='Make the material colors random' + ) + # Material custom Properties properties + mat_simple_adv_toggle = EnumProperty( + name="Material Options", + items=(("SIMPLE", "Simple", "Show Simple Material Options"), + ("ADVANCED", "Advanced", "Show Advanced Material Options")), + description="Choose which Material Options to show", + default="SIMPLE" + ) + mat_run_color_blender = BoolProperty( + name="Run Color Blender", + default=False, + description="Generate colors from a color scheme" + ) + mmColors = EnumProperty( + items=(("RANDOM", "Random", "Use random colors"), + ("CUSTOM", "Custom", "Use custom colors"), + ("BW", "Black/White", "Use Black and White"), + ("BRIGHT", "Bright Colors", "Use Bright colors"), + ("EARTH", "Earth", "Use Earth colors"), + ("GREENBLUE", "Green to Blue", "Use Green to Blue colors")), + description="Choose which type of colors the materials uses", + default="BRIGHT", + name="Define a color palette" + ) + # Custom property for how many keyframes to skip + mmSkip = IntProperty( + name="frames", + min=1, max=500, + default=20, + description="Number of frames between each keyframes" + ) + # Custom property to enable/disable random order for the + mmBoolRandom = BoolProperty( + name="Random Order", + default=False, + description="Randomize the order of the colors" + ) + # Custom Color properties + mmColor1 = FloatVectorProperty( + min=0, max=1, + default=(0.8, 0.8, 0.8), + description="Custom Color 1", subtype="COLOR" + ) + mmColor2 = FloatVectorProperty( + min=0, max=1, + default=(0.8, 0.8, 0.3), + description="Custom Color 2", + subtype="COLOR" + ) + mmColor3 = FloatVectorProperty( + min=0, max=1, + default=(0.8, 0.5, 0.6), + description="Custom Color 3", + subtype="COLOR" + ) + mmColor4 = FloatVectorProperty( + min=0, max=1, + default=(0.2, 0.8, 0.289), + description="Custom Color 4", + subtype="COLOR" + ) + mmColor5 = FloatVectorProperty( + min=0, max=1, + default=(1.0, 0.348, 0.8), + description="Custom Color 5", + subtype="COLOR" + ) + mmColor6 = FloatVectorProperty( + min=0, max=1, + default=(0.4, 0.67, 0.8), + description="Custom Color 6", + subtype="COLOR" + ) + mmColor7 = FloatVectorProperty( + min=0, max=1, + default=(0.66, 0.88, 0.8), + description="Custom Color 7", + subtype="COLOR" + ) + mmColor8 = FloatVectorProperty( + min=0, max=1, + default=(0.8, 0.38, 0.22), + description="Custom Color 8", + subtype="COLOR" + ) + # BW Color properties + bwColor1 = FloatVectorProperty( + min=0, max=1, + default=(0.0, 0.0, 0.0), + description="Black/White Color 1", + subtype="COLOR" + ) + bwColor2 = FloatVectorProperty( + min=0, max=1, + default=(1.0, 1.0, 1.0), + description="Black/White Color 2", + subtype="COLOR" + ) + # Bright Color properties + brightColor1 = FloatVectorProperty( + min=0, max=1, + default=(1.0, 0.0, 0.75), + description="Bright Color 1", + subtype="COLOR" + ) + brightColor2 = FloatVectorProperty( + min=0, max=1, + default=(0.0, 1.0, 1.0), + description="Bright Color 2", + subtype="COLOR" + ) + brightColor3 = FloatVectorProperty( + min=0, max=1, + default=(0.0, 1.0, 0.0), + description="Bright Color 3", + subtype="COLOR" + ) + brightColor4 = FloatVectorProperty( + min=0, max=1, + default=(1.0, 1.0, 0.0), + description="Bright Color 4", subtype="COLOR" + ) + # Earth Color Properties + earthColor1 = FloatVectorProperty( + min=0, max=1, + default=(0.068, 0.019, 0.014), + description="Earth Color 1", + subtype="COLOR" + ) + earthColor2 = FloatVectorProperty( + min=0, max=1, + default=(0.089, 0.060, 0.047), + description="Earth Color 2", + subtype="COLOR" + ) + earthColor3 = FloatVectorProperty( + min=0, max=1, + default=(0.188, 0.168, 0.066), + description="Earth Color 3", + subtype="COLOR" + ) + earthColor4 = FloatVectorProperty( + min=0, max=1, + default=(0.445, 0.296, 0.065), + description="Earth Color 4", + subtype="COLOR" + ) + earthColor5 = FloatVectorProperty( + min=0, max=1, + default=(0.745, 0.332, 0.065), + description="Earth Color 5", + subtype="COLOR" + ) + # Green to Blue Color properties + greenblueColor1 = FloatVectorProperty( + min=0, max=1, + default=(0.296, 0.445, 0.074), + description="Green/Blue Color 1", + subtype="COLOR" + ) + greenblueColor2 = FloatVectorProperty( + min=0, max=1, + default=(0.651, 1.0, 0.223), + description="Green/Blue Color 2", + subtype="COLOR" + ) + greenblueColor3 = FloatVectorProperty( + min=0, max=1, + default=(0.037, 0.047, 0.084), + description="Green/Blue Color 3", + subtype="COLOR" + ) + + # Toolbar show/hide booleans for tool options + btrace_menu_items = [ + ('tool_help', "Help", + "Pick one of the options below", "INFO", 0), + ('tool_objectTrace', "Object Trace", + "Trace selected mesh object with a curve", "FORCE_MAGNETIC", 1), + ('tool_objectsConnect', "Objects Connect", + "Connect objects with a curve controlled by hooks", "OUTLINER_OB_EMPTY", 2), + ('tool_meshFollow', "Mesh Follow", + "Follow selection items on animated mesh object", "DRIVER", 3), + ('tool_handwrite', "Handwriting", + "Create and Animate curve using the grease pencil", "BRUSH_DATA", 4), + ('tool_particleTrace', "Particle Trace", + "Trace particle path with a curve", "PARTICLES", 5), + ('tool_particleConnect', "Particle Connect", + "Connect particles with a curves and animated over particle lifetime", "MOD_PARTICLES", 6), + ('tool_growCurve', "Grow Curve", + "Animate curve bevel over time by keyframing points radius", "META_BALL", 7), + ('tool_fcurve', "F-Curve Noise", + "Add F-Curve noise to selected objects", "RNDCURVE", 8), + ('tool_colorblender', "Color Blender", + "Pick the color of the created curves", "COLOR", 9), + ] + btrace_toolmenu = EnumProperty( + name="Tools", + items=btrace_menu_items, + description="", + default='tool_help' + ) + + +# Draw Brush panel in Toolbar +class addTracerObjectPanel(Panel): + bl_label = "Btrace" + bl_space_type = "VIEW_3D" + bl_region_type = "TOOLS" + bl_context = "objectmode" + bl_category = "Create" + bl_options = {'DEFAULT_CLOSED'} + + def draw(self, context): + layout = self.layout + Btrace = context.window_manager.curve_tracer + addon_prefs = context.user_preferences.addons["btrace"].preferences + switch_expand = addon_prefs.expand_enum + obj = context.object + + # Color Blender Panel options + def color_blender(): + # Buttons for Color Blender + row = box.row() + row.label("Color palette") + row.prop(Btrace, "mmColors", text="") + + # Show Custom Colors if selected + if Btrace.mmColors == 'CUSTOM': + row = box.row(align=True) + for i in range(1, 9): + row.prop(Btrace, "mmColor" + str(i), text="") + # Show Earth Colors + elif Btrace.mmColors == 'BW': + row = box.row(align=True) + row.prop(Btrace, "bwColor1", text="") + row.prop(Btrace, "bwColor2", text="") + # Show Earth Colors + elif Btrace.mmColors == 'BRIGHT': + row = box.row(align=True) + for i in range(1, 5): + row.prop(Btrace, "brightColor" + str(i), text="") + # Show Earth Colors + elif Btrace.mmColors == 'EARTH': + row = box.row(align=True) + for i in range(1, 6): + row.prop(Btrace, "earthColor" + str(i), text="") + # Show Earth Colors + elif Btrace.mmColors == 'GREENBLUE': + row = box.row(align=True) + for i in range(1, 4): + row.prop(Btrace, "greenblueColor" + str(i), text="") + elif Btrace.mmColors == 'RANDOM': + row = box.row() + + # Curve noise settings + def curve_noise(): + row = box.row() + row.label(text="F-Curve Noise", icon='RNDCURVE') + row = box.row(align=True) + row.prop(Btrace, "fcnoise_rot", toggle=True) + row.prop(Btrace, "fcnoise_loc", toggle=True) + row.prop(Btrace, "fcnoise_scale", toggle=True) + + col = box.column(align=True) + col.prop(Btrace, "fcnoise_amp") + col.prop(Btrace, "fcnoise_timescale") + box.prop(Btrace, "fcnoise_key") + + # Curve Panel options + def curve_settings(): + # Button for curve options + row = self.layout.row() + row = box.row(align=True) + + row.prop(Btrace, "show_curve_settings", + icon='CURVE_BEZCURVE', text="Curve Settings") + row.prop(Btrace, "material_settings", + icon='MATERIAL_DATA', text="Material Settings") + + if Btrace.material_settings: + row = box.row() + row.label(text="Material Settings", icon='COLOR') + row = box.row() + row.prop(Btrace, "trace_mat_random") + if not Btrace.trace_mat_random: + row = box.row() + row.prop(Btrace, "trace_mat_color", text="") + else: + row.prop(Btrace, "mat_run_color_blender") + if Btrace.mat_run_color_blender: + row = box.row() + row.operator("object.colorblenderclear", + text="Reset Material Keyframes", + icon="KEY_DEHLT") + row.prop(Btrace, "mmSkip", text="Keyframe every") + color_blender() + row = box.row() + + if Btrace.show_curve_settings: + # selected curve options + if len(context.selected_objects) > 0 and obj.type == 'CURVE': + col = box.column(align=True) + col.label(text="Edit Curves for:", icon='IPO_BEZIER') + col.separator() + col.label(text="Selected Curve Bevel Options") + row = col.row(align=True) + row.prop(obj.data, "bevel_depth", text="Depth") + row.prop(obj.data, "bevel_resolution", text="Resolution") + row = col.row(align=True) + row.prop(obj.data, "resolution_u") + else: # For new curve + box.label(text="New Curve Settings", icon='CURVE_BEZCURVE') + box.prop(Btrace, "curve_spline") + box.prop(Btrace, "curve_handle") + box.label(text="Bevel Options") + col = box.column(align=True) + row = col.row(align=True) + row.prop(Btrace, "curve_depth", text="Depth") + row.prop(Btrace, "curve_resolution", text="Resolution") + row = col.row(align=True) + row.prop(Btrace, "curve_u") + + # Grow Animation Panel options + def add_grow(): + # Button for grow animation option + row = box.row() + row.label(text="Animate Final Curve", icon="SPACE2") + row = box.row() + row.prop(Btrace, "animate", text="Add Grow Curve Animation", icon="META_BALL") + box.separator() + if Btrace.animate: + box.label(text="Frame Animation Settings:", icon="META_BALL") + col = box.column(align=True) + col.prop(Btrace, "anim_auto") + if not Btrace.anim_auto: + row = col.row(align=True) + row.prop(Btrace, "anim_f_start") + row.prop(Btrace, "anim_length") + row = col.row(align=True) + row.prop(Btrace, "anim_delay") + row.prop(Btrace, "anim_f_fade") + + box.label(text="Additional Settings") + row = box.row() + row.prop(Btrace, "anim_tails") + row.prop(Btrace, "anim_keepr") + + # Start Btrace Panel + if switch_expand == 'list': + layout.label(text="Available Tools:", icon="COLLAPSEMENU") + col = layout.column(align=True) + col.prop(Btrace, "btrace_toolmenu", text="") + elif switch_expand == 'col': + col = layout.column(align=True) + col.prop(Btrace, "btrace_toolmenu", expand=True) + elif switch_expand == 'row': + row = layout.row(align=True) + row.alignment = 'CENTER' + row.prop(Btrace, "btrace_toolmenu", text="", expand=True) + + # Start Object Tools + sel = context.selected_objects + + # Default option (can be expanded into help) + if Btrace.btrace_toolmenu == 'tool_help': + row = layout.row() + row.label("Pick an option", icon="HELP") + + # Object Trace + elif Btrace.btrace_toolmenu == 'tool_objectTrace': + row = layout.row() + row.label(text=" Trace Tool:", icon="FORCE_CURVE") + box = self.layout.box() + row = box.row() + row.label(text="Object Trace", icon="FORCE_MAGNETIC") + row.operator("object.btobjecttrace", text="Run!", icon="PLAY") + row = box.row() + row.prop(Btrace, "settings_toggle", icon="MODIFIER", text="Settings") + myselected = "Selected %d" % len(context.selected_objects) + row.label(text=myselected) + if Btrace.settings_toggle: + box.label(text="Edge Type for Curves:", icon="IPO_CONSTANT") + row = box.row(align=True) + row.prop(Btrace, "convert_edgetype", text="") + box.prop(Btrace, "object_duplicate") + if len(sel) > 1: + box.prop(Btrace, "convert_joinbefore") + else: + Btrace.convert_joinbefore = False + row = box.row() + row.prop(Btrace, "distort_curve") + if Btrace.distort_curve: + col = box.column(align=True) + col.prop(Btrace, "distort_modscale") + col.prop(Btrace, "distort_noise") + row = box.row() + curve_settings() # Show Curve/material settings + add_grow() # Grow settings here + + # Objects Connect + elif Btrace.btrace_toolmenu == 'tool_objectsConnect': + row = layout.row() + row.label(text=" Trace Tool:", icon="FORCE_CURVE") + box = self.layout.box() + row = box.row() + row.label(text="Objects Connect", icon="OUTLINER_OB_EMPTY") + row.operator("object.btobjectsconnect", text="Run!", icon="PLAY") + row = box.row() + row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') + row.label(text="") + if Btrace.settings_toggle: + row = box.row() + row.prop(Btrace, "respect_order", text="Selection Options") + if Btrace.respect_order: + box.operator("object.select_order", + text="Click to start order selection", + icon='UV_SYNC_SELECT') + row = box.row() + row.prop(Btrace, "connect_noise", text="Add F-Curve Noise") + if Btrace.connect_noise: + curve_noise() # Show Curve Noise settings + + curve_settings() # Show Curve/material settings + add_grow() # Grow settings here + + # Mesh Follow + elif Btrace.btrace_toolmenu == 'tool_meshFollow': + row = layout.row() + row.label(text=" Trace Tool:", icon="FORCE_CURVE") + box = self.layout.box() + row = box.row() + row.label(text="Mesh Follow", icon="DRIVER") + row.operator("object.btmeshfollow", text="Run!", icon="PLAY") + row = box.row() + if Btrace.fol_mesh_type == 'OBJECT': + a, b = "Trace Object", "SNAP_VOLUME" + if Btrace.fol_mesh_type == 'VERTS': + a, b = "Trace Verts", "SNAP_VERTEX" + if Btrace.fol_mesh_type == 'EDGES': + a, b = "Trace Edges", "SNAP_EDGE" + if Btrace.fol_mesh_type == 'FACES': + a, b = "Trace Faces", "SNAP_FACE" + row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') + row.label(text=a, icon=b) + if Btrace.settings_toggle: + col = box.column(align=True) + row = col.row(align=True) + row.prop(Btrace, "fol_mesh_type", expand=True) + row = col.row(align=True) + if Btrace.fol_mesh_type != 'OBJECT': + row.prop(Btrace, "fol_sel_option", expand=True) + row = box.row() + if Btrace.fol_sel_option == 'RANDOM': + row.label("Random Select of Total") + row.prop(Btrace, "fol_perc_verts", text="%") + if Btrace.fol_sel_option == 'CUSTOM': + row.label("Choose selection in Edit Mode") + if Btrace.fol_sel_option == 'ALL': + row.label("Select All items") + col = box.column(align=True) + col.label("Time Options", icon="TIME") + col.prop(Btrace, "particle_step") + row = col.row(align=True) + row.prop(Btrace, "fol_start_frame") + row.prop(Btrace, "fol_end_frame") + curve_settings() # Show Curve/material settings + add_grow() # Grow settings here + + # Handwriting Tools + elif Btrace.btrace_toolmenu == 'tool_handwrite': + row = layout.row() + row.label(text=" Trace Tool:", icon="FORCE_CURVE") + box = self.layout.box() + row = box.row() + row.label(text='Handwriting', icon='BRUSH_DATA') + row.operator("curve.btwriting", text="Run!", icon='PLAY') + row = box.row() + row = box.row() + row.label(text='Grease Pencil Writing Tools') + col = box.column(align=True) + row = col.row(align=True) + row.operator("gpencil.draw", text="Draw", icon='BRUSH_DATA').mode = 'DRAW' + row.operator("gpencil.draw", text="Poly", icon='VPAINT_HLT').mode = 'DRAW_POLY' + row = col.row(align=True) + row.operator("gpencil.draw", text="Line", icon='ZOOMOUT').mode = 'DRAW_STRAIGHT' + row.operator("gpencil.draw", text="Erase", icon='TPAINT_HLT').mode = 'ERASER' + row = box.row() + row.operator("gpencil.data_unlink", text="Delete Grease Pencil Layer", icon="CANCEL") + row = box.row() + curve_settings() # Show Curve/material settings + add_grow() # Grow settings here + + # Particle Trace + elif Btrace.btrace_toolmenu == 'tool_particleTrace': + row = layout.row() + row.label(text=" Trace Tool:", icon="FORCE_CURVE") + box = self.layout.box() + row = box.row() + row.label(text="Particle Trace", icon="PARTICLES") + row.operator("particles.particletrace", text="Run!", icon="PLAY") + row = box.row() + row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') + row.label(text="") + if Btrace.settings_toggle: + box.prop(Btrace, "particle_step") + row = box.row() + row.prop(Btrace, "curve_join") + curve_settings() # Show Curve/material settings + add_grow() # Grow settings here + + # Connect Particles + elif Btrace.btrace_toolmenu == 'tool_particleConnect': + row = layout.row() + row.label(text=" Trace Tool:", icon="FORCE_CURVE") + box = self.layout.box() + row = box.row() + row.label(text='Particle Connect', icon='MOD_PARTICLES') + row.operator("particles.connect", icon="PLAY", text='Run!') + row = box.row() + row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') + row.label(text="") + if Btrace.settings_toggle: + box.prop(Btrace, "particle_step") + row = box.row() + row.prop(Btrace, 'particle_auto') + if not Btrace.particle_auto: + row = box.row(align=True) + row.prop(Btrace, 'particle_f_start') + row.prop(Btrace, 'particle_f_end') + curve_settings() # Show Curve/material settings + add_grow() # Grow settings here + + # Grow Animation + elif Btrace.btrace_toolmenu == 'tool_growCurve': + row = layout.row() + row.label(text=" Curve Tool:", icon="OUTLINER_OB_CURVE") + box = self.layout.box() + row = box.row() + row.label(text="Grow Curve", icon="META_BALL") + row.operator("curve.btgrow", text="Run!", icon="PLAY") + row = box.row() + row.prop(Btrace, "settings_toggle", icon="MODIFIER", text="Settings") + row.operator("object.btreset", icon="KEY_DEHLT") + if Btrace.settings_toggle: + box.label(text="Frame Animation Settings:") + col = box.column(align=True) + col.prop(Btrace, "anim_auto") + if not Btrace.anim_auto: + row = col.row(align=True) + row.prop(Btrace, "anim_f_start") + row.prop(Btrace, "anim_length") + row = col.row(align=True) + row.prop(Btrace, "anim_delay") + row.prop(Btrace, "anim_f_fade") + + box.label(text="Additional Settings") + row = box.row() + row.prop(Btrace, "anim_tails") + row.prop(Btrace, "anim_keepr") + + # F-Curve Noise Curve + elif Btrace.btrace_toolmenu == 'tool_fcurve': + row = layout.row() + row.label(text=" Curve Tool:", icon="OUTLINER_OB_CURVE") + box = self.layout.box() + row = box.row() + row.label(text="F-Curve Noise", icon='RNDCURVE') + row.operator("object.btfcnoise", icon='PLAY', text="Run!") + row = box.row() + row.prop(Btrace, "settings_toggle", icon='MODIFIER', text='Settings') + row.operator("object.btreset", icon='KEY_DEHLT') + if Btrace.settings_toggle: + curve_noise() + + # Color Blender + elif Btrace.btrace_toolmenu == 'tool_colorblender': + row = layout.row() + row.label(text=" Curve/Object Tool:", icon="OUTLINER_OB_CURVE") + box = self.layout.box() + row = box.row() + row.label(text="Color Blender", icon="COLOR") + row.operator("object.colorblender", icon='PLAY', text="Run!") + row = box.row() + row.operator("object.colorblenderclear", text="Reset Keyframes", icon="KEY_DEHLT") + row.prop(Btrace, "mmSkip", text="Keyframe every") + color_blender() |