Huge refactoring

2021-06-22 12:23:03 +02:00
parent a286c08b72
commit 0dc9e8fc7b
12 changed files with 343 additions and 361 deletions
@@ -0,0 +1,36 @@
+# coding: utf-8
+
+import os
+import re
+
+import munch
+import yaml
+
+_dir_path = os.path.dirname(os.path.realpath(__file__))
+
+# Add support for '1e3' kind of float
+_loader = yaml.SafeLoader
+_loader.add_implicit_resolver(
+    u"tag:yaml.org,2002:float",
+    re.compile(
+        u"""^(?:
+     [-+]?(?:[0-9][0-9_]*)\\.[0-9_]*(?:[eE][-+]?[0-9]+)?
+    |[-+]?(?:[0-9][0-9_]*)(?:[eE][-+]?[0-9]+)
+    |\\.[0-9_]+(?:[eE][-+][0-9]+)?
+    |[-+]?[0-9][0-9_]*(?::[0-5]?[0-9])+\\.[0-9_]*
+    |[-+]?\\.(?:inf|Inf|INF)
+    |\\.(?:nan|NaN|NAN))$""",
+        re.X,
+    ),
+    list(u"-+0123456789."),
+)
+
+
+def load_params(filename):
+    with open(filename) as f:
+        para_disk = yaml.load(f, Loader=_loader)
+    return munch.munchify(para_disk)
+
+
+def default_params():
+    return load_params(_dir_path + "/params.yml")