import scitools.filetable as ft
import matplotlib.pyplot as plt
import glob

## file1 = open("resp1/energy.dat")
## ener = ft.read(file1)
## time = ener[:, 0]
## energy = ener[:, 1]
## enstrophy = ener[:, 2]
## #ratio = ener[:,3]
## #dedt = ener[:,4]
## #nuS = ener[:,5]
## #nuES = ener[:,6]
VISCOSITY = 1. / 1600.
## nuS_calc = VISCOSITY * enstrophy[2:]
import numpy as np
# multiproc #
filepath = './res_'
plt.ioff()
plt.figure(1)

error = []

filelist = glob.glob(filepath + '*')
for f in filelist:
    filename = open(f + '/ener')
    data = ft.read(filename)
    time = data[:, 0]
    energy = data[:, 1]
    enstrophy = data[:, 2]
    plt.plot(time, energy, '+-', label=f)
    error.append(np.sum(energy))

plt.legend()
plt.show()

print error