Add UI for making Slab reinforcement

BentShapeRebar.py

@@ -388,7 +388,7 @@ def makeBentShapeRebar(
     rounding,
     amount_spacing_check,
     amount_spacing_value,
-    orientation="Bottom Left",
+    orientation="Bottom",
     structure=None,
     facename=None,
 ):

RebarData.py

@@ -29,7 +29,7 @@
 import enum
 
 
-class RebarTypes(enum.Enum):
+class RebarTypes(str, enum.Enum):
     """A primitive rebar types enumeration object."""
 
     straight = "StraightRebar"

RebarTools.py

@@ -281,6 +281,35 @@ def Activated():
         MainBeamReinforcement.CommandBeamReinforcement()
 
 
+class SlabReinforcementTool:
+    @staticmethod
+    def GetResources():
+        return {
+            "Pixmap": str(
+                Path(__file__).parent / "icons" / "SlabReinforcement.png"
+            ),
+            "MenuText": QT_TRANSLATE_NOOP(
+                "Reinforcement_SlabRebars", "Slab Reinforcement"
+            ),
+            "ToolTip": QT_TRANSLATE_NOOP(
+                "Reinforcement_SlabRebars",
+                "Creates a Slab Reinforcement from the selected face of the "
+                "Structural element.",
+            ),
+        }
+
+    @staticmethod
+    def IsActive():
+        return True if FreeCADGui.activeDocument() else False
+
+    @staticmethod
+    def Activated():
+        from SlabReinforcement import MainSlabReinforcement
+
+        # Call to CommandSlabReinforcement() function
+        MainSlabReinforcement.CommandSlabReinforcement()
+
+
 class BillOfMaterialTool:
     @staticmethod
     def GetResources():
@@ -417,6 +446,7 @@ def updateLocale():
 FreeCADGui.addCommand("Reinforcement_HelicalRebar", HelicalRebarTool())
 FreeCADGui.addCommand("Reinforcement_ColumnRebars", ColumnReinforcementTool())
 FreeCADGui.addCommand("Reinforcement_BeamRebars", BeamReinforcementTool())
+FreeCADGui.addCommand("Reinforcement_SlabRebars", SlabReinforcementTool())
 FreeCADGui.addCommand("Reinforcement_BillOfMaterial", BillOfMaterialTool())
 FreeCADGui.addCommand("Reinforcement_BarShapeCutList", RebarShapeCutListTool())
 FreeCADGui.addCommand(
@@ -437,6 +467,7 @@ def updateLocale():
     "Reinforcement_HelicalRebar",
     "Reinforcement_ColumnRebars",
     "Reinforcement_BeamRebars",
+    "Reinforcement_SlabRebars",
 ]
 
 # Initialize "Arch_Rebar" command
@@ -455,6 +486,7 @@ def updateLocale():
     "Reinforcement_HelicalRebar",
     "Reinforcement_ColumnRebars",
     "Reinforcement_BeamRebars",
+    "Reinforcement_SlabRebars",
     "Arch_Rebar",
     "Reinforcement_BillOfMaterial",
     "Reinforcement_BarShapeCutList",

Rebarfunc.py

@@ -38,7 +38,7 @@
 
 
 def getEdgesAngle(edge1, edge2):
-    """ getEdgesAngle(edge1, edge2): returns a angle between two edges."""
+    """getEdgesAngle(edge1, edge2): returns a angle between two edges."""
     vec1 = vec(edge1)
     vec2 = vec(edge2)
     angle = vec1.getAngle(vec2)
@@ -73,7 +73,7 @@ def getBaseStructuralObject(obj):
 
 
 def getBaseObject(obj):
-    """ getBaseObject(obj): This function will return last base object."""
+    """getBaseObject(obj): This function will return last base object."""
     if hasattr(obj, "Base"):
         return getBaseObject(obj.Base)
     else:
@@ -860,7 +860,7 @@ def translate(context, text, disambig=None):
 
 
 def print_in_freecad_console(*msg):
-    """ Print given arguments on FreeCAD console."""
+    """Print given arguments on FreeCAD console."""
     s = ""
     for m in msg:
         s += str(m) + ", "