################################################
### 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")
anz = 36*6
depth = 4.5#5.5
#rad = 5.0
p0 = gC( 5, 1, 0)
lenX = rs.Distance(gC( 5, 0, 0), gC( 0, 0, 0))+depth*2  ### laenge des anfangs center point zur aussenkante d gebaeudes

#deltaLen = (lenX-rad)/(anz+1)
deltaAng = 10.0 ## grad
deltaZ = 156/(anz)#+0)
coords = []
for i in range(anz+1):
#    rad += deltaLen
    vec = rs.VectorScale(lVec, lenX)
    vec = rs.VectorRotate( vec, i*deltaAng, [0,0,1])
    vec[2] = i*deltaZ
    #rs.AddLine( p0, rs.VectorAdd(p0, vec))
    if i > 2 and i < 15:
        vec = rs.VectorAdd(vec, rs.VectorScale( dVec, 50.0 ) )
    if i > 21 and i < 36:
        vec = rs.VectorAdd(vec, rs.VectorScale( dVec, -50.0 ) )
    coords.append(rs.VectorAdd(p0, vec))        ### vektoren verschieben, 6x

top = gC(10,0,39)
coords.append(rs.VectorAdd(top, rs.VectorScale( dVec3, 10.0 ) ))
height = 39
rows = 10
z = 1
lenY = rs.Distance(gC( 10, 3, 0), gC( 10, 0, 0))
for i in range(15):
    p = rs.VectorScale(lVec, lenY)
    p = gC (rows, 3, height)
    #print "p",p
    coords.append(p)
    height = height-1
    p = gC (rows, 0, height)
    coords.append(p)
    height = height-1
    #print "i",i

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:
    #print i
    middle = gC(5,2,39)
    bottom = gC(8,0,0)
    paraX = rs.CurveClosestPoint( crv, cor )
    planeX = rs.CurvePerpFrame( crv, paraX )
    circ = rs.AddCircle( planeX, depth)
    circCoords = rs.DivideCurve( circ, circDiv, 0)
    if i < 30:
        rs.ObjectColor(rs.AddLine (middle, coords[i]),[0,0,0])
    if i > 40 and i < 80:
        rs.ObjectColor(rs.AddLine (bottom, coords[i]),[0,0,0])
    rs.DeleteObject( circ )
    allCoordsLists.append( circCoords )
    i = i + 1

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 )
    #print curveDiv
#    if i <5:
#        rs.ObjectColor(rs.AddLine (middle, coords[i]),[0,0,0])
#    if i > 17 and i < 22:
#        rs.ObjectColor(rs.AddLine (bottom, 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")