###################################
### DM2_w25 # AGruber@tugraz.at ###
### hu_06 UN_headquaters  NYC   ### CRVs 4 ROLLER.COASTER | DEMO
###################################


##############################
import rhinoscriptsyntax as rs
import random, time, sys   ###
sys.path.append("P:/")     ###
sys.path.append("P:/WWW/lukschme/dm2/")
import DM_lib as dm        ### dm.reload_lib(dm)
##############################      

rs.UnitSystem(4)                                            # meters = 4, cm = 3 etc
rs.ShowGrid(None, False)                                    # grid > False = off
rs.ShowGridAxes(None, True)                                 # y/y/z axen display > False/True = off/on
rs.ViewDisplayMode(rs.CurrentView(), "wireframe")
rs.Command("cplane w t enter", False)				        # cPlane World Top
dm.PointRadius(displayModeX=0, rad=3, styl=3)               # 0 => wireframe | info: (.., verbose=1)
dm.PointRadius(displayModeX=1, rad=4, styl=1)               # 1 => shaded    | info: (.., verbose=1)
dm.PointRadius(displayModeX=2, rad=2, styl=0)               # 2 => rendered  | info: (.., verbose=1)
dm.printDisplay(state=False)                                # nomen est omen
rs.EnableRedraw(False)



###_________________________________________#
### basic settings for grid to fit UN_slab  #
###                                         # ! no need 2 change !
floors = H  = dm.H = 40                     # default=40 / incl roof
slabs  = L  = dm.L = 11                     # default=11
depth  = D  = dm.D =  4                     # default= 4 / division in building_depth
floorHeight = fH = dm.fH = 4.0              # default= 4.0 / 4.0*(H-1) = 156 meters
                                            #
#############                               # get gridCoords L*D*H = 1760
UnoGridCoords = dm.getUnoGridCoords()       # get from DM_lib as dm
#################___________________________#


#######################
### my little helper: ###############
### gC (get_Coord) im UN_grid !     #
def gC(L,D,H): return dm.gC(L,D,H)  #
#####################################

##################
depthVec  = dVec = rs.VectorUnitize(rs.VectorSubtract( gC(0, 0, 0), gC(0, 2, 0) ))
lengthVec = lVec = rs.VectorUnitize(rs.VectorSubtract( gC(0, 0, 0), gC(1, 0, 0) ))
##################



dm.newEmptyLayer("UNO::setUp", [120,120,140])
if 1: ### SETUP >> dont' exec @ homework !
    if 0: rs.AddPoints(UnoGridCoords)       ### just 4 reference
    if 1:
        rs.AddLine( gC(0, 0, 0), gC(10, 0, 0) )
        rs.AddLine( gC(0, 0, 0), gC(0,  3, 0) )
        rs.AddLine( gC(0, 0, 0), gC(0, 0, 39) )
        rs.ObjectColor(rs.AllObjects()[2], [0, 222, 0] ), rs.CurveArrows(rs.AllObjects()[0], 2)
        rs.ObjectColor(rs.AllObjects()[1], [222, 0, 0] ), rs.CurveArrows(rs.AllObjects()[1], 2)
        rs.ObjectColor(rs.AllObjects()[0], [0, 0, 222] ), rs.CurveArrows(rs.AllObjects()[2], 2)
        rs.ObjectPrintWidth( rs.AllObjects()[0:3], 0.5 )
        
        p0 = gC(  0, 0, 0)
        p1 = gC( 10, 0, 0)
        p2 = gC( 10, 3, 0)
        p3 = gC(  0, 3, 0)
        unoBaseCoords = [p0, p1, p2, p3, p0]
        unoBaseCrv = rs.AddCurve( unoBaseCoords, 1)
        unoTopCoords = [dm.gC(  0, 0, 39),
                        dm.gC( 10, 0, 39),
                        dm.gC( 10, 3, 39),
                        dm.gC(  0, 3, 39),
                        dm.gC(  0, 0, 39),
                       ]
        unoTopCrv = rs.AddCurve( unoTopCoords, 1)
        rs.ZoomExtents()
    if 0: ### long version
        coords = []
        for y in range(11):
            for z in range(0, 40, 2):
                for x in range(0, 4, 3):
                    coords.append( dm.gC(y,x,z) )
        rs.AddPoints( coords )
        dm.SetObjDisplayModeAllViewports(rs.AddPoints( coords ), displaymodeX=0, verbose=1)
    
    if 1: ### smart version / comprehension
        pnts = rs.AddPoints([ dm.gC(y,x,z) for y in range(11) for z in range(0, 40, 2) for x in range(0, 4, 3) ])
        dm.SetObjDisplayModeAllViewports(pnts, displaymodeX=2, verbose=0) ### 2 == rendered


    
    
    #### CAMERA
    #### set / get camera
    if 0:
        print ("ViewNames :"),
        print (rs.ViewNames())
        
        cam = gC(5, -20, 19.5)
        tar = gC(5, 0, 19.5)
        rs.AddLine( cam, tar )
        
        #dm.setCameraTarget( cam, tar, 500.0, 0, 0, 0)
        rs.ViewProjection(view='Perspective', mode=1)   ### 1 = parallel, 2 = perspective, 3 = two point perspective
        dm.getCameraTarget(view=rs.CurrentView(), verbose=1)
        #dm.setCameraTarget( [334.42526127, 493.18649387, 171.88081703], [756.84885751, 592.83573956, 82.0], lens=50.0, rota=0, upVec=[0,0,1] ) # ... danke, andi !

#######################################
################ HERE YOU GO AS YOU GO:
dm.newEmptyLayer("myPROJ", [111,111,111])

#############
### just demo
        ### get coords for leitCurve
if 1:   ### if 1/True: neue coords werden berechnet und in den (speicher)scope der DM_lib geschrieben
        ### siehe dm.coordsX  = coordsX
        ### wenn curve passt >> 0/False
        ### dann werden die zuletzt nach dm.coordsX gespeicherten importiert/verwendet 
    coordsX = [  gC( random.randint(-10,20), random.randint(-10,10), random.randint(10,39) ) for i in range(8) ]
    coordsX[0]  = gC(2,0,2)
    coordsX[1]  = gC(2,-10,2)
    coordsX[-2] = gC(20,1.5,39)
    coordsX[-1] = gC(10,1.5,39)
    dm.coordsX  = coordsX
crv_1 = rs.AddCurve( dm.coordsX, 1)
crv_2 = rs.AddCurve( dm.coordsX, 2)
crv_3 = rs.AddCurve( dm.coordsX, 3)
print ("crv_Domain = "+str(rs.CurveDomain(crv_1)[1])+" crv_1")
print ("crv_Domain = "+str(rs.CurveDomain(crv_2)[1])+" crv_2")
print ("crv_Domain = "+str(rs.CurveDomain(crv_3)[1])+" crv_3")

##### most important 4 2_day :)
dom = rs.CurveDomain(crv_2)[1]

radius = 11.0
anzahl = 16         ### higher >> rutsche closer 2 crv_2
number = 2**6


coords= rs.DivideCurve(crv_2, anzahl, False)
print ("len coords = "+str(len(coords)))
#dm.textDots(coords)


coords = []
params = []
fac = dom/anzahl
for i in range(anzahl+1):
    #print (i, fac*i)
    cor = rs.EvaluateCurve( crv_2, fac*i)
    #rs.AddPoint( cor )
    rs.ObjectColor( rs.AllObjects()[0], [222,0,0])
    params.append(fac*i)
    coords.append(cor)
print ("len coords = "+str(len(coords)))
print ("len params = "+str(len(params)))
#rs.ObjectColor(dm.textDots(coords), [200,0,0])

deltaAng = -360/number
allCoordsLists = []
for param in params:
    #nPlane = rs.CurvePerpFrame(crv, param)
    #circ = rs.AddCircle(nPlane, radius)
    #allCoordsLists.append( rs.DivideCurve(circ, number, False))
    ###
    pnt  = rs.EvaluateCurve( crv_2, param)
    #rs.AddPoint( pnt )
    hVec = dm.vecCurvePerpXY(crv_2, param)
    hVec = rs.VectorScale( hVec, radius)
    hPnt = rs.VectorAdd( pnt, hVec)
    #rs.AddLine( pnt, hPnt )
    coords = []
    for a in range(number):
        dm.esc()
        ang = deltaAng * a
        tan = rs.CurveTangent(crv_2, param)
        vec = rs.VectorRotate(hVec, ang, tan )
        cor = rs.VectorAdd( pnt, vec)
        #rs.AddLine( pnt, cor )
        coords.append( cor )
    allCoordsLists.append( coords )


if 1:
    i=0
    for i in range(int(number/2), int(number/1)):
        dm.esc()
        coords = []
        for coordList in allCoordsLists:
            coords.append(coordList[i])
        rs.AddCurve(coords, 3)
        rs.Redraw()

###################
### abstuetzung u.a.
if 1:
    rs.SelectObject( rs.AllObjects()[int(number/4)] )
    middleCrv = rs.AllObjects()[int(number/4)]
    midCoords = rs.CurveEditPoints( middleCrv )
    middleCrv = rs.AddCurve( midCoords, 1)
    rs.ObjectColor( middleCrv, [0,222,0])
    rs.ObjectPrintWidth( middleCrv, 1.0)
    
    baseCrv = rs.AddCurve( [gC(-2,-2,0), gC(-2,5,0),  gC(12,5,0), gC(12,-2,0),gC(-2,-2,0)], 3)
    for cor in midCoords:
        param = rs.CurveClosestPoint(baseCrv, cor)
        nPnt  = rs.EvaluateCurve( baseCrv, param )
        if rs.Distance( cor, nPnt ) > 30.0:
            stuetze = rs.AddLine( cor, nPnt )
            rs.ObjectColor( stuetze, [0,111,111])
            rs.ObjectPrintWidth( stuetze, 0.5)


#############################################
### read coords from surface or curve or else 
if 0:
    trump_tower = rs.ObjectsByName("_bldg3D_higBy_hig_261243324x")
    if trump_tower:
        coords = dm.getSurfacePoints( trump_tower[0] )
        dm.textDots( coords )
    else:
        print ("### must import https://iam.tugraz.at/dm2/w25/data/tutorials/assignment11/data_ag/UN_OSM.3dm ###")
    
    brunnen = rs.ObjectsByName("crv_121257600x")
    if brunnen:
        coords = rs.CurveEditPoints( brunnen[0] )
        dm.textDots( coords )
    else:
        print ("### must import https://iam.tugraz.at/dm2/w25/data/tutorials/assignment11/data_ag/UN_OSM.3dm ###")


########## EOS / EndOfScript
rs.EnableRedraw(True)     ### 4_the_MACs
dm.eDup()                 ### delete duplicate objects 
dm.printDisplay(state=True, scale=1000)
dm.zA( 0.9 )
dm.newEmptyLayer("Default")