"""
Kachelung 7
10.09.2015
"""

import c4d
import math

# Variablen und Konstanten
N = 32      # Anzahl Punkte pro Kante
r = 1.330   # Kantenlaenge
h = 0.5*math.sqrt(3.0) *r      # Hoehe Dreieck    
rid = math.sqrt(3.0)/6.0 * r   # Inkreisradius Dreieck

Abstand_x = r
Abstand_y = h
Versatz_x = 0.5*r

NS = 7            # Anzahl Wiederholungen in eine Richtung
if NS%2 ==0:      # nur ungerade NS zulassen
    NS = NS +1

Faktor = 1000        # Skalierungsfaktor


Dreieck1 = []
Dreieck2 = []

Inversion = True
#Inversion = False

def Invers(z):
    return 1/z

def CreateNullobjekt():    
    obj = c4d.BaseObject(c4d.Onull)   # Nullobjekt erzeugen
    obj.SetName("Kachelung 7") 
    obj.Message(c4d.MSG_UPDATE)
    return obj

def Drehung(z, Grad):
    Winkel = Grad * math.pi/180
    x = z.real
    y = z.imag     
    x1 = math.cos(Winkel) * x + math.sin(Winkel) * y  
    y1 = -math.sin(Winkel) * x + math.cos(Winkel) * y 
    return complex(x1, y1)
     
# Dreiecke 1 **************************************************************************************
def CreateSplineObjectD1():    
    obj = c4d.BaseObject(c4d.Ospline)   # Splineobjekt erzeugen
    obj.ResizeObject(N*3*(2*NS+1)*(2*NS+1))      
    dt= r/N

    zz = 0   # Zaehler Punkte    

    # Dreiecksdaten erzeugen
    for i in range(0,N):
        z = complex(-0.5 * r + i * dt, -rid)              
        Dreieck1.append(z)
    for i in range(0,N):
        z = complex(i * dt, 0)          
        z = Drehung(z, -120) + complex(0.5*r,  -rid)    
        Dreieck1.append(z)
    for i in range(0,N):
        z = complex(i * dt, 0)    
        z = Drehung(z, 120) + complex(0, h-rid)
        Dreieck1.append(z)
      
    for k in range(0,2*NS+1): 
        y0 = -Abstand_y*NS + k*Abstand_y  
        for j in range(0,2*NS+1):                           
            if k%2 == 0.0: 
                x0 = -Abstand_x*NS + j*Abstand_x + Versatz_x      
            else:    
                x0 = -Abstand_x*NS + j*Abstand_x
            for i in range(0,3*N):
                z = Dreieck1[i] + complex(x0, y0)   
                if Inversion:
                    z = Invers(z)
                obj.SetPoint(zz, c4d.Vector(z.real*Faktor, z.imag*Faktor, 0.0))            
                zz = zz + 1

    obj.MakeVariableTag(c4d.Tsegment, (2*NS+1)*(2*NS+1))     
    for i in range(0, (2*NS+1)*(2*NS+1)):
        obj.SetSegment(i, 3*N, True)  

    obj[c4d.SPLINEOBJECT_CLOSED] = True
    obj.SetName("Dreiecke 1")
       

    obj.Message(c4d.MSG_UPDATE)
    return obj


# Dreiecke 2 ************************************************************************************
def CreateSplineObjectD2():    
    obj1 = c4d.BaseObject(c4d.Ospline)   # Splineobjekt erzeugen
    obj1.ResizeObject(N*3*(2*NS+1)*(2*NS+1))      
    dt= r/N

    zz = 0   # Zähler Punkte    

    # Dreiecksdaten erzeugen
    for i in range(0,N):
        z = complex(-0.5 * r + i * dt, rid)               
        Dreieck2.append(z)
    for i in range(0,N):
        z = complex(i * dt, 0)    
        z = Drehung(z, 120) + complex (0.5*r, rid)
        Dreieck2.append(z)
    for i in range(0,N):
        z = complex(i * dt, 0)   
        z = Drehung(z, -120) + complex(0, -(h-rid))
        Dreieck2.append(z)
       
    for k in range(0,2*NS+1): 
        y0 = -Abstand_y*NS + k*Abstand_y  
        for j in range(0,2*NS+1):                          
            if k%2 == 0.0: 
                x0 = -Abstand_x*NS + j*Abstand_x + Versatz_x      
            else:    
                x0 = -Abstand_x*NS + j*Abstand_x
            for i in range(0,3*N):
                z = Dreieck2[i] +  complex(0.5 * r, rid) + complex(x0, y0)   
                if Inversion:
                    z = Invers(z)
                obj1.SetPoint(zz, c4d.Vector(z.real*Faktor, z.imag*Faktor, 0.0))            
                zz = zz + 1

    obj1.MakeVariableTag(c4d.Tsegment, (2*NS+1)*(2*NS+1))     
    for i in range(0, (2*NS+1)*(2*NS+1)):
        obj1.SetSegment(i, 3*N, True)  

    obj1[c4d.SPLINEOBJECT_CLOSED] = True
    obj1.SetName("Dreiecke 2")
       
    obj1.Message(c4d.MSG_UPDATE)
    return obj1



def main():
    nullobj = CreateNullobjekt()        
    slinobj1 = CreateSplineObjectD1()
    slinobj2 = CreateSplineObjectD2()
       
    doc.InsertObject(nullobj, None, None, True)
    doc.InsertObject(slinobj2, nullobj, None, True)
    doc.InsertObject(slinobj1, nullobj, None, True)   
 
        
    c4d.EventAdd()


if __name__=='__main__':
    main()