import rhinoscriptsyntax as rs
import random as ran

rs.DeleteObjects(rs.AllObjects())
rs.EnableRedraw(False)

################################################################################
# domino variables
A = 5           # distance between columns
B = A/3         # distance of columns to end of plate
thick = 0.2    # thickness of all slabs
hgt = 2.5       # height of room
xcol = 5        # columns in x
ycol = 3         # columns in y
levels = 4      # number of floor plates
f_height = 0.5  # foundation height
f_size = 0.8    # foundation edge size
level= 0.7

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

##variables facade
xcor = 0.5
ycor = 0.5
zcor = 0.5
dist_x = 0.5      #distance between boxes x direction
dist_z = 0.2      #distance between boxes z direction
################################################################################

# function to create box at centralpoint
def make_box(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, (-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=A, f_size=f_size, f_height=f_height, 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=level, thick=thick, hgt=hgt, xcol=xcol, ycol=ycol):
    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)

#make a terrace
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=A, level=level, thick=thick, hgt=hgt, xcol=xcol, ycol=ycol):
    cls=[]
    for i in range(xcol-1):
        for j in range(ycol-1):
            ori = ran.randint(0,5)
            if ori<4:
                orientation = ori*90
                cls.append(make_arc([i*A+A/2,j*A+A/2,level], A/3, thick, hgt, orientation))
    return(cls)

#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)
    rs.DeleteObject(s_outline)
    rs.DeleteObject(path)
    return (hull)

# building dom-ino
def make_domino(levels=levels, thick=thick, hgt=hgt, A=A, f_size=f_size, f_height=f_height, xcol=xcol, ycol=ycol):
    f_list=[]
    c_list=[]
    p_list=[]
    B = A/3
    center_pt = [A*(xcol-1)/2, A*(ycol-1)/2, f_height]
    p_width = A*(xcol-1) + 2*B 
    p_length = A*(ycol-1) + f_size 
    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))
            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]  # start point of stair
    
    #loop to create staircase
    stair_l = []
    for i in range (levels):
        start[2] = f_height+thick + i*(thick+hgt)   #z_wert
        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))     #special podest
        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)

# make facade
def make_facade1 (A=A, xcol=xcol, ycol=ycol, xcor=xcor, ycor=ycor, zcor=zcor, dist_x=dist_x, dist_z=dist_z, f_size=f_size, f_height=f_height, levels=levels, hgt=hgt, thick=thick):
    B = A/3
    center_pt = [A*(xcol-1)/2, A*(ycol-1)/2, f_height]
    p_width = A*(xcol-1) + 2*B 
    p_length = A*(ycol-1) + f_size 
    corners_fac = [(0,0,0), (xcor,0,0), (xcor,ycor,0), (0,ycor,0),
       (0,0,zcor), (xcor,0,zcor), (xcor,ycor,zcor), (0,ycor,zcor)]
    num_box = (p_width/2)/(xcor+dist_x)
    num_box_hgt = (hgt*(levels-1))/(zcor+dist_z)
    if (xcol<4):
        num_box_v = (p_width/3)/(xcor+dist_x)
        my_line1= rs.AddLine([-B+(p_width/1.5), -f_size/2-ycor, f_height+thick],[-B+p_width-xcor, -f_size/2-ycor, f_height+thick])
        my_pts1 = rs.DivideCurve(my_line1, num_box_v, False, True)
    if (xcol>3):
        my_line1= rs.AddLine([-B+(p_width/2), -f_size/2-ycor, f_height+thick],[-B+p_width-xcor, -f_size/2-ycor, f_height+thick])
        my_pts1 = rs.DivideCurve(my_line1, num_box, False, True)
    my_line2= rs.AddLine([-B, -f_size/2, 0],[-B, -f_size/2, ((levels-1)*hgt)])
    my_pts2 = rs.DivideCurve(my_line2, num_box_hgt, False, True)
    for p in my_pts1:
        for i in my_pts2:
                box_fac = rs.AddBox(corners_fac)
                rs.MoveObject(box_fac, p)
                rs.MoveObject(box_fac, (0,0,i[2]))
    rs.DeleteObject(my_line1)
    rs.DeleteObject(my_line2)

def make_facade2 (A=A, xcol=xcol, ycol=ycol, xcor=xcor, ycor=ycor, zcor=zcor, dist_x=dist_x, dist_z=dist_z, f_size=f_size, f_height=f_height, levels=levels, hgt=hgt, thick=thick):
    B = A/3
    center_pt = [A*(xcol-1)/2, A*(ycol-1)/2, f_height]
    p_width = A*(xcol-1) + 2*B 
    p_length = A*(ycol-1) + f_size 
    corners_fac = [(0,0,0), (xcor,0,0), (xcor,ycor,0), (0,ycor,0),
       (0,0,zcor), (xcor,0,zcor), (xcor,ycor,zcor), (0,ycor,zcor)]
    num_box = p_width/(xcor+dist_x)
    num_box_hgt = (hgt*(levels-1))/(zcor+dist_z)
    my_line1= rs.AddLine([-B, (A*(ycol-1))+(f_size/2), f_height+thick],[-B+p_width-xcor, (A*(ycol-1))+(f_size/2), f_height+thick])
    my_pts1 = rs.DivideCurve(my_line1, num_box, False, True)
    my_line2= rs.AddLine([-B, (A*(ycol-1))+(f_size/2)-ycor, 0],[-B, (A*(ycol-1))+(f_size/2)-ycor, ((levels-1)*hgt)])
    my_pts2 = rs.DivideCurve(my_line2, num_box_hgt, False, True)
    for p in my_pts1:
        for i in my_pts2:
                box_fac = rs.AddBox(corners_fac)
                rs.MoveObject(box_fac, p)
                rs.MoveObject(box_fac, (0,0,i[2]))
    rs.DeleteObject(my_line1)
    rs.DeleteObject(my_line2)


# action
(f_list, c_list, p_list, stair_l) = make_domino(levels, thick, hgt, A, f_size, f_height, xcol, ycol)
make_facade1(A, xcol, ycol, xcor, ycor, zcor, dist_x, dist_z, f_size, f_height, levels, hgt, thick)
make_facade2(A, xcol, ycol, 0.2, ycor, 0.7, dist_x, dist_z, f_size, f_height, levels, hgt, thick)



# make layers
rs.AddLayer("foundation")
rs.LayerColor("foundation", (220,60,60))
rs.ObjectLayer(f_list, "foundation")
rs.AddLayer("columns")
rs.LayerColor("columns", (60,220,220))
rs.ObjectLayer(c_list, "columns")
rs.AddLayer("plates")
rs.LayerColor("plates", (60,220,60))
rs.ObjectLayer(p_list, "plates")
rs.AddLayer("stair")
rs.LayerColor("stair", (150,150,220))
rs.ObjectLayer(stair_l, "stair")


rs.EnableRedraw(True)

#rs.MoveObject(my_rect, [-B, -f_size/2, f_height+thick])