import rhinoscriptsyntax as rs
import random as ran

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

##############################################################
# derived variables 
##############################################################

A= 5                #A-Module Size (distance between columns)
B= A/3              #B = Distance of Columns to end of plate
thick= 0.3          #thickness of all slabs
hgt= 3.0            #height of room
xcol= 7             #columns in x-direction
ycol= 4             #columns in y-direction
levels= 6           #number of floor plates
f_size= 0.8         #f_size = foundation edge size
f_height= 0.5       #f_height = foundation height

###############################################################
# 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                          # length of floor plate (y)
p_hight = (levels-1)*(hgt+thick)+thick



# funtion to create box at centerpoint
def make_box(insertion= [0,0,0],xsize=10,ysize=10,zsize=10):
    #createbox
        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):
    #createbox
        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 field of foundations
def make_foundation(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)

# make field of columns
def make_column(A=5.0, level=0.5, 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)

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

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


rs.EnableRedraw(False)

#build dom-ino
def make_domino(A=A, B=B, thick=thick, hgt=hgt, levels=levels, xcol=xcol, ycol=ycol, f_height=f_height, f_size=f_size):
    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_foundation(A,f_size, f_height, xcol, ycol)
        else:
            c_list.extend(make_column(A, level, thick, hgt, xcol, ycol))  
            if(i%2):
                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               #pod_w = depth of landing
    start = [0,-s_width*2+f_size/2, f_height+thick]     #startpoint of stair
    
    #make_stair(start, th, tt, steps, thick, s_width)
    for i in range(levels):
        start[2] = f_height+thick + i*(thick+hgt)   #z-wert bei jeder Iteration neu gesetzt
        if i==levels-1:                     #letztes Podest
            make_podest([start[0]-pod_w, start[1]+s_width, start[2]-thick], pod_w, s_width, thick)      #Sonderpodest
        else:
            make_podest([start[0]-pod_w, start[1], start[2]-thick], pod_w, s_width*2, thick)      #Sonderpodest
            make_stair(start, th, tt, int(steps/2), thick, s_width)
            make_podest([start[0]+(steps/2)*tt, start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick)      #Sonderpodest
            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)

(f_list, c_list, p_list)=make_domino()


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

rs.AddLayer("column")
rs.LayerColor("column", (60,120,60))
rs.ObjectLayer(c_list, "column")

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

#facade
bsize= 0.2                 
dist = 0.5                 
offs = bsize + dist
num=14
numx = int(p_width/offs)+1    
numz = int(p_hight/offs)

corners = [(0,0,0), (bsize,0,0), (bsize,bsize,0), (0,bsize,0),
           (0,0,bsize), (bsize,0,bsize), (bsize,bsize,bsize), (0,bsize,bsize)]

"""
#level01
rs.EnableRedraw(False)
for i in range(numx):
    for j in range(numz):
            if ((i+j)%3) and ((i-j)%4):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (-B,0,1))


rs.EnableRedraw(False)
for i in range (numx):
    for j in range (numz): 
            if ((i-j)%3)and((i-j)%4):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box, (i*offs,0,j*offs))
                    rs.MoveObject(box,(5,0,1))


rs.EnableRedraw(False)
for i in range (num):
    for j in range (num): 
            if ((i-j)%3)and((i-j)%4):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box,(i*offs,0,j*offs))
                    rs.MoveObject(box,(-B,0,1))
"""

#level02
rs.EnableRedraw(False)
for i in range(numx):
    for j in range(numz):
            if ((i+j)%6) and ((i-j)%2):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (-B,5,1))

"""
rs.EnableRedraw(False)
for i in range(num):
    for j in range(num):
            if ((i+j)%3) and ((i-j)%4):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (10,5,1))

rs.EnableRedraw(False)
for i in range(num):
    for j in range(num):
            if ((i+j)%3) and ((i-j)%4):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (15,5,1))


#level03
rs.EnableRedraw(False)
for i in range(numx):
    for j in range(numz):
            if ((i-j)%3) and ((i+j)%4):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (-B,10,1))


rs.EnableRedraw(False)
for i in range (num):
    for j in range (num): 
            if ((i-j)%3)and((i-j)%4):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box, (i*offs,0,j*offs))
                    rs.MoveObject(box,(5,10,1))

rs.EnableRedraw(False)
for i in range (num):
    for j in range (num): 
            if ((i-j)%3)and((i-j)%4):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box, (i*offs,0,j*offs))
                    rs.MoveObject(box,(20,10,1))
"""

#level04
rs.EnableRedraw(False)
for i in range(numx):
    for j in range(numz):
            if ((i+j)%2) and ((i-j)%2):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (-B,15,1))

"""
rs.EnableRedraw(False)
for i in range(num):
    for j in range(num):
            if ((i+j)%3) and ((i-j)%4):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (10,15,1))

rs.EnableRedraw(False)
for i in range(num):
    for j in range(num):
            if ((i+j)%3) and ((i-j)%4):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs,0,j*offs))
                rs.MoveObject(box, (15,15,1))
"""