"""
Koch Kurve 2
24.08.2019
www.3d-meier.de
"""

import c4d
import random
import math
import cmath

# Variablen und Konstanten
Titel = 'Koch Kurve' # Name
N = 1000000          # Anzahl Punkte
Nv = 100             # Anzahl Punkte verwerfen
dx = 0.6             # Breite Polygon
dy = 0.6             # Hoehe Polygon
Faktor = 1000        # Skalierungsfaktor
s = math.sqrt(3.0)   # Konstante


def CreatePolygonObject():
    obj = c4d.BaseObject(c4d.Opolygon)
    obj.ResizeObject(4*N, N)
    obj.SetName(Titel)
  # Startwert der Berechnung
    z = complex(0.50, 0.50)
  # Zähler für Punkte setzten
    zz = 0
  # Zähler für Polygone setzten
    zzz = 0
    for i in xrange(0,Nv):
        zufall = random.randint(0,3)
        if zufall == 0:
           z = z/3.0
        if zufall == 1:
           z = (cmath.exp(math.pi*complex(0, 1)/3.0)/3.0)*z + 1/3.0
        if zufall == 2:
           z = (cmath.exp(-math.pi*complex(0, 1)/3.0)/3.0)*z + ((cmath.exp(math.pi*complex(0, 1)/3.0) + 1)/3.0)
        if zufall == 3:
           z = z/3.0 + 2/3.0

    for i in xrange(0,N):
        zufall = random.randint(0,3)
        if zufall == 0:
           z = z/3.0
        if zufall == 1:
           z = (cmath.exp(math.pi*complex(0, 1)/3.0)/3.0)*z + 1/3.0
        if zufall == 2:
           z = (cmath.exp(-math.pi*complex(0, 1)/3.0)/3.0)*z + ((cmath.exp(math.pi*complex(0, 1)/3.0) + 1)/3.0)
        if zufall == 3:
           z = z/3.0 + 2/3.0

        x = z.real*Faktor
        y = z.imag*Faktor
        obj.SetPoint(zz, c4d.Vector((x-dx/2.0),(y+dy/2.0),0))
        zz=zz+1
        obj.SetPoint(zz, c4d.Vector((x-dx/2.0),(y-dy/2.0),0))
        zz=zz+1
        obj.SetPoint(zz, c4d.Vector((x+dx/2.0),(y+dy/2.0),0))
        zz=zz+1
        obj.SetPoint(zz, c4d.Vector((x+dx/2.0),(y-dy/2.0),0))
        zz=zz+1
        obj.SetPolygon(zzz, c4d.CPolygon(zz-1,zz-2,zz-4,zz-3))
        zzz=zzz+1

    #
    obj.Message(c4d.MSG_UPDATE)
    return obj


def main():
    plyobj = CreatePolygonObject()

    #
    doc.InsertObject(plyobj, None, None, True)

    c4d.EventAdd()


if __name__=='__main__':
    main()