# import Rhino Commands
import rhinoscriptsyntax as rs
import random as ran

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

#Set Variables
xsize = 10
ysize = 12
zsize = 5

#create a 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                           # A = Module size (distance between columns)
B = A/3                         # B = Distance of columns to end of plate
thick = 0.2                    # thickness of all slabs
hgt = 2.7                       # height of room
xcol = 3                        # columns in x direction 
ycol = 2                        # columns in y direction
levels = 4                      # 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)
p_height = (hgt+thick)*(levels-1)+thick
#center = [int(p_width/asize)/2,int(p_length/asize)/2,0]
#center = [p_width/2,p_length/2,0]

#facade
#num = 42          
#num2 = 12 

offs = xsize = asize = 0.2            
ysize = 0.1 
off = zsize = csize = 0.5
bsize = -0.1 
    

num = int(p_width/asize)
offs = xsize = asize = p_width/num

num2 = int(p_height/csize)
off = zsize = csize = p_height/num2

numx = int(p_length/asize)


corners = [(0,bsize,0),(xsize,bsize,0),(xsize,0,0),(0,0,0),
           (0,0,csize),(asize,0,csize),(asize,ysize,csize),(0,ysize,csize)]



def make_box(insertion=[0,0,0],xsize=10, ysize=10, zsize=10):

    #create a 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(6,9,12)
#make_box([10,-3,-4],zsize=1)

# function to create field of foundations

def make_foundations(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)
# function to create 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)

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 make 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)

"""
# function to make 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)
"""
# function to create 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,7)
            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)

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

# 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_foundations(A, f_size, f_height, xcol, ycol)
        else:
            c_list.extend(make_columns(A, level, thick, hgt, xcol, ycol))
            if(i%3):
                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                                                  # 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


#stairs

    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))     #extrapodest
        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.AddLayer("stairs1")
rs.LayerColor("stairs1", (255,0,0))
rs.ObjectLayer(stair_l, "stairs1")


print f_list
print c_list
print p_list









# loop to create storeys
for i in range(levels):
    center_pt[2] = f_height+i*(thick+hgt)
    level = f_height+thick+(i-1)*(hgt+thick)
#    if i==0:


#Facade
num= 20

bsize = p_length/num
dist = 5
offs = bsize
num1 = int(p_width/bsize)+1
num2= int((f_height+ hgt*(levels-1)+ thick*levels)/bsize+1)


corners = [ (0,0,0), (0,0,bsize), (0,bsize,bsize), (0,bsize,0),
            (bsize,0,0), (bsize,0,bsize), (bsize,bsize,bsize), (bsize,bsize,0)]

# facade 1 kurz

for i in range(num):
    for j in range (num2):
        rand= ran.randint(0,9)
        if rand<8 or i==2:
            wuerfel = rs.AddBox(corners)
            rs.MoveObject(wuerfel, (0, i*offs, j*offs))
            rs.MoveObjects(wuerfel, (-B, -f_size/2, 0)) 
            if rand>2:
                rs.MoveObject(wuerfel, (-0.1,0,0))
            if rand>2 and rand<5:
                rs.MoveObject(wuerfel, (-0.2,0,0))

# facade 2 kurz

for i in range(num):
    for j in range (num2):
        rand= ran.randint(0,9)
        if rand<8 or i==2:
            wuerfel = rs.AddBox(corners)
            rs.MoveObject(wuerfel, (0, i*offs, j*offs))
            rs.MoveObjects(wuerfel, (p_width-B-bsize, -f_size/2, 0)) 
            if rand>2:
                rs.MoveObject(wuerfel, (0.1,0,0))
            if rand>2 and rand<5:
                rs.MoveObject(wuerfel, (0.2,0,0))

# facade 1 lang

for i in range(num1):
    for j in range (num2):
        #if 3>i>5:
            rand= ran.randint(0,9)
            if rand<8 or i==2:
                wuerfel = rs.AddBox(corners)
                rs.MoveObject(wuerfel, ( i*offs,0, j*offs))
                rs.MoveObjects(wuerfel, (-B, -f_size/2, 0)) 
                if rand>2:
                    rs.MoveObject(wuerfel, (0,-0.1,0))
                if rand>2 and rand<5:
                    rs.MoveObject(wuerfel, (0,-0.2,0))

# facade 2 lang

for i in range(num1):
    for j in range (num2):
        rand= ran.randint(0,9)
        if rand<8 or i==2:
            wuerfel = rs.AddBox(corners)
            rs.MoveObject(wuerfel, ( i*offs,0, j*offs))
            rs.MoveObjects(wuerfel, (-B,p_length-f_size/2-bsize, 0)) 
            if rand>2:
                rs.MoveObject(wuerfel, (0,0.1,0))
            if rand>2 and rand<5:
                rs.MoveObject(wuerfel, (0,0.2,0))

#make city of Dom
#rotate buildings 

buildings = []
current_building = rs.AllObjects()

for _ in range(15):
    current_building = rs.RotateObjects(current_building, [0, 0, 0], 3, copy=True)
    rs.MoveObjects(current_building, (3, 25, 0))
    buildings.append(current_building)

rs.EnableRedraw(True)


"""
radius = A*xcol*2                        
part = 7
rotate = ran.randint(-45,45)
box = rs. AllObjects()

cirl = rs.AddCircle([0,0,0], radius)
pnt = rs.DivideCurve(cirl, part)
for i in pnt:
    rotate = ran.randint(-30,60)
    rs.CopyObjects(box, i)
"""