import rhinoscriptsyntax as rs
import scriptcontext as sc
import random as ran

allobjs = rs.AllObjects()
rs.DeleteObjects(allobjs)

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

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

# 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]]
#rs.AddBox(corners)

# create a box
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)

# make_box([10,-3,0])

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

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

# Builing Dom-ino
f_list = []
c_list = []
p_list = []
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))
    p_list.append(make_box(center_pt, p_width, p_length, thick))

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

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

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