From 912feb17708f00fd8b6c30433ad6985c2ca1cf55 Mon Sep 17 00:00:00 2001
From: Stefan Reck <stefan.reck@fau.de>
Date: Tue, 13 Apr 2021 19:33:11 +0200
Subject: [PATCH] draft for toml parser
---
orcasong/toml.py | 37 +++++++++++++++++++++++++++++++++++++
1 file changed, 37 insertions(+)
create mode 100644 orcasong/toml.py
diff --git a/orcasong/toml.py b/orcasong/toml.py
new file mode 100644
index 0000000..94e4374
--- /dev/null
+++ b/orcasong/toml.py
@@ -0,0 +1,37 @@
+import os
+import toml
+from orcasong.core import FileGraph
+from orcasong.extractors import get_neutrino_mc_info_extr
+
+
+EXTRACTORS = {
+ "neutrino": get_neutrino_mc_info_extr,
+}
+
+
+def add_parser(subparsers):
+ parser = subparsers.add_parser(
+ "graph",
+ description='Produce graph dl file from aanet file.')
+ parser.add_argument('infile', type=str)
+ parser.add_argument('config', type=str)
+ parser.add_argument('detx_file', type=str)
+ parser.add_argument('--outfile', type=str, default=None)
+ return parser.parse_args()
+
+
+def make_graph(infile, config, detx_file, outfile=None):
+ if outfile is None:
+ outfile = f"{os.path.splitext(os.path.basename(infile))[0]}_dl.h5"
+
+ cfg = toml.load(config)
+ extractor_cfg = cfg.pop("extractor")
+ extractor_name = extractor_cfg.pop("name")
+ inps = {"infile": infile, "config": config, "detx_file": detx_file, "outfile": outfile}
+ extractor = EXTRACTORS[extractor_name](inps, **extractor_cfg)
+
+ FileGraph(
+ det_file=detx_file,
+ extractor=extractor,
+ **cfg,
+ ).run(infile=infile, outfile=outfile)
--
GitLab