################################################
### DM2_w24 # tanja.konrad@student.tugraz.at ###
### ue_05 UN_headquaters  NYC                ###
################################################


##############################
import rhinoscriptsyntax as rs
import random, time, sys   ###
sys.path.append("P:/")     ###
import DM_lib as dm        ### 
reload(dm)
##############################      

rs.UnitSystem(4)                                        # km = 5, meters = 4, cm = 3 etc
rs.ShowGrid(None, 0)                                    # grid > 0 = off
rs.ShowGridAxes(None, 1)                                # y/y/z axen display > 0/1 = off/on
rs.ViewDisplayMode(rs.CurrentView(), "wireframe")
rs.Command("cplane w t enter", 0)						# cPlane World Top
dm.PointRadius(displayModeX=0, rad=3, styl=3)
dm.printDisplay(state=1) ### nomen est omen / printDisplay(state=1, scale=1000, thickness=1, color="Display")
rs.EnableRedraw(0)       ### MAC >> problem !


#dm.PointRadius(displayModeX=0, rad=2, styl=0, verbose=0) # 0 wireframe
#dm.PointRadius(displayModeX=1, rad=17, styl=2, verbose=0) # 1 shaded
#dm.PointRadius(displayModeX=2, rad=9, styl=3, verbose=1) # rendered

###_________________________________________#
### 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
#################___________________________#


dm.newEmptyLayer("UNO::setUp::points", [150,150,200]  )
dm.newEmptyLayer("UNO::setUp::geo",    [150, 20,150]  )
dm.newEmptyLayer("UNO::setUp::crv",    [250, 10, 10])
dm.newEmptyLayer("UNO::setUp",         [120,120,140]  )


################
lengthVec = lVec = rs.VectorUnitize(rs.VectorSubtract( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord(1, 0, 0) )) ## rs.AddPoint( dm.getUnoCoord(1, 0, 0) )
depthVec  = dVec = rs.VectorUnitize(rs.VectorSubtract( dm.getUnoCoord(0, 1, 0), dm.getUnoCoord(0, 0, 0) )) ## rs.AddPoint( dm.getUnoCoord(0, 1, 0) )
dVec2 = rs.VectorUnitize(rs.VectorSubtract( dm.getUnoCoord(1, 0, 0), dm.getUnoCoord(0, 0, 0) ))
dVec3 = rs.VectorUnitize(rs.VectorSubtract( dm.getUnoCoord(0, 0, 1), dm.getUnoCoord(0, 0, 0) ))

################

if 1: ### DEMOs / SETUP >> dont' exec @ homework !
    if 1:
        rs.AddLine( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord(10, 0,  0) )
        rs.AddLine( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord( 0, 3,  0) )
        rs.AddLine( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord( 0, 0, 39) )
        
        for y in range(slabs):
            for x in range(depth):
                rs.AddLine( dm.getUnoCoord(y, x, 0), dm.getUnoCoord( y, x,  39) )

            #rs.AddLine( dm.getUnoCoord(0, 0, 0), dm.getUnoCoord( 0, 0,  0) )

#        rs.CurveArrows(rs.AllObjects()[0], arrow_style=2)
#        rs.CurveArrows(rs.AllObjects()[1], arrow_style=2)
#        rs.CurveArrows(rs.AllObjects()[2], arrow_style=2)
        
#        rs.ObjectColor(rs.AllObjects()[2], [0, 222, 0] )
#        rs.ObjectColor(rs.AllObjects()[1], [222, 0, 0] )
#        rs.ObjectColor(rs.AllObjects()[0], [0, 0, 222] )
        
        rs.ObjectPrintWidth( rs.AllObjects()[0:3], 1.0 )

        p0 = dm.getUnoCoord(  0, 0, 0)
        p1 = dm.getUnoCoord( 10, 0, 0)
        p2 = dm.getUnoCoord( 10, 3, 0)
        p3 = dm.getUnoCoord(  0, 3, 0)
        unoBaseCoords = [p0, p1, p2, p3, p0]
        unoBaseCrv = rs.AddCurve( unoBaseCoords, 1)
        unoTopCoords = [dm.getUnoCoord(  0, 0, 39),
                        dm.getUnoCoord( 10, 0, 39),
                        dm.getUnoCoord( 10, 3, 39),
                        dm.getUnoCoord(  0, 3, 39),
                        dm.getUnoCoord(  0, 0, 39),
                       ]
        unoTopCrv = rs.AddCurve( unoTopCoords, 1)
        rs.ZoomExtents()
    if 1:
        ##############################
        rs.CurrentLayer("UNO::setUp::points")
        #rs.AddPoints( UnoGridCoords )
        #rs.AddLine(UnoGridCoords)
        #print "UnoGridCoords",UnoGridCoords
#        for i in range(L): rs.ObjectColor( rs.AddTextDot( str(i), dm.getUnoCoord(i, 0, 0) ), [0, 222, 0] )
#        for i in range(D): rs.ObjectColor( rs.AddTextDot( str(i), dm.getUnoCoord(0, i, 0) ), [222, 0, 0] )
#        for i in range(H): rs.ObjectColor( rs.AddTextDot( str(i), dm.getUnoCoord(0, 0, i) ), [0, 0, 222] )
        
        ##############################
    
    
    if 0:
        rs.CurrentLayer("UNO::setUp::crv")
        dm.UN_slab(showAll=1)
        for i in range(8):
            p0 = dm.getUnoCoord( random.randint(0,L-1), random.randint(0,D-1), random.randint(0,H-1) )
            p1 = dm.getUnoCoord( random.randint(0,L-1), random.randint(0,D-1), random.randint(0,H-1) )
            rs.AddCurve( [p0, p1] )
            dm.textDots(  [p0, p1] )
        rs.ZoomExtents()
    if 0:
        coords = UnoGridCoords[:] ### make COPY - not equal !
        random.shuffle( coords )
        rs.AddCurve( coords[0:], random.randint(1, 3) )

#################
### abbreveation:
def gC (l=1, d=2, h=3):
    cor = dm.getUnoCoord(l, d, h)
    return cor
## rs.AddPoint( gC(1,0,1) )
#################

#_____________________________here you go:
dm.newEmptyLayer("myPROJ", [100,100,100])

rs.CurrentLayer("UNO::setUp::crv")

p0 = gC (0, 0, 0)
p1 = rs.VectorAdd(p0, rs.VectorScale( dVec2, -50.0 ) )
p2 = rs.VectorAdd(p1, rs.VectorScale( dVec, -100.0 ) )
p3 = gC (3, 0, 3)
#p4 = rs.VectorAdd(p3, rs.VectorScale( dVec, -50.0 ) )

p5 = gC (0, 3, 10)
p6 = rs.VectorAdd(p5, rs.VectorScale( dVec, 50.0 ) )

p7 = gC (5, 3, 0)
p8 = gC (6, 3, 10)
p9 = gC (10, 0, 0)
p10 = rs.VectorAdd(p9, rs.VectorScale( dVec2, 50.0 ) )
p11 = rs.VectorAdd(p10, rs.VectorScale( dVec, 100.0 ) )

p12 = gC (0, 0, 20)
p13 = rs.VectorAdd(p12, rs.VectorScale( dVec2, 70.0 ) )
p14 = gC (0, 0, 39)
p15 = rs.VectorAdd(p14, rs.VectorScale( dVec3, 50.0 ) )
p16 = rs.VectorAdd(p15, rs.VectorScale( dVec, 50.0 ) )
p17 = gC (5, 3, 39)
p18 = rs.VectorAdd(p17, rs.VectorScale( dVec3, 30.0 ) )
p19 = rs.VectorAdd(p18, rs.VectorScale( dVec2, 50.0 ) )
p20 = gC (10, 3, 20)

coords = [p17, p18, p19, p0, p1, p2, p3,p5, p6,p7,p8,p9,p10,p11,p12, p13, p14,p15,p16,p17]
crvDiv = 88
crv = rs.AddCurve( coords, 2 )
coords = rs.DivideCurve( crv, crvDiv, create_points=0 )
#rs.AddPoints( coords ) 

for cor in coords:
    pass
    #rs.AddCircle( cor, 5.5 )

#rs.CurvePerpFrame() ##CurvePerpFrame(curve_id, parameter) 

dom = rs.CurveDomain( crv )
dom = rs.CurveDomain( crv )[1]

circDiv = 33
allCoordsLists = []
for cor in coords:
    paraX = rs.CurveClosestPoint( crv, cor )
    planeX = rs.CurvePerpFrame( crv, paraX )
    circ = rs.AddCircle( planeX, 5.5)
    circCoords = rs.DivideCurve( circ, circDiv, 0)
    rs.DeleteObject( circ )
    allCoordsLists.append( circCoords )

i = 0

for i in range(circDiv):
    coords = []
    for list in allCoordsLists:
        pX = list[i]
        coords.append( pX )
    curve = rs.AddCurve(coords,2)
    start = rs.CurveStartPoint(curve)

    #Streben
    middle = gC(5,2,39)
    bottom = gC(8,0,0)
    coords.append(start)
    curveDiv = rs.DivideCurve( curve, crvDiv, create_points=0 )
    
    if i < 5:
        rs.ObjectColor(rs.AddLine (middle, curveDiv[i]),[0,0,0])
    if i > 17 and i < 22:
        rs.ObjectColor(rs.AddLine (middle, curveDiv[i]),[0,0,0])

    rs.DeleteObject(curve)
    curve = rs.AddCurve(coords,2)
    if pX[2] < 168:
        rs.ObjectColor(curve,[0,0,0])
    else:
        rs.ObjectColor(curve,[255,0,100])
    rs.DeleteObject(crv)

#____________________________:here you end

###################################
dm.zA( proz=0.95 )
rs.EnableRedraw(1) ### 4 the MACs
#dm.printDisplay(1)
rs.CurrentLayer("Default")