Feat/line mapping initialization

discopop_explorer/discopop_explorer.py

@@ -18,7 +18,8 @@
 import pstats2  # type:ignore
 from pluginbase import PluginBase  # type:ignore
 
-from discopop_library.PathManagement.PathManagement import get_path
+from discopop_library.LineMapping.initialize import initialize_line_mapping
+from discopop_library.PathManagement.PathManagement import get_path, load_file_mapping
 from discopop_library.discopop_optimizer.Microbench.ExtrapInterpolatedMicrobench import (
     ExtrapInterpolatedMicrobench,
 )
@@ -215,6 +216,9 @@ def run(arguments: ExplorerArguments):
             stats = pstats2.Stats(profile, stream=f).sort_stats("time").reverse_order()
             stats.print_stats()
 
+    # initialize the line_mapping.json
+    initialize_line_mapping(load_file_mapping(arguments.file_mapping_file), arguments.project_path)
+
     print("Time taken for pattern detection: {0}".format(end - start))
 
     # demonstration of Microbenchmark possibilities

discopop_library/LineMapping/initialize.py

@@ -0,0 +1,40 @@
+# This file is part of the DiscoPoP software (http://www.discopop.tu-darmstadt.de)
+#
+# Copyright (c) 2020, Technische Universitaet Darmstadt, Germany
+#
+# This software may be modified and distributed under the terms of
+# the 3-Clause BSD License.  See the LICENSE file in the package base
+# directory for details.
+
+import json
+import os.path
+from pathlib import Path
+from typing import Dict
+
+from discopop_library.LineMapping.load import load_line_mapping
+from discopop_library.LineMapping.save import save_line_mapping
+
+
+def initialize_line_mapping(
+    file_mapping: Dict[int, Path],
+    discopop_path: str = ".discopop",
+):
+    """initializes the line mapping dictionary to track line shifts due to inserted pragmas.
+    The Dictionary will be stored in .discopop/line_mapping.json.
+    Line ids start with 1."""
+    line_mapping_dict: Dict[int, Dict[int, int]] = dict()
+
+    # initialize line mapping (1->1, 2->2, ...)
+    for file_id in file_mapping:
+        if file_id not in line_mapping_dict:
+            line_mapping_dict[file_id] = dict()
+        line_id = 1
+        with open(file_mapping[file_id], "r") as f:
+            for line in f.readlines():
+                line_mapping_dict[file_id][line_id] = line_id
+                line_id += 1
+
+    save_line_mapping(line_mapping_dict, discopop_path)
+
+    # debug
+    load_line_mapping(discopop_path)

discopop_library/LineMapping/load.py

@@ -0,0 +1,18 @@
+# This file is part of the DiscoPoP software (http://www.discopop.tu-darmstadt.de)
+#
+# Copyright (c) 2020, Technische Universitaet Darmstadt, Germany
+#
+# This software may be modified and distributed under the terms of
+# the 3-Clause BSD License.  See the LICENSE file in the package base
+# directory for details.
+import json
+import os.path
+from typing import Dict
+
+
+def load_line_mapping(discopop_path: str = ".discopop") -> Dict[int, Dict[int, int]]:
+    """Loads and returns the line_mapping dictionary"""
+    line_mapping_dict: Dict[int, Dict[int, int]] = dict()
+    with open(os.path.join(discopop_path, "line_mapping.json")) as f:
+        line_mapping_dict = json.load(f)
+    return line_mapping_dict

discopop_library/LineMapping/save.py

@@ -0,0 +1,17 @@
+# This file is part of the DiscoPoP software (http://www.discopop.tu-darmstadt.de)
+#
+# Copyright (c) 2020, Technische Universitaet Darmstadt, Germany
+#
+# This software may be modified and distributed under the terms of
+# the 3-Clause BSD License.  See the LICENSE file in the package base
+# directory for details.
+import json
+import os
+from typing import Dict
+
+
+def save_line_mapping(line_mapping_dict: Dict[int, Dict[int, int]], discopop_path: str = ".discopop"):
+    """dumps line_mapping_dict to line_mapping.json"""
+    # dump line mapping to json file
+    with open(os.path.join(discopop_path, "line_mapping.json"), "w+") as f:
+        json.dump(line_mapping_dict, f)