"""
8 Oktaeder
09.02.2021
www.3d-meier.de
"""

import c4d
import math

# Variablen und Konstanten
Titel = '8 Oktaeder' # Name

NP = 24    # Anzahl Punkte
N3 = 48    # Anzahl Dreiecke

Radius = 0.3 # Radius der Eckpunktkugeln   

a = 1
h = a/math.sqrt(2)
c = 5.0*a/6.0
d = a*math.sqrt(2)/6.0


Punkte = [(    -c - a,      0.0, -a/2.0),
          (        -c,      0.0, -a/2.0),
          (    -a/2.0,    d + h, -a/2.0),
          (     a/2.0,    d + h, -a/2.0),
          (    -a/2.0,   -d - h, -a/2.0),
          (     a/2.0,   -d - h, -a/2.0),
          (     c + a,      0.0, -a/2.0),
          (         c,      0.0, -a/2.0),
          (    -c - a,      0.0,  a/2.0),
          (        -c,      0.0,  a/2.0),
          (    -a/2.0,    d + h,  a/2.0),
          (     a/2.0,    d + h,  a/2.0),
          (    -a/2.0,   -d - h,  a/2.0),
          (     a/2.0,   -d - h,  a/2.0),
          (     c + a,      0.0,  a/2.0),
          (         c,      0.0,  a/2.0),
          (-c - a/2.0,        h,    0.0),
          (-c - a/2.0,       -h,    0.0),
          ( c + a/2.0,        h,    0.0),
          ( c + a/2.0,       -h,    0.0),
          (       0.0,        d,    0.0),
          (       0.0,  d + 2*h,    0.0),
          (       0.0,       -d,    0.0),
          (       0.0, -d - 2*h,    0.0)]

Dreiecke = [(16,  0,  8,  8),
            ( 8,  0, 17, 17),
            (16,  1,  0,  0),
            ( 1, 17,  0,  0),
            (17,  9,  8,  8),
            ( 8,  9, 16, 16),
            (18, 14,  6,  6),
            ( 6, 14, 19, 19),
            (18,  6,  7,  7),
            ( 7,  6, 19, 19),
            (18, 15, 14, 14),
            (14, 15, 19, 19),
            (10, 11, 21, 21),
            ( 2, 10, 21, 21),
            ( 3,  2, 21, 21),
            (11,  3, 21, 21),
            (23, 12,  4,  4),
            (23,  4,  5,  5),
            (23,  5, 13, 13),
            (23, 13, 12, 12),
            (12, 13, 22, 22),
            (22,  5,  4,  4),
            (20, 11, 10, 10),
            ( 2,  3, 20, 20),
            (16, 10,  2,  2),
            (16,  2,  1,  1),
            ( 9, 10, 16, 16),
            (17,  4, 12, 12),
            (17, 12,  9,  9),
            ( 4, 17,  1,  1),
            ( 4,  1, 22, 22),
            (12, 22,  9,  9),
            (13,  5, 19, 19),
            (19, 15, 13, 13),
            (19,  5,  7,  7),
            ( 3, 11, 18, 18),
            (18,  7,  3,  3),
            (11, 15, 18, 18),
            (15, 22, 13, 13),
            ( 5, 22,  7,  7),
            (22, 15,  7,  7),
            (22,  1,  9,  9),
            ( 9,  1, 20, 20),
            ( 7, 15, 20, 20),
            ( 7, 20,  3,  3),
            ( 2, 20,  1,  1),
            (20, 15, 11, 11),
            (20, 10,  9,  9)]


Faktor = 100   # Skalierungsfaktor
k = 1          # Korrekturfaktor fuer Kantenlaenge 1


#************************************************************************
def CreateNullobjekt1():
    obj = c4d.BaseObject(c4d.Onull)
    obj.SetName(Titel)
    obj.Message(c4d.MSG_UPDATE)
    return obj

#************************************************************************
def CreateNullobjekt2():
    obj = c4d.BaseObject(c4d.Onull)
    obj.SetName('Ecken')
    obj.Message(c4d.MSG_UPDATE)
    return obj

#************************************************************************
def CreateNullobjekt3():
    obj = c4d.BaseObject(c4d.Onull)
    obj.SetName('Kanten')
    obj.Message(c4d.MSG_UPDATE)
    return obj

#************************************************************************
def CreateNullobjekt4():
    obj = c4d.BaseObject(c4d.Onull)
    obj.SetName('Polygone')
    obj.Message(c4d.MSG_UPDATE)
    return obj

#************************************************************************
def CreateDreiecke():
    obj = c4d.BaseObject(c4d.Opolygon)
    obj.ResizeObject(NP,N3)
    obj.SetName('Dreiecke')
  # Punkte uebergeben
    for i in xrange(NP):
        obj.SetPoint(i, c4d.Vector(Punkte[i][0]*Faktor*k, Punkte[i][1]*Faktor*k, Punkte[i][2]*Faktor*k))
  # Dreiecke setzen
    for i in xrange(N3):
        obj.SetPolygon( i, c4d.CPolygon(Dreiecke[i][0], Dreiecke[i][1], Dreiecke[i][2], Dreiecke[i][3]))

    obj.Message(c4d.MSG_UPDATE)
    return obj

#************************************************************************
def CreateSplineDreiecke():
    obj = c4d.BaseObject(c4d.Ospline)
    obj.SetName("Spline-Dreiecke")
    obj.ResizeObject(N3*3)
    zz = 0
    for i in xrange(N3):
        for j in xrange(3):
            x = Punkte[Dreiecke[i][j]][0]
            y = Punkte[Dreiecke[i][j]][1]
            z = Punkte[Dreiecke[i][j]][2]
            obj.SetPoint(zz, c4d.Vector(x*Faktor*k, y*Faktor*k, z*Faktor*k))
            zz = zz + 1
  # Segmente erzeugen
    obj.MakeVariableTag(c4d.Tsegment, N3)
    for i in range(0, N3):
        obj.SetSegment(i, 3, True)
  # Spline schliessen
    obj[c4d.SPLINEOBJECT_CLOSED] = True

    obj.Message(c4d.MSG_UPDATE)
    return obj

#************************************************************************
def main():
    nullobj1 = CreateNullobjekt1()
    nullobj2 = CreateNullobjekt2()   # Ecken
    nullobj3 = CreateNullobjekt3()   # Kanten
    nullobj4 = CreateNullobjekt4()   # Polygone    

    plyobj1 = CreateDreiecke()
    
    splobj1 = CreateSplineDreiecke()

    doc.InsertObject(nullobj1, None, None, True)
    doc.InsertObject(nullobj4, nullobj1, None, True)
    doc.InsertObject(nullobj3, nullobj1, None, True)
    doc.InsertObject(nullobj2, nullobj1, None, True)
    
    doc.InsertObject(splobj1, nullobj3, None, True)
        
    doc.InsertObject(plyobj1, nullobj4, None, True)

  # Kugeln auf Eckpunkte setzen
    for i in range(0, NP):
        obj = c4d.BaseObject(c4d.Osphere)
        obj[c4d.PRIM_SPHERE_RAD] = Radius
        obj.SetName(str(i))
        x = Punkte[i][0]
        y = Punkte[i][1]
        z = Punkte[i][2]
        obj.SetAbsPos(c4d.Vector(x*Faktor*k, y*Faktor*k, z*Faktor*k))
        doc.InsertObject(obj, nullobj2, None, True)
    c4d.EventAdd()


if __name__=='__main__':
    main()