import rhinoscriptsyntax as rs
import random as ran

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

"""
#set Variables/Parameters
xsize = 8      #xside of box
ysize = 12      #yside of box
zsize = 10      #zside of box

#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]]

rs.AddBox(corners)
"""

#domino variables
A = 5           # Module size (distance between columns)
B = A/3         # Distance of columns to end of plate
thick = 0.2     # thickness of all slabs
hgt = 2.7       # hight of room
xcol = 3        # columns in x direction
ycol = 3        # columns in y direction
levels = 5      # number of floor plates
f_height = 0.5  # foundation hight
f_size = 0.8    # 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 plates (x)
p_length = A*(ycol-1)+f_size                        #length of floor plates (y)

#function to create box at centre point
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)

#make_box([10,-3,-4],zsize=2)

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

#function to create field of foundations
def make_foundation(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([A*i,A*j,0], f_size, f_size, f_height))
    return(fns)

#function to create field of columns
def make_columns(A=5,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([A*i,A*j,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=5,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<7:
                orientation = ori*140
                cls.append(make_arc([i*A+A/2,j*A+A/2,level], A/2, thick, hgt, orientation))
    return(cls)

# function to 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)
rs.EnableRedraw(False)

#make facade
def make_pillar (insertion=[0,0,0], xs=0.1, ys=0.1, zs=ran.randint(5,27)):
    corners = [(0,0,0),(xs,0,0), (xs,ys,0), (0,ys,0),
               (0,0,zs), (xs,0,zs), (xs,ys,zs), (0,ys,zs)]
    box = rs.AddBox(corners)
    rs.MoveObject(box, insertion)
    
def make_facade():
    
    baseline_length= rs.AddLine((-B-0.1, -0.4,thick+f_height ), (-B-0.1, p_length - 0.4, thick+f_height))
    pts_length = rs.DivideCurve(baseline_length,ran.randint(15,45))
    for p in pts_length:
        make_pillar(p)
        
    baseline_width= rs.AddLine([-B,-0.4+p_length,thick+f_height], [p_width-B,-0.4+p_length,thick+f_height])
    pts_width = rs.DivideCurve(baseline_width,ran.randint(10,22))
    for p in pts_width:
        make_pillar(p)
make_facade() 

#building dom-ino
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)
        level = f_height + thick + (i-1)*(hgt+thick)
        if i == 0:
            f_list = make_foundation(A, xcol=xcol, ycol=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))
    #############
    # 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
    #########
    #stair parameters
    #########
    tt = 0.3            #steps size
    s_width = 1.2       #stair width
    pod_w = B           #podest width
    start = [pod_w,-(s_width*2+f_size/2), f_height+thick]   #startpoint of stairs
    
    #loop to create staircase
    stair_1=[]
    for i in range(levels):
        start[2] = f_height+thick+ i*(thick+hgt)           #z-Wert bei jeder Literation 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))
    
    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")

    return(f_list, c_list, p_list, stair_1)



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


#make_domino()

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


for i in range(1):
    pt=rs.GetPoint()
    angel=rs.GetAngle(pt,)
    (f_list, c_list, p_list, stair_1)=make_domino()
    rs.MoveObjects(f_list+c_list+p_list+stair_1, pt)
    rs.RotateObjects(f_list+c_list+p_list+stair_1, pt, angel)
"""


curve=rs.AddSpiral([0,0,0], [0,0,1], 0, 1, 50,10)
pts= rs.DivideCurve(curve,10)

#functiion to create a dominoblock
def make_dominoblock():
    for i in pts:
        kuchen= ran.randint(2,10)
        (f_list, c_list, p_list, stair_1)=make_domino(levels=kuchen)
        rs.MoveObjects(f_list+c_list+p_list+stair_1, i)
    rs.DeleteObject(curve)

#dominoblock=make_dominoblock()
make_domino()


city_curve=rs.AddSpiral([0,0,0], [0,0,1], 0, 5, 500,10)
points=rs.DivideCurve(city_curve,50)