bracket units in axis labels

Author: betaBison

gnss_lib_py/utils/visualizations.py

@@ -579,6 +579,16 @@ def _get_label(inputs):
     if not isinstance(inputs,dict):
         raise TypeError("_get_label input must be dictionary.")
 
+    # handle units specially.
+    units = {"m","km",
+             "deg","rad",
+             "sec","s","hr","min",
+             "mps",
+             }
+    unit_replacements = {
+                         "mps" : "m/s",
+                        }
+
     label = ""
     for key, value in inputs.items():
 
@@ -593,8 +603,14 @@ def _get_label(inputs):
         except ValueError:
             pass
 
-        value = value.upper()
-        value = value.replace("_"," ")
+        value = value.split("_")
+        if value[-1] in units:
+            # make units lowercase and bracketed.
+            if value[-1] in unit_replacements:
+                value[-1] = unit_replacements[value[-1]]
+            value = " ".join(value[:-1]).upper() + " [" + value[-1] + "]"
+        else:
+            value = " ".join(value).upper()
 
         if key == "gnss_id": # use GNSS specific capitalization
             constellation_map = {"GALILEO" : "Galileo",

tests/utils/test_visualizations.py

@@ -415,6 +415,10 @@ def test_get_label():
     assert viz._get_label({"gnss_id" : "galileo",
                            "signal_type" : "b1i"}) == "Galileo B1i"
 
+    assert viz._get_label({"row" : "x_rx_m"}) == "X RX [m]"
+    assert viz._get_label({"row" : "lat_rx_deg"}) == "LAT RX [deg]"
+    assert viz._get_label({"row" : "vx_sv_mps"}) == "VX SV [m/s]"
+
     with pytest.raises(TypeError) as excinfo:
         viz._get_label(["should","fail"])
     assert "dictionary" in str(excinfo.value)