From 00fbf20f0321c8d724ad74e3520295e689e244c1 Mon Sep 17 00:00:00 2001
From: L-Nafaryus <l.nafaryus@gmail.com>
Date: Fri, 9 Apr 2021 22:28:15 +0500
Subject: [PATCH] Destruction [3]

---
 samples/test.py | 121 +++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 120 insertions(+), 1 deletion(-)

diff --git a/samples/test.py b/samples/test.py
index 852cccb..2cfef0f 100644
--- a/samples/test.py
+++ b/samples/test.py
@@ -37,7 +37,6 @@ def simpleCubic(theta = 0.01, fillet = False):
     ###
     #   Bounding box
     ##
-    # YZ 
     sk = geompy.Sketcher3D()
     sk.addPointsAbsolute(xl, 0, 0)
     sk.addPointsAbsolute(0, yw, 0)
@@ -123,3 +122,123 @@ def simpleCubic(theta = 0.01, fillet = False):
 
     return shape
     
+
+def simpleHexagonalPrism(theta = 0.01, fillet = False):
+    
+    ###
+    #   Parameters
+    ##
+    r0 = 1.0
+    L = 2 * r0
+
+    radius = r0 / (1 - theta)
+    xn, yn, zn = 3, 3, 3 
+    
+    length = L * sqrt(2)
+    width = L * sqrt(2)
+    height = L
+
+    point = []
+    xl, yw, zh = -L - L / 6, -L - L / 6, -L / 6
+    point.append((L + xl, L + yw, L + zh))
+    point.append((5 * L / 3 + xl, 2 * L / 3 + yw, 2 * L / 3 + zh))
+    point.append((2 * L + xl, L + yw, 0 + zh))
+    point.append((5 * L / 3 + xl, 5 * L / 3 + yw, -L / 3 + zh))
+    point.append((L + xl, 2 * L + yw, 0 + zh))
+    point.append((2 * L / 3 + xl, 5 * L / 3 + yw, 2 * L / 3 + zh))
+    point.append((L + xl, L + yw, L + zh))
+
+    C1, C2 = 0.8, 0.05
+    theta1, theta2 = 0.01, 0.28
+    Cf = C1 + (C2 - C1) / (theta2 - theta1) * (theta - theta1)
+    filletradius = Cf * (radius - r0)
+    
+    scale = 100
+    oo = geompy.MakeVertex(0, 0, 0)
+
+    ###
+    #   Bounding box
+    ## 
+    sk = geompy.Sketcher3D()
+
+    for p in point:
+        sk.addPointsAbsolute(*p)
+    
+    inletface = geompy.MakeFaceWires([sk.wire()], False)
+    vecflow = geompy.GetNormal(inletface)
+    hexagonPrism = geompy.MakePrismVecH(inletface, vecflow, L * sqrt(3))
+
+    inletface = geompy.MakeScaleTransform(inletface, oo, scale)
+    hexagonPrism = geompy.MakeScaleTransform(hexagonPrism, oo, scale)
+
+    faces = geompy.ExtractShapes(hexagonPrism, geompy.ShapeType["FACE"], False)
+    symetryface = []
+
+    for face in faces:
+        norm = geompy.GetNormal(face)
+        angle = round(geompy.GetAngle(norm, vecflow), 0)
+
+        if (angle == 0 or angle == 180) and not face == inletface:
+            outletface = face
+        
+        else:
+            symetryface.append(face)
+    
+    ###
+    #   Grains
+    ##
+    ox = geompy.MakeVectorDXDYDZ(1, 0, 0)
+    oy = geompy.MakeVectorDXDYDZ(0, 1, 0)
+    oz = geompy.MakeVectorDXDYDZ(0, 0, 1)
+
+    grain = geompy.MakeSphereR(radius)
+    lattice = geompy.MakeMultiTranslation2D(grain, ox, L, xn, oy, L, yn)
+    lattice = geompy.MakeMultiTranslation1D(lattice, oz, L, zn)
+    
+    grains = geompy.ExtractShapes(lattice, geompy.ShapeType["SOLID"], True)
+    grains = geompy.MakeFuseList(grains, False, False)
+
+    grains = geompy.MakeScaleTransform(grains, oo, scale)
+
+    if fillet:
+        grains = geompy.MakeFilletAll(grains, filletradius * scale)
+
+    ###
+    #   Groups
+    ##
+    shape = geompy.MakeCutList(hexagonPrism, [grains])
+    shape = geompy.MakeScaleTransform(shape, oo, 1 / scale, theName = "simpleCubic")
+
+    sall = geompy.CreateGroup(shape, geompy.ShapeType["FACE"])
+    geompy.UnionIDs(sall,
+        geompy.SubShapeAllIDs(shape, geompy.ShapeType["FACE"]))
+
+    inlet = geompy.CreateGroup(shape, geompy.ShapeType["FACE"], theName = "inlet")
+    inletshape = geompy.MakeCutList(inletface, [grains])
+    inletshape = geompy.MakeScaleTransform(inletshape, oo, 1 / scale)
+    geompy.UnionList(inlet, geompy.SubShapeAll(
+        geompy.GetInPlace(shape, inletshape, True), geompy.ShapeType["FACE"]))
+
+    outlet = geompy.CreateGroup(shape, geompy.ShapeType["FACE"], theName = "outlet")
+    outletshape = geompy.MakeCutList(outletface, [grains])
+    outletshape = geompy.MakeScaleTransform(outletshape, oo, 1 / scale)
+    geompy.UnionList(outlet, geompy.SubShapeAll(
+        geompy.GetInPlace(shape, outletshape, True), geompy.ShapeType["FACE"]))
+    
+    symetry = []
+    for (n, face) in enumerate(symetryface):
+        name = "symetry" + str(n)
+        symetry.append(geompy.CreateGroup(shape, geompy.ShapeType["FACE"], theName = name))
+        symetryshape = geompy.MakeCutList(face, [grains])
+        symetryshape = geompy.MakeScaleTransform(symetryshape, oo, 1 / scale)
+        geompy.UnionList(symetry[n], geompy.SubShapeAll(
+            geompy.GetInPlace(shape, symetryshape, True), geompy.ShapeType["FACE"]))
+
+    groups = []
+    groups.append(inlet)
+    groups.append(outlet)
+    groups.extend(symetry)
+    wall = geompy.CutListOfGroups([sall], groups, theName = "wall")
+
+    return shape
+