##############################

import rhinoscriptsyntax as rs

import random, time, sys   ###

sys.path.append("P:/www/argetlam/DM2")     ###

import DM_lib as dm        ### reload( dm )

##############################

rs.UnitSystem(3)                       

rs.ShowGrid(view=None, show=0)

rs.ShowGridAxes(view=None, show=0)

rs.ViewDisplayMode(view=None, mode="Wireframe")

rs.EnableRedraw(0)

dm.PointRadius(displayModeX=0, rad=3, styl=1)          

##Kreise





dm.newEmptyLayer("PROJ::crv", [0,0,200]) 
rs.ViewDisplayMode("Perspective", "Shaded")
rs.ViewDisplayMode("Front", "Shaded")
rs.ViewDisplayMode("Top", "Shaded")
rs.ZoomExtents()

rs.EnableRedraw(1) ### 4 the MACs

dm.printDisplay(1)






################KoordinatenSuche


def generateZwiebel(ID,hohe,krum1,v,jung):
    haus = rs.ObjectsByName(ID)[0]
    
    #print haus
    
    coords = dm.getSurfacePoints( haus )
    
    #dm.textDots(coords) 
    
    #print rs.CurveEditPoints( crv )
    
    
    #####################Variablen
    
    leng = rs.Distance(coords[1], coords[2])
    bre = rs.Distance(coords[2], coords[3])
    hoeh = hohe
    
    center = dm.pntInbetween(coords [1],coords [3])
    spitze = rs.VectorAdd(center,[0,0,hoeh])
    rs.AddPoint(spitze)
    rs.DeleteObject(rs.AllObjects()[0])
    distVec1 = rs.VectorSubtract(coords[1],center)
    distVec2 = rs.VectorSubtract(coords[2],center)
    distVec3 = rs.VectorSubtract(coords[3],center)
    distVec4 = rs.VectorSubtract(coords[4],center)
    
    num = 27 ### max 32 / min ~16
    anz = 22 ### sinnvoll ab ~12
    
    step = (hoeh*1)/ num
    
    krum=0.03   
    
    ######

    if 1:
    
        center_index = num // 2
        allCoords = []
        sprung = 20 #11
        #sprung2 = 20
        for i in range(num):
    
            distance = abs(i - center_index)
    
            if i==0:
                rs.AddCurve([coords [1],coords [2],coords [3],coords [4],coords [1]],1)
                pnts_0 = rs.DivideCurve(rs.AllObjects()[0], anz, create_points=0)
                rs.DeleteObject(rs.AllObjects()[0])
                
            elif i<sprung:
                centNeu = rs.VectorAdd(center,[0,0,step*i])
                vergr = (i/(v*sprung))**(-krum1*i/(v*sprung))
                p1 = rs.VectorAdd(centNeu,rs.VectorScale(distVec1,vergr))
                p2 = rs.VectorAdd(centNeu,rs.VectorScale(distVec2,vergr))
                p3 = rs.VectorAdd(centNeu,rs.VectorScale(distVec3,vergr))
                p4 = rs.VectorAdd(centNeu,rs.VectorScale(distVec4,vergr))
                rs.AddCurve([p1,p2,p3,p4,p1],1)
                pnts_0 = rs.DivideCurve(rs.AllObjects()[0], anz, create_points=0)
                rs.DeleteObject(rs.AllObjects()[0])
                


            else:
                centNeu = rs.VectorAdd(center,[0,0,step*i])
                verkl = ((1/v)**(-krum1/v)+(sprung-i)*((1/v)**(-krum1/v)/(num-sprung)))*jung**(sprung-i)
                #verkl = (1/v)**(-krum1/v)+(1/v)**(-krum1/v)/(num-sprung)*(sprung-i)
                #verkl = (1/v)**(-krum1/v)*krum2**(sprung-i)
                p1 = rs.VectorAdd(centNeu,rs.VectorScale(distVec1,verkl))
                p2 = rs.VectorAdd(centNeu,rs.VectorScale(distVec2,verkl))
                p3 = rs.VectorAdd(centNeu,rs.VectorScale(distVec3,verkl))
                p4 = rs.VectorAdd(centNeu,rs.VectorScale(distVec4,verkl))
                rs.AddCurve([p1,p2,p3,p4,p1],1)
                pnts_0 = rs.DivideCurve(rs.AllObjects()[0], anz, create_points=0)
                rs.DeleteObject(rs.AllObjects()[0])
                
            allCoords.append( pnts_0 )



####Spiral-Muster
    Zufall=random.randint(0,100)
    
    if Zufall<25:
        lenx = len(allCoords)
        for i in range(anz):
            coords = []
            for j in range(lenx):
                k = (i+j)%anz
                coords.append( allCoords[j][k] )
            coords.append( spitze )
            coords_=[]
            for j in range(len(coords)):
                coords_.append([rs.VectorDotProduct(coords[j],[1,0,0]),rs.VectorDotProduct(coords[j],[0,1,0]),rs.VectorDotProduct(coords[j],[0,0,1])])

            rs.AddCurve( coords_, 2)
    elif Zufall <50:
        lenx = len(allCoords)
        for i in range(anz):
            coords = []
            for j in range(lenx):
                k = (j-i)%anz
                coords.append( allCoords[j][k] )
            coords.append( spitze )
            coords_=[]
            for j in range(len(coords)):
                coords_.append([rs.VectorDotProduct(coords[j],[1,0,0]),rs.VectorDotProduct(coords[j],[0,1,0]),rs.VectorDotProduct(coords[j],[0,0,1])])
    
            kurve=rs.AddCurve( coords_, 2)
            rs.ObjectColor(kurve,[40,100,70])

    elif Zufall < 75:
        lenx = len(allCoords)
        for i in range(anz):
            coords = []
            for j in range(lenx):
                k = (i-j)%anz
                coords.append( allCoords[j][k] )
            coords.append( spitze )
            coords_=[]
            for j in range(len(coords)):
                coords_.append([rs.VectorDotProduct(coords[j],[1,0,0]),rs.VectorDotProduct(coords[j],[0,1,0]),rs.VectorDotProduct(coords[j],[0,0,1])])
            kurve=rs.AddCurve( coords_, 2)
            rs.ObjectColor(kurve,[80,0,120])
        for i in range(anz):
            coords = []
            for j in range(lenx):
                k = (j-i)%anz
                coords.append( allCoords[j][k] )
            coords.append( spitze )
            coords_=[]
            for j in range(len(coords)):
                coords_.append([rs.VectorDotProduct(coords[j],[1,0,0]),rs.VectorDotProduct(coords[j],[0,1,0]),rs.VectorDotProduct(coords[j],[0,0,1])])
            kurve=rs.AddCurve( coords_, 2)
            rs.ObjectColor(kurve,[150,0,70])
    else:
        lenx = len(allCoords)
        for i in range(anz):
            coords = []
            for j in range(lenx):
                k = (i)%anz
                coords.append( allCoords[j][k] )
            coords.append( spitze )
            coords_=[]
            for j in range(len(coords)):
                coords_.append([rs.VectorDotProduct(coords[j],[1,0,0]),rs.VectorDotProduct(coords[j],[0,1,0]),rs.VectorDotProduct(coords[j],[0,0,1])])
            kurve=rs.AddCurve( coords_, 2)
            rs.ObjectColor(kurve,[150,0,50])

######Generate
generateZwiebel("_bldg3D_higBy_hig_137425142",15000, 1.2,0.4,1)
rs.ZoomExtents()
generateZwiebel("_bldg3D_higBy_flo_265260901",10000,1.3,0.6,1)
generateZwiebel("_bldg3D_higBy_hig_265260824",5000,0.7,0.4,1.1)
generateZwiebel("_bldg3D_higBy_ran_265260933",7000,0.6,0.3,1.3)
generateZwiebel("_bldg3D_higBy_flo_159866206",12000,0.6,0.3,1.3)
generateZwiebel("_bldg3D_higBy_hig_265260825",4000,0.35,0.3,1)