def tta_precip():
import Windprof2 as wp
print "BBY-tta | BBY-notta | CZD-tta | CZD-notta | FRS-tta | FRS-notta "
o = "{:11.1f} {:11.1f} {:11.1f} {:11.1f} {:11.1f} {:11.1f}"
for case in range(8, 15):
bby, czd, frs, _ = get_data(str(case))
tta_times = wp.get_tta_times(case=str(case), continuous=True)
if len(tta_times) > 0:
tta_idx_bby = (bby.index >= tta_times[0]) & (bby.index <= tta_times[-1])
tta_idx_czd = (czd.index >= tta_times[0]) & (czd.index <= tta_times[-1])
tta_idx_frs = (frs.index >= tta_times[0]) & (frs.index <= tta_times[-1])
bby_tta = bby.precip[tta_idx_bby].sum()
bby_notta = bby.precip[~tta_idx_bby].sum()
czd_tta = czd.precip[tta_idx_czd].sum()
czd_notta = czd.precip[~tta_idx_czd].sum()
frs_tta = frs.precip[tta_idx_frs].sum()
frs_notta = frs.precip[~tta_idx_frs].sum()
else:
bby_tta = -999
bby_notta = bby.precip.sum()
czd_tta = -999
czd_notta = czd.precip.sum()
frs_tta = -999
frs_notta = frs.precip.sum()
print o.format(bby_tta, bby_notta, czd_tta, czd_notta, frs_tta, frs_notta)
