import rhinoscriptsyntax as rs
import random as ran

# delete everything
allobjs = rs.AllObjects()
rs.DeleteObjects(allobjs)

rs.EnableRedraw(False)

### variables
#Dom-ino
A = 5                                               # A = Module size (distance between columns)
B = A/3                                             # B = Distance of columns to end of plate
thick = 0.2                                         # thickness of all slabs
block = 0.2                                         # measure of column
hgt = 2.7                                           # height of room
xcol = 4                                            # columns in x direction
ycol = 3                                            # columns in y direction
levels = 3                                          # number of floor plates
f_height = 0.5                                      # f_height = foundation height
f_size = 0.8                                        # f_size   = foundation edge size
#facade
fac_width = 0.05                                     # width of facade blocks
fac_depth = 0.05                                     # depth of facade blocks
fac_height = hgt                                    # height of facade blocks
fac_dist = 0.1                                      # distance between pieces


### derived values
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 (x)
p_length = A*(ycol-1)+f_size                           # width of floor plate (y)

# function to create box at centerpoint
def make_box(insertion = [0,0,0],xsize=10, ysize=10, zsize=10):
    # 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))
    rs.MoveObject(box, insertion)
    return(box)

#function to create podest
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)

#function to create a field of foundations
def make_foundations(A=5.0, f_size=0.8, f_height=0.5, xcol=xcol, ycol=ycol):
    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=A, level=0.7, block=block, hgt=2.7, xcol=xcol, ycol=ycol):
    cls = []
    for i in range(xcol):
        for j in range (ycol):
            cls.append(make_box([i*A,j*A,level], block, block, hgt))
    return(cls)

# Function to create a Arc
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,0,0],[-(rad-thick),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)

# Function to create a terrace
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-1):
            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 create a facade
def make_facade(fac_width = fac_width, fac_height = fac_height, fac_depth = fac_depth):
    wls = []
    for level in range(levels-1):
        z_base = f_height + thick + level * (hgt + thick)
        for i in range(int(A * (xcol - 1)/fac_dist)):
            for z in range(int(hgt / fac_height)):
                random_fac_height = fac_height * ran.uniform(0.2,1)  # Random height
                wls.append(make_box([i * fac_dist, 0, z_base + z * random_fac_height], fac_width, fac_depth, random_fac_height))
                wls.append(make_box([i * fac_dist, A * (ycol - 1), z_base + z * random_fac_height], fac_width, fac_depth, random_fac_height))
        for y in range(int(A * (ycol - 1)/fac_dist)):
            for z in range(int(hgt / fac_height)):
                random_fac_height = fac_height * ran.uniform(0.2,1)  # Random height
                wls.append(make_box([0, y * fac_dist, z_base + z * random_fac_height], fac_depth, fac_width, random_fac_height))
                wls.append(make_box([A * (xcol - 1), y * fac_dist, z_base + z * random_fac_height], fac_depth, fac_width, random_fac_height))
    return(wls)

"""
# function to create a facade outside
def make_facade(fac_width = fac_width, fac_height = fac_height, fac_depth = fac_depth):
    wls = []
    for level in range(levels-1):
        z_base = f_height + level * (hgt + thick)
        for i in range(int((A+(B*2)) * (xcol - 1)/fac_dist)):
            for z in range(int(1)):
                random_fac_height = fac_height * ran.uniform(0.2,1) # Random height
                wls.append(make_box([i * fac_dist, -(f_size/2)-(fac_depth/2), z_base + z * random_fac_height], fac_width, fac_depth, random_fac_height))
                wls.append(make_box([i * fac_dist, A * (ycol - 1)+(f_size/2)+(fac_depth/2), z_base + z * random_fac_height], fac_width, fac_depth, random_fac_height))
        for y in range(int(A * (ycol - 1)/fac_dist)):
            for z in range(int(1)):
                random_fac_height = fac_height * ran.uniform(0.2,1)  # Random height
                wls.append(make_box([-B-(fac_depth/2), y * fac_dist, z_base + z * random_fac_height], fac_depth, fac_width, random_fac_height))
                wls.append(make_box([A * (xcol - 1)+B+(fac_depth/2), y * fac_dist, z_base + z * random_fac_height], fac_depth, fac_width, random_fac_height))
    return(wls)
"""

#function to create 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)
    return(hull)

# Function to create 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=[]                           # list of foundations
    c_list=[]                           # list of columns
    p_list=[]                           # list of plates
    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_size, f_height, xcol, ycol)
        else:
            c_list.extend(make_columns(A, level, thick, hgt, xcol, ycol))
            if(i%2):
                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.28                                                     #step size
    s_width = 1.2                                               #s_width = stair width 
    pod_w = B                                                   #pod_w = depth of landing
    start = [pod_w,-(s_width*2+f_size/2), f_height+thick]       #startpoint of stair
    
    #loop to create staircase
    s_list = []
    for i in range(levels):
        start[2] = f_height+thick + i*(thick+hgt)    
        if i==levels-1:       
            s_list.append(make_podest([start[0]-pod_w, start[1]+s_width, start[2]-thick], pod_w, s_width, thick)) 
        else:
            s_list.append(make_podest([start[0]-pod_w, start[1], start[2]-thick], pod_w, s_width*2, thick))
            s_list.append(make_stair(start, th, tt, int(steps/2), thick, s_width))
            s_list.append(make_podest([start[0]+(steps/2)*tt, start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick))
            s_list.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, s_list)

(f_list, c_list, p_list, s_list)= make_domino()

w_list = make_facade()

rs.AddLayer("foundation")
rs.LayerColor("foundation", (220,60,60))
rs.ObjectLayer(f_list, "foundation")

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

rs.AddLayer("plates")
rs.LayerColor("plates", (180,245,30))
rs.ObjectLayer(p_list, "plates")

rs.AddLayer("facade")
rs.LayerColor("facade", (70,50,45))
rs.ObjectLayer(w_list, "facade")

rs.AddLayer("stairs")
rs.LayerColor("stairs", (65,70,20))
rs.ObjectLayer(s_list, "stairs")


rs.EnableRedraw(True)