print 'fields: ', data.GetFieldNames()
print 'extract coords'
coord = data.GetLocations()
print 'extract U'
U = data.GetVectorField('Velocity')
print 'done.'
del data
id_U = np.where(coord[:,2] == 0) # values from surface
coord = coord[id_U]
c = fio.fluidityclean_coord(coord,0)
Us = fio.fluidityclean_2D(coord,U[id_U,0].flatten(),0)
Vs = fio.fluidityclean_2D(coord,U[id_U,1].flatten(),0)
S = np.sqrt(Us**2 + Vs**2)
x = c[:,0]
y = c[:,1]
## PRINT
#triang = tri.Triangulation(x, y)
plt.figure()
plt.gca().set_aspect('equal')
plt.autoscale(enable=True, axis='both', tight=True)
if time == 0:
print 'extract U'
U = data.GetVectorField('Velocity')
print 'done.'
#T = data.GetScalarField('T')
#Tmax = np.max(T)
#Tmin = np.min(T)
print 'using: ', Tmin,Tmax, ' while max is', np.max(T), ' and min is ', min(T)
del data
id_T = np.where(coord[:,2] == 0) # values from surface
coord = coord[id_T]
c = fio.fluidityclean_coord(coord,0)
T = fio.fluidityclean_2D(coord,T[id_T],0)
Us = fio.fluidityclean_2D(coord,U[id_T,0].flatten(),0)
Vs = fio.fluidityclean_2D(coord,U[id_T,1].flatten(),0)
x = c[:,0]
y = c[:,1]
# T[np.asarray(np.where(T > Tmax))] = Tmax
# T[np.asarray(np.where(T < Tmin))] = Tmin
## PRINT
#triang = tri.Triangulation(x, y)
plt.figure()
plt.gca().set_aspect('equal')
