import rhinoscriptsyntax as rs
import random as ran

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

##############################################################
# domino variables                                           #
##############################################################
A = 5                                # A = Module size (distance between columns)
B = A/3                              # B = Distance of columns to end of plate
thick = 0.18                         # thickness of all slabs
hgt = 2.7                            # height of room
xcol = 2                             # columns in x direction
ycol = 3                             # columns in y direction
levels = 5                           # number of floor plates
f_height = 0.5                       # f_height = foundation height
f_size = 0.8                         # f_size = foundation edge size

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

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

#function to creater stair
def make_stair(start, th, tt, steps, thick, s_width):
    steps = 14
    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])
    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)


rs.EnableRedraw(False)

# 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 = []            # 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.3                  # step size
    s_width = 1.2             # s_width = stair width
    pod_w = B                 # pod_w = depth of landing
    start = [pod_w, -(s_width*2 + fsize / 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:    # letztes podest
            stair_l.append(make_podest([start[0] - pod_w, start[1] + s_width, start[2] - thick], pod_w, s_width, thick))  # sonderpodest
        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()

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")

rs.EnableRedraw(True)