import matplotlib.pyplot as plt
import os.path

import tephi

dew_point = os.path.join(tephi.DATA_DIR, 'dews.txt')
dry_bulb = os.path.join(tephi.DATA_DIR, 'temps.txt')
column_titles = [('pressure', 'dewpoint'), ('pressure', 'temperature')]
dew_data, temp_data = tephi.loadtxt(dew_point, dry_bulb, column_titles=column_titles)
dews = zip(dew_data.pressure, dew_data.dewpoint)
temps = zip(temp_data.pressure, temp_data.temperature)

tpg = tephi.Tephigram()
dprofile = tpg.plot(dews)
dbarbs = [(0, 0, 900), (15, 120, 600), (35, 240, 300)]
dprofile.barbs(dbarbs)
tprofile = tpg.plot(temps)
tbarbs = [(10, 15, 900), (21, 45, 600), (25, 135, 300)]
tprofile.barbs(tbarbs)
plt.show()