Refstate gravity support

Author: amorison

stagpy/phyvars.py

@@ -239,6 +239,7 @@
     ('Cp', Varr('Heat capacity', 'Heat capacity', 'J/(kg.K)')),
     ('Tcond', Varr('Conductivity', 'Conductivity', 'W/(m.K)')),
     ('P', Varr('Pressure', 'Pressure', 'Pa')),
+    ('grav', Varr('Gravity', 'Gravity', 'm/s2')),
 ))
 
 Varp = namedtuple('Varp', ['description'])
@@ -266,6 +267,7 @@
     '1/s': lambda scl: 1 / scl.time,
     'K/s': lambda scl: scl.temperature / scl.time,
     'm/s': lambda scl: scl.length / scl.time,
+    'm/s2': lambda scl: scl.length / scl.time**2,
 }
 
 PREFIXES = ('k', 'M', 'G')

stagpy/stagyyparsers.py

@@ -251,7 +251,7 @@ def _clean_names_refstate(names):
     return [to_clean.get(n, n) for n in names]
 
 
-def refstate(reffile, ncols=7):
+def refstate(reffile, ncols=8):
     """Extract reference state profiles.
 
     Args: