Merge pull request #20 from nemocrys:arvedes/issue7

Read output of SaveData/SaveScalars solver to pandas dataframe

CHANGES.txt

@@ -1,3 +1,6 @@
+v1.0.3
+- post: add function to read output of SaveData/SaveScalars solver to pandas dataframe
+
 v1.0.2
 - add referencing to other bodies / boundaries (issue #17) using the StringFromList class
 

pyelmer/post.py

@@ -2,6 +2,7 @@
 import re
 import numpy as np
 import matplotlib.pyplot as plt
+import pandas as pd
 from dataclasses import dataclass
 
 
@@ -224,3 +225,25 @@ def plot_residuals(sim_dir, solvers, save=False):
             fig.savefig(fig_dir + fig.title + ".png")
 
     return figs, axes
+
+
+def dat_to_dataframe(dat_file):
+    """Read a .dat file generated by elmer (e.g. SaveData/SaveScalars
+    solver) into a pandas dataframe.
+
+    Args:
+        dat_file (str): file path of elmer .dat file
+
+    Returns:
+        pandas.core.frame.DataFrame: content of dat file with header
+    """
+    with open(f"{dat_file}.names") as f:
+        lines = f.readlines()
+    names = []
+    names_start = False
+    for line in lines:
+        if names_start == True:
+            names.append(line.split(":")[-1].strip())
+        if "Data on different columns" in line:
+            names_start = True
+    return pd.read_table(dat_file, names=names, delim_whitespace=True)

setup.py

@@ -27,6 +27,7 @@
         "matplotlib",
         "numpy",
         "objectgmsh",
+        "pandas",
     ],
     python_requires=">=3.7",
 )