"""
Kachelung 1
10.09.2015
"""

import c4d
import math

# Variablen und Konstanten
N = 5            # Anzahl Punkte pro Kante
r = 1.0          # Kantenlaenge
h = 0.5*math.sqrt(3.0) *r      

Abstand_x = 2*r
Abstand_y = 2*h
Versatz_x = r

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

Faktor = 1000        # Skalierungsfaktor

Sexeck = []
Dreieck = []

#Inversion = True
Inversion = False

def Invers(z):
    return 1/z

def CreateNullobjekt():    
    obj = c4d.BaseObject(c4d.Onull)   # Nullobjekt erzeugen
    obj.SetName("Kachelung 1") 
    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)
     
# Sechsecke **************************************************************
def CreateSplineObjectS():    
    # Splineobjekte erzeugen
    obj = c4d.BaseObject(c4d.Ospline)   
    obj.ResizeObject(N*6*(2*NS+1)*(2*NS+1))      
    dt= r/N  # Unterteilung Kante
    zz = 0   # Zaehler Punkte    
    
    # Sechseckdaten erzeugen
    for i in range(0,N):
        z = complex(-0.5 * r + i * dt, -h)              
        Sexeck.append(z)
    for k in range(1,6):        
        for j in range(0,N):
            z = Drehung(Sexeck[j], -60*k)                         
            Sexeck.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,6*N):
                z = Sexeck[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, 6*N, True) 

    obj[c4d.SPLINEOBJECT_CLOSED] = True
    obj.SetName("Sechsecke")
       
    obj.Message(c4d.MSG_UPDATE)
    return obj
     

# Dreiecke *************************************************************************************
def CreateSplineObjectD():    
    obj = c4d.BaseObject(c4d.Ospline)   # Splineobjekt erzeugen
    obj.ResizeObject(N*3*(2*NS+1)*(2*NS+1)*2)      
    dt= r/N
    zz = 0   # Zaehler Punkte    

    # Dreiecksdaten erzeugen
    for i in range(0,N):
        z = complex(-0.5 * r + i * dt, -h)              
        Dreieck.append(z)
    for i in range(0,N):
        z = complex(i * dt, 0)          
        z = Drehung(z, -120) + complex(0.5*r, - h)    
        Dreieck.append(z)
    for i in range(0,N):
        z = complex(i * dt, 0)    
        z = Drehung(z, 120)            
        Dreieck.append(z)
    # Dreieck verschieben und spiegeln    
    for i in range(0,3*N):        
         Dreieck[i] =  Dreieck[i] + complex(r, 0)
    for i in range(0,3*N):         
        z =  complex( Dreieck[i].real, - Dreieck[i].imag) 
        Dreieck.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*2):
                z = Dreieck[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)*2)     
    for i in range(0, (2*NS+1)*(2*NS+1)*2):
        obj.SetSegment(i, 3*N, True)  

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


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



if __name__=='__main__':
    main()