plot_channels.py

import sys
import matplotlib.pyplot as plt
import numpy as np
from scipy.signal import argrelextrema
import csv

num_channels = int(sys.argv[1])
data1 = []
data2 = []
window_size = 50

with open("results/" + sys.argv[2] + ".csv") as csvfile:
    for row in csv.reader(csvfile, delimiter=","):
        for i in range(0, num_channels):
            if len(data1) == i:
                data1.append([])
            data1[i].append(row[i])

if not sys.argv[3].isdigit():
    with open("results/" + sys.argv[3] + ".csv") as csvfile:
        for row in csv.reader(csvfile, delimiter=","):
            for i in range(0, num_channels):
                if len(data2) == i:
                    data2.append([])
                data2[i].append(row[i])

    points = int(sys.argv[4])
    chan = int(sys.argv[5])
else:
    points = int(sys.argv[3])

dist = []

with open("results/distance.csv") as csvfile:
    for row in csv.reader(csvfile, delimiter=","):
        dist.append(float(row[0]))

dist = np.array(dist)
maxI = argrelextrema(dist, np.greater)
maxI = list(maxI[0])

gTruths = []


if not sys.argv[3].isdigit():
    plt.plot(data1[chan][:points], label=sys.argv[2])
    plt.plot(data2[chan][:points], label=sys.argv[3])
else:
    for i in range(0, num_channels):
        plt.plot(data1[i][:points], label=sys.argv[2])

for i in maxI:
    if maxI.index(i) != 0 and maxI.index(i) != len(maxI) - 1:
        prevMax = dist[maxI[maxI.index(i) - 1]]
        nextMax = dist[maxI[maxI.index(i) + 1]]

        if dist[i] * 0.3 > prevMax or dist[i] * 0.3 > nextMax:
            plt.axvline(x = i * window_size, color='r', ymin=0.5, ymax=1.0)

for i in gTruths:
    plt.axvline(x = i, color='b', ymin=-0.5, ymax=0.5)

plt.legend(loc='upper left')
plt.axis([0, points, -0.5, 1.5])
plt.show()

#python plot_channels.py 8 original decoded 5000 1
#python plot_channels.py 8 original 5000