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])
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")