"""
hyper-diamond class
"""
from data import *

class HyperDiamond():
    def __init__(self):
        self.vertices3d =  t = [[0.0]*3 for i in range(24)]
        
        # construction of lines
        self.lines = []
        
        # three tesseracts
        for start in range(0, 24, 8):
            stop = start + 16           
            for v1 in range(start, stop - 1):
                for v2 in range(v1 + 1, stop): 
                    # distance calculation
                    vert1 = vertices4d[v1%24]
                    vert2 = vertices4d[v2%24]
                    distance = 0.0
                    for coord in range(4):
                        distance += (vert2[coord] - vert1[coord])**2
                    if distance == 1.0:
                        self.lines.append([self.vertices3d[v1%24], self.vertices3d[v2%24]])
        
        # transform to 3D
        self.perspective = True
        self.projectionTo3D()
    
    def projectionTo3D(self):
        if self.perspective:
            self.perspectiveTo3D()
        else:
            self.isometricTo3D()
    
    #perspective projection to 3D (w,x,y,z) -> (x,y,z), -w is direction of camera, camera w = 3
    def perspectiveTo3D(self):
        i = 0
        for vert in vertices4d:
            cam_dist = 2.0 - vert[0]
            for c in range(3):
                self.vertices3d[i][c] = vert[c+1]/cam_dist
            i += 1
        
    #icometric projection to 3D (w,x,y,z) -> (x,y,z), 4D z = 3D z 
    def isometricTo3D(self):
        i = 0
        for vert in vertices4d:
            for c3 in range(3):
                self.vertices3d[i][c3] = 0.0
                for c4 in range(4):
                    self.vertices3d[i][c3] += 0.5*iso_axes[c4][c3]*vert[c4]
            i += 1