Analysis: record files that were indexed

src/kaskara/analysis.py

@@ -3,6 +3,7 @@
 __all__ = ("Analysis",)
 
 import json
+import os
 import typing as t
 
 import attr
@@ -21,21 +22,31 @@ class Analysis:
 
     Attributes
     ----------
+    files: t.FrozenSet[str]
+        The set of files that were analyzed within the program.
     loops: ProgramLoops
         The set of loop control-flow statements within the program.
     functions: ProgramFunctions
         The set of functions within the program.
     statements: ProgramStatements
         The set of statements within the program.
     """
+    files: frozenset[str]
     loops: ProgramLoops
     functions: ProgramFunctions
     statements: ProgramStatements
     insertions: ProgramInsertionPoints
 
     def with_relative_locations(self, base: str) -> Analysis:
+        files: set[str] = set()
+        for filename in self.files:
+            if os.path.isabs(filename):
+                filename = os.path.relpath(filename, base)  # noqa: PLW2901
+            files.add(filename)
+
         return attr.evolve(
             self,
+            files=frozenset(files),
             loops=self.loops.with_relative_locations(base),
             functions=self.functions.with_relative_locations(base),
             statements=self.statements.with_relative_locations(base),

src/kaskara/clang/analyser.py

@@ -51,6 +51,7 @@ def run(self) -> Analysis:
         statements = self._find_statements()
         insertions = statements.insertions()
         return Analysis(
+            files=self._project.files,
             loops=loops,
             functions=functions,
             statements=statements,

src/kaskara/python/analyser.py

@@ -37,6 +37,7 @@ def run(self) -> Analysis:
         loops = collect_loops(self._container)
         insertions = statements.insertions()
         return Analysis(
+            files=self._project.files,
             functions=functions,
             statements=statements,
             insertions=insertions,

src/kaskara/spoon/analyser.py

@@ -112,6 +112,7 @@ def run(self) -> Analysis:
         insertions = statements.insertions()
 
         return Analysis(
+            files=self._project.files,
             loops=loops,
             functions=functions,
             statements=statements,