diff options
Diffstat (limited to 'mkcuts_dvpT.py')
-rw-r--r-- | mkcuts_dvpT.py | 49 |
1 files changed, 48 insertions, 1 deletions
diff --git a/mkcuts_dvpT.py b/mkcuts_dvpT.py index 682ea4f..8affbdc 100644 --- a/mkcuts_dvpT.py +++ b/mkcuts_dvpT.py @@ -67,7 +67,23 @@ def main(): op_pre = ScalarOperator(func_pre) op_cs2 = FractionOperator(func_pre, func_rho) - # The Master Loop + # Init Master Lists + rhoZall = [] + rhoXall = [] + velZall = [] + velXall = [] + preZall = [] + preXall = [] + TZall = [] + TXall = [] + lTZall = [] + lTXall = [] + + print "" + print "// Loading Snapshots" + print "" + + # Load Snapshots for iout in iouts: # Give Feedback @@ -131,7 +147,38 @@ def main(): velX = np.ma.masked_array(velX, mask=np.isnan(velX)) preZ = np.ma.masked_array(preZ, mask=np.isnan(preZ)) preX = np.ma.masked_array(preX, mask=np.isnan(preX)) + + # Append to Master List + rhoZall.append(rhoZ) + rhoXall.append(rhoX) + velZall.append(velZ) + velXall.append(velX) + preZall.append(preZ) + preXall.append(preX) + TZall.append(TZ) + TXall.append(TX) + lTZall.append(lTZ) + lTXall.append(lTX) + + print "" + print "// Plotting Snapshots" + print "" + + # Plot Snapshots + for ii, iout in enumerate(iouts): + # Load from Master List + rhoZ = rhoZall[ii] + rhoX = rhoXall[ii] + velZ = velZall[ii] + velX = velXall[ii] + preZ = preZall[ii] + preX = preXall[ii] + TZ = TZall[ii] + TX = TXall[ii] + lTZ = lTZall[ii] + lTX = lTXall[ii] + # Set Masked Colormap # cmap = mpl.cm.hot cmap = mpl.cm.jet |