import rhinoscriptsyntax as rs
import random as ran

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

rs.EnableRedraw(False)

# Define global variables
A = 5                             # Base size of each unit
B = A / 3                          # Thickness used for foundations and other structures
thick = 0.2                        # Thickness for some elements like columns and stairs
hgt = 2.7                          # Height of each level
xcol = 3                           # Number of columns in the X direction
ycol = 4                           # Number of columns in the Y direction
levels = 12                        # Number of levels in the structure
f_height = 0.5                     # Height of the foundation elements
f_size = 0.8                       # Size of the foundation elements
rotation_angle = 30                # Rotation angle between each building copy (in degrees)
num_copies = 15                    # Number of building copies

# Center of the structure
center_pt = [A * (xcol - 1) / 2, A * (ycol - 1) / 2, f_height]  
p_width = A * (xcol - 1) + 2 * B                                  
p_length = A * (ycol - 1) + f_size                             

rs.EnableRedraw(False)

# function to create a box (used for columns, podests, etc.)
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 the foundations (1st level)
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 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 create a terrace (an arc-shaped structure)
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 terraces (multiple arcs at different positions)
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 stairs (geometry)
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

# function to create a podest (platform)
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 the whole building structure
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 = []  # Foundation
    c_list = []  # Columns
    p_list = []  # Platforms
    stair_1 = []  # Stairs
    facade_list = []  # Facades

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

    steps = int((hgt + thick) / 0.17)
    if steps % 2:
        steps -= 1
    th = (hgt + thick) / steps
    if th > 0.19:
        steps += 2
        th = (hgt + thick) / steps
    
    tt = 0.3  # Step depth
    s_width = 2  # Stair width
    pod_w = B  # Podest width
    start = [pod_w, -(s_width * 2 + f_size / 2), f_height + thick] 

    for i in range(levels):
        start[2] = f_height + thick + i * (thick + hgt)
        
        if i == levels - 1:  
            stair_1.append(make_podest([start[0] - pod_w, start[1] + s_width, start[2] - thick], pod_w, s_width, thick)) 
        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))

    # Create facades
    facade_height = f_height + (levels * (hgt + thick))  # Calculating facade height based on levels

    # Front facade
    front = rs.AddLine((-1.67, -0.4, 0.5), (p_width - 1.67, -0.4, 0.5))
    pts_front = rs.DivideCurve(front, 11)
    for p in pts_front:
        facade_list.append(make_box([p[0], p[1], 0], 0.2, 0.2, facade_height))

    # Back facade
    depth = 15.8
    back = rs.AddLine((-1.67, -0.4 + depth, 0.5), (p_width - 1.67, -0.4 + depth, 0.5))
    pts_back = rs.DivideCurve(back, 35)
    for p in pts_back:
        facade_list.append(make_box([p[0], p[1], 0], 0.2, 0.2, facade_height))

    # Left facade
    left = rs.AddLine((-B, -0.4, 0.5), (-B, p_length - 0.4, 0.5))
    pts_1 = rs.DivideCurve(left, 70)
    for p in pts_1:
        facade_list.append(make_box([p[0], p[1], 0], 0.2, 0.2, facade_height))

    # Right facade
    right = rs.AddLine((p_width - B, -0.4, 0.5), (p_width - B, p_length - 0.4, 0.5))
    pts_r = rs.DivideCurve(right, 20)
    for p in pts_r:
        facade_list.append(make_box([p[0], p[1], 0], 0.2, 0.2, facade_height))

    return f_list, c_list, p_list, stair_1, facade_list


# Function to duplicate and rotate the building with facades
def duplicate_building(num_copies, rotation_angle):
    f_list, c_list, p_list, stair_1, facade_list = make_domino()

    for i in range(num_copies):
        # Copy the entire building (foundations, columns, platforms, stairs, facades)
        f_list_copy = [rs.CopyObject(obj) for obj in f_list]
        c_list_copy = [rs.CopyObject(obj) for obj in c_list]
        p_list_copy = [rs.CopyObject(obj) for obj in p_list]
        stair_1_copy = [rs.CopyObject(obj) for obj in stair_1]
        facade_copy = [rs.CopyObject(obj) for obj in facade_list]

        # Randomly generate the new position for the building copy
        rand_x = ran.uniform(-80, 130)
        rand_y = ran.uniform(-80, 130)
        rand_z = ran.uniform(0, 0)

        # Move the copied building to a random position
        rs.MoveObject(f_list_copy + c_list_copy + p_list_copy + stair_1_copy + facade_copy, [rand_x, rand_y, rand_z])

        # rotate the building around the center
        angle = rotation_angle * i
        rotation_center = [A * (xcol - 1) / 2, A * (ycol - 1) / 2, 0]  # Center point for rotation
        rs.RotateObject(f_list_copy + c_list_copy + p_list_copy + stair_1_copy + facade_copy, rotation_center, angle)

# function to duplicate and rotate the building
duplicate_building(num_copies, rotation_angle)

rs.EnableRedraw(True)