import rhinoscriptsyntax as rs
import random as ran

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


# dom-ino variables:
A = 5.0             # 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 = 2            # columns in x direction
ycol = 4            # columns in y direction
levels = 5         # number of floor plates
f_height = 0.5      # foundation height
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 plate (x)
p_length = A*(ycol-1) + f_size                      #lengh of floor plate (y)

p_height= f_height+ (hgt+thick)*(levels-1)

"""
#Set Variables/Parameters
xsize = 8         #xside length of box
ysize = 12          #yside length of box
zsize = 5         #zside length of box

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


#function to create a 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 a 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_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=[]
    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)

#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[-2][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.DeleteObject(s_outline)
    rs.DeleteObject(path)
    return(hull)


#function to create podest 
def make_podest(insertion=[0,0,0],xsize=1,ysize=1,zsize=1):
    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]


rs.EnableRedraw(False)


#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%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 value 
    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   #stair width
    pod_w = B       #depth of landing
    start = [pod_w,-(s_width*2+f_size/2),f_height+thick]    #startpoint of stair

    #loop for staircase
    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], 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)

(f_list, c_list, p_list)= 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,200))
rs.ObjectLayer(p_list,"plates")

rs.EnableRedraw(False)

print f_list
print c_list
print p_list


# facade 1
dis=0.4
facadexsize=0.2
facadeysize=0.2
facade_length = int(p_length/dis)
facade_hgt = 2.8
place = A*(xcol-1)+B+facadexsize/2

def make_facade (insertion=[0,0,0], facadexsize=0.2, facadeysize=0.2, facadezsize=(facade_hgt+thick)*(levels-1), dis=0.4):
    facade = []
    facade_length = int(p_length/dis)
    for i in range (facade_length):
        facade.append(make_box([insertion[0],i*dis-f_size/2,f_height], facadexsize, facadeysize, facadezsize))
    return (facade)

make_facade(insertion=[A*(xcol-1)+B+facadexsize/2,0,0])
make_facade(insertion=[-B-facadexsize/2,0,0],facadezsize=((facade_hgt+thick)*(levels-1))/2, dis=0.9857)


#facade 2

#variables
bsize = 0.3              #side length of box
dist = 0.02              #distance between boxes
offs = bsize + dist      #offset of boxes
num = 50                #number of boxes in one row
facade_length = int((p_width+0.3)/offs)
facade_hgt = 2.8
facade_z= int(((p_height+0.3)/offs))
place = A*(xcol-1)+2 
num1= num+20
num2=50

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


for i in range(facade_length):
    for p in range(facade_z):
        myran = ran.randint(0,10)
        if (myran >= 3):
            if ((i*p)%4) and ((i-p)%6):
                box = rs.AddBox(corners)
                rs.MoveObject(box, (i*offs, 0, p*offs))
                rs.MoveObject(box, (-B,-0.5,0))
                rs.CopyObject(box, (0,p_length-f_size/2,0))


#for random creation of buildings 
for i in range (2): 
    xran= ran.randint(0,150)
    yran= ran.randint(0,200)
    angle= ran.uniform(0,180)
#    diva= ran.randint(2,5)
    (f_list, c_list, p_list)= make_domino(levels) 
    rs.RotateObjects(f_list+ c_list+ p_list, (0,0,0),angle)
    rs.MoveObjects(f_list+ c_list+ p_list,(xran,yran,0))

curv= rs.AddSpiral((0,0,0), (0,0,3), 0 ,6 ,180 ,55)
pts= rs.DivideCurve(curv, 70)

for p in pts:

    (f_list, c_list, p_list,)= make_domino()
    rs.MoveObject(f_list+ c_list+ p_list,p)
    param= rs.CurveClosestPoint(curv, p)
    normal= rs.CurveTangent(curv, param)
    angle= rs.Angle([0,0,0], normal)[0]
    rs.RotateObjects(f_list+ c_list+ p_list, p, angle)






"""
#Skripte die notwendig sein koennten
#define spiral 

my_c = rs.AddSpiral([0,0,0],[0,0,1], 10,15, 100, 20)
plane = rs.WorldXYPlane()

for i in range(100):
    xran = ran.randint(0,100)
    yran = ran.randint(0,300)
    zran = 0
    if not(rs.PointInPlanarClosedCurve([xran,yran,zran], my_c,plane,0)):
        mybox = rs.AddBox(corners)
        rs.RotateObjects(f_list+ c_list+ p_list, (0,0,0),angel)
        rs.MoveObjects(f_list+ c_list+ p_list,(xran,yran,0))
        #rs.MoveObject(mybox, [xran,yran,zran])


# facade 2
facadexsize=0.2
facadeysize=0.2
facade_length = 12
facade_hgt = 2.8
place = -B-facadexsize/2

def make_facade2 (insertion=[0,0,0], facadexsize=0.2, facadeysize=0.2, facadezsize=(facade_hgt+thick)*(levels-1), dis=0.9857):
    facade = []
    for i in range (facade_length):
        facade.append(make_box([place,i*dis-f_size/2,f_height], facadexsize, facadeysize, facadezsize))
    return (facade)



#define points of the building without them colliding 
for i in range (7):
    pt=rs.GetPoint()
    angel=rs.GetAngle(pt,)
    (f_list, c_list, p_list)= make_domino()
    rs.MoveObjects(f_list+ c_list+ p_list,(pt))
    rs.RotateObjects(f_list+ c_list+ p_list, pt,angel)
"""