From ce4cd882c2ab27ab2e26aa6d1cb68e08332e5692 Mon Sep 17 00:00:00 2001
From: podlesny <podlesny@zedat.fu-berlin.de>
Date: Sun, 28 Mar 2021 23:56:45 +0200
Subject: [PATCH] now plots colorbar as legend

---
 data/tools/support/slip_rates.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/data/tools/support/slip_rates.py b/data/tools/support/slip_rates.py
index d122199d..1002e0aa 100644
--- a/data/tools/support/slip_rates.py
+++ b/data/tools/support/slip_rates.py
@@ -2,13 +2,13 @@ import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.cm as cm
 
-TANGENTIAL_COORDS = 1
+SORT_COORD = 0
 
 def local_to_global(coords, SORT_COORD=0):
     return coords[:, SORT_COORD].argsort()
 
-def slip_rates(h5file, body_ID, FINAL_TIME, patch=[], interval=[], SORT_COORD=0):
-    body = 'body' + str(body_ID) # 'frictionalBoundary'     'body' + str(body_ID)
+def slip_rates(h5file, body_ID, FINAL_TIME, patch=[], interval=[], TANGENTIAL_COORDS=1):
+    body = 'body' + str(body_ID)   # 'frictionalBoundary'     'body' + str(body_ID)
 
     coords = np.array(h5file[body + '/coordinates'])
     if len(patch) == 0:
@@ -37,10 +37,12 @@ def slip_rates(h5file, body_ID, FINAL_TIME, patch=[], interval=[], SORT_COORD=0)
 
     # plot
     ax_slip = fig.add_subplot(1, 1, 1)
-    ax_slip.imshow(v_tx, interpolation='nearest', cmap=cm.Greys_r, aspect='auto', origin='lower', extent=[0, patch_size, time[0], time[-1]])
+    pos = ax_slip.imshow(v_tx, interpolation='nearest', cmap=cm.Greys_r, aspect='auto', origin='lower', extent=[0, patch_size, time[0], time[-1]])
     #ax_slip.set_ylim([time[-1], time[0]])
     ax_slip.set_ylabel('time [s]')
     ax_slip.set_xlabel('horizontal coordinate [m]')
+    fig.colorbar(pos, ax=ax_slip)
+
     #ax_slip.set_yscale('log')
    # ax_slip.set_ylim([1e-6,1e-2])
     #-------------------------
-- 
GitLab