#!/usr/bin/env python

#
# $Id: pts.py,v 1.4 2009/04/29 08:21:32 dintrans Exp $
# Plot the data/time_series.dat file: bmx, bmy, bmz and brms
#

import numpy as N
import pylab as P
import pencil as pc

par2=pc.read_param(quiet=True, param2=True)

ts=pc.read_ts(plot_data=False)

P.semilogy(ts.t, ts.bmx, label='bmx')
P.semilogy(ts.t, ts.bmy, label='bmy')
P.semilogy(ts.t, ts.bmz, label='bmz')
P.semilogy(ts.t, ts.brms, label='brms')
P.xlabel('time')

cte=ts.bmx[-1]
k1=1.
tsat=280. ; f=cte*(1.-N.exp(-2*par2.eta*k1**2*(ts.t-tsat)))
P.plot(ts.t, f, label='tsat=280')
tsat=300. ; f=cte*(1.-N.exp(-2*par2.eta*k1**2*(ts.t-tsat)))
P.plot(ts.t, f, label='tsat=300')

P.legend(loc='best')

P.show()