Addon: Mesh Extra Objects: Gears, Wallfactory, Beam, Pipe: Implemented add object in edit mode

1 files changed, 15 insertions, 4 deletions

diff --git a/add_mesh_extra_objects/add_mesh_beam_builder.py b/add_mesh_extra_objects/add_mesh_beam_builder.py
index 9aa89ef1..623fc0ca 100644
--- a/add_mesh_extra_objects/add_mesh_beam_builder.py
+++ b/add_mesh_extra_objects/add_mesh_beam_builder.py
@@ -790,9 +790,20 @@ class addBeam(Operator):
                 obj.data[prm] = getattr(self, prm)
 
             return {'FINISHED'}
-
-        self.report({'WARNING'}, "Option only valid in Object mode")
-        return {'CANCELLED'}
+            
+        if bpy.context.mode == "EDIT_MESH":
+            active_object = context.active_object
+            name_active_object = active_object.name
+            bpy.ops.object.mode_set(mode='OBJECT')
+            mesh = addBeamMesh(self, context)
+            obj = object_utils.object_data_add(context, mesh, operator=None)
+            obj.select_set(True)
+            active_object.select_set(True)
+            bpy.ops.object.join()
+            context.active_object.name = name_active_object
+            bpy.ops.object.mode_set(mode='EDIT')
+
+        return {'FINISHED'}
 
 def BeamParameters():
     BeamParameters = [
@@ -803,4 +814,4 @@ def BeamParameters():
             "edgeA",
             "Cursor",
             ]
-    return BeamParameters
\ No newline at end of file
+    return BeamParameters