import rhinoscriptsyntax as rs
import random as ran
import fantastic_stairs_lib as fs
reload(fs)
# delete everything and start from scratch
rs.DeleteObjects(rs.AllObjects())
rs.EnableRedraw(False)

def make_v_grid(x_cnt,z_cnt,xsize,zsize,l_len):
    pts=[]
    lines=[]
    for i in range(x_cnt):
        for j in range(z_cnt):
            lines.append(rs.AddLine([i*xsize,0,j*zsize],[i*xsize,l_len,j*zsize] ))
            pts.append(rs.AddPoint([i*xsize,0,j*zsize]))
    return(pts,lines)

x_cnt, z_cnt, xsize, zsize=3, 4, 4.0, 4.0
l_len=1.0
dist_x, dist_y, dist_z=20, 20, 0
angle= ran.uniform(90,180)
(pts1,lines1)= make_v_grid(x_cnt,z_cnt,xsize,zsize,l_len)
rs.RotateObjects(pts1, [0,0,0], angle, [0,0,1])
rs.RotateObjects(lines1, [0,0,0], angle, [0,0,1])
rs.MoveObjects(pts1, (dist_x, dist_y, dist_z))
rs.MoveObjects(lines1, (dist_x, dist_y, dist_z))
(pts2,lines2) = make_v_grid(x_cnt,z_cnt,xsize,zsize,l_len)

ptl1= range(0,len(pts1))
ptl2= range(0,len(pts2))
#print ptl1
#print ptl2

crv_list=[]
while len(ptl1)>0: 
    r_init = ran.randint(0,len(ptl1)-1)
    r_end = ran.randint(0,len(ptl2)-1)
    init = lines1[ptl1[r_init]]
    end = lines2[ptl2[r_end]]
    #rs.AddLine(init,end)
    crv_list.append(fs.blend_curves(init,end,2))
    ptl1.pop(r_init)
    ptl2.pop(r_end)
    #print ptl1
    #print ptl2
rs.DeleteObjects(pts1)
rs.DeleteObjects(pts2)

for curve in crv_list:
    (st_crv, tothgt) = fs.curve_to_stairline(curve)
    len_c = rs.CurveLength(st_crv)
    if tothgt<=zsize:
        fs.make_curved_ramp(st_crv,tothgt)
    else:
        (steps,th) = fs.calculate_steps_height(tothgt)
        An = 0
        tt = (len_c-An)/steps
        if tt>0.32:
            tt=fs.tt
            P=steps/2
            P = 3
            pod_l=fs.calculate_podlen(len_c,tt,steps,P,An)
            print "stair values (length, tread height, tread depth, steps", len_c, th, tt, steps
            fs.make_curved_podeststair(st_crv, steps=steps, th=th, tt=tt, pod_l=pod_l, P=P, An=An, DC=steps+1, s_width=1.2)


connected_dominos = fs.make_domino(xsize, xsize/3, thick=fs.thick, hgt=zsize-fs.thick,levels=z_cnt, xcol=x_cnt,ycol=fs.ycol)
for o_list in connected_dominos:
    rs.MoveObjects(o_list,[0,-(fs.ycol-1)*xsize-fs.f_size/2,-(fs.f_height+fs.thick)])
    co_list= rs.MirrorObjects(o_list,[0,0,0], [dist_x,0,0], copy=True)
    rs.MoveObjects(co_list,[dist_x-(x_cnt-1)*xsize, dist_y, dist_z])
    rs.RotateObjects(co_list,[dist_x, dist_y, dist_z],(angle-180), [0,0,1])

rs.EnableRedraw(True)