import rhinoscriptsyntax as rs 
import random as ran 

#delete everything and start from scratch
allobjs = rs.AllObjects()
rs.DeleteObjects(allobjs)

###
# domino variables
###
A=5
B=A/3
thick=0.2
hgt=2.7
xcol=2
ycol=5
levels=5
f_height=0.5
f_size=0.8

###
#derived variables
###
center_pt=[A*(xcol-1)/2, A*(ycol-1)/2, f_height]   #insertion point of floor plate
p_width=A*(xcol-1)+2*B                      #width of floor plate
p_length=A*(ycol-1)+f_size                  #length of floor plate

def make_box(insertion=[0,0,0],xsize=10,ysize=10,zsize=10):       #default werte
    #create a box
    corners = [[0,0,0], [xsize,0,0], [xsize,ysize,0], [0,ysize,0], 
            [0,0,zsize], [xsize,0,zsize], [xsize,ysize,zsize], [0,ysize,zsize]]
    box = rs.AddBox(corners)
    rs.MoveObject(box, (-xsize/2,-ysize/2,0))   #stuetzen zentrieren
    rs.MoveObject(box, insertion)
    return(box)

#function to create box at cornerpoint
def make_podest(insertion=[0,0,0], xsize=10, ysize=10, zsize=10):
    corners = [[0,0,0], [xsize,0,0], [xsize,ysize,0], [0,ysize,0],
            [0,0,zsize], [xsize,0,zsize], [xsize,ysize,zsize], [0,ysize,zsize]]
    box = rs.AddBox(corners)
    rs.MoveObject(box,insertion)
    return(box)

#funtion to create a field of foundations
def make_foundations(A=5.0, f_size=0.8, f_height=0.5, xcol=2, ycol=3,):
    fns = []
    for i in range(xcol):
        for j in range(ycol):
            fns.append(make_box([i*A,j*A,0], f_size,f_size,f_height))
    return(fns)

#function to create a field of columns
def make_columns(A=5.0, level=0.7, thick=0.2, hgt=3.0, xcol=2, ycol=3):
    cls=[]
    for i in range(xcol):
        for j in range(ycol):
            cls.append(make_box([i*A,j*A,level], thick,thick,hgt))
    return(cls)

def make_arc(insertion, rad, thick, hgt, orientation):
    segs=[]
    segs.append(rs.AddArc([0,0,0],rad,180))
    segs.append(rs.AddArc([0,0,0],rad-thick,180))
    segs.append(rs.AddLine([rad,0,0],[rad-thick,0,0]))
    segs.append(rs.AddLine([-(rad-thick),0,0], [-rad,0,0]))
    crv=rs.JoinCurves(segs, delete_input=True)
    path=rs.AddLine([0,0,0],[0,0,hgt])
    arc=rs.ExtrudeCurve(crv,path)
    rs.CapPlanarHoles(arc)
    rs.DeleteObject(crv)
    rs.DeleteObject(path)
    rs.RotateObject(arc,[0,0,0],orientation)
    rs.MoveObject(arc,insertion)
    return(arc)

def make_terrace(A=5.0, level=0.7, thick=0.2, hgt=3.0, xcol=2, ycol=3):
    cls=[]
    for i in range(xcol-1):
        for j in range(ycol):
            ori=ran.randint(0,6)
            if ori<4:
                orientation=ori*90
                cls.append(make_arc([i*A+A/2,j*A+A/2,level], A/2,thick,hgt,orientation))
    return(cls)

#function to creat stair
def make_stair(start, th, tt, steps, thick, s_width):
    pointlist=[start]
    for i in range (steps):
        pointlist.append([pointlist[-1][0], pointlist[-1][1], pointlist[-1][2]+th])
        pointlist.append([pointlist[-1][0]+tt, pointlist[-1][1], pointlist[-1][2]])
    pointlist.append([pointlist[-1][0], pointlist[-1][1], pointlist[-1][2]-thick])
    pointlist.append([pointlist[0][0], pointlist[0][1], pointlist[0][2]-thick])
    pointlist.append([pointlist[0][0], pointlist[0][1], pointlist[0][2]])
    s_outline = rs.AddPolyline(pointlist)
    path=rs.AddLine(start, [start[0], start[1]+s_width, start[2]])
    hull = rs.ExtrudeCurve(s_outline, path)
    rs.CapPlanarHoles(hull)
    rs.DeleteObjects((s_outline,path))
    return(hull)

#building dom-ino
def make_domino(A=A, B=B, thick=thick, hgt=hgt, levels=levels, xcol=xcol, ycol=ycol, f_height=f_height, f_size=f_size):
    f_list=[]
    c_list=[]
    p_list=[]
    center_pt=[A*(xcol-1)/2, A*(ycol-1)/2, f_height]    #insertion point of floor plate
    p_width=A*(xcol-1)+2*B                              #width of floor plate
    p_length=A*(ycol-1)+f_size                          #length of floor plate

    for i in range(levels):
        center_pt[2]=f_height+i*(thick+hgt)
        level=f_height+thick+(i-1)*(hgt+thick)
        if i==0:
            f_list=make_foundations(A,f_height,f_size,xcol,ycol)
        else:
            c_list.extend(make_columns(A,level,thick,hgt,xcol,ycol))
            c_list.extend(make_terrace(A,level,thick,hgt,xcol,ycol))
        p_list.append(make_box(center_pt,p_width,p_length,thick))
    level=f_height+thick+(levels-1)*(hgt+thick)
    c_list.extend(make_terrace(A,level,thick,hgt,xcol,ycol))

#calculate stair values
    steps = int((hgt+thick)/0.17)
    if steps%2:
        steps = steps-1
    th= (hgt+thick)/steps
    if (th>0.19):
        steps = steps+2
        th= (hgt+thick)/steps

    #stair parameters
    tt=0.3                                                  #step size
    s_width=1.2                                             #stair width
    pod_w=B                                                 #depth of landing
    start = [pod_w,-(s_width*2+f_size/2), f_height+thick]   #startpoint of stair
    
    #loop to create staircase
    stair_l = []
    for i in range (levels):
        start[2]=f_height+thick + i*(thick+hgt)             #z-wert bei jeder iteration neu gesetzt
        if i==levels-1:                                     #last podest
            stair_l.append(make_podest([start[0]-pod_w, start[1]+s_width, start[2]-thick], pod_w, s_width, thick))
        else:
            stair_l.append(make_podest([start[0]-pod_w, start[1], start[2]-thick], pod_w, s_width*2, thick))
            stair_l.append(make_stair(start, th, tt, int(steps/2), thick, s_width))
            stair_l.append(make_podest([start[0]+(steps/2)*tt, start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick))
            stair_l.append(make_stair([start[0]+(steps/2)*tt, start[1]+s_width, start[2]+(steps/2)*th], th, -tt, int(steps/2), thick, s_width))

    return(f_list, c_list, p_list, stair_l)

#(f_list, c_list, p_list, stair_l) = make_domino()  

my_c = rs.AddSpiral([0,0,0], [0,0,10], 0,3, 100,50)   #spirale erstellen
my_pts = rs.DivideCurve(my_c, 10, True, True)

for p in my_pts:
    levels=ran.randint(2,7)
    xcol=ran.randint(2,7)
    ycol=ran.randint(2,7)
    (f_list, c_list, p_list, stair_l)=make_domino(levels=levels, xcol=xcol, ycol=ycol)
    rs.MoveObject(f_list+c_list+p_list+stair_l, p)

"""
for p in my_pts:
   my_dom = make_domino
   rs.MoveObject(my_dom, p) 
"""

"""
for i in range(5):
    xran=ran.randint(0,50)
    yran=ran.randint(0,70)
    levels=ran.randint(2,7)
    (f_list, c_list, p_list, stair_l)=make_domino(levels=levels)
    rs.MoveObject(f_list+c_list+p_list+stair_l, (xran,yran,0))
"""

#create layers and layer colors
rs.AddLayer("foundation")
rs.LayerColor("foundation", (220,60,60))
rs.ObjectLayer(f_list, "foundation")

rs.AddLayer("columns")
rs.LayerColor("columns", (60,220,60))
rs.ObjectLayer(c_list, "columns")

rs.AddLayer("plates")
rs.LayerColor("plates", (60,60,220))
rs.ObjectLayer(p_list, "plates")

"""
# facade one
width= 0.2
depth= 12.6
f_length = 52
off = A*(xcol-2.7)+B+depth/1.5

# facade offset
def make_facade(insertion= [0,0,0] ,depth=0.3, width=0.1, height=(levels-1)*(hgt+thick)+thick, dist=0.3):
    fcd = []
    for i in range(f_length):
        fcd.append(make_box([off,i*dist-f_size/4,f_height], depth, width, height))
    return(fcd)
    
make_facade()

# facade two 
width= 0.2
depth= 0.4
f_length = 52
off = A*(xcol-2.7)+B+depth/1.5 

# facade offset
def make_facade(insertion= [0,0,0] ,depth=0.3, width=0.1, height=(levels-1)*(hgt+thick)+thick, dist=0.3):
    fcd = []
    for i in range(f_length):
        fcd.append(make_box([off,i*dist-f_size/4,f_height], depth, width, height))
    return(fcd)
    
make_facade()
"""

"""
for i in range(200):                #boxen im rechteck erstellen auf random
    xran = ran.randint(-100,400)
    yran = ran.randint(-200,600)
    zran = 0
    ranscale = ran.uniform(1,6)    #die boxen werden verschieden gross
    if not (rs.PointInPlanarClosedCurve([xran,yran,zran], my_rect, plane, 0)):
        my_box = rs.AddBox(corners)
        rs.MoveObject(my_box, [xran,yran,zran])
        rs.ScaleObject(my_box, [xran,yran,zran], [1,1,ranscale])
"""