refactor from cube to box and mesh

Author: monkstone

3DSketches/menger2.py

@@ -1,12 +1,11 @@
 from pyprocessing import *
-from util.cube import Cube
+from util.box import Box
 
 FOV = PI/3.0
 angle = 0.0
 ANGLE_STEP = PI / 180.0
 menger = []
-
-             
+ 
 
 def setup():
     size(800,600)
@@ -23,9 +22,9 @@ def draw():
     angle = (angle + ANGLE_STEP) % TWO_PI
     rotateZ(angle)
     rotateY(angle)
-    export_menger()
-    #for cub in menger:
-    #    draw_cube(cub)
+    #export_menger()
+    for cub in menger:
+        draw_cube(cub)
 
 
 def draw_cube(cube):
@@ -54,8 +53,8 @@ def create_menger(xx, yy, zz, sz):
     Create a recursive menger sponge using my_cube
     """	
     u = sz / 3.0
-    if (sz < 10):
-        menger.append(Cube(xx, yy, zz, sz))
+    if (sz < 20):
+        menger.append(Box.createAcube(xx, yy, zz, sz).toMesh())
     else:
         for i in xrange(-1, 2, 1):
             for j in xrange(-1, 2, 1):

3DSketches/util/box.py

@@ -0,0 +1,58 @@
+"""
+box.py contains use lighweight Vec3D class to store indexed vector points 
+by having public attributes (as does python default)
+"""
+from vec3d import Vec3D
+from mesh import Mesh
+
+class Box(object):
+    """
+    A lightweight convenience class to store box point data
+    With a function to return indexed vector points 
+    """
+    
+    def __init__(self, x = 0, y = 0,  z = 0,  xd = 1,  yd = 1,  zd = 1):
+        """
+        Initialise the new Box object
+        """
+        self.xx= x
+        self.yy = y 
+        self.zz = z 
+        self.xd = xd
+        self.yd = yd
+        self.zd = zd
+        self.idx_array  = [1,  0,  2,  3,  1,  2,  5,  4,  6,  7,  5,  6,  4,  5,  0,  1,  4,  0,  7,  6,  2,  2,  6,  3,  0,  5,  7,  2,  0,  7,  4,  1, 6,  6,  1,  3]
+        self.normal_array = [0, 0, 1, 0, 0, -1, 1, 0, 0, -1, 0, 0, 0, 1, 0, 0, -1, 0]
+    def points(self):
+        """
+        The unique corner points of box (front + back order)
+        """
+        cpoints = [] 
+        cpoints.append( Vec3D(-self.xd * 0.5 + self.xx, -self.yd * 0.5 + self.yy, -self.zd * 0.5 + self.zz) )  
+        cpoints.append( Vec3D(+self.xd * 0.5 + self.xx, -self.yd * 0.5 + self.yy, -self.zd * 0.5 + self.zz) )
+        cpoints.append( Vec3D(-self.xd * 0.5 + self.xx, +self.yd * 0.5 + self.yy, -self.zd * 0.5 + self.zz) )
+        cpoints.append( Vec3D(+self.xd * 0.5 + self.xx, +self.yd * 0.5 + self.yy, -self.zd * 0.5 + self.zz) )
+        cpoints.append( Vec3D(+self.xd * 0.5 + self.xx, -self.yd * 0.5 + self.yy, +self.zd * 0.5 + self.zz) )
+        cpoints.append( Vec3D(-self.xd * 0.5 + self.xx, -self.yd * 0.5 + self.yy, +self.zd * 0.5 + self.zz) )
+        cpoints.append( Vec3D(+self.xd * 0.5 + self.xx, +self.yd * 0.5 + self.yy, +self.zd * 0.5 + self.zz) )
+        cpoints.append( Vec3D(-self.xd * 0.5 + self.xx, +self.yd * 0.5 + self.yy, +self.zd * 0.5 + self.zz) )
+        return cpoints   
+        
+        
+    def toMesh(self):
+        return Mesh.meshFromVertices(self.points(), self.idx_array)
+ 
+                    
+    @classmethod         
+    def boxFromMaxMin(clas, bmin, bmax):
+        """
+        Alternative constructor creates an instance of from min and max Vec3D
+        """
+        return Box((bmin.x + bmax.x/2.0),  (bmin.y +bmax.y/2.0),  (bmin.z + bmax.z/2.0),  abs(bmin.x - bmax.x),  abs(bmin.y - bmax.y),  abs(bmin.z - bmax.z))
+        
+    @classmethod         
+    def createAcube(clas,  x,  y,  z,  sz):
+        """
+        Alternative constructor creates a special box, where side have equal length
+        """
+        return Box(x,  y,  z,  sz,  sz,  sz)    

3DSketches/util/mesh.py

@@ -0,0 +1,81 @@
+"""
+mesh.py contains use lighweight Vec3D class to store unique vector points
+and correspondindg mesh indices 
+"""
+
+class Mesh(object):
+    """
+    A lightweight convenience class to store mesh point data
+    With a function to return indexed vector points 
+    """
+      
+    def __init__(self, name = "mesh0"):
+        """
+        Initialise the new Mesh object
+        """
+        self.name = name  
+        self.vertex_index = []
+        self.vertices = []
+        
+    def setVertices(self,  vertices):
+       self.vertices = vertices
+       
+    def setVertexIndex(self,  idx):
+       self.vertex_index = idx   
+        
+    def mesh_array(self):
+        """
+        Returns array of Vec3D for use in pyprocessing
+        """
+        temp = []
+        for i in self.vertex_index:
+            temp.append(self.vertices[i])
+        return temp        
+        
+    def vert_string(self,  pt):
+        """
+        helper function that returns PovRAY vector string from a Vec3D
+        """
+        return "\t<{0}, {1}, {2}>,\n".format(pt.x,  pt.y,  pt.z)
+        
+    def vect_string(self,  x,  y,  z):
+        return "\t<{0}, {1}, {2}>,\n".format(x,  y,  z)    
+        
+    def index_string(self,  idx):
+        """
+        PovRAY formatted face indices as string
+        """
+        temp = []
+        temp.append("\tface_indices {\n")
+        temp.append("\t{0:d},\n".format(int(len(idx)/3)))   # need to coerce int?       
+        for i in xrange(0, len(idx),  3):
+            temp.append(self.vect_string(idx[i],  idx[i +1],  idx[i +2]))
+        temp.append("\t}\n")    
+        return "".join(temp)   
+             
+    def mesh2(self):
+        """
+        Returns a string of type mesh2 array for use in PovRAY
+        """
+        temp = []
+        temp.append("mesh2{\n")
+        temp.append("\tvertex_vectors {\n")
+        temp.append("\t{0},\n".format(len(self.vertices)))
+        for pt in self.points():
+            temp.append(self.vert_string(pt))
+        temp.append("\t}\n") 
+        temp.append(self.index_string(self.vertex_index))
+        temp.append("}\n") 
+        return "".join(temp)
+                    
+    @classmethod         
+    def meshFromVertices(clas, vertices, vertex_index):
+        """
+        Alternative constructor creates an instance of from vertex array and vertex indices
+        """
+        temp = Mesh()
+        temp.setVertexIndex(vertex_index)
+        temp.setVertices(vertices)
+        return temp
+        
+