"""
Kachelung 2
10.09.2015
"""

import c4d
import math

# Variablen und Konstanten
N = 32      # Anzahl Punkte pro Kante
r = 1.0    # Kantenlaenge

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

Faktor = 1000        # Skalierungsfaktor

Quadrat = []

#Inversion = True
Inversion = False

def Invers(z):
    return 1/z

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

# Quadrate 1 *********************************************************************************
def CreateSplineObjectQ1():    
    # Splineobjekte erzeugen
    obj1 = c4d.BaseObject(c4d.Ospline)   
    obj1.ResizeObject(N*4*(2*NS+1)*(4*NS+1))      
    dt= r/N  # Unterteilung Kante
    zz = 0   # Zähler Punkte        
    # Quadratdaten erzeugen
    for i in range(0,N):
        z = complex(-0.5 * r + i * dt, -0.5 * r)              
        Quadrat.append(z)
    for i in range(0,N):
        z = complex(0.5 * r, -0.5 * r + i * dt)            
        Quadrat.append(z)
    for i in range(0,N):
        z = complex(0.5 * r - i * dt, 0.5 * r)             
        Quadrat.append(z)
    for i in range(0,N):
        z = complex(-0.5 * r, 0.5 * r - i * dt)               
        Quadrat.append(z)
    
    for k in range(0,4*NS+1):     
        y0 = -2*NS + k*r
        for j in range(0,2*NS+1):                         
            if k%2 == 0.0: 
                x0 = -2*r*NS + j*2*r                               
            else:    
                x0 = -2*r*NS + j*2*r + r               
            for i in range(0,4*N):
                z = Quadrat[i] + 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)*(4*NS+1))     
    for i in range(0, (2*NS+1)*(4*NS+1)):
        obj1.SetSegment(i, 4*N, True)  

    obj1[c4d.SPLINEOBJECT_CLOSED] = True
    obj1.SetName("Quadrate 1")
       
    obj1.Message(c4d.MSG_UPDATE)
    return obj1


# Quadrate 2 ********************************************************************************
def CreateSplineObjectQ2():    
    # Splineobjekte erzeugen
    obj2 = c4d.BaseObject(c4d.Ospline)   
    obj2.ResizeObject(N*4*(2*NS+1)*(4*NS+1))      
    dt= r/N  # Unterteilung Kante
    zz = 0   # Zaehler Punkte        
    
    for k in range(0,4*NS+1):     
        y0 = -2*NS + k*r
        for j in range(0,2*NS+1):                         
            if k%2 == 0.0: 
                x0 = -2*r*NS + j*2*r + r                               
            else:    
                x0 = -2*r*NS + j*2*r               
            for i in range(0,4*N):
                z = Quadrat[i] + complex(x0, y0)   
                if Inversion:
                    z = Invers(z)
                obj2.SetPoint(zz, c4d.Vector(z.real*Faktor, z.imag*Faktor, 0.0))            
                zz = zz + 1
    obj2.MakeVariableTag(c4d.Tsegment, (2*NS+1)*(4*NS+1))     
    for i in range(0, (2*NS+1)*(4*NS+1)):
        obj2.SetSegment(i, 4*N, True)  

    obj2[c4d.SPLINEOBJECT_CLOSED] = True
    obj2.SetName("Quadrate 2")
       
    obj2.Message(c4d.MSG_UPDATE)
    return obj2


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



if __name__=='__main__':
    main()