Initial implementation of CompSolid

Author: adam-urbanczyk

cadquery/occ_impl/shapes.py

@@ -103,6 +103,7 @@
     TopoDS_Vertex,
     TopoDS_Solid,
     TopoDS_Shell,
+    TopoDS_CompSolid,
 )
 
 from OCP.GC import GC_MakeArcOfCircle, GC_MakeArcOfEllipse  # geometry construction
@@ -219,6 +220,7 @@
     ta.TopAbs_FACE: "Face",
     ta.TopAbs_SHELL: "Shell",
     ta.TopAbs_SOLID: "Solid",
+    ta.TopAbs_COMPSOLID: "CompSolid",
     ta.TopAbs_COMPOUND: "Compound",
 }
 
@@ -241,6 +243,7 @@
     ta.TopAbs_FACE: TopoDS.Face_s,
     ta.TopAbs_SHELL: TopoDS.Shell_s,
     ta.TopAbs_SOLID: TopoDS.Solid_s,
+    ta.TopAbs_COMPSOLID: TopoDS.CompSolid_s,
     ta.TopAbs_COMPOUND: TopoDS.Compound_s,
 }
 
@@ -251,6 +254,7 @@
     ta.TopAbs_FACE: BRepAdaptor_Surface,
     ta.TopAbs_SHELL: "Shell",
     ta.TopAbs_SOLID: "Solid",
+    ta.TopAbs_SOLID: "CompSolid",
     ta.TopAbs_COMPOUND: "Compound",
 }
 
@@ -280,7 +284,9 @@
     ga.GeomAbs_OtherCurve: "OTHER",
 }
 
-Shapes = Literal["Vertex", "Edge", "Wire", "Face", "Shell", "Solid", "Compound"]
+Shapes = Literal[
+    "Vertex", "Edge", "Wire", "Face", "Shell", "Solid", "CompSolid", "Compound"
+]
 Geoms = Literal[
     "Vertex",
     "Wire",
@@ -386,6 +392,7 @@ def cast(
             ta.TopAbs_FACE: Face,
             ta.TopAbs_SHELL: Shell,
             ta.TopAbs_SOLID: Solid,
+            ta.TopAbs_COMPSOLID: CompSolid,
             ta.TopAbs_COMPOUND: Compound,
         }
 
@@ -726,6 +733,13 @@ def Solids(self) -> List["Solid"]:
 
         return [Solid(i) for i in self._entities("Solid")]
 
+    def CompSolids(self) -> List["CompSolid"]:
+        """
+        :returns: All the compsolids in this Shape
+        """
+
+        return [CompSolid(i) for i in self._entities("CompSolid")]
+
     def Area(self) -> float:
         """
         :returns: The surface area of all faces in this Shape
@@ -2746,6 +2760,14 @@ def dprism(
         return Solid(shape)
 
 
+class CompSolid(Shape, Mixin3D):
+    """
+    a single compsolid
+    """
+
+    wrapped: TopoDS_CompSolid
+
+
 class Compound(Shape, Mixin3D):
     """
     a collection of disconnected solids

tests/test_cadquery.py

@@ -4255,3 +4255,17 @@ def testFindFace(self):
         self.assertTrue(isinstance(w2.findFace(searchParents=True), Face))
         with raises(ValueError):
             w2.findFace(searchParents=False)
+
+    def testCompSolid(self):
+
+        from OCP.BRepPrimAPI import BRepPrimAPI_MakePrism
+
+        tool = Solid.makeSphere(1, angleDegrees3=120)
+        shell = tool.Shells()[0]
+        v = Vector(0, 0, 1)
+
+        builder = BRepPrimAPI_MakePrism(shell.wrapped, v.wrapped)
+        result = Shape.cast(builder.Shape())
+
+        self.assertEqual(len(result.CompSolids()), 1)
+        self.assertEqual(len(result.Solids()), 4)