import rhinoscriptsyntax as rs
import random as ran

allobjs = rs.AllObjects()
rs.DeleteObjects(allobjs)
rs.EnableRedraw(False)

A= 5
B= A/3
thick= 0.2
hgt= 3.0
xcol= 5
ycol= 4
levels= 5
f_size= 0.8
f_height= 0.5


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
p_height= (levels-1)*(hgt+thick)+f_height+thick

def make_box(insertion= [0,0,0],xsize=10,ysize=10,zsize=10):
    #createbox
        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)
        
#create box at corner point
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)
#
  
  # make field of foundations
def make_foundation(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)
#

  # make field of columns
def make_column(A=5.0, level=0.5, 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)
#

  # build domino
#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_foundation(A,f_size, f_height, xcol, ycol)
#    else:
#        c_list.extend(make_column(A, level, thick, hgt, xcol, ycol))  
#    p_list.append(make_box(center_pt, p_width, p_length, thick))
#    
#rs.AddLayer("foundation")
##rs.LayerColor("foundation", (220,60,60))
#rs.ObjectLayer(f_list, "foundation")
#
#rs.AddLayer("column")
##rs.LayerColor("column", (60,220,60))
#rs.ObjectLayer(c_list, "column")
#
#rs.AddLayer("plate")
##rs.LayerColor("plate", (60,60,220))
#rs.ObjectLayer(p_list, "plate")

#facade


#level01
rs.EnableRedraw(False)

def make_fassade(p_width=p_width, p_length=p_length, p_height=p_height, center_pt=center_pt):
    bsize= 0.2                 
    dist = 0.5                 
    offs = bsize + dist
    numx = int(p_width/(bsize+dist))
    #numx = 5*xcol
    numz = int(p_height/(bsize+dist)) 
    numy = int(p_length/(bsize+dist))
    
    corners = [(0,0,0), (bsize,0,0), (bsize,bsize,0), (0,bsize,0),
           (0,0,bsize), (bsize,0,bsize), (bsize,bsize,bsize), (0,bsize,bsize)]
           
    fas_list=[]
    for i in range(numx):
        for j in range(numz):
                if ((i+j)%4) and ((i-j)%6):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box, (i*offs-B,-f_size/2-bsize,j*offs+f_height))
                    rs.MoveObject(box, (0,0,0))
                    fas_list.append(box)
    
    for i in range(numx):
        for j in range(numz):
                if ((i+j)%4) and ((i-j)%6):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box, (i*offs-B,-f_size/2-bsize,j*offs+f_height))
                    rs.RotateObject(box, center_pt, 180)
                    fas_list.append(box)
    
    for i in range(numy):
       for j in range(numz):
                if ((i+j)%4) and ((i-j)%6):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box, (-B-bsize,i*offs-f_size/2,j*offs+f_height))
                    fas_list.append(box)
    
    for i in range(numy):
       for j in range(numz):
                if ((i+j)%4) and ((i-j)%6):
                    box = rs.AddBox(corners)
                    rs.MoveObject(box, (-B-bsize,i*offs-f_size/2,j*offs+f_height))
                    rs.RotateObject(box, center_pt, 180)
                    fas_list.append(box)
    return(fas_list)



"""
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_terrase(A=5.0,level=0.7, thick=0.2, hgt=3, 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)
"""
tt=0.3
s_width=1.2
pod_w=B 
th=0.19

#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[0][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)


#build domino
def make_domino (A=A, thick=thick, hgt=hgt, levels=levels, xcol=xcol, ycol=ycol, f_height=f_height, f_size=f_size, th=th, tt=tt):
    f_list=[]   #list of foundations
    c_list=[]   #list of columns
    p_list=[]   #list of plates
    
    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
    p_height= (levels-1)*(hgt+thick)+f_height+thick
    
    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.extend(make_foundation(A, f_size, f_height,xcol, ycol))
        else:
            c_list.extend(make_column(A, level, thick, hgt, xcol,ycol))
            #c_list.extend(make_terrase(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_terrase(A, level, thick, hgt, xcol,ycol))
    
    (fas_list)=make_fassade(p_width=p_width,p_length=p_length, p_height=p_height, center_pt=center_pt)

    #calculator stair values
    steps=int((hgt+thick)/0.15)
    if steps%2:
        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=1.2     #stair witdth
    pod_w=B 
    start = [+pod_w,-(s_width*2+f_size/2), p_height+thick] #start point of stair

    #loop staircase
    stair_list = []
    for i in range(levels):
        start[2]= f_height+thick+i*(thick+hgt)
        if i ==levels-1:      
    
    #letzets podest
            stair_list.append(make_podest([start[0]-pod_w, start[1]+s_width, start[2]-thick], pod_w, s_width, thick)) # sonderpodest
        else:
            stair_list.append(make_podest([start[0]-pod_w,start[1], start[2]-thick], pod_w, s_width*2, thick))
            stair_list.append(make_stair(start, th, tt, int(steps/2), thick, s_width))
            stair_list.append(make_podest([start[0]+(steps/2)*tt,start[1], start[2]+(steps/2)*th-thick], pod_w, s_width*2, thick))
            stair_list.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_list,fas_list)    
    #facade


#
#(f_list, c_list, p_list, stair_list) = make_domino()

"""
#spiral
for i in range(5):
        pt=rs.GetPoint()
        angel=rs.GetAngle(pt,)
        (f_list, c_list, p_list, stair_list)=make_domino()
        rs.MoveObjects(f_list+c_list+p_list+stair_list,(pt))
        rs.RotateObjects(f_list+c_list+p_list+stair_list, pt, angel)

for i in range(10):
        xran=ran.randint(0,200)
        yran=ran.randint(0,200)
        angel=ran.uniform(0,180)
        (f_list, c_list, p_list, stair_list,fas_list)=make_domino()
        rs.RotateObjects(f_list+ c_list+ p_list+ stair_list+fas_list, (0,0,0), angel)
        rs.MoveObjects(f_list+ c_list+ p_list+ stair_list+fas_list,(xran,yran,0))
"""

# Initialize lists
f_list = []
c_list = []
p_list = []
fa_list = []

#for i in range(levels):
#    center_pt[2] = f_height + i * (thick + hgt)
#    level = f_height + thick + (i - 1) * (hgt + thick)
#    
#    # Create foundations at the first level
#    if i == 0:
#        f_list = make_foundation(A, f_size, f_height, xcol, ycol)
#    else:
#        # Create columns and add them to the list
#        c_list.extend(make_column(A, level, thick, hgt, xcol, ycol))
#    
#    # Create plates (floors)
#    p_list.append(make_box(center_pt, p_width, p_length, thick))
"""

cornerl_pt = [p_width-B-0.1, A*(ycol-1)/2, 0]
corner2_pt = [A*(xcol-1)/2, p_length-0.3, 0]
"""

curve = rs.AddCurve([[-40,-30,0], [30,80,0], [100,-10,0], [160,70,0]], 3)
crv_pts = rs.DivideCurve(curve, 20)


for p in crv_pts:
    levels = ran.randint(3,10)
    xcol= ran.randint(2,5)
    (f_list, c_list, p_list, stair_l, fas_list)= make_domino(levels = levels, xcol=xcol)
    rs.MoveObjects(f_list, p)
    rs.MoveObjects(c_list, p)
    rs.MoveObjects(p_list, p)
    rs.MoveObjects(stair_l, p)
    rs.MoveObjects(fas_list, p)

rs.DeleteObject(curve)