import rhinoscriptsyntax as rs
import random as ran
rs.EnableRedraw(False)

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

################################################
# dom-ino varibles 
################################################

A =8                                                # A = module size (distance between columns)
B = A/3                                             # B = distance of columns to end of plate
thick = 0.18                                        # thickness of all slabs
hgt = 2.7                                           # height of room
xcol = 2                                            # collums in x direction
ycol = 3                                            # collums in y direction
levels = 4                                          # number of floor plates
f_height = 0.5                                      # f_hight = foundation height
f_size = 0.9                                        # f_size = 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 plate (x)
p_length = A*(ycol-1) + f_size                      #lenght of ffloor plate (y)

# create pox at centerpoint
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)

#create box at cornerpoint
def make_podest(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, 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=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)

# function to a 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)

# function to create a 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)
    rs.DeleteObject(s_outline)
    return(hull)

rs.EnableRedraw(False)

# building 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_foundations(A, f_size, f_height, xcol, ycol)
        else:
            c_list.extend(make_columns(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    # s_width = stair width
    pod_w = B        # depth of landing
    start = [pod_w,-(s_width*2+f_size/2), f_height+thick]     #startpoint of stair
    
    #loop to create staircase
    stair_l = []
    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
            stair_l.append(make_podest([start[0]-pod_w, start[1]+s_width,start[2]-thick], pod_w, s_width, thick)) #sonderpodest
        else:
            stair_l.append(make_podest([start[0]-pod_w, start[1],start[2]-thick], pod_w, s_width*2, thick))
            stair_l.append(make_stair(start, th, tt, int(steps/2), thick, s_width))
            stair_l.append(make_podest([start[0]+(steps/2)*tt, start[1],start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick))
            stair_l.append(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, stair_l)

(f_list, c_list, p_list, stair_l) = make_domino()

make_domino()


################################################
#facade 
################################################

xnum = 15                                          # number of boxes in x direction
ynum = 20                                          # number of boxes in y direction
znum = 10                                          # number of boxes in z direction
xsize =  thick                                     # degth of boxes
zsize = (((levels-1)*hgt)+(levels*thick))/znum     # hight of boxes
ysize = (2*A+f_size)/ynum                          # thickness of boxes
x_wth = (A+2*B)/xnum
y_wth = xsize

# first side
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)]

# left
for i in range(ynum):     
    for j in range(znum):
        my_ran = ran.randint(0,9)
        if (my_ran >= 3):
            box = rs.AddBox(corners)
            rs.MoveObject(box, (0,i*ysize,j*zsize))
            rs.MoveObject(box, ((-B-xsize), (-f_size/2),(f_height)))

# right
for i in range(ynum):     
    for j in range(znum):
        my_ran = ran.randint(0,9)
        if (my_ran >= 3):
            box = rs.AddBox(corners)
            rs.MoveObject(box, (0,i*ysize,j*zsize))
            rs.MoveObject(box, ((A+B), (-f_size/2),(f_height)))

# second side
corners = [(0,0,0), (x_wth,0,0), (x_wth,y_wth,0), (0,y_wth,0),
           (0,0,zsize), (x_wth,0,zsize), (x_wth,y_wth,zsize), (0,y_wth,zsize)]

# right
"""
for i in range(xnum):
    for j in range(znum):
        my_ran = ran.randint(0,9)
        if (my_ran >= 3):
            box = rs.AddBox(corners)
            rs.MoveObject(box, (i*x_wth,0,j*zsize))
            rs.MoveObject(box, (-B ,(-f_size/2)-y_wth,(f_height)))
"""
# left
for i in range(xnum):
    for j in range(znum):
        my_ran = ran.randint(0,9)
        if (my_ran >= 3):
            box = rs.AddBox(corners)
            rs.MoveObject(box, (i*x_wth,0,j*zsize))
            rs.MoveObject(box, (-B ,(A*2+f_size/2),(f_height)))

#########################################
# spiral
#########################################


my_spi = rs.AddSpiral([0,0,0],(0,0,1), 0, 2, 60, 120)
#startpoint, endpoint, height, rotation, radius1, radius 2
my_s = rs.MoveObject(my_spi, center_pt)
rs.MoveObject(my_s, (0,15,-9))

my_pts = rs.DivideCurve(my_spi, 30, False, True) 

for p in my_pts: 
    (f_list, c_list, p_list, stair_1) = make_domino()
    rs.MoveObject((f_list+c_list+p_list+stair_1),p)
    param= rs.CurveClosestPoint(my_spi, p)
    normal= rs.CurveTangent(my_spi, param)
    angle= rs.Angle([0,0,0], normal)[0]
    rs.RotateObjects(f_list+c_list+p_list+stair_1, p, angle)