Refactor qto utils and migrate to bmesh volume calculation. See #1222.

src/ifcblenderexport/blenderbim/bim/module/qto/__init__.py

@@ -0,0 +1,18 @@
+import bpy
+from . import ui, prop, operator
+
+classes = (
+    operator.CalculateEdgeLengths,
+    operator.CalculateFaceAreas,
+    operator.CalculateObjectVolumes,
+    prop.BIMQtoProperties,
+    ui.BIM_PT_qto_utilities,
+)
+
+
+def register():
+    bpy.types.Scene.BIMQtoProperties = bpy.props.PointerProperty(type=prop.BIMQtoProperties)
+
+
+def unregister():
+    del bpy.types.Scene.BIMQtoProperties

src/ifcblenderexport/blenderbim/bim/module/qto/operator.py

@@ -0,0 +1,51 @@
+import bpy
+import bmesh
+
+
+class CalculateEdgeLengths(bpy.types.Operator):
+    bl_idname = "bim.calculate_edge_lengths"
+    bl_label = "Calculate Edge Lengths"
+
+    def execute(self, context):
+        result = 0
+        for obj in bpy.context.selected_objects:
+            if not obj.data or not obj.data.edges:
+                continue
+            for edge in obj.data.edges:
+                if edge.select:
+                    result += (obj.data.vertices[edge.vertices[1]].co - obj.data.vertices[edge.vertices[0]].co).length
+        bpy.context.scene.BIMQtoProperties.qto_result = str(round(result, 3))
+        return {"FINISHED"}
+
+
+class CalculateFaceAreas(bpy.types.Operator):
+    bl_idname = "bim.calculate_face_areas"
+    bl_label = "Calculate Face Areas"
+
+    def execute(self, context):
+        result = 0
+        for obj in bpy.context.selected_objects:
+            if not obj.data or not obj.data.polygons:
+                continue
+            for polygon in obj.data.polygons:
+                if polygon.select:
+                    result += polygon.area
+        bpy.context.scene.BIMQtoProperties.qto_result = str(round(result, 3))
+        return {"FINISHED"}
+
+
+class CalculateObjectVolumes(bpy.types.Operator):
+    bl_idname = "bim.calculate_object_volumes"
+    bl_label = "Calculate Object Volumes"
+
+    def execute(self, context):
+        result = 0
+        for obj in bpy.context.selected_objects:
+            if not obj.data or not isinstance(obj.data, bpy.types.Mesh):
+                continue
+            bm = bmesh.new()
+            bm.from_mesh(obj.data)
+            result += bm.calc_volume()
+            bm.free()
+        bpy.context.scene.BIMQtoProperties.qto_result = str(round(result, 3))
+        return {"FINISHED"}

src/ifcblenderexport/blenderbim/bim/module/qto/prop.py

@@ -0,0 +1,17 @@
+import bpy
+from blenderbim.bim.prop import StrProperty, Attribute
+from blenderbim.bim.module.owner.data import Data
+from bpy.types import PropertyGroup
+from bpy.props import (
+    PointerProperty,
+    StringProperty,
+    EnumProperty,
+    BoolProperty,
+    IntProperty,
+    FloatProperty,
+    FloatVectorProperty,
+    CollectionProperty,
+)
+
+class BIMQtoProperties(PropertyGroup):
+    qto_result: StringProperty(default="", name="Qto Result")

src/ifcblenderexport/blenderbim/bim/module/qto/ui.py

@@ -0,0 +1,23 @@
+from bpy.types import Panel
+
+
+class BIM_PT_qto_utilities(Panel):
+    bl_idname = "BIM_PT_qto_utilities"
+    bl_label = "Quantity Take-off"
+    bl_space_type = "VIEW_3D"
+    bl_region_type = "UI"
+    bl_category = "BlenderBIM"
+
+    def draw(self, context):
+        layout = self.layout
+        props = context.scene.BIMQtoProperties
+
+        row = layout.row()
+        row.prop(props, "qto_result", text="Results")
+
+        row = layout.row(align=True)
+        row.operator("bim.calculate_edge_lengths")
+        row = layout.row(align=True)
+        row.operator("bim.calculate_face_areas")
+        row = layout.row(align=True)
+        row.operator("bim.calculate_object_volumes")