Info learned +dimensional switch

Author: amorison

stagpy/args.py

@@ -26,7 +26,7 @@ def _sub(cmd, *sections):
     ('time', _sub(time_series, 'core', 'plot', 'scaling')),
     ('refstate', _sub(refstate, 'core', 'plot')),
     ('plates', _sub(plates, 'core', 'plot', 'scaling')),
-    ('info', _sub(commands.info_cmd, 'core')),
+    ('info', _sub(commands.info_cmd, 'core', 'scaling')),
     ('var', _sub(commands.var_cmd)),
     ('version', _sub(commands.version_cmd)),
     ('config', _sub(commands.config_cmd)),

stagpy/commands.py

@@ -7,6 +7,7 @@
 import sys
 
 import loam.tools
+import pandas
 
 from . import conf, phyvars, __version__
 from . import stagyydata
@@ -49,7 +50,23 @@ def info_cmd():
             print(', snapshot {}/{}'.format(step.isnap, lsnap.isnap))
         else:
             print()
-        print(indent(step.timeinfo.loc[varlist].to_string(), '  '))
+        series = step.timeinfo.loc[varlist]
+        if conf.scaling.dimensional:
+            series = series.copy()
+            dimensions = []
+            for var, val in series.iteritems():
+                dim = phyvars.TIME.get(var) or phyvars.TIME_EXTRA.get(var)
+                dim = dim.dim if dim is not None else '1'
+                if dim == '1':
+                    dimensions.append('')
+                else:
+                    series[var], dim = sdat.scale(val, dim)
+                    dimensions.append(dim)
+            series = pandas.concat(
+                [series,
+                 pandas.DataFrame(data=dimensions, index=series.index)],
+                axis=1)
+        print(indent(series.to_string(header=False), '  '))
         print()