def special_BetaEq(EOSlist,temps,yes,mytype):
for ieos in range(len(EOSlist)):
myeos = eosDriver(EOSlist[ieos][1])
rhomax = (10.0e0**max(myeos.h5file['logrho']))*0.995
eostablename = EOSlist[ieos][0]+\
"_eostable_BetaEq_"+mytype+".dat"
energy_shift = myeos.h5file['energy_shift'][0]
if(mytype == "c30p5"):
tempfunc = kentaDataTofLogRhoFit2()
elif(mytype == "c30p10"):
tempfunc = kentaDataTofLogRhoFit1()
elif(mytype == "c30p0"):
tempfunc = getTRollFunc(30.0,0.01,14.055,0.375)
elif(mytype == "c20p0"):
tempfunc = getTRollFunc(20.0,0.01,14.0-0.07,0.125)
elif(mytype == "c40p0"):
tempfunc = getTRollFunc(40.0,0.01,14.25-0.07,0.5)
logrhos = linspace(log10(rhomin),log10(rhomax),nrhos)
eostable = zeros((nrhos,2))
for i in range(nrhos):
temp = tempfunc(logrhos[i])
ye = myeos.setBetaEqState({'rho': 10.0**logrhos[i],
'temp': temp})
(press,eps) = myeos.query(['logpress','logenergy'])
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
if (i>0 and eostable[i,0] < eostable[i-1,0]):
eostable[i,0] = eostable[i-1,0]
print "Making EOS: %15.6E %15.6E %15.6E" % (10.0**logrhos[i],temp,ye)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
del myeos
colors = {'c30p0': 'g',
'c20p0': 'b',
'c40p0': 'r',
'cold': 'm',
'c30p5': 'c',
'c30p10': 'k'}
symbols = {'c30p0': 's',
'c20p0': 'v',
'c40p0': '^',
'c30p5': 'p',
'c30p10': 'H',
'cold': '*'}
scriptsList = ['c40p0', 'c30p0', 'c20p0', 'cold']
cXXp0_params = [(40.0, 14.18, 0.5,), (30.0, 14.055, 0.375), (20.0, 13.93, 0.25)]
tempFuncs = [getTRollFunc(params[0], 0.01, params[1], params[2]) for params in cXXp0_params]
tempFuncs.append(kentaDataTofLogRhoFit1())
tempFuncs.append(kentaDataTofLogRhoFit2())
tempFuncs.append(lambda x: 0.01)
tempFuncsDict = {scriptsList[i]: tempFuncs[i] for i in range(len(scriptsList))}
#############################################################
# First plot
#############################################################
filters = ('edMax>2.0e14', 'ToverW<0.25', 'baryMass<2.6')
plotList = []
# coldTovSet = cstDataset('cold', eosName, ye, sourceDb)
# coldTovSeq = cstSequence(coldTovSet, tovSlice, filters)
# theEos.resetCachedBetaEqYeVsRhobs(tempFuncsDict['cold'], 13.5, 16.0)
# coldTovPlot = \
symbols = {
'c30p0': 's',
'c20p0': 'v',
'c40p0': '^',
'c30p5': 'p',
'c30p10': 'H',
'cold': '*'
}
scriptsList = ['c40p0', 'c30p0', 'c20p0', 'cold']
cXXp0_params = [(
40.0,
14.18,
0.5,
), (30.0, 14.055, 0.375), (20.0, 13.93, 0.25)]
tempFuncs = [
getTRollFunc(params[0], 0.01, params[1], params[2])
for params in cXXp0_params
]
tempFuncs.append(kentaDataTofLogRhoFit1())
tempFuncs.append(kentaDataTofLogRhoFit2())
tempFuncs.append(lambda x: 0.01)
tempFuncsDict = {scriptsList[i]: tempFuncs[i] for i in range(len(scriptsList))}
#############################################################
# First plot
#############################################################
filters = ('edMax>2.0e14', 'ToverW<0.25', 'baryMass<2.6')
plotList = []
# coldTovSet = cstDataset('cold', eosName, ye, sourceDb)
# coldTovSeq = cstSequence(coldTovSet, tovSlice, filters)
# theEos.resetCachedBetaEqYeVsRhobs(tempFuncsDict['cold'], 13.5, 16.0)
def special_NFBetaEq_pnu(EOSlist,temps,mytype):
for ieos in range(len(EOSlist)):
myeos = eosDriver(EOSlist[ieos][1])
rhomax = (10.0e0**max(myeos.h5file['logrho']))*0.995
eostablename = EOSlist[ieos][0]+\
"_eostable_NFBetaEq_pnu_"+mytype+".dat"
energy_shift = myeos.h5file['energy_shift'][0]
logrhos = linspace(log10(rhomin),log10(rhomax),nrhos)
eostable = zeros((nrhos,2))
energy_shift = energy_shift*eps_gf
rhotrap = 10.0e0**12.5
if(mytype == "c30p5"):
tempfunc = kentaDataTofLogRhoFit2()
elif(mytype == "c30p10"):
tempfunc = kentaDataTofLogRhoFit1()
elif(mytype == "c30p0"):
tempfunc = getTRollFunc(30.0,0.01,14.055,0.375)
elif(mytype == "c20p0"):
tempfunc = getTRollFunc(20.0,0.01,14.0-0.07,0.125)
elif(mytype == "c40p0"):
tempfunc = getTRollFunc(40.0,0.01,14.25-0.07,0.5)
for i in range(nrhos):
temp_cold = 0.5e0
temp = tempfunc(logrhos[i])
rho = 10.0**logrhos[i]
ylep = myeos.setBetaEqState({'rho': rho,
'temp': temp_cold})
ye2 = myeos.setBetaEqState({'rho': rho,
'temp': temp})
ye1 = myeos.setNuFullBetaEqState({'rho': rho,
'temp': temp},ylep)
ye = ye2*(1-exp(-rhotrap/rho)) + ye1*(exp(-rhotrap/rho))
myeos.setState({'rho': rho,
'temp': temp,
'ye' : ye})
print "Making EOS: %15.6E %15.6E %15.6E %15.6E %15.6E %15.6E" %\
(10.0**logrhos[i],temp,ye,ye1,ye2,ylep)
(press,eps,munu) = myeos.query(['logpress','logenergy','munu'])
eta = munu/temp
pnu = get_pnu(eta,temp,rho,rhotrap)
epsnu = 3.0e0*pnu / rho
# convert units
eostable[i,0] = log10((10.0**press+pnu) * press_gf)
eostable[i,1] = log10((10.0**eps+epsnu) * eps_gf)
if (i>0 and eostable[i,0] < eostable[i-1,0]):
eostable[i,0] = eostable[i-1,0]
eostable[i,0] = eostable[i-1,0]
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
del myeos
#print logrhos
def entropyOfT(logtemp, logrho, target):
return multidimInterp((ye, logtemp, logrho), [theEos.h5file['ye'][:],
theEos.h5file['logtemp'][:],
theEos.h5file['logrho'][:]],
theEos.h5file['entropy'][:, :, :],
linInterp, 2) - target
esses=[0.5, 1.0, 2.0, 3.0]
tRollFunc = getTRollFunc(40.0, tcold, 14.0, 0.5)
escolors=['r', 'g', 'b', 'k', 'c', 'm', 'y', 'orange', 'violet']
proll = []
prollBeta = []
legends = []
lrsList = []
pBetaList = []
for i, s in enumerate(esses):
ts=[]
tsBeta=[]
lps =[]
lpsBeta = []
for lr in logrhos:
rho = numpy.power(10.0, lr)
theEos.setState({'rho': rho, 'ye': ye, 'temp': tcold})
pCold = theEos.query('logpress')
slicesToPlot = [a7, diffMaxRotSlice]
plateauPlots = True
tFuncs = []
scripts = []
if plateauPlots:
scripts = ['c30p10', 'c30p5', 'cold']
tFuncs = [kentaDataTofLogRhoFit1(),
kentaDataTofLogRhoFit2(),
lambda x: 0.01]
else:
params40 = (40.0, 14.18, 0.5,)
params20 = (20.0, 13.93, 0.25,)
scripts = ['c40p0', 'c20p0', 'cold']
tFuncs = [getTRollFunc(params40[0], 0.01, params40[1], params40[2]),
getTRollFunc(params20[0], 0.01, params20[1], params20[2]),
lambda x: 0.01]
ye = 'BetaEq'
colorLegs = []
for i, script in enumerate(scripts):
tempFunc = tFuncs[i]
theEos.resetCachedBetaEqYeVsRhobs(tempFunc, 13.5, 16.0)
xFunc = lambda x: theEos.rhobFromEnergyDensityWithTofRho(x, ye, tempFunc) / 1.0e15
#xFunc = lambda x: x
filters = ('ToverW<0.25',)
a6 = {'a': 0.6, 'rpoe': 'min'}
a5 = {'a': 0.5, 'rpoe': 'min'}
a4 = {'a': 0.4, 'rpoe': 'min'}
colors = {'c30p0': 'g',
'c20p0': 'b',
'c40p0': 'r',
'cold': 'k',
'c30p5': 'c',
'c30p10': 'm'}
scripts = ['c40p0', 'cold']
params40 = (40.0, 14.18, 0.5,)
params20 = (20.0, 13.93, 0.25,)
tFuncs = [#getTRollFunc(params20[0], 0.01, params20[1], params20[2]),
getTRollFunc(params40[0], 0.01, params40[1], params40[2]),
#kentaDataTofLogRhoFit2(),
lambda x: 0.01]
ye = 'BetaEq'
colorLegs = []
pltsForLeg = []
#############################################################
# First lay down RotNS results
#############################################################
for i, script in enumerate(scripts):
#break
tempFunc = tFuncs[i]
theEos.resetCachedBetaEqYeVsRhobs(tempFunc, 13.5, 16.0)
xFunc = lambda x: theEos.rhobFromEnergyDensityWithTofRho(x, ye, tempFunc)
#xFunc = lambda x: x
'c20p0': 'v',
'c40p0': '^',
'c30p5': 'p',
'c30p10': 'H',
'cold': '*'}
lineWidths = {'c30p0': 2,
'c20p0': 2,
'c40p0': 2,
'c30p5': 3,
'c30p10': 4,
'cold': 2}
scriptsList = ['c40p0', 'c30p0', 'c20p0', 'c30p10', 'c30p5', 'cold']
cXXp0_params = [(40.0, 14.18, 0.5,), (30.0, 14.055, 0.375), (20.0, 13.93, 0.25)]
tempFuncs = [getTRollFunc(params[0], 0.01, params[1], params[2]) for params in cXXp0_params]
tempFuncs.append(kentaDataTofLogRhoFit1())
tempFuncs.append(kentaDataTofLogRhoFit2())
tempFuncs.append(lambda x: 0.01)
tempFuncsDict = {scriptsList[i]: tempFuncs[i] for i in range(len(scriptsList))}
mgShift = 0.0
mgYfunc = lambda x: x + mgShift
#############################################################
# First plot: Just Shen, frac diffs
#############################################################
markerSize = 6
edgeColor = 'k'
for script in colors.keys():
thisSet = cstDataset(script, eosName, ye, sourceDb)
xaxisData, xlabels = readFile('/home/jeff/work/Shen135135_x_v2.dat')
#yaxisData, ylabels = readFile('/home/jeff/work/Shen135135_y_v2.dat')
#zaxisData, zlabels = readFile('/home/jeff/work/Shen135135_z_v2.dat')
##############################################################################
#Paper temperature plot
##############################################################################
fig = plt.figure(figsize=(10, 8))
fig.subplots_adjust(left=0.14)
fig.subplots_adjust(bottom=0.14)
fig.subplots_adjust(top=0.967)
fig.subplots_adjust(right=0.97)
plateau10 = kentaDataTofLogRhoFit1()
plateau5 = kentaDataTofLogRhoFit2()
core20 = getTRollFunc(20.0, 0.01, 14.0 - .07, .25)
core30 = getTRollFunc(30.0, 0.01, 14.125 - .07, .375)
core40 = getTRollFunc(40.0, 0.01, 14.25 - 0.07, .5)
lrs = numpy.log10([xaxisData['rho'][i] for i in range(len(xaxisData['rho'])) if xaxisData['d'][i] > 0.0])
simTemps = [xaxisData['T'][i] for i in range(len(xaxisData['rho'])) if xaxisData['d'][i] > 0.0]
lrsMore = numpy.array([15.1, 15.0, 14.9, 14.8] + lrs.tolist())
fig.add_subplot(111).plot(lrs, simTemps, plot_defaults.kotzbraun, lw=4, zorder=4,
dashes=plot_defaults.longDashes)
fig.add_subplot(111).plot(
lrsMore, core40(lrsMore), 'r',
lrsMore, core30(lrsMore), 'g',
lrsMore, core20(lrsMore), 'b',
lrsMore, plateau10(lrsMore), 'm',
lrsMore, plateau5(lrsMore), 'c')
legends = ["Sekiguchi et al.", "c40p0", "c30p0", "c20p0", "c30p10", "c30p5"]
xaxisData, xlabels = readFile("/home/jeff/work/Shen135135_x_v2.dat")
# yaxisData, ylabels = readFile('/home/jeff/work/Shen135135_y_v2.dat')
# zaxisData, zlabels = readFile('/home/jeff/work/Shen135135_z_v2.dat')
##############################################################################
# Paper temperature plot
##############################################################################
fig = plt.figure(figsize=(10, 8))
fig.subplots_adjust(left=0.14)
fig.subplots_adjust(bottom=0.14)
fig.subplots_adjust(top=0.967)
fig.subplots_adjust(right=0.97)
plateau10 = kentaDataTofLogRhoFit1()
plateau5 = kentaDataTofLogRhoFit2()
core20 = getTRollFunc(20.0, 0.01, 14.0 - 0.07, 0.25)
core30 = getTRollFunc(30.0, 0.01, 14.125 - 0.07, 0.375)
core40 = getTRollFunc(40.0, 0.01, 14.25 - 0.07, 0.5)
lrs = numpy.log10([xaxisData["rho"][i] for i in range(len(xaxisData["rho"])) if xaxisData["d"][i] > 0.0])
simTemps = [xaxisData["T"][i] for i in range(len(xaxisData["rho"])) if xaxisData["d"][i] > 0.0]
lrsMore = numpy.array([15.1, 15.0, 14.9, 14.8] + lrs.tolist())
fig.add_subplot(111).plot(lrs, simTemps, plot_defaults.kotzbraun, lw=4, zorder=4, dashes=plot_defaults.longDashes)
fig.add_subplot(111).plot(
lrsMore,
core40(lrsMore),
"r",
lrsMore,
core30(lrsMore),
"g",
lrsMore,
def makeeostable(nrhos,rhomin,rhomax,myeos,myeosname,mytype,par1,par2):
assert isinstance(myeos, eosDriver)
logrhos = linspace(log10(rhomin),log10(rhomax),nrhos)
eostable = zeros((nrhos,2))
eostable2 = zeros((nrhos,4))
energy_shift = 0.0
print mytype
if(mytype == 'fixed_ye_temp'):
ye = par2
temp = par1
eostablename = myeosname+"_eostable_Ye=%06.3f_T=%06.3f.dat" % \
(par2,par1)
energy_shift = myeos.h5file['energy_shift'][0]
for i in range(nrhos):
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
press = myeos.query('logpress')
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
eps = myeos.query('logenergy')
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'c30p0_fixed_Ye'):
ye = par2
tempfunc = getTRollFunc(30.0,0.01,14.055,0.375)
eostablename = myeosname+"_eostable_Ye=%06.3f_c30p0.dat" % \
(par2)
energy_shift = myeos.h5file['energy_shift'][0]
for i in range(nrhos):
temp = tempfunc(logrhos[i])
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
press = myeos.query('logpress')
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
eps = myeos.query('logenergy')
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'c20p0_fixed_Ye'):
ye = par2
tempfunc = getTRollFunc(20.0,0.01,14.0-0.07,0.125)
eostablename = myeosname+"_eostable_Ye=%06.3f_c20p0.dat" % \
(par2)
energy_shift = myeos.h5file['energy_shift'][0]
for i in range(nrhos):
temp = tempfunc(logrhos[i])
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
press = myeos.query('logpress')
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
eps = myeos.query('logenergy')
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'c40p0_fixed_Ye'):
ye = par2
tempfunc = getTRollFunc(40.0,0.01,14.25-0.07,0.5)
eostablename = myeosname+"_eostable_Ye=%06.3f_c40p0.dat" % \
(par2)
energy_shift = myeos.h5file['energy_shift'][0]
for i in range(nrhos):
temp = tempfunc(logrhos[i])
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
press = myeos.query('logpress')
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
eps = myeos.query('logenergy')
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'c30p10_fixed_Ye'):
ye = par2
tempfunc = kentaDataTofLogRhoFit1()
eostablename = myeosname+"_eostable_Ye=%06.3f_c30p10.dat" % \
(par2)
energy_shift = myeos.h5file['energy_shift'][0]
for i in range(nrhos):
temp = tempfunc(logrhos[i])
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
press = myeos.query('logpress')
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
eps = myeos.query('logenergy')
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'c30p5_fixed_Ye'):
ye = par2
tempfunc = kentaDataTofLogRhoFit2()
eostablename = myeosname+"_eostable_Ye=%06.3f_c30p5.dat" % \
(par2)
energy_shift = myeos.h5file['energy_shift'][0]
for i in range(nrhos):
temp = tempfunc(logrhos[i])
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
press = myeos.query('logpress')
myeos.setState({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp})
eps = myeos.query('logenergy')
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'fixed_temp_betaeq'):
eostablename = myeosname+"_eostable_BetaEq_T=%06.3f.dat" % (par1)
energy_shift = myeos.h5file['energy_shift'][0]
temp = par1
lastPress = -1e-300
lastEps = None
for i in range(nrhos):
ye = myeos.setBetaEqState({'rho': 10.0**logrhos[i],
'temp': temp})
(press,eps) = myeos.query(['logpress','logenergy'])
if press < lastPress:
print "WOW, PRESSURE DECREASE WHEN INCREASING DENSITY, THAT'S F****D"
print "MANUALLY SETTING VALUES TO LAST VALUES"
press = lastPress
eps = lastEps
lastPress = press
lastEps = eps
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
eostable2[i,0] = eostable[i,0]
eostable2[i,1] = eostable[i,1]
eostable2[i,2] = ye
eostable2[i,3] = temp
print "Making EOS: %15.6E %15.6E %15.6E" % (10.0**logrhos[i],temp,ye)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
esstring = "%18.9E\n" % (energy_shift/eps_gf)
eosfile.write(esstring)
for i in range(len(eostable2[:,0])):
sline = "%22.14E %22.14E %22.14E %22.14E %22.14E\n" % \
(logrhos[i],eostable2[i,0],eostable2[i,1],eostable2[i,2],eostable2[i,3])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'fixed_ye_entropy'):
ye = par2
entropy = par1
energy_shift = myeos.h5file['energy_shift'][0]
eostablename = "eostable_ye=%06.3f_s=%06.3f.dat" % (par2,par1)
print eostablename
try:
eosin = (numpy.loadtxt(eostablename))
eostable[:,0] = eosin[:,1]
eostable[:,1] = eosin[:,2]
except IOError:
for i in range(nrhos):
temp = 0.0
temp = myeos.getTemperatureFromQuantityTYe({'rho': 10.0**logrhos[i],
'ye': ye,
'temp': temp},'entropy', entropy)
myeos.setState({'rho': 10.0**logrhos[i],
'ye' : ye,
'temp': temp})
(press,eps) = myeos.query(['logpress','logenergy'])
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
print "Making EOS: %15.6E %15.6E %15.6E" % (10.0**logrhos[i],temp,ye)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'fixed_entropy_betaeq'):
energy_shift = myeos.h5file['energy_shift'][0]
entropy = par1
eostablename = "eostable_betaeq_s=%06.3f.dat" % (par1)
print eostablename
try:
eosin = (numpy.loadtxt(eostablename))
eostable[:,0] = eosin[:,1]
eostable[:,1] = eosin[:,2]
except IOError:
for i in range(nrhos):
(ye,temp) = \
myeos.setConstQuantityAndBetaEqState({'rho': 10.0**logrhos[i]},\
'entropy',par1)
(press,eps) = myeos.query(['logpress','logenergy'])
# convert units
eostable[i,0] = log10(10.0**press * press_gf)
eostable[i,1] = log10(10.0**eps * eps_gf)
print "Making EOS: %15.6E %15.6E %15.6E" % (10.0**logrhos[i],temp,ye)
energy_shift = energy_shift*eps_gf
# write EOS table
eosfile=open(eostablename,"w")
for i in range(len(eostable[:,0])):
sline = "%15.6E %15.6E %15.6E\n" % \
(logrhos[i],eostable[i,0],eostable[i,1])
eosfile.write(sline)
eosfile.close()
elif(mytype == 'poly_G2_K100'):
eostable[:,0] = log10(100.0*(10.0**logrhos[:]*rho_gf)**2.0)
eostable[:,1] = log10(10.0**eostable[:,0]/(10.0**logrhos[:]*rho_gf) )
else:
print "This kind of table can't be done yet: ",mytype
sys.exit()
logrhos = log10(10.0**logrhos * rho_gf)
dlrho = logrhos[1]-logrhos[0]
return (eostable,energy_shift,dlrho,logrhos)
'c20p0': 'v',
'c40p0': '^',
'T=00010': '*',
'c30p5': 'p',
'c30p10': 'H'}
scripts = {'c30p0': deepcopy(modelParamsTemplate.copy()),
'c20p0': deepcopy(modelParamsTemplate.copy()),
'c40p0': deepcopy(modelParamsTemplate.copy()),
'T=00010': deepcopy(modelParamsTemplate.copy()),
'c30p5': deepcopy(modelParamsTemplate.copy()),
'c30p10': deepcopy(modelParamsTemplate.copy())}
shen = eosDriver.eosDriver('/home/jeff/work/HShenEOS_rho220_temp180_ye65_version_1.1_20120817.h5')
tfuncs = {'c30p0': eosDriver.getTRollFunc(30.0, 0.01, 14.055, .375),
'c20p0': eosDriver.getTRollFunc(20.0, 0.01, 13.93, .25),
'c40p0': eosDriver.getTRollFunc(40.0, 0.01, 14.18, .5),
'T=00010': lambda x: 0.01,
'c30p5': eosDriver.kentaDataTofLogRhoFit2(),
'c30p10': eosDriver.kentaDataTofLogRhoFit1()}
ye = 0.1
massFraction = .5
def thermalPressureSupport(rho, tfunc):
shen.setState({'rho': rho, 'temp': 0.01, 'ye': ye})
cold = shen.query('logpress', deLog10Result=True)
shen.setState({'rho': rho, 'temp': tfunc(rho), 'ye': ye })
hot = shen.query('logpress', deLog10Result=True)
print ls220.query(["logpress", "entropy", "logenergy"])
max = 30.0
min = 0.01
logrhos = numpy.arange(10.0, 16.0, 0.05)
rhos = numpy.power(10.0, logrhos)
midsAndScales = [(14.0, 0.5), (13.5, 0.5), (14.0, 0.25)]
# ls220.writeRotNSeosfile("test10WithNuYeLess-2.eos", {'funcTofLogRho': 'kentaDataTofLogRhoFit1'}, None, True)
labels = []
# print shen.getBetaEqYeVsRhobTable(kentaDataTofLogRhoFit1(), 14, 16)
for mid, scale in midsAndScales:
mpl.plot(logrhos, getTRollFunc(max, min, mid, scale)(logrhos), logrhos, kentaDataTofLogRhoFit1()(logrhos))
labels.append("mid=" + str(mid) + " scale=" + str(scale))
mpl.legend(labels, loc=2)
mpl.ylabel("T (MeV)")
mpl.xlabel(r"log10($\rho_b$ CGS)")
# mpl.show()
# print ls220.solveForQuantity({'rho': 1e7, 'temp': 1.0}, 'munu', 0., bounds=None)
# print ls220.solveForQuantity({'rho': 1e15, 'ye': 0.1}, 'entropy', 1., bounds=None)
ye = "BetaEq"
# shen.resetCachedRhobVsEds(getTRollFunc(20., .01, 13.93,.25), ye)
# print shen.cachedRhobVsEd
# mpl.loglog(numpy.power(10.0, shen.cachedRhobVsEd[1]), (numpy.power(10.0,shen.cachedRhobVsEd[0])-numpy.power(10.0, shen.cachedRhobVsEd[1]))/numpy.power(10.0, shen.cachedRhobVsEd[1]))
# mpl.show()
# exit()
