import rhinoscriptsyntax as rs
import random as ran

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

###########################################
# Domino Variables
###########################################
A = 5                          # 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 = 4                       # columns in x direction
ycol = 5                       # columns in y direction
levels = 4                     # number of floor plates
f_height = 0.5                 # f_height = foundation height
f_size = 0.8                   # 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               # length of floor plate (y)


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


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,0,0], [-(rad-thick),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 make 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/4,j*A+A/4,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))
        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 parameter
    #############################################
    tt=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_1 = []
    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_1.append(make_podest([start[0]-pod_w, start[1]+s_width, start[2]-thick], pod_w, s_width, thick))  # sonderpodest
        else:
            stair_1.append(make_podest([start[0]-pod_w, start[1], start[2]-thick], pod_w, s_width*2, thick))
            stair_1.append(make_stair(start, th, tt, int(steps/2), thick, s_width))
            stair_1.append(make_podest([start[0]+(steps/2)*tt, start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick))
            stair_1.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_1)

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


rs.AddLayer("foundation")
rs.LayerColor("foundation", (100,50,50))
rs.ObjectLayer(f_list, "foundation")
rs.AddLayer("columns")
rs.LayerColor("columns", (50,100,50))
rs.ObjectLayer(c_list, "columns")
rs.AddLayer("plates")
rs.LayerColor("plates", (50,50,100))
rs.ObjectLayer(p_list, "plates")

# facade 1
width= 0.3
depth= 17
dist= 0.5
#f_length = 22
f_length = int(p_length/dist)+1
off = A*(xcol-2.7)+B+depth/2 

def make_facade(insertion= [0,0,0] ,depth=0.3, width=0.1, height=(levels-0.7)*(hgt+thick)+thick, dist=0.5):
    fcd = []
    f_length = int(p_length/dist)+1
    for i in range(f_length):
        fcd.append(make_box([off,i*dist-f_size/2,f_height], depth, width, height))
    return(fcd)

make_facade()

# facade 2
width= 0.3
depth= 16.5
f_length = 33
off2 = -B+6.7 

def make_facade(insertion= [0,0,0], depth=0.7, width=0.1, height=(levels-1.03)*(hgt+thick)+thick, dist=0.2):
    fcd = []
    f_length = int(p_length/dist/1)
    for i in range(f_length):
        fcd.append(make_box([off2,i*dist-f_size/2,f_height], depth, width, height))
    return(fcd)

make_facade()

# facade 2 part 2
width= 0.3
depth= 17
f_length = 22
off2 = -B

def make_facade(insertion= [0,0,0], depth=0.3, width=0.1, height=(levels-0.7)*(hgt+thick)+thick, dist=0.4):
    fcd = []
    f_length = int(p_length/dist/1)+1
    for i in range(f_length):
        fcd.append(make_box([off2,i*dist-f_size/2,f_height], depth, width, height))
    return(fcd)

make_facade()


# create dom-ino city
"""
plane = rs.WorldXYPlane()
#f = make_domino()
for i in range(30):
    xran = ran.randint(-100,200)
    yran = ran.randint(-100,200)
    stock = ran.randint(2,7)
    angle = ran.uniform(0,180)
    (f_list, c_list, p_list, stair_1) = make_domino(levels=stock)
    rs.MoveObject(f_list+c_list+ p_list+ stair_1,[xran,yran,0])
    rs.RotateObjects(f_list+c_list+ p_list+ stair_1,[xran,yran,0], angle)
"""

# make buildings follow a curve
crv = rs.AddSpiral((0,0,0), (0,0,1), 0, 2, 150, 10)
my_pts = rs.DivideCurve(crv, 60, True, 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(crv,p)
   normal = rs.CurveTangent(crv,param)
   angle2 = ran.uniform(0,180)
   rs.RotateObjects(f_list+c_list+ p_list+stair_1, p, angle2)