"""
Rhombic Propello Octahedron 2
04.04.2021
www.3d-meier.de
"""

import c4d
import math

# Variablen und Konstanten
Titel = 'Rhombic Propello Octahedron 2' # Name

NP = 30    # Anzahl Punkte
N3 =  8    # Anzahl Dreiecke
N4 = 24    # Anzahl Vierecke

Radius = 3 # Radius der Eckpunktkugeln

Punkte = [( 0,  0,  4),
          ( 0,  0, -4),
          ( 0,  4,  0),
          ( 0, -4,  0),
          ( 4,  0,  0),
          (-4,  0,  0),
          ( 1,  2, -3),
          ( 1, -2,  3),
          (-1,  2,  3),
          (-1, -2, -3),
          ( 2, -3,  1),
          (-2,  3,  1),
          ( 2,  3, -1),
          (-2, -3, -1),
          (-3,  1,  2),
          ( 3,  1, -2),
          ( 3, -1,  2),
          (-3, -1, -2),
          ( 3,  2,  1),
          (-3, -2,  1),
          (-3,  2, -1),
          ( 3, -2, -1),
          ( 2,  1,  3),
          (-2,  1, -3),
          ( 2, -1, -3),
          (-2, -1,  3),
          ( 1,  3,  2),
          ( 1, -3, -2),
          (-1, -3,  2),
          (-1,  3, -2)]

Dreiecke = [(20, 29, 23, 23),
            ( 6, 12, 15, 15),
            (18, 26, 22, 22),
            ( 8, 11, 14, 14),
            (13, 17,  9,  9),
            (27, 24, 21, 21),
            (19, 28, 25, 25),
            ( 7, 10, 16, 16)]

Vierecke = [( 9, 17, 23,  1),   
            ( 1, 23, 29,  6), 
            ( 6, 15, 24,  1), 
            ( 1, 24, 27,  9), 
            ( 8, 26,  2, 11), 
            (11,  2, 29, 20), 
            (26, 18, 12,  2), 
            ( 2, 12,  6, 29), 
            (25, 28,  7,  0), 
            ( 0,  7, 16, 22), 
            (14, 25,  0,  8), 
            ( 8,  0, 22, 26), 
            (13,  9, 27,  3), 
            ( 3, 27, 21, 10), 
            (19, 13,  3, 28), 
            (28,  3, 10,  7), 
            (14, 11, 20,  5), 
            ( 5, 20, 23, 17), 
            (25, 14,  5, 19), 
            (19,  5, 17, 13), 
            (16, 10, 21,  4), 
            ( 4, 21, 24, 15), 
            (18,  4, 15, 12), 
            (22, 16,  4, 18)] 

Faktor = 100         # Skalierungsfaktor
#k = 1/math.sqrt(6)   # Korrekturfaktor fuer Kantenlaenge 1
k=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 CreateVierecke():
    obj = c4d.BaseObject(c4d.Opolygon)
    obj.ResizeObject(NP, N4)
    obj.SetName('Vierecke')
  # 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))
  # Vierecke setzen
    for i in xrange(N4):
        obj.SetPolygon(i, c4d.CPolygon(Vierecke[i][0], Vierecke[i][1], Vierecke[i][2], Vierecke[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 CreateSplineVierecke():
    obj = c4d.BaseObject(c4d.Ospline)
    obj.SetName("Spline-Vierecke")
    obj.ResizeObject(N4*4)
    zz = 0
    for i in xrange(N4):
        for j in xrange(4):
            x = Punkte[Vierecke[i][j]][0]
            y = Punkte[Vierecke[i][j]][1]
            z = Punkte[Vierecke[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, N4)
    for i in range(0, N4):
        obj.SetSegment(i, 4, 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()
    plyobj2 = CreateVierecke()

    splobj1 = CreateSplineDreiecke()
    splobj2 = CreateSplineVierecke()

    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(splobj2, nullobj3, None, True)
    doc.InsertObject(splobj1, nullobj3, None, True)

    doc.InsertObject(plyobj2, nullobj4, 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
        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()