#!/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

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}

def readBikeCountFile(filename):
    file = open(filename)
    reader = csv.reader(file, delimiter=';')
    data = {}
    header = reader.next()
    for i in range(len(header)):
        header[i] = header[i].replace('\xf4','o')
        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)
    return data,header

filename = '/home/nicolas/Research/Data/ville-montreal/open-data/2011.csv'
data, header = readBikeCountFile(filename)


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