import rhinoscriptsyntax as rs
import random as ran

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

rs.EnableRedraw(False)

### variables
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
hgt = 2.7           # height of room
xcol = 2            # 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
fac_width = 0.1     # width of facade blocks
fac_depth = 0.1     # depth of facade blocks
fac_height = hgt    # height of facade blocks
fac_num = 8         # number of facade pieces
fac_dist = 0.2       # 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)

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 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=2.7, 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 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)

# building dom-ino
f_list=[]               # list of foundations
c_list=[]               # list of columns
p_list=[]               # list of plates
w_list=[]               # list of facade

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=2, ycol=3))
    p_list.append(make_box(center_pt, p_width, p_length, thick))
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,105,45))
rs.ObjectLayer(w_list, "facade")

rs.EnableRedraw(True)