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

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

num = 20          
num2 = 30         

xsize = p_width/num                                 # degth of boxes          
ysize = -0.3                                        # hight of boxes             
zsize = p_height/num2                               # thickness of boxes
asize = p_width/num                                 # number of boxes in x direction
bsize = 0.5                                         # number of boxes inn y directio 
csize = p_height/num2                               # number of boxes in z direction 
dist = 0           
offs = xsize + dist  
off = zsize + dist  

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]]


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

# function to create a podest
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, treppenhoehe, treppentiefe, steps, thick, s_width):
    pointlist=[start]
    for i in range(steps):
        pointlist.append([pointlist[-1][0],pointlist[-1][1],pointlist[-1][2]+treppenhoehe])
        pointlist.append([pointlist[-1][0]+treppentiefe,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)

#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 fns
    c_list = []         #list of cls
    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=xcol, ycol=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
    treppenhoehe = (hgt+thick)/steps
    if(treppenhoehe>0.19):
        steps = steps+2
        treppenhoehe = (hgt+thick)/steps
    
    ############################################
    # stair parameters
    ############################################
    treppentiefe=0.3            #step size
    s_width = 1.2               #s_width = stair width
    pod_w = B                   #pod_w = 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, treppenhoehe, treppentiefe, int(steps/2), thick, s_width))
            stair_l.append(make_podest([start[0]+(steps/2)*treppentiefe, start[1], start[2]+(steps/2)*treppenhoehe-thick], pod_w, s_width*2, thick))
            stair_l.append(make_stair([start[0]+(steps/2)*treppentiefe, start[1]+s_width, start[2]+(steps/2)*treppenhoehe], treppenhoehe, -treppentiefe, 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()

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

rs.EnableRedraw(False)
############################################
# facade
############################################

y_width = ((ycol-1)*A+f_size)/num

step_size_directions = [[xsize, 0, zsize], [0, y_width, zsize], [xsize, 0, zsize], [0, y_width, zsize], [xsize, 0, zsize]] 
start_points = [[-B, -f_size/2, f_height], [(xcol-1)*A+B, -f_size/2, f_height], [-B , (ycol-1)*A+f_size/2, f_height], [-B, -f_size/2, f_height], [A, -f_size/2, f_height]] 
number_of_blocks_w = [4, 20, 20, 20, 4] 

rs.EnableRedraw(False)

for h in range(len(start_points)): 
    for i in range (number_of_blocks_w[h]):
        for j in range (num2):
            myran = ran.randint(0, 4)
            if (myran >= 2): 
                if h in [0,4]:
                    x_size = step_size_directions[h][0]
                    y_size = ran.uniform(ysize/3, ysize)
                elif h == 1:
                    x_size = ran.uniform(xsize/3, xsize)
                    y_size = step_size_directions[h][1]
                elif h == 2:
                    x_size = step_size_directions[h][0]
                    y_size = -ran.uniform(ysize/3, ysize)
                else:
                    x_size = -ran.uniform(xsize/3, xsize)
                    y_size = step_size_directions[h][1]
                corners = [[0,0,0],[x_size,0,0],[x_size,y_size,0],[0,y_size,0],
                    [0,0,zsize],[x_size,0,zsize],[x_size,y_size,zsize],[0,y_size,zsize]]
                box = rs.AddBox(corners)
                rs.MoveObject(box,(i*step_size_directions[h][0] + start_points[h][0], i*step_size_directions[h][1] + start_points[h][1], j*step_size_directions[h][2] + start_points[h][2]))
                rs.ObjectColor(box,(i*(220/num),j*(60/num),60))

################################################
# rotate buildings
################################################

build_a = rs.AllObjects()

build_b = rs.RotateObjects(build_a, [0,0,0], 60, copy=True)
rs.MoveObjects(build_b, (10,20,0))
build_c = rs.RotateObjects(build_b, [0,0,0], 60, copy=True)
rs.MoveObjects(build_c, (10,20,0))
build_d = rs.RotateObjects(build_c, [0,0,0], 60, copy=True)
rs.MoveObjects(build_d, (10,20,0))
build_e = rs.RotateObjects(build_d, [0,0,0], 60, copy=True)
rs.MoveObjects(build_e, (10,20,0))
build_f = rs.RotateObjects(build_e, [0,0,0], 60, copy=True)
rs.MoveObjects(build_f, (10,20,0))