diff --git a/disk_postprocess.py b/disk_postprocess.py index da62519..9872fd9 100644 --- a/disk_postprocess.py +++ b/disk_postprocess.py @@ -341,7 +341,7 @@ def make_image_aux( dmap_omega = rt_omega.process(cam) dmap_cs = rt_cs.process(cam) dmap_Q = (lbox * dmap_cs.map.T) * dmap_omega.map.T / (np.pi * G * dmap_col) - map_Q = np.log10(dmap_Q) + map_Q = dmap_Q im = P.imshow( map_Q, @@ -352,6 +352,7 @@ def make_image_aux( (radius + center[1]) * lbox_units, ], origin="lower", + norm=mpl.colors.LogNorm(), ) P.locator_params(axis="x", nbins=ntick) @@ -362,7 +363,7 @@ def make_image_aux( P.xlabel(title_ax[ax_h]) P.ylabel(title_ax[ax_v]) cbar = P.colorbar(im) - cbar.set_label(r"$log(Q)$") + cbar.set_label(r"$Q$") name = directory + "/Q_" + ax_los + "_" + tag + "_" + format(num, "05") name_im = name + out_ext @@ -747,7 +748,8 @@ def plot_disk_prop(path, num, force=False, tag="", interactive=False): P.close() # Q - P.ylim([0, 10.0]) + P.ylim([0, 5.0]) + P.grid() P.plot(prop_disk["rad"], abs(prop_disk["Q_kepl"]), color="b", linewidth=2) P.ylabel(r"$Q$") P.xlabel("disk radius ")