import numpy as np
import csv
import matplotlib.pyplot as plt
from plot_data import plot_data

def plot_data(data):
  """ simple plot for x, y """
  
  plt.title('BLE Plot');
  # plt.plot([row[0] for row in data], label="x")
  plt.plot(data, label="rssi")
  plt.legend(loc='upper right')
  plt.show()


def get_data(file_name):
  """reads the file and returns """
  inputfile = open(file_name, mode="r")
  data = []

  # start reading
  for line in inputfile:
    # print(line)
    data.append(line.rstrip())
  return data

def run():
  # TODO: place your csv file in the same directory as task1-1.csv
  file_name = 'task1-1.csv'
  data = get_data(file_name)

  print(data)
  plot_data(data)

if __name__ == '__main__':
  run()