###################################
### DM2_w24 # AGruber@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 !



###_________________________________________#
### 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) )
################

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) )
        
        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 0:
        ##############################
        rs.CurrentLayer("UNO::setUp::points")
        rs.AddPoints( 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 (l=0, d=3, h=39)
rs.AddPoint( p0 )
p1 = rs.VectorAdd(p0, rs.VectorScale( dVec, 50.0 ) )
p2 = gC (0, 0, 30)
p21 = rs.VectorAdd(p1, rs.VectorScale( dVec, 50.0 ) )
p3 = gC (8, 0, 20)
p4 = rs.VectorAdd(p3, rs.VectorScale( dVec, -100.0 ) )
p5 = gC (0, 0, 0)

coords = [p0, p1, p2, p3, p4, p5]
#rs.AddCurve( coords, 1 )
crvDiv = 44
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 )


dom = rs.CurveDomain( crv )
print "dom =", dom
dom = rs.CurveDomain( crv )[1]
print "dom =", dom


circDiv = 16
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 )
    rs.ObjectColor( rs.AddCurve( coords, 2 ), [0, 100, 100] )
    #rs.ObjectColor( rs.AddCurve( coords, 1 ), [220, 100, 220] )
    
    

e0 = gC (l=10, d=3, h=39)
rs.AddPoint( e0 )
e1 = rs.VectorAdd(e0, rs.VectorScale( dVec, 50.0 ) )
e2 = gC (1, 1, 30)
e3 = gC (1, 0, 1)
e4 = rs.VectorAdd(e3, rs.VectorScale( dVec, -100.0 ) )
e5 = gC (10, 3, 0)

coords = [e0, e1, e2, e3, e4, e5]
#rs.AddCurve( coords, 1 )
crvDiv = 24
crv = rs.AddCurve( coords, 1 )
coords = rs.DivideCurve( crv, crvDiv, create_points=0 )
#rs.AddPoints( coords )
for cor in coords:
    pass
dom = rs.CurveDomain( crv )
print "dom =", dom
dom = rs.CurveDomain( crv )[1]
print "dom =", dom

circDiv = 16
allCoordsLists = []
for cor in coords:
    paraX = rs.CurveClosestPoint( crv, cor )
    planeX = rs.CurvePerpFrame( crv, paraX )
    circ = rs.AddCircle( planeX, 10.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 )
    rs.ObjectColor( rs.AddCurve( coords, 2 ), [200, 100, 100] )
    #rs.ObjectColor( rs.AddCurve( coords, 1 ), [220, 100, 220] )
    
#########################################BANTIKI
import math

def demoCoaster(anzahl=128, circDiv=64, rad=8.5, makeAll=0, crvDeg=3, delCircs=6, 
                amplitude=6, frequency=6.0, pattern="wave", num_turns=6, color=[173, 216, 230]):
    karlaCoords = []
    p0 = rs.VectorAdd(dm.getUnoCoord(0, 0, 10), rs.VectorScale(lVec, 10))
    karlaCoords = [p0]
    counter = 0

    for h in range(0, H-1, 2):     # Generate the main spiral trajectory
        counter += 1
        
        
        if pattern == "wave":
            offset_x = amplitude * math.sin(frequency * h)
            offset_y = amplitude * math.cos(frequency * h)
        elif pattern == "zigzag":
            offset_x = amplitude if counter % 4 < 2 else -amplitude
            offset_y = amplitude if counter % 4 >= 2 else -amplitude
        elif pattern == "spiral_out":
            offset_x = h * frequency * math.cos(counter)
            offset_y = h * frequency * math.sin(counter)
        else:
            offset_x = offset_y = 0

        if counter % 4 == 1:
            karlaCoor = rs.VectorAdd(dm.getUnoCoord(int((L-1)/2), 0, h), rs.VectorScale(dVec, offset_x))
        elif counter % 4 == 2:
            karlaCoor = dm.getUnoCoord(L-1, int((D-1)/2), h + int(offset_y))
        elif counter % 4 == 3:
            karlaCoor = rs.VectorAdd(dm.getUnoCoord(int((L-1)/2), D-1, h), rs.VectorScale(dVec, int(offset_y * -1)))
        elif counter % 4 == 0:
            karlaCoor = dm.getUnoCoord(0, int((D-1)/2), h)

        karlaCoords.append(karlaCoor)

    karlaCoords.append(karlaCoords[0])
    crv = rs.AddCurve(karlaCoords, 2)

    rs.ObjectColor(crv, color)
#ROMBUS
    dom = rs.CurveDomain(crv)[1]
    allCoords = []

    for p in range(anzahl + 1):
        fac = p / anzahl
        coord = rs.EvaluateCurve(crv, dom * fac)
        planeX = rs.CurvePerpFrame(crv, dom * fac)
        vecXY = dm.vecCurvePerpXY(crv, dom * fac)
        vecXY = rs.VectorScale(vecXY, rad)

        # Define the rhombus vertices
        rhombus_coords = [
            rs.VectorAdd(coord, rs.VectorScale(vecXY, 1)),          # Top point
            rs.VectorAdd(coord, rs.VectorScale(vecXY, -1)),         # Bottom point
            rs.VectorAdd(coord, rs.VectorRotate(vecXY, 90, planeX[3])),  # Left point
            rs.VectorAdd(coord, rs.VectorRotate(vecXY, -90, planeX[3]))  # Right point
        ]
        
        rhombus_coords.append(rhombus_coords[0])  # Close the rhombus

        if not delCircs:
            rs.AddCurve(rhombus_coords, 1)  # Draw rhombus shape
        allCoords.append(rhombus_coords)

    allSpirals = []

    for i in range(1, len(allCoords)):
        rhombus = allCoords[i]



# Call the demoCoaster function with updated parameters to create a wider spiral with more turns
rs.CurrentLayer("UNO::setUp::crv")
demoCoaster(anzahl=32, circDiv=32, rad=3.5, makeAll=2, crvDeg=2, delCircs=0, 
            amplitude=1.5, frequency=0.7, pattern="wave", num_turns=8, color=[173, 216, 230])

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



#____________________________:here you end

###################################
dm.zA( proz=0.95 )
rs.EnableRedraw(1) ### 4 the MACs
#dm.printDisplay(1)
rs.CurrentLayer("Default")