###################################
### 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:/")     ###
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 0: ### 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 0:   ### 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
    amount_segments = 16         ### higher >> rutsche closer 2 crv_2
    number = 2**6


    coords= rs.DivideCurve(crv_2, amount_segments, False)
    print ("len coords = "+str(len(coords)))
    #dm.textDots(coords)


    coords = []
    params = []
    fac = dom/amount_segments
    for i in range(amount_segments+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")

#point1 = [UnoGridCoords]
#rs.AddSphere(point1, 5)

#print(UnoGridCoords)

## HU_05
UnoPanelCoords = dm.UnoPanelCoords
allPanels     = UnoPanelCoords[4]

if 0:
    for panel in allPanels:

        #print panel #ein  Panel ist eine Liste an Punkten
        #print len(panel) #amount_segments der Punkte von einem Panel

        randomOffset = random.randint(0, 2000)/1000
        
        vec1 = rs.VectorCreate(panel[1], panel[0])
        vec2 = rs.VectorCreate(panel[3], panel[0])
        normal = rs.VectorCrossProduct(vec1, vec2)
        normal = rs.VectorUnitize(normal)

        xNormalOffset = normal[0]
        yNormalOffset = normal[1]
        zNormalOffset = -normal[2]

        for i in range(4):
            startPoint = rs.CreatePoint(panel[i])
            endPoint = rs.CreatePoint(panel[(i+1) % 4])

            randomOffsetStartPoint = rs.CreatePoint(panel[i])
            randomOffsetStartPoint[0] += xNormalOffset * randomOffset
            randomOffsetStartPoint[1] += yNormalOffset * randomOffset
            randomOffsetStartPoint[2] += zNormalOffset * randomOffset

            line1 = rs.AddLine(startPoint, randomOffsetStartPoint)

            endPoint[0] += xNormalOffset * randomOffset
            endPoint[1] += yNormalOffset * randomOffset
            endPoint[2] += zNormalOffset * randomOffset

            line2 = rs.AddLine(randomOffsetStartPoint, endPoint)
            rs.ObjectColor([line1, line2], (0, 250, 250)) #cyan

# HU_06
if 1:
    upper_points = UnoGridCoords[len(UnoGridCoords) - 44: len(UnoGridCoords)]

    first_point = upper_points[0]
    second_point = upper_points[10]
    third_point = upper_points[len(upper_points)-11]
    last_point = upper_points[len(upper_points)-1]

    x_direction = last_point - third_point
    x_norm = rs.VectorUnitize(x_direction)
    y_direction = last_point - second_point
    y_norm = rs.VectorUnitize(y_direction)

    last_point = third_point + x_direction + rs.VectorUnitize(x_direction) * 2
    second_point = first_point + x_direction + rs.VectorUnitize(x_direction) * 2

    first_point  += - x_norm * 10 - y_norm * 10
    second_point +=   x_norm * 10 - y_norm * 10
    third_point  += - x_norm * 10 + y_norm * 10
    last_point   +=    x_norm* 10 + y_norm * 10

    height_of_building = first_point[2] - UnoGridCoords[0][2]

    edge_points = [first_point, second_point, last_point, third_point]

    if 0:
        for i in range(len(edge_points)):
            rs.AddSphere(edge_points[i], 1)

    def addSlide(upper_edge_points, height_of_building, amount_segments):
        middle = (upper_edge_points[0] + upper_edge_points[len(upper_edge_points) - 1]) / 2
        
        middle += x_norm * 20
        points = [rs.CreatePoint(middle)]
        middle -= x_norm * 20

        points.append(rs.CreatePoint(middle))

        height_diff_per_segment = height_of_building / amount_segments

        current_point = upper_edge_points[0]
        segment_y = current_point[2]
        
        for i in range(amount_segments):
            for i2 in range(len(upper_edge_points)):
                p = rs.CreatePoint(upper_edge_points[i2])
                p[2] = segment_y
                segment_y -= height_diff_per_segment / len(upper_edge_points)

                points.append(p)

        middle[2] -= height_of_building
        points.append(rs.CreatePoint(middle))
        middle += x_norm * 20
        points.append(rs.CreatePoint(middle))

        radius = 4.0

        for p in points:
            p[2] += radius

        curve = rs.AddCurve(points, 2)

        domain = rs.CurveDomain(curve)
        t_start = domain[0]
        t_end = domain[1]
        dom_len = t_end - t_start

        amount_segments = 40
        number = 30

        allCoordsLists = []

        fac = dom_len / amount_segments

        for i in range(amount_segments + 1):
            param = t_start + (fac * i) # t parameter (wo sind wir auf der kurve)
            point = rs.EvaluateCurve(curve, param) # Punkt bei t holen
            tangent = rs.CurveTangent(curve, param) # Tangente bei t holen
            
            hVec = rs.VectorCrossProduct(tangent, [0, 0, 1]) # Vector der nach oben schaut
            hVec = rs.VectorUnitize(hVec) # Normieren
            hVec = rs.VectorScale(hVec, radius) # Auf Radius skalieren
            
            # Höhen Vektor im kreis rotieren
            coords = []
            deltaAng = 360.0 / number
            
            for a in range(number):
                ang = deltaAng * a - 90 # sonst schaut Kurve nach links
                
                vec = rs.VectorRotate(hVec, ang, tangent)
                rotated_point = rs.PointAdd(point, vec)
                
                coords.append(rotated_point)
            
            # Ring in die Liste hinzufügen
            allCoordsLists.append(coords)

        start_index = int(number * 0.25) # Nur der untere Halbkreis
        end_index = int(number * 0.75)

        midCoords = []        

        # Kurven entlang der Linie einfügen
        for k in range(start_index, end_index):
            points = []
            for ring in allCoordsLists:
                points.append(ring[k])
                if k == number * 0.5:
                    midCoords.append(ring[k])
            
            slide_curve = rs.AddCurve(points, 2)
            rs.ObjectColor(slide_curve, (255, 0, 0)) #red
        
        if 1:
            baseCrv = rs.AddCurve( [gC(-4,-4,0), gC(-4,7,0),  gC(14,7,0), gC(14,-4,0),gC(-4,-4,0)], 3)
            for cor in midCoords:
                param = rs.CurveClosestPoint(baseCrv, cor)
                nPnt  = rs.EvaluateCurve( baseCrv, param )
                if rs.Distance( cor, nPnt ) > 35.0:
                    stuetze = rs.AddLine( cor, nPnt )
                    rs.ObjectColor( stuetze, [0,111,111])
                    rs.ObjectPrintWidth( stuetze, 0.5)


    addSlide(edge_points, height_of_building, 5)

    print(height_of_building)
