"""
Tetraeder 1
25.03.2021
www.3d-meier.de
"""

import c4d
import math

# Variablen und Konstanten
Titel = 'Tetraeder 1' # Name

NP = 4    # Anzahl Punkte
N3 = 4    # Anzahl Dreiecke

Radius = 2.5 # Radius der Eckpunktkugeln

Punkte = [( 1,  1,  1),
          ( 1, -1, -1),
          (-1,  1, -1),
          (-1, -1,  1)]

Dreiecke = [(0,  1,  2,  2),
            (0,  1,  3,  3),
            (1,  2,  3,  3),
            (2,  0,  3,  3)]

Faktor = 100                    # Skalierungsfaktor
k = 1/(2*math.sqrt(2.0))        # Korrekturfaktor fuer Kantenlaenge 1
#k = 1

ku = 128                        # Unterteilung der Kugeln
ru = Faktor*math.sqrt(6)/4      # Umkugelradius
ri = Faktor*math.sqrt(6)/12     # Innenkugelkradius
rk = Faktor*math.sqrt(2)/4      # Kantenkugelradius


#************************************************************************
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 CreateNullobjekt5():
    obj = c4d.BaseObject(c4d.Onull)
    obj.SetName('Kugeln')
    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
    nullobj5 = CreateNullobjekt5()   # Umkugel, etc.    

    plyobj1 = CreateDreiecke()
    splobj1 = CreateSplineDreiecke()

    doc.InsertObject(nullobj1, None, None, True)
    doc.InsertObject(nullobj5, nullobj1, 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)

  # Umkugel erzeugen
    obj = c4d.BaseObject(c4d.Osphere)
    obj[c4d.PRIM_SPHERE_SUB] = ku
    obj[c4d.PRIM_SPHERE_RAD] = ru
    obj.SetName('Umkugel')  
    obj[c4d.ID_BASEOBJECT_VISIBILITY_EDITOR] = 1
    obj[c4d.ID_BASEOBJECT_VISIBILITY_RENDER] = 1        
    doc.InsertObject(obj, nullobj5, None, True)

  # Innenkugel erzeugen
    obj = c4d.BaseObject(c4d.Osphere)
    obj[c4d.PRIM_SPHERE_SUB] = ku
    obj[c4d.PRIM_SPHERE_RAD] = ri
    obj.SetName('Inkugel')  
    obj[c4d.ID_BASEOBJECT_VISIBILITY_EDITOR] = 1
    obj[c4d.ID_BASEOBJECT_VISIBILITY_RENDER] = 1        
    doc.InsertObject(obj, nullobj5, None, True)
    
  # Kantenkugel erzeugen
    obj = c4d.BaseObject(c4d.Osphere)
    obj[c4d.PRIM_SPHERE_SUB] = ku
    obj[c4d.PRIM_SPHERE_RAD] = rk
    obj.SetName('Kantenkugel')  
    obj[c4d.ID_BASEOBJECT_VISIBILITY_EDITOR] = 1
    obj[c4d.ID_BASEOBJECT_VISIBILITY_RENDER] = 1        
    doc.InsertObject(obj, nullobj5, None, True)
        
    c4d.EventAdd()

if __name__=='__main__':
    main()