import rhinoscriptsyntax as rs
import random as ran

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

A = 5               #A=module sze (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 = 3            #columns in x direction
ycol = 4            #columns in y direction
levels = 7          #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)


rs.EnableRedraw(False)

#funtion to create box at centerpoint
def make_box(insertion = [0,0,0],xsize=10,ysize=10,zsize=10):

    #create box
    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, (-xsize/2, -ysize/2,0))
    rs.MoveObject(box, insertion)
    return(box)

#function to create a field of foundations
def make_foundations(A=5, 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 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)

#make_arc([0,0,0], 2, 0.18, 7.0, 90)


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


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


#fuction domino
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 fo 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):   #only every second floor
                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
    ###########################
    #stairparameters
    ###########################
    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()

#make bruestung

##totalheight
#total_hgt = f_height + levels * (thick + hgt)

##facade
#def make_facade(total_hgt, p_width, p_length):
#    facade_thick = 0.2
#    facade = make_box([0,0,0], p_width, p_length, total_hgt)
#    return facade
#facade = make_facade(total_hgt, p_width, p_length)


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)

def make_multiple_domino(offset=[0, 0, 0], A=A, B=B, thick=thick, hgt=hgt, levels=levels, xcol=xcol, ycol=ycol, f_height=f_height, f_size=f_size):
    translation = rs.VectorCreate(offset, [0, 0, 0])  
    (f_list, c_list, p_list, stair_1) = make_domino(A, B, thick, hgt, levels, xcol, ycol, f_height, f_size)
    
    # move object
    rs.MoveObjects(f_list + c_list + p_list + stair_1, translation)
    return (f_list, c_list, p_list, stair_1)

# Parameter for raster
num_houses_x = 3  #number of houses
num_houses_y = 4  

# distance beteen dom inos
x_spacing = p_width + 8  
y_spacing = p_length + 4

for i in range(num_houses_x):
    for j in range(num_houses_y):
        x_offset = i * x_spacing
        y_offset = j * y_spacing
        make_multiple_domino(offset=[x_offset, y_offset, 0])

"""
#plane=rs.WorldXYPlane()
#rect = rs.AddRectangle(plane, p_width, p_length)
#translation = [-B, -f_size/2, f_height+thick]
#rs.MoveObject(rect, translation) 
#
#Achsabst = 2
#c_len = rs.CurveLength(rect)
#c_div = int(c_len/Achsabst)
#
#pts = rs.DivideCurve(rect, c_div)
##for p in pts:
##    make_box(p, 0.2, 0.2, (hgt+thick)*(levels-1))
#
#for L in range (levels-1):
#    for p in pts:
#        make_box([p[0],p[1], p[2]+L*(hgt+thick)], 0.4-L*0.05, 0.4-L*0.05, hgt+thick)
#"""

import math

width = p_width  
length = p_length  
height = hgt  
num_levels = levels-3  
num_divisions = 100  
amplitude = 1  
frequency = 1  

plane = rs.WorldXYPlane()
rect = rs.AddRectangle(plane, p_width, p_length)  
translation = [-B, -f_size/2, f_height+thick]
rs.MoveObject(rect, translation) 


edges = rs.ExplodeCurves(rect, delete_input=False)  

#start_z = f_height + thick

for edge in edges:
    pts = rs.DivideCurve(edge, num_divisions)

pts = rs.DivideCurve(rect, num_divisions)
#for p in pts:
#    make_box(p, 0.2, 0.2, (hgt+thick)*(levels-1))

for L in range(num_levels):
    scale_factor = 1 - (L / (num_levels * 2))  
    for p in pts:
        x = p[0]
        y = p[1]


        z = amplitude * math.sin(frequency * x + L * math.pi / num_levels)    #start_z

        box_width = 0.3 * scale_factor  
        box_depth = 0.2 * scale_factor  
        box_height = hgt * scale_factor

        make_box([x, y, z + L * (hgt * 2)], box_width, box_depth, box_height)