#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
#sys.path.append('/home/nicolas/Research/Code/traffic-intelligence/python/')

import matplotlib.pyplot as plt
import matplotlib.mlab as pylab
import numpy as np

# data = csv2rec('2009.csv', delimiter=';') # problem 

import csv
import datetime

filename = '/home/nicolas/Research/Data/ville-montreal/open-data/2009.csv'
file = open(filename)

frString2Month = {'janv':1, 'f\xe9vr':2, 'mars':3, 'avr':4, 'mai':5, 'juin':6, 'juil':7, 'ao\xfbt':8, 'sept':9, 'oct':10, 'nov':11, 'd\xe9c':12}

reader = csv.reader(file, delimiter=';')
data = {}
header = reader.next()
for i in range(len(header)):
    header[i] = header[i].replace('\xf4','')
    data[header[i]] = []

for row in reader:
    date = row[0].split(' ')
    data[header[0]].append(datetime.date(int(date[3]), frString2Month[date[2]], int(date[1])))
    for i in range(1,len(row)):
        if len(row[i]) > 0:
            data[header[i]].append(int(row[i].replace(' ','')))
        else:
            data[header[i]].append(0)

fig = plt.figure()
for i in range(1,len(header)):
    plt.subplot(len(header)-1,1,i)
    plt.plot(data['Date'], data[header[i]])
    plt.ylabel(header[i])
    fig.autofmt_xdate()
plt.show()
