generates square shaped transitions for a list of given data points.

genSquareTransitions.py

'''
This script generates square shaped transitions
for a list of given data points.

### Example 
Input:
x,y
0,100
5,430
10,200

Output:
x,y
0,100
5,100
5,430
10,430
10,200

'''

import csv 
import sys 

if(len(sys.argv) < 3): 
    print("Usage: " + sys.argv[0] + " <input csv file> <output csv file>")
    sys.exit(1)

infile = open(sys.argv[1],"r")
reader = csv.reader(infile,delimiter="\t")

outfile = open(sys.argv[2],"w")
writer = csv.writer(outfile)


last = None
skipHeader = True
for row in reader:
    if(skipHeader):
        writer.writerow(row)
        skipHeader = False
        continue
    if(last != None):
        writer.writerow([row[0], last])

    last = row[1]
    writer.writerow(row)

infile.close()
outfile.close()