Ejemplo n.º 1
0
inputfile = args.inputfile
outputfile = 'four-panel-champagne.' + figformat

###	Data set up.
datacubes = []
vs_types = []
vsminmax = []

for i in range(4):
	datacubes.append(torch.CFD_Data(inputfile, axial=False))

color_maps = []
color_maps.append(hgspy.get_density_cmap())
color_maps.append(hgspy.get_water_cmap())
color_maps.append(hgspy.get_temperature_cmap())
color_maps.append(hgspy.get_par_cmap())


vs_types = []
vs_types.append(torch.VarType("nh", isLog10=datacubes[0].appropriate_to_log("nh")))
vs_types.append(torch.VarType("pre", isLog10=datacubes[1].appropriate_to_log("pre")))
vs_types.append(torch.VarType("tem", isLog10=datacubes[2].appropriate_to_log("tem")))
vs_types.append(torch.VarType("hii", isLog10=datacubes[3].appropriate_to_log("hii")))

vsminmax = []
#vsminmax.append([3.5, 4.8])
#vsminmax.append([-10, -5])
#vsminmax.append([2   , 4])
#vsminmax.append([1, 5])
Ejemplo n.º 2
0
load_src("torch", "../torchpack/torch.py")
import torch
load_src("hgspy", "../torchpack/hgspy.py")
import hgspy
load_src("fmt", "../torchpack/formatting.py")
import fmt
load_src("mp1", "../torchpack/modelpaper1.py")
import mp1
mp1_data = mp1.ModelData()

DPI = 300
figformat = 'png'
plot_size = 10
torch.set_font_sizes(fontsize=8)
cmap = hgspy.get_temperature_cmap()

def addImage(index, grid, xrange):
	col = int(index / 9)
	row = index % 9

	filename = mp1_data.getDataFilename(row, col, 25)

	ax = grid.grid[col][row]
	cbar_ax = grid.cgrid[col][row]


	data = torch.CFD_Data(filename, axial=True)

	vs = data.get_var(torch.VarType("tem", True))
	vsi = data.interpolate(vs, "linear")