import rhinoscriptsyntax as rs
import random as ran

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

#-------------------------------------------------------------------------------
# dom-ino variables:
#-------------------------------------------------------------------------------
A = 5.0             # 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      # foundation height
f_size = 0.8        # foundation edge size

#-------------------------------------------------------------------------------
# derived values:
#-------------------------------------------------------------------------------
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


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 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_height, 0.5))
    return(fns)
                        

# function to create a field of columns
def make_columns (A = 5, level=0.7, thick=0.2, hgt=3,  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)
    

# building dom-ino 
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))
    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")


# facade west
facadexsize=0.2
facadeysize=0.2
facade_length = 28
facade_hgt = 2.8
place = A*(xcol-1)+B+facadexsize

def make_facade (insertion=[0,0,0], facadexsize=0.2, facadeysize=0.2, facadezsize=(facade_hgt+thick)*(levels-1), dis=0.4):
    facade = []
    for i in range (facade_length):
        facade.append(make_box([place,i*dis-f_size/2,f_height], facadexsize, facadeysize, facadezsize))
    return (facade)
        
make_facade()

# facade east
facadexsize=0.2
facadeysize=0.2
facade_length = 12
facade_hgt = 1.4
place = A*(xcol-2.75)+B+facadexsize

def make_facade (insertion=[0,0,0], facadexsize=0.2, facadeysize=0.2, facadezsize=(facade_hgt+thick)*(levels-1), dis=0.9857):
    facade = []
    for i in range (facade_length):
        facade.append(make_box([place,i*dis-f_size/2,f_height], facadexsize, facadeysize, facadezsize))
    return (facade)
        
make_facade()