import rhinoscriptsyntax as rs
import random as ran

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

# variables and values
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 = 6                # columns in y direction
levels = 10             # number of levels
f_height = 0.6          # foundation height
f_size = 0.8            # foundation edge size
num_facs = 300          # number of facade elements

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)

rs.EnableRedraw(enable=False)

# function to creat a box
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 foundations on the edges
def make_foundations(A=5, f_size=0.8, f_height=0.8, xcol=4, ycol=6):
    fns=[]
    for i in range(xcol):
        for j in range(ycol):
            if i == 0 or i == xcol-1 or j == 0 or j == ycol-1:
                fns.append(make_box([i*A, j*A, 0], f_size, f_size, f_height))
    return(fns)

# function to create columns on the edges
def make_columns(A=5, level=0.7, thick=0.2, hgt=3.0, xcol=4, ycol=6):
    cls = []
    for i in range(xcol):
        for j in range(ycol):
            if i == 0 or i == xcol-1 or j == 0 or j == ycol-1:
                cls.append(make_box([i*A, j*A, level], thick, thick, hgt))
    return cls

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


# create a modified polyline around plate with an opening for stairs
def create_polyline_around_plate(center, p_width, p_length):
    # define four points around the center, creating an open side for stairs
    corner1 = [center[0] - p_width / 2, center[1] - p_length / 2, center[2]]
    corner2 = [center[0] + p_width / 2, center[1] - p_length / 2, center[2]]
    corner3 = [center[0] + p_width / 2, center[1] + p_length / 2, center[2]]
    corner4 = [center[0] - p_width / 2, center[1] + p_length / 2, center[2]]
    # define a polyline with an open edge for stairs on one side (remove corner1-corner2)
    polyline_points = [corner2, corner3, corner4]
    polyline = rs.AddPolyline(polyline_points + [corner1])  
    return polyline

# facade
def create_polyline_around_plate2(polyline, num_facs=250):
    fas_list=[]
    r_length = rs.CurveLength(polyline)
    my_pts = rs.DivideCurve(polyline, num_facs)
    for pt in my_pts:
        width = ran.uniform(0.2, 0.6)   
        depth = ran.uniform(0.2, 0.6)   
        height = ran.uniform(1.0, (levels*hgt)+levels)  
        fas_list.append(make_box([pt[0], pt[1], 0], width, depth, height))
    return(fas_list)

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))
    polyline = create_polyline_around_plate(center_pt, p_width, p_length)
    fas_list = create_polyline_around_plate2(polyline, num_facs=250)
    rs.DeleteObject(polyline)

    ##############################################
    # 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, fas_list


# make curve
point0 = (0,0,0)
point1 = (0,0,10)
pitch = 1
turns = 3
radius0 = 50.0
radius1 = 200.0
spiral = rs.AddSpiral(point0, point1, pitch, turns, radius0, radius1)
spiral_points = rs.DivideCurve(spiral, 30, create_points=False, return_points=True)
rs.DeleteObject(spiral)

# create more buildings
for i, pt in enumerate(spiral_points):
    levels = ran.randint(2, 7)
    f_list, c_list, p_list, stair_1, fas_list = make_domino(levels=levels)
    rs.MoveObjects(f_list + c_list + p_list + stair_1 + fas_list, pt)

# create layers
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.AddLayer("facade")
rs.LayerColor("facade", (0,255,255))
rs.ObjectLayer("fas_list", "facade")
rs.AddLayer("stairs")
rs.LayerColor("stairs", (63, 0, 255))  
rs.ObjectLayer(stair_1, "stairs")       

rs.EnableRedraw(True)