import direct.directbase.DirectStart
from pandac.PandaModules import *
from math import sqrt

def IcoSphere(radius, subdivisions):
  ico_path = NodePath('ico_path')
  ico_node = GeomNode('ico_node')
  ico_path.attachNewNode(ico_node)
  gvd = GeomVertexData('gvd', GeomVertexFormat.getV3(), Geom.UHStatic)
  geom = Geom(gvd)
  gvw = GeomVertexWriter(gvd, 'vertex')
  ico_node.addGeom(geom)
  prim = GeomTriangles(Geom.UHStatic)
  verts = []
  phi = .5*(1.+sqrt(5.))
  invnorm = 1/sqrt(phi*phi+1)
  verts.append(Vec3(-1,  phi, 0) * invnorm)
  verts.append(Vec3( 1,  phi, 0) * invnorm)
  verts.append(Vec3(0,   1,  -phi) * invnorm)
  verts.append(Vec3(0,   1,   phi) * invnorm)
  verts.append(Vec3(-phi,0,  -1) * invnorm)
  verts.append(Vec3(-phi,0,   1) * invnorm)
  verts.append(Vec3( phi,0,  -1) * invnorm)
  verts.append(Vec3( phi,0,   1) * invnorm)
  verts.append(Vec3(0,   -1, -phi) * invnorm)
  verts.append(Vec3(0,   -1,  phi) * invnorm)
  verts.append(Vec3(-1,  -phi,0) * invnorm)
  verts.append(Vec3( 1,  -phi,0) * invnorm)

  faces = [
   0,1,2,
   0,3,1,
   0,4,5,
   1,7,6,
   1,6,2,
   1,3,7,
   0,2,4,
   0,5,3,
   2,6,8,
   2,8,4,
   3,5,9,
   3,9,7,
   11,6,7,
   10,5, 4,
   10,4,8,
   10,9,5,
   11,8,6,
   11,7,9,
   10,8,11,
   10,11,9
  ]

  size = 60

  for subdivision in range(0,subdivisions):
   size*=4;
   newFaces = []
   for i in range(0,size/12):
    i1 = faces[i*3]
    i2 = faces[i*3+1]
    i3 = faces[i*3+2]
    i12 = len(verts)
    i23 = i12+1
    i13 = i12+2
    v1 = verts[i1]
    v2 = verts[i2]
    v3 = verts[i3]
    vt = v1+v2
    vt.normalize()
    verts.append(vt)
    vt = v2+v3
    vt.normalize()
    verts.append(vt)
    vt = v1+v3
    vt.normalize()
    verts.append(vt)
    newFaces.append(i1)
    newFaces.append(i12)
    newFaces.append(i13)
    newFaces.append( i2)
    newFaces.append(i23)
    newFaces.append(i12)
    newFaces.append(i3)
    newFaces.append(i13)
    newFaces.append(i23)
    newFaces .append(i12)
    newFaces.append(i23)
    newFaces.append(i13)
   faces = newFaces
   
  for i in range(0,len(verts)):
   gvw.addData3f(VBase3(verts[i]))
  for i in range(0, len(faces)/3):
   prim.addVertices(faces[i*3],faces[i*3+1],faces[i*3+2])

  prim.closePrimitive()
  geom.addPrimitive(prim)
   
  return ico_path

ico = IcoSphere(1,6)
ico.reparentTo(render)
ico.setRenderModeWireframe()
ico.analyze()
base.setFrameRateMeter(True)
run()

vt = v1+v2
vt.normalize()
verts.append(vt)

verts.append((v1+v2).normalize())

(Point3(0, 1, 2)+Point3(0, 1, 3)).normalize()

v = Point3(0, 1, 2)+Point3(0, 1, 3)
v.normalize()

class vec3:
  def __init__(self, x, y, z):
   self.x = x
   self.y = y
   self.z = z
  def __add__(self, other):
   v = vec3(self.x, self.y, self.z)
   v.x += other.x
   v.y += other.y
   v.z += other.z
   return v
  def normalize_1(self):
   self.x = 1
   self.y = 1
   self.z = 1
  def normalize_2(self):
   self.x = 1
   self.y = 1
   self.z = 1
   return self
  def __str__(self):
   return "vec3({0}, {1}, {2})".format(self.x, self.y, self.z)

v1 = (vec3(2,3,4)+vec3(5,6,7)).normalize_1()
print "v1 = {0}".format(v1)
v2 = (vec3(2,3,4)+vec3(5,6,7)).normalize_2()
print "v2 = {0}".format(v2)

ico = procedural.IcoSphere(1,8)
ico.writeBamFile("icosphere.bam")