op_set_edge_linear.py

import bpy
from bpy.props import BoolProperty, FloatProperty, EnumProperty
import bmesh

from . import util
from . import op_set_edge_flow


class SetEdgeLinearOP(bpy.types.Operator, op_set_edge_flow.SetEdgeLoopBase):
    bl_idname = "mesh.set_edge_linear"
    bl_label = "Set edge linear"
    bl_options = {'REGISTER', 'UNDO'}
    bl_description = "makes edge loops linear"

    space_evenly : BoolProperty(name="Space evenly", default=False)
    distance : FloatProperty(name="Distance", default=1.0, min=0)

    def draw(self, context):
        layout = self.layout
        column = layout.column()
        if self.do_straighten:
            column.prop(self, "distance")
        else:
            column.prop(self, "space_evenly")

    def invoke(self, context, event):
        #print("--------------------")
        super(SetEdgeLinearOP, self).invoke(context)

        self.do_straighten = self.can_straighten()
        if self.do_straighten:
            distance = 0
            edge_count = 0
            for obj in self.objects:
                for edgeloop in self.edgeloops[obj]:
                    distance += edgeloop.get_average_distance()

                edge_count += len(self.edgeloops[obj])

            distance /= edge_count

            self.distance = distance * 0.35

        return self.execute(context)

    def can_straighten(self):
        for obj in self.objects:
            for edgeloop in self.edgeloops[obj]:
                if len(edgeloop.edges) != 1:
                    return False

        return True

    def execute(self, context):
        #print("execute")
        
        if not hasattr(self, "objects") or not self.objects:
            return self.invoke(context, None)

        bpy.ops.object.mode_set(mode='OBJECT')

        self.revert()

        for obj in self.objects:
            for edgeloop in self.edgeloops[obj]:
                if self.do_straighten:
                    edgeloop.straighten(self.distance)
                else:
                    edgeloop.set_linear(self.space_evenly)

            self.bm[obj].to_mesh(obj.data)

        bpy.ops.object.mode_set(mode='EDIT')
        return {'FINISHED'}