#!/usr/bin/python
# -*- coding: utf-8 -*-
#plot polar curves for gliders
from numpy import *
from matplotlib.pyplot import *
import csv
import re

def addPolars(regexStr) :
	a = re.compile(regexStr); 

	i = 0;
	for n in names :
		if a.match(n) :
			print "Matched " + str(i) + ": '" + n + "'";
			plot(data[0, :], data[i, :], label=n);

		i += 1;
		
	legend();
	draw();


csvReader = csv.reader(open("polars.csv","rb"));

names = [];
data = [];

for row in csvReader : 
    names.append(row[0]);
    data.append(array(row[1:]));

data = array(data, dtype=float);

clf();
plot(data[0, :], data[42, :], "--", label = "Ref: DG 100");
plot(data[0, :], data[81, :], "--", label = "Ref: Discus B");
legend();
draw();

print "Call addPolars(reg exp)"