Script to create Menger sponge

src/Mod/TemplatePyMod/MengerSponge.py

@@ -0,0 +1,120 @@
+# Script to create a Menger sponge
+# (c) 2012 Werner Mayer LGPL
+
+# The script is based on the work of daxmick at
+# https://sourceforge.net/apps/phpbb/free-cad/viewtopic.php?f=3&t=2307
+
+import threading
+import Mesh, MeshGui
+from FreeCAD import Base
+
+# Create a Box and Place it a coords (x,y,z)
+def PlaceBox(x,y,z):
+   box = Mesh.createBox(1,1,1)
+   box.translate(x,y,z)
+   return box
+
+def Sierpinski(level,x0,y0,z0):
+   #print threading.current_thread().name
+   boxnums = pow(3,level)
+   thirds = boxnums / 3
+   twothirds = thirds * 2
+   if(level == 0):
+      rangerx = [x0]
+      rangery = [y0]
+      rangerz = [z0]
+   else:
+      rangerx = [ x0, x0 + thirds, x0 + twothirds ]
+      rangery = [ y0, y0 + thirds, y0 + twothirds ]
+      rangerz = [ z0, z0 + thirds, z0 + twothirds ]
+   block = 1
+   skip=[5,11,13,14,15,17,23]
+   mesh=Mesh.Mesh()
+   for i in rangerx:
+      for j in rangery:
+         for k in rangerz:
+            if block not in skip:
+               if(level > 0):
+                  mesh.addMesh(Sierpinski(level-1,i,j,k))
+               else:
+                  mesh.addMesh(PlaceBox(i,j,k))
+            block+=1
+   return mesh
+
+### Multi-threaded ###
+
+class MengerThread(threading.Thread):
+    def __init__(self,args):
+        self.args=args
+        self.mesh=Mesh.Mesh()
+        threading.Thread.__init__(self)
+    def run(self):
+        for i in self.args:
+            self.mesh.addMesh(Sierpinski(*i))
+
+def makeMengerSponge_mt(level=3,x0=0,y0=0,z0=0):
+    """
+    Is much slower than makeMengerSponge!!! :(
+    """
+    if level == 0:
+        mesh=Sierpinski(level,x0,y0,z0)
+        Mesh.show(mesh)
+        return
+
+    boxnums = pow(3,level)
+    thirds = boxnums / 3
+    twothirds = thirds * 2
+    
+    rangerx = [ x0, x0 + thirds, x0 + twothirds ]
+    rangery = [ y0, y0 + thirds, y0 + twothirds ]
+    rangerz = [ z0, z0 + thirds, z0 + twothirds ]
+    block = 1
+    skip=[5,11,13,14,15,17,23]
+
+    # collect the arguments for the algorithm in a list
+    args=[]
+    for i in rangerx:
+        for j in rangery:
+            for k in rangerz:
+                if block not in skip:
+                    args.append((level-1,i,j,k))
+                block+=1
+
+    numJobs = 4
+    threads=[]
+    while numJobs > 0:
+        size = len(args)
+        count = size / numJobs
+        numJobs-=1
+        thr=MengerThread(args[:count])
+        threads.append(thr)
+        args=args[count:]
+
+    print "Number of threads: %i" % (len(threads))
+    for thr in threads:
+        thr.start()
+    for thr in threads:
+        thr.join()
+
+    mesh=Mesh.Mesh()
+    for thr in threads:
+        mesh.addMesh(thr.mesh)
+        del thr.mesh
+        
+    print mesh
+    mesh.removeDuplicatedPoints()
+    mesh.removeFacets(mesh.getInternalFacets())
+    mesh.rebuildNeighbourHood()
+    print "Mesh is solid: %s" % (mesh.isSolid())
+    Mesh.show(mesh)
+
+
+### Single-threaded ###
+
+def makeMengerSponge(level=3,x0=0,y0=0,z0=0):
+    mesh=Sierpinski(level,x0,y0,z0)
+    mesh.removeDuplicatedPoints()
+    mesh.removeFacets(mesh.getInternalFacets())
+    mesh.rebuildNeighbourHood()
+    print "Mesh is solid: %s" % (mesh.isSolid())
+    Mesh.show(mesh)