def minj_mflux(*args,**kwargs):
Qu = jana.quantities()
Minj_List =[]
Minj_MHD_List=[]
for i in range(len(args[1])):
Minj_List.append(Qu.Mflux(args[1][i],Mstar=args[0][i],scale=True)['Mfr']+Qu.Mflux(args[1][i],Mstar=args[0][i],scale=True)['Mfz'])
MHD_30minj = Qu.Mflux(args[2],Mstar=30.0,scale=True)['Mfr']+Qu.Mflux(args[2],Mstar=30.0,scale=True)['Mfz']
for i in args[0]:
Minj_MHD_List.append(MHD_30minj*np.sqrt(i/30.0))
f1 = plt.figure(num=1)
ax1 = f1.add_subplot(211)
plt.axis([10.0,70.0,2.0e-5,7.99e-5])
plt.plot(args[0],Minj_MHD_List,'k*')
plt.plot(args[0],Minj_List,'ko')
plt.minorticks_on()
locs,labels = plt.yticks()
plt.yticks(locs, map(lambda x: "%.1f" % x, locs*1e5))
plt.text(0.0, 1.03, r'$10^{-5}$', transform = plt.gca().transAxes)
plt.xlabel(r'Stellar Mass : $M_{*} [M_{\odot}]$')
plt.ylabel(r' $\dot{M}_{\rm vert} + \dot{M}_{\rm rad}\, [M_{\odot}\,\rm yr^{-1}]$')
ax2 = f1.add_subplot(212)
plt.axis([10.0,70.0,0.0,50.0])
plt.plot(args[0],100*((np.array(Minj_List)-np.array(Minj_MHD_List))/np.array(Minj_MHD_List)),'k^')
plt.minorticks_on()
plt.xlabel(r'Stellar Mass : $M_{*} [M_{\odot}]$')
plt.ylabel(r'$\%$ Change in Total Mass Outflow Rates')
def conto_plots(Data,DataMHD,**kwargs):
Qu = jana.quantities()
Cur = Qu.Current(Data)
CurMHD = Qu.Current(DataMHD)
Fastsp = Qu.Magspeed(Data)['fast']
Slowsp = Qu.Magspeed(Data)['slow']
Alfvsp = Qu.Magspeed(Data)['alfven']
Vpol = np.sqrt(Data.v1**2 + Data.v2**2)
FastspMHD = Qu.Magspeed(DataMHD)['fast']
SlowspMHD = Qu.Magspeed(DataMHD)['slow']
AlfvspMHD = Qu.Magspeed(DataMHD)['alfven']
VpolMHD = np.sqrt(DataMHD.v1**2 + DataMHD.v2**2)
fastrat = Fastsp/Vpol
slowrat = Slowsp/Vpol
Alfvrat = Alfvsp/Vpol
fastratMHD = FastspMHD/VpolMHD
slowratMHD = SlowspMHD/VpolMHD
AlfvratMHD = AlfvspMHD/VpolMHD
f1 = plt.figure(num=1)
ax1 = f1.add_subplot(121)
currcont=plt.contour(Data.x1,Data.x2,Cur.T,kwargs.get('Currents',[-0.7,-0.6,-0.5,-0.4,-0.3]),colors=kwargs.get('colors','r'),linestyles=kwargs.get('ls','-'),lw=kwargs.get('lw',2.0))
plt.clabel(currcont,manual=True)
currmhdcont = plt.contour(DataMHD.x1,DataMHD.x2,CurMHD.T,kwargs.get('Currents',[-0.7,-0.6,-0.5,-0.4,-0.3]),colors=kwargs.get('colors','k'),linestyles=kwargs.get('ls','-'),lw=kwargs.get('lw',2.0))
plt.clabel(currmhdcont,manual=True)
plt.xlabel(r'r [AU]')
plt.ylabel(r'z [AU]')
plt.title(r'Total Current -- $\int\int J_{z} r dr d\phi$')
ax2 = f1.add_subplot(122)
plt.xlabel(r'r [AU]')
#plt.ylabel(r'$z [AU]$')
plt.title(r'Critical Surfaces')
fcont=plt.contour(Data.x1,Data.x2,fastrat.T,[1.0],colors='r',linestyles='solid')
plt.clabel(fcont,inline=1,fmt=r'Fast')
scont=plt.contour(Data.x1,Data.x2,slowrat.T,[1.0],colors='r',linestyles='dashdot')
plt.clabel(scont,inline=1,fmt=r'Slow')
acont=plt.contour(Data.x1,Data.x2,Alfvrat.T,[1.0],colors='r',linestyles='dashed')
plt.clabel(acont,inline=1,fmt=r'Alfv$\acute{e}$n')
mfcont=plt.contour(DataMHD.x1,DataMHD.x2,fastratMHD.T,[1.0],colors='k',linestyles='solid')
plt.clabel(mfcont,manual=1,fmt=r'Fast')
mscont=plt.contour(DataMHD.x1,DataMHD.x2,slowratMHD.T,[1.0],colors='k',linestyles='dashdot')
plt.clabel(mscont,manual=1,fmt=r'Slow')
macont=plt.contour(DataMHD.x1,DataMHD.x2,AlfvratMHD.T,[1.0],colors='k',linestyles='dashed')
plt.clabel(macont,manual=1,fmt=r'Alfv$\acute{e}$n')
