import rhinoscriptsyntax as rs
import grid_array as ga
import random
import math
rs.EnableRedraw(False)

#input variables - grid_count, grid_size, rec_size, rot_factor
#ga.attractor_grid(20, 10, 8, 1000)


def ran():
    return random.uniform(-1,1)

def random_vec(length):
    start_point = [0,0,0]
    end_point = [ran(), ran(), ran()]
    ran_vec = rs.VectorCreate(start_point, end_point)
    ran_vec = rs.VectorUnitize(ran_vec)
    ran_vec = rs.VectorScale(ran_vec, length)
    return ran_vec


"""
iterations = 1000
step_len = 5
start_point = [0,0,0]
points = []
points.append(start_point)

for i in range(iterations):
    next_point = rs.VectorAdd(start_point,random_vec(step_len))
    points.append(next_point)
    #rs.AddPoint(next_point)
    start_point = next_point

rs.AddPolyline(points)
"""


iterations = 2048
step_len = 5
start_point = [0,0,0]
points = []
points.append(start_point)

for n in range(10):
    for i in range(iterations):
        next_point = rs.VectorAdd(start_point,random_vec(step_len))
        line = rs.AddLine(start_point,next_point)
        
        rs.ObjectColor(line,[0, 0, i/8])
        points.append(next_point)
        start_point = next_point
    
    start_point = [0,0,0]


"""
iterations = 1024
step_len = 5
start_point = [0,0,0]
points = []
points.append(start_point)

for n in range(10):
    for i in range(iterations):
        ran_vec = random_vec(step_len)
        next_point = rs.VectorAdd(start_point,ran_vec)
        line = rs.AddLine(start_point,next_point)
        r = math.fabs(ran_vec[0])*50
        g = math.fabs(ran_vec[1])*50
        b = math.fabs(ran_vec[2])*50
        
        rs.ObjectColor(line,[r, g, b])
        points.append(next_point)
        start_point = next_point
    
    start_point = [0,0,0]
"""