def azabel(ax, axHist, azdata, **kwargs):
l = kwargs.get("labelstr", None)
c = kwargs.get("col", None)
which = kwargs.get("which", "Int")
cInt = kwargs.get("colInt", "blue")
cRad = kwargs.get("colRad", "red")
if c is not None:
cInt = c
cRad = c
xscale = kwargs.get("xs", 1.0)
yscale = kwargs.get("ys", 1.0)
abdat = inverAbelDat(azdata[0], azdata[1])
tS = kwargs.get("trimStart", 0)
# Direct integral
datInt = np.column_stack((xscale * abdat[0][0][tS:], yscale * abdat[0][1][tS:]))
# Radon transform
datRad = np.column_stack((xscale * abdat[1][0][tS:], yscale * abdat[1][1][tS:]))
if ax is not None:
if "both" == which or "Int" == which:
ax.plot(datInt[:, 0], datInt[:, 1], "-", marker=".", ms=5, color=cInt, label=l)
# ms=5, color=cInt, label=l+'Integral' )
if "both" == which or "Rad" == which:
ax.plot(datRad[:, 0], datRad[:, 1], ":", marker=".", ms=5, color=cRad, label=l + "Radon")
# axHist.hist( datInt[:,1] , bins=10, weights=4*np.pi*np.power(datInt[:,0],2) )
return datInt, datRad
cddata = coldens.average_cd( os.getcwd(), shots )
print "\nSum of cddata = ", np.sum(cddata)
else:
coldens.Plot_CD_AZ( figure, gs0[0,0], dirpath=os.getcwd(), \
shots=shots,
title='column density', magnif=magnif, \
lattice_d=lattice_d,\
cutPos='manual', r0=r0, c0=c0, **azdict)
cddata = coldens.average_cd( os.getcwd(), shots )
print "\nSum of cddata = ", np.sum(cddata)
azdata = coldens.average_azcd( os.getcwd(), shots, trimStart=0)
r = azdata[0]
az = azdata[1]
from scipy import integrate
print "\nIntegral of azdata = ", 2*np.pi * integrate.simps( az*r, r)
from inverabel import inverAbelDat
abdat = inverAbelDat( azdata[0], azdata[1] )
r = abdat[0][0]
ab = abdat[0][1]
print "\nIntegral of abdata = ", 4*np.pi * integrate.simps( ab*(r**2.), r)
figure.savefig( outfile, dpi=250 )
