import rhinoscriptsyntax as rs
import random as ran

# alles loeschen
allobjs = rs.AllObjects()
rs.DeleteObjects(allobjs)


###############################
#Variablen
###############################
A = 5               # Module size (distance between columns)
B = A/3             # Distance of columns to the end of plate
thick = 0.2         # thickness of all slabs
hgt = 2.7           # height of room
xcol = 5            # columns in x direction
ycol = 3            # columns in y direction
levels = 5          # number of floor plates
f_height = 0.5      # f_height = foundation height
f_size = 0.8        # f_size = foundation edge size
level = 0.7         # plattenhoehe
pts = 1000          # anzahl der punkte
y1= 0
y2= A*(ycol-1)
x1= 0
x2= A*(xcol-1)


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

# funktion um box zu erstellen - 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)

# funktion um box zu erstellen - 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)

# funktion um feld von gruendungen zu erstellen
def make_foundations(A=A,f_size=f_size,f_height=f_height, xcol=xcol, ycol=ycol):
    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)

# funktion um feld von saeulen zu erstellen
def make_columns(A=A, level=level, thick=thick, hgt=hgt, xcol=xcol, ycol=ycol):
    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)

def make_terrace(A=A, level=level, thick=thick, hgt=hgt, xcol=xcol, ycol=ycol):
    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)

# funktion um treppe zu erstellen
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.DeleteObjects((s_outline, path))
    return(hull)
rs.EnableRedraw(False)


# building dom-ino
def make_domino(A=A, B=B, thick=thick, hgt=hgt, xcol=xcol, ycol=ycol, levels=levels, f_height=f_height, f_size=f_size, level=level, pts=pts, y1=y1, y2=y2, x1=x1, x2=x2):
    f_list=[]       # liste von gruendungen
    c_list=[]       # liste von saeulen
    p_list=[]       # liste von platten
    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  
    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))
    
    ##################################
    # treppe berechnen
    ##################################
    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
    
    ##################################
    # treppen parameter
    ##################################
    tt=0.3                                                      # treppentiefe
    s_width = 1.2                                               # s_width = treppenbreite
    pod_w = B                                                   # pod_w = laenge von podest
    start = [A*(xcol-2)+B,-(s_width*2+f_size/2), f_height+thick]       # startpunkt treppe
    
    #loop um stiege zu erstellen
    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, th, tt, int(steps/2), thick, s_width))
            stair_l.append(make_podest([start[0]+(steps/2)*tt, start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick))
            stair_l.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_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(True)
"""


# fassade erstellen
for i in range(pts):
    x= ran.uniform((A*(xcol-2)),0)
    y1= y1
    z= ran.uniform(0.7,(levels-1+thick)*(hgt+thick))
    rs.AddPoint([x,y1,z])

for i in range(pts):
    x1= x1
    y= ran.uniform((A*(ycol-2)),0)
    z= ran.uniform(0.7,(levels-1+thick)*(hgt+thick))
    rs.AddPoint([x1,y,z])

for i in range(pts):
    x= ran.uniform((A*(xcol-1)),A)
    y2= y2
    z= ran.uniform(0.7,(levels-1+thick)*(hgt+thick))
    rs.AddPoint([x,y2,z])

for i in range(pts):
    x2= x2
    y= ran.uniform((A*(ycol-1)),A)
    z= ran.uniform(0.7,(levels-1+thick)*(hgt+thick))
    rs.AddPoint([x2,y,z])

my_c = rs.AddSpiral([0,0,0], [0,0,10], 0, 3, 100, 25)
my_pts = rs.DivideCurve(my_c, 20, True, True)


# dom-inos erstellen
for p in my_pts:
    xran = ran.randint(0,100)
    yran = ran.randint(0,200)
    levels = ran.randint(2,7)
    xcol = ran.randint(3,6)
    ycol = ran.randint(3,6)
    (f_list, c_list, p_list, stair_l) = make_domino(levels=levels, xcol=xcol, ycol=ycol)
    rs.MoveObject(f_list+c_list+p_list+stair_l, p)
    rs.DeleteObject(my_c)