Merge pull request #248 from kedoxey/dev

Added function to return state variables

Author: rgerkin

neuronunit/models/lems.py

@@ -206,4 +206,21 @@ def has_pulse_generator(self, tree=None):
     def temp_dir(self):
         if not hasattr(self, '_temp_dir'):
             self._temp_dir = TemporaryDirectory()
-        return self._temp_dir
+        return self._temp_dir
+
+    def get_state_variables(self):
+        """
+        Parses LEMS xml file and gets simulation's state variables from
+        OutputFile element.
+
+        Returns:
+            Dict of type "string: string" with the format "id: quantity"
+
+         """
+        lems_tree = etree.parse(self.lems_file_path)
+        state_variables = {'t': 't'}
+        for output_column in lems_tree.iter('OutputColumn'):
+            id = output_column.attrib['id']
+            quantity = output_column.attrib['quantity']
+            state_variables[id] = quantity
+        return state_variables