Include station location in annotation, if present

uafgeotools · Aug 27, 2023 · 5ff1a72 · 5ff1a72
1 parent 7aa099a
commit 5ff1a72
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/mtuq/graphics/annotations.py b/mtuq/graphics/annotations.py
@@ -14,8 +14,13 @@ def station_label_writer(ax, station, origin, units='km'):
         station.latitude,
         station.longitude)
 
-    # display station name
-    label = '.'.join([station.network,station.station])
+    # construct station label
+    if len(station.location) > 1:
+        label = station.network + '.' + station.station + '.' + station.location
+    else:
+        label = station.network + '.' + station.station
+
+    # display station label
     pyplot.text(0.2,0.50, label, fontsize=11, transform=ax.transAxes)
 
     # display distance

diff --git a/mtuq/graphics/waveforms.py b/mtuq/graphics/waveforms.py
@@ -329,7 +329,7 @@ def plot_data_greens2(filename,
 def _initialize(nrows=None, ncolumns=None, column_width_ratios=None, 
     header=None, height=None, width=None, margin_top=0.25, margin_bottom=0.25,
     margin_left=0.25, margin_right=0.25, header_height=1.5, 
-    station_labels=True, station_label_width=0.4):
+    station_labels=True, station_label_width=0.5):
 
     if header:
         height += header_height