added ability to merge analyses

src/kaskara/analysis.py

@@ -53,6 +53,16 @@ def with_relative_locations(self, base: str) -> Analysis:
             insertions=self.insertions.with_relative_locations(base),
         )
 
+    def merge(self, other: Analysis) -> Analysis:
+        """Merges the results of this analysis and another analysis together."""
+        return Analysis(
+            files=self.files.union(other.files),
+            loops=self.loops.merge(other.loops),
+            functions=self.functions.merge(other.functions),
+            statements=self.statements.merge(other.statements),
+            insertions=self.insertions.merge(other.insertions),
+        )
+
     def is_inside_loop(self, location: FileLocation) -> bool:
         return self.loops.is_within_loop(location)
 

src/kaskara/functions.py

@@ -4,6 +4,7 @@
 __all__ = ("Function", "ProgramFunctions")
 
 import abc
+import itertools
 import os
 import typing as t
 from dataclasses import dataclass
@@ -82,6 +83,13 @@ def to_dict(self) -> dict[str, t.Any]:
             "functions": [f.to_dict() for f in self],
         }
 
+    def merge(self, other: ProgramFunctions) -> ProgramFunctions:
+        """Merges the results of this analysis and another analysis together."""
+        return self.from_functions(
+            project_directory=self._project_directory,
+            functions=itertools.chain(self, other),
+        )
+
     def with_relative_locations(self, base: str) -> ProgramFunctions:
         """Creates a new instance with relative file locations."""
         functions = [f.with_relative_locations(base) for f in self]

src/kaskara/insertions.py

@@ -36,6 +36,9 @@ def __init__(self, contents: list[InsertionPoint]) -> None:
                 self.__file_insertions[filename] = []
             self.__file_insertions[filename].append(ins)
 
+    def merge(self, other: ProgramInsertionPoints) -> ProgramInsertionPoints:
+        return ProgramInsertionPoints(list(self) + list(other))
+
     def with_relative_locations(self, base: str) -> ProgramInsertionPoints:
         return ProgramInsertionPoints([
             insertion_point.with_relative_location(base)

src/kaskara/loops.py

@@ -41,6 +41,15 @@ def from_body_location_ranges(
             FileLocationRangeSet(bodies),
         )
 
+    def merge(self, other: ProgramLoops) -> ProgramLoops:
+        covered_by_loop_bodies = self._covered_by_loop_bodies.union(
+            other._covered_by_loop_bodies,
+        )
+        return ProgramLoops(
+            _covered_by_loop_bodies=covered_by_loop_bodies,
+            _project_directory=self._project_directory,
+        )
+
     def with_relative_locations(self, base: str) -> ProgramLoops:
         """Creates a new instance with relative file locations."""
         covered_by_loop_bodies = self._covered_by_loop_bodies.with_relative_locations(

src/kaskara/statements.py

@@ -3,6 +3,7 @@
 __all__ = ("Statement", "ProgramStatements")
 
 import abc
+import itertools
 import os
 import typing as t
 from dataclasses import dataclass, field
@@ -97,6 +98,12 @@ def __post_init__(self) -> None:
         )
         logger.debug(f"indexed statements by file:\n{summary}")
 
+    def merge(self, other: ProgramStatements) -> ProgramStatements:
+        return self.build(
+            project_directory=self._project_directory,
+            statements=itertools.chain(self, other),
+        )
+
     def with_relative_locations(self, base: str) -> ProgramStatements:
         """Creates a new instance with relative file locations."""
         return self.build(