main_ot.py

import bpy
from . properties import NTZEDGCRV_ignitproperties
from . import misc_functions
from . import misc_layout

from bpy.props import (StringProperty, BoolProperty, IntProperty, FloatProperty, FloatVectorProperty, EnumProperty, PointerProperty)
from bpy.types import (Panel, Operator, AddonPreferences, PropertyGroup)

# -----------------------------------------------------------------------------
#    Main Addon Operator
# -----------------------------------------------------------------------------    

class NTZEDGCRV_OT_insertedges(bpy.types.Operator):
    """Tooltip"""
    bl_idname = "ntzedgcrv.insertedges"
    bl_label = "Neltulz - Edge Curve"
    bl_description = "Allows you to quick insert edges with flow"
    bl_options = {'REGISTER', 'UNDO', 'PRESET'}

    bForcePanelOptions : BoolProperty(
        name="Force Panel Options",
        description='Gets all of the operator properties from the sidebar panel options.  This will override any settings in the user customized keymap.',
        default = False
    )

    numSegments : IntProperty(
        name="Segment Number",
        description="Number of segments to add. (Default: 1)",
        default = 1,
        min = 1,
        soft_max = 16
    )

    useEdgeFlow : BoolProperty(
        name="Use Edge Flow",
        description='Allows you to use the EdgeFlow options (Tension, Num of iterations, Min Angle).  If this is disabled, then edges will be added without edge flow adjustments. (Default: True)',
        default = True
    )

    showEdgeFlowOptions : BoolProperty(
        name="Show Edge Flow Options",
        default = True
    )

    tension : IntProperty(
        name="Tension",
        description="Tension (Default: 180)",
        default = 180,
        max = 500,
        min = -500
    )

    numIterations : IntProperty(
        name="Number of Iterations",
        description="Number of iterations. (Default: 1)",
        default = 1,
        min = 1,
        soft_max = 128
    )


    minAngle : IntProperty(
        name="Minimum Angle",
        description="Minimum Angle (Default: 0)",
        default = 0,
        max = 180,
        min = 0
    )

    @classmethod
    def poll(cls, context):
        return bpy.context.mode == "EDIT_MESH"


    def draw(self, context):
        scn = context.scene
        layout = self.layout.column(align=True)

        layout.separator()

        layout.prop(self, "numSegments", slider=True)

        layout.separator()

        edgeFlowOptionsEnabled = False #declare
        if self.useEdgeFlow:
            edgeFlowOptionsEnabled = True

        misc_layout.createShowHide(self, context, None, None, "showEdgeFlowOptions", "useEdgeFlow", "Use Edge Flow", layout)

        if self.showEdgeFlowOptions:

            edgeFlowSection = layout.column(align=True)

            edgeFlowSection.separator()

            edgeFlowOptionsCol = edgeFlowSection.column(align=True)
            edgeFlowOptionsCol.enabled = edgeFlowOptionsEnabled

            edgeFlowOptionsCol.prop(self, "tension", slider=edgeFlowOptionsEnabled)
            edgeFlowOptionsCol.prop(self, "numIterations", slider=edgeFlowOptionsEnabled)
            edgeFlowOptionsCol.prop(self, "minAngle", slider=edgeFlowOptionsEnabled)

    #END draw()

    def execute(self, context):
        scn = context.scene

        selObjs = bpy.context.selected_objects
        activeObjAtBegin = bpy.context.view_layer.objects.active

        try:
            #try to determine objectMode
            modeAtBegin = bpy.context.object.mode
        except:
            modeAtBegin = "OBJECT"

        if activeObjAtBegin is not None:
            if not activeObjAtBegin in selObjs:
                if modeAtBegin == "EDIT":
                    activeObjAtBegin.select_set(True)
                    selObjs.append(activeObjAtBegin)

        

        for obj in selObjs:
            
            #if more than 1 object is selected, then we need to iterate them one at a time.
            if len(selObjs) > 1:
                bpy.ops.object.mode_set(mode='OBJECT') #switch back to object mode
                bpy.ops.object.select_all(action='DESELECT') #deselect all objects
                
                bpy.context.view_layer.objects.active = obj
                obj.select_set(True)

            selectedVerts = misc_functions.getSelectedVerts(self, context, obj)
            if len(selectedVerts) > 0:
                
                selectedEdges = misc_functions.getSelectedEdges(self, context, obj)
                selectedEdgeIndices = list()

                for edge in selectedEdges:
                    selectedEdgeIndices.append(edge.index)


                

                totalNewEdgeLoopIndices = list()


                for edgeIndex in set(selectedEdgeIndices):

                    bpy.ops.mesh.loopcut(number_cuts=self.numSegments, object_index=0, edge_index=edgeIndex, mesh_select_mode_init=(False, True, False))

                    newEdges = misc_functions.getSelectedEdges(self, context, obj)
                    newEdgeIndices = list()
                    for edge in newEdges:
                        newEdgeIndices.append(edge.index)

                    totalNewEdgeLoopIndices = list( set(totalNewEdgeLoopIndices).union(newEdgeIndices) )

                bpy.ops.mesh.select_all(action='DESELECT')
                
                totalEdges = misc_functions.getAllEdges(self, context, obj)
                
                #switch to object mode
                bpy.ops.object.mode_set(mode='OBJECT')

                for i in range( len(totalNewEdgeLoopIndices) ):
                    obj.data.edges[ totalNewEdgeLoopIndices[i] ].select = True

                #switch to edit mode
                bpy.ops.object.mode_set(mode='EDIT')
                
                
                #detect if EdgeFlow addon is installed and enabled
                if self.useEdgeFlow:
                    try:
                        bpy.ops.mesh.set_edge_flow(tension=self.tension, iterations=self.numIterations, min_angle=self.minAngle)
                    except:
                        self.report({'ERROR'}, 'The "EdgeFlow" add-on could not be found.  Please check to see if the Edge Flow addon is installed and enabled.' )


        #reselect all originally selected objects, and then set the active object as the original active object
        if len(selObjs) > 1:
            bpy.ops.object.mode_set(mode='OBJECT') #switch back to object mode
            bpy.ops.object.select_all(action='DESELECT') #deselect all objects

            for obj in selObjs:
                obj.select_set(True)

            bpy.context.view_layer.objects.active = activeObjAtBegin

            #switch to edit mode
            bpy.ops.object.mode_set(mode='EDIT')

        #final step - Reset bForcePanelOptions so that it doesn't interfere when it is not explicitly set via keymap
        self.bForcePanelOptions = False

        return {'FINISHED'}
    # END execute()

    def invoke(self, context, event):
        scn = context.scene
        
        if self.bForcePanelOptions:

            if scn.ntzedgcrv.customEdgeCurveSettings == "USE":
            
                self.numSegments    = scn.ntzedgcrv.numSegmentsSlider
                self.useEdgeFlow    = scn.ntzedgcrv.useEdgeFlowCheckbox
                self.tension        = scn.ntzedgcrv.tensionSlider
                self.numIterations  = scn.ntzedgcrv.numIterationsSlider
                self.minAngle       = scn.ntzedgcrv.minAngleSlider

        return self.execute(context)
    #END invoke()

# END Operator()