import os, csv, random

# Generates random vectors and writes them to a CSV file

WriteFile = True # Write CSV file if true - useful for testing


CSVFileName ="DataOutput.csv"
CSVfile = open(os.path.join('/Users/Si/Desktop/', CSVFileName), 'w')

def genlist():
    # Generates a list of random vectors

    global v, ListLength
    ListLength = 25 #Amount of vectors to be produced

    Max = 100 #Maximum range value

    x = [] #Empty x vector list
    y = [] #Empty y vector list
    z = [] #Empty x vector list

    v = [] #Empty xyz vector list

    for i in xrange (ListLength):
        rnd = random.randrange(0,(Max)) #Generate random number
        x.append(rnd) #Add it to x list

    for i in xrange (ListLength):
        rnd = random.randrange(0,(Max))
        y.append(rnd) #Add it to y list

    for i in xrange (ListLength):
        rnd = random.randrange(0,(Max)) #Generate random number
        z.append(rnd) #Add it to z list

    for i in xrange (ListLength):
        merge = x[i], y[i],z[i] # Merge x[i], y[i], x[i]
        v.append(merge) #Add merged list into v list

def writeCSV():
    # Write Vectors to CSV file

    wr = csv.writer(CSVfile, quoting = csv.QUOTE_MINIMAL, dialect='excel')
    wr.writerow(('Point Number', 'X Vector', 'Y Vector', 'Z Vector'))

    for i in xrange (ListLength):
        wr.writerow((i+1, v[i][0], v[i][1], v[i][2]))

    print"Data written to", CSVfile

genlist()

if WriteFile is True:
   writeCSV()