import rhinoscriptsyntax as rs
import random as ran

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

"""
#set Variables/Parameters
xsize = 8      #xside of box
ysize = 12      #yside of box
zsize = 10      #zside of box

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

#domino variables
A = 5           # Module size (distance between columns)
B = A/3         # Distance of columns to end of plate
thick = 0.2    # thickness of all slabs
hgt = 2.7       # hight of room
xcol = 2        # columns in x direction
ycol = 3        # columns in y direction
levels = 3      # number of floor plates
f_height = 0.5  # foundation hight
f_size = 0.8    # 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 plates (x)
p_length = A*(ycol-1)+f_size                #length of floor plates (y)


def make_box(insertion,xsize=10,ysize=10,zsize=10):
    #create 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)

#make_box([10,-3,-4],zsize=2)

#function to create field of foundations
def make_foundation(A=5,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([A*i,A*j,0], f_size, f_size, f_height))
    return(fns)

#function to create field of columns
def make_columns(A=5,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([A*i,A*j,level], thick, thick, hgt))
    return(cls)

#building dom-ino
for i in range(levels):
    f_list=[]           #list of foundations
    c_list=[]           #list of columns
    p_list=[]           #list of plates
    center_pt[2] = f_height + i*(thick+hgt)
    level = f_height + thick + (i-1)*(hgt+thick)
    if i == 0:
        f_list = make_foundation(A, xcol=xcol, ycol=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("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")