#DM2 - WS2024-25 - Group Hirschberg 3
#Raphael Wieser 
#ue4 - domino & stairs

#importlibrarys
import rhinoscriptsyntax as rs
import random as ran

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

rs.EnableRedraw(0)  #disables the live view of the code execution

####
# domino variables
####

A = 5                       # A = Module size (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 = 5                    # columns in x direction
ycol = 6                    # 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

####
# derived values
####
center_pt = [A*(xcol-1)/2,A*(ycol-1)/2, 0] #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)

#define function to make box at centerpoint
def make_box(insertion=[0,0,0], xsize=10, ysize=10, zsize=10): 
    #by giving default values in the bracket when definening the func instead of
    #lower down you can only give the first value when executing the func
    
    #setup corners list
    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))   #moves object to center of insert point
    rs.MoveObject(box, insertion)
    return(box)

#define function to make 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)


def make_foundations(A=5.0, f_size=0.8, f_height=0.5, xcol=2, ycol=3):
    fns = [] #create empty list fns (foundations)
    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=[] #create empty list cns (columns)
    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-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)

def make_terrace(A=5.0, level=0.7, thick=0.2, hgt=3.0, xcol=2, ycol=3,):
    cls=[] #create empty list cns (columns)
    for i in range(xcol-1):
        for j in range (ycol-1):
            ori = ran.randint(0,6)  #random integer for orientation
            if ori<4:   #if random integer is smaller than 4 the terrace is generated, bigger than 4 and its not generated
                orientation = ori*90
                cls.append(   make_arc([i*A+A/2, j*A+A/2, level], A/2, thick, hgt, orientation   ))
    return(cls)

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)

#set default stair parameters
steps=8
th=.18
tt=0.3
thick=0.2
s_width=1.2

#buildingdomino
#loop for creating storeys
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)   #z-wert des center_pt uebergeben
        level = f_height+thick+(i-1)*(hgt+thick)
        if i==0:
            f_list.extend(   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)  #set height for terrace
    c_list.extend(   make_terrace(A, level, thick, hgt, xcol, ycol)   )
    #################################################################################
    # calculate stair values
    #################################################################################
    steps=int((hgt+thick)/0.17)
    if steps%2:  #check if steps is an even number
        steps=steps-1
    th=(hgt+thick)/steps
    if (th>0.19):
        steps= steps+2
        th=(hgt+thick)/steps
    #################################################################################
    # stair parameters
    #################################################################################
    tt=0.3      #step size
    s_width     #s_width = stair width
    pod_w = B   #pod_w = podest width - distance from columns to free plate edges
    start = [pod_w,-(s_width*2+f_size/2), f_height+thick]    #startpoint of stair
    
    #loop to create stair case
    #make_stair(start, th, tt, steps, thick, s_width)
    stair_l=[]
    for i in range (levels):
        start[2]=f_height+thick + i*(thick+hgt) #z-value that gets reset every iteration
        
        if i==levels-1:  #last podest
            stair_l.append(   make_podest([start[0] -pod_w, start[1]+s_width, start[2]-thick], pod_w, s_width, thick)   ) #special podest (uppermost, last podest)
        else:
            stair_l.append(   make_podest([start[0] -pod_w, start[1], start[2]-thick], pod_w, s_width*2, thick)   )   #wide podest
            stair_l.append(   make_stair(start, th, tt, int(steps/2), thick, s_width)   )    #half a stair run
            stair_l.append(   make_podest([start[0] +(steps/2)*tt, start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick)   ) #intermediate podest
            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)   )    #half a stair run
    
    ####second flight of stairs
        #loop to create stair case
    #make_stair(start, th, tt, steps, thick, s_width)
    
    start = [p_width-pod_w*2-steps*tt,-(s_width*2+f_size/2)+p_length+s_width*2, f_height+thick]
    for i in range (levels):
        start[2]=f_height+thick + i*(thick+hgt) #z-value that gets reset every iteration
        
        if i==levels-1:  #last podest
            stair_l.append(   make_podest([start[0] -pod_w, start[1]+2*s_width, start[2]-thick], pod_w, -s_width*2, thick)   ) #special podest (uppermost, last podest)
        else:
            stair_l.append(   make_podest([start[0] -pod_w, start[1], start[2]-thick], pod_w, s_width*2, thick)   )   #wide podest
            stair_l.append(   make_stair(start, th, tt, int(steps/2), thick, s_width)   )    #half a stair run
            stair_l.append(   make_podest([start[0] +(steps/2)*tt, start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick)   ) #intermediate podest
            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)   )    #half a stair run
    return(f_list, c_list, p_list, stair_l)
    
    
    return(f_list, c_list, p_list, stair_l)



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


#create layer
rs.AddLayer("foundation")
#change color of existing layer
rs.LayerColor("foundation",(220,60,60)   )
#assign objects to existing layer
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(c_list,"plates")

#create city
for i in range(5):
    xran = ran.randint(200,500)
    yran = ran.randint(200,500)
    zran = 0
    angle= ran.uniform(0,180)
    levels= ran.randint(3,11)
    (f_list, c_list, p_list, stair_1)= make_domino(levels=levels)
    rs.MoveObject(f_list+ c_list+ p_list+ stair_1, [xran,yran,zran])
    rs.RotateObject(f_list+ c_list+ p_list+ stair_1, [xran,yran,zran], angle)


curv= rs.AddSpiral((0,0,0), (0,0,1), 0, 2, 200, 100)
pts= rs.DivideCurve(curv, 7)

for p in 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(curv, p)
    normal= rs.CurveTangent(curv, param)
    angle= rs.Angle([0,0,0], normal) [0]
    rs.RotateObject(f_list+ c_list+ p_list+ stair_1, p, angle)


"""
#defineletter
def make_letter(insertion=(0,0,0), letter="1", xsize=10, ysize=10, zsize=10):
    mytext = rs.AddText(letter,(0,zsize,0),height=zsize, font = "Arial")
    crv = rs.ExplodeText(mytext,True)[0]
    path= rs.AddLine([0,0,0],[0,0,ysize])
    obj = rs.ExtrudeCurve(crv, path)
    rs.CapPlanarHoles(obj)
    rs.DeleteObjects([crv, path])
    rs.RotateObject(obj, [0,0,0], 90,[1,0,0])
    bbx = rs.BoundingBox(obj)
    #print bbx
    #rs.AddPolyline(bbx)
    rs.MoveObject(obj, (-bbx[0][0], -bbx[0][1], -bbx[0][2]))
    scalex= xsize/rs.Distance(bbx[0],bbx[1])
    scalez= zsize/rs.Distance(bbx[0],bbx[4])
    rs.ScaleObject(obj,(0,0,0),(scalex,1,scalez))
    rs.MoveObject(obj, (-xsize/2, -ysize/2,0))
    rs.MoveObject(obj, insertion)


#buildingfacade
div1=5
div2=1

for i in range(levels-1):
    for j in range(div1-1):

        center_pt[2] = f_height+i*(2*thick+hgt)   #z-wert des center_pt uebergeben
        center_pt[1] = -thick*2   #y-wert des center_pt uebergeben
        center_pt[0] = -f_size+(f_size+A*xcol+thick*2)/div1*j   #x-wert des center_pt uebergeben
        level = f_height+thick+(i-1)*(hgt+thick)
        make_letter(insertion=center_pt,letter="v", xsize=p_width/div1, ysize=thick/2, zsize=hgt+thick,)

for i in range(levels-1):
    for j in range(div1-1):
        center_pt[2] = f_height+i*(2*thick+hgt)   #z-wert des center_pt uebergeben
        center_pt[1] = -thick/2*5+p_length   #y-wert des center_pt uebergeben
        center_pt[0] = -f_size+(f_size+A*xcol+thick*2)/div1*j   #x-wert des center_pt uebergeben
        level = f_height+thick+(i-1)*(hgt+thick)
        make_letter(insertion=center_pt,letter="Y", xsize=p_width/div1, ysize=thick/2, zsize=hgt+thick,)
"""


rs.EnableRedraw(1)  #enables the live view of the code execution