cool = TorchCool(args.coolfile, axial=True)
nh = cfd.get_var('nh')
tem = cfd.get_var('tem')
crate = cool.get_var('cool')
L = np.absolute(cool.get_var('lcool')/(nh*nh))
E = 1.5*nh*1.3806488e-16*tem
with np.errstate(divide='ignore', over='ignore'):
dt = E / np.absolute(crate) / 3.15569e7
dt[crate == 0] = 1e10
dt[~np.isfinite(dt)] = 1e10
L[dt > 20] = L[dt < 20].min()
vs = [nh, tem, L, dt]
for i in range(3):
vs[i] = cfd.safe_log10(vs[i])
vsminmax = [[-0.87, 6.89],
[None, None],
[None, None],
[0, 20]]
### Plotting
plotter = TorchPlotter(cfd, plot_size=5, figformat=fformat, dpi=300)
### Interpolation set up.
vsi = []
for i in range(4):
vsi.append(cfd.interpolate(vs[i], 'nearest'))
vsminmax[i][0] = (vs[i].min() if vsminmax[i][0] == None else vsminmax[i][0])
vsminmax[i][1] = (vsi[i].max() if vsminmax[i][1] == None else vsminmax[i][1])
