colorscale all plots the same

1 files changed, 34 insertions, 10 deletions

diff --git a/mkcuts_dvpT.py b/mkcuts_dvpT.py
index 319b7fa..ca48547 100644
--- a/mkcuts_dvpT.py
+++ b/mkcuts_dvpT.py
@@ -193,6 +193,20 @@ def main():
     print "// Plotting Snapshots"
     print ""
 
+    # Log10?
+    rhoZmax = np.log10(rhoZmax)
+    rhoXmax = np.log10(rhoXmax)
+    preZmax = np.log10(preZmax)
+    preXmax = np.log10(preXmax)
+    lTZmax = np.log10(TZmax)
+    lTXmax = np.log10(TXmax)
+    rhoZmin = np.log10(rhoZmin)
+    rhoXmin = np.log10(rhoXmin)
+    preZmin = np.log10(preZmin)
+    preXmin = np.log10(preXmin)
+    lTZmin = np.log10(TZmin)
+    lTXmin = np.log10(TXmin)
+
     # Plot Snapshots
     for ii, iout in enumerate(iouts):
         
@@ -243,7 +257,8 @@ def main():
         # Plot Density
         if args.d:
             f1 = plt.figure(1)
-            plt.imshow(rhoZ, cmap=rho_cm, extent=ext, interpolation='none')
+            plt.imshow(rhoZ, cmap=rho_cm, extent=ext, interpolation='none', \
+                       vmin=rhoZmin, vmax=rhoZmax)
             plt.colorbar()
             plt.grid()
             plt.title('Log10 Density Cut [g/cc], t=%.2f' % output.info["time"])
@@ -251,7 +266,8 @@ def main():
             plt.ylabel('Y [AU]')
 
             f2 = plt.figure(2)
-            plt.imshow(rhoX, cmap=rho_cm, extent=ext, interpolation='none')
+            plt.imshow(rhoX, cmap=rho_cm, extent=ext, interpolation='none', \
+                       vmin=rhoXmin, vmax=rhoXmax)
             plt.colorbar()
             plt.grid()
             plt.title('Log10 Density Cut [g/cc], t=%.2f' % output.info["time"])
@@ -261,7 +277,8 @@ def main():
         # Plot Velocity
         if args.v:
             f3 = plt.figure(3)
-            plt.imshow(velZ, cmap=vel_cm, extent=ext)
+            plt.imshow(velZ, cmap=vel_cm, extent=ext, \
+                       vmin=velZmin, vmax=velZmax)
             plt.colorbar()
             plt.grid()
             plt.title('Total Flow Speed Cut [km/s], t=%.2f' % output.info["time"])
@@ -269,7 +286,8 @@ def main():
             plt.ylabel('Y [AU]')
 
             f4 = plt.figure(4)
-            plt.imshow(velX, cmap=vel_cm, extent=ext)
+            plt.imshow(velX, cmap=vel_cm, extent=ext, \
+                       vmin=velXmin, vmax=velXmax)
             plt.colorbar()
             plt.grid()
             plt.title('Total Flow Speed Cut [km/s], t=%.2f' % output.info["time"])
@@ -279,7 +297,8 @@ def main():
         # Plot Pressure
         if args.p:
             f5 = plt.figure(5)
-            plt.imshow(preZ, cmap=pre_cm, extent=ext, interpolation='none')
+            plt.imshow(preZ, cmap=pre_cm, extent=ext, interpolation='none', \
+                       vmin=preZmin, vmax=preZmax)
             plt.colorbar()
             plt.grid()
             plt.title('Log10 Pressure Cut [debye], t=%.2f' % output.info["time"])
@@ -287,7 +306,8 @@ def main():
             plt.ylabel('Y [AU]')
 
             f6 = plt.figure(6)
-            plt.imshow(preX, cmap=pre_cm, extent=ext, interpolation='none')
+            plt.imshow(preX, cmap=pre_cm, extent=ext, interpolation='none', \
+                       vmin=preXmin, vmax=preXmax)
             plt.colorbar()
             plt.grid()
             plt.title('Log10 Pressure Cut [debye], t=%.2f' % output.info["time"])
@@ -297,7 +317,8 @@ def main():
         # Plot Temperature
         if args.T:
             f7 = plt.figure(7)
-            plt.imshow(TZ, cmap=tmp_cm, extent=ext, interpolation='none')
+            plt.imshow(TZ, cmap=tmp_cm, extent=ext, interpolation='none', \
+                       vmin=TZmin, vmax=TZmax)
             plt.colorbar()
             plt.grid()
             plt.title('Temperature Cut [K], t=%.2f' % output.info["time"])
@@ -305,7 +326,8 @@ def main():
             plt.ylabel('Y [AU]')
 
             f8 = plt.figure(8)
-            plt.imshow(TX, cmap=tmp_cm, extent=ext, interpolation='none')
+            plt.imshow(TX, cmap=tmp_cm, extent=ext, interpolation='none', \
+                       vmin=TXmin, vmax=TXmax)
             plt.colorbar()
             plt.grid()
             plt.title('Temperature Cut [K], t=%.2f' % output.info["time"])
@@ -315,7 +337,8 @@ def main():
         # Plot Log10 of Temperature
         if args.lT:
             f9 = plt.figure(9)
-            plt.imshow(lTZ, cmap=tmp_cm, extent=ext, interpolation='none')
+            plt.imshow(lTZ, cmap=tmp_cm, extent=ext, interpolation='none', \
+                       vmin=lTZmin, vmax=lTZmax)
             plt.colorbar()
             plt.grid()
             plt.title('Log10 Temperature Cut [K], t=%.2f' % output.info["time"])
@@ -323,7 +346,8 @@ def main():
             plt.ylabel('Y [AU]')
 
             f10 = plt.figure(10)
-            plt.imshow(lTX, cmap=tmp_cm, extent=ext, interpolation='none')
+            plt.imshow(lTX, cmap=tmp_cm, extent=ext, interpolation='none', \
+                       vmin=lTXmin, vmax=lTXmax)
             plt.colorbar()
             plt.grid()
             plt.title('Log10 Temperature Cut [K], t=%.2f' % output.info["time"])