From 3acb6e59cf8325be3e2bd9f6a70a922f2399e9fa Mon Sep 17 00:00:00 2001
From: Harshil Patel <hpppatel@ucdavis.edu>
Date: Thu, 27 Jun 2024 20:02:57 -0700
Subject: [PATCH] resources: Update elfie.py to work with obtain_resources
 (#1289)

Change-Id: I08c5e50a150c8434c6c2ca36af81fb6ec3915af8
---
 src/python/gem5/resources/elfie.py    | 29 +++++++++++++++++++++++++--
 src/python/gem5/resources/resource.py |  2 ++
 2 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/src/python/gem5/resources/elfie.py b/src/python/gem5/resources/elfie.py
index d90930d3f0..840a664a97 100644
--- a/src/python/gem5/resources/elfie.py
+++ b/src/python/gem5/resources/elfie.py
@@ -24,7 +24,10 @@
 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
-from typing import List
+from typing import (
+    List,
+    Optional,
+)
 
 from m5.objects import PcCountTrackerManager
 from m5.params import PcCountPair
@@ -36,9 +39,31 @@ class ELFieInfo:
     See https://github.com/intel/pinball2elf for more information.
     """
 
-    def __init__(self, start: PcCountPair, end: PcCountPair):
+    def __init__(
+        self,
+        start_pc: Optional[str] = None,
+        end_pc: Optional[str] = None,
+        start_pc_count: Optional[str] = None,
+        end_pc_count: Optional[str] = None,
+        start: Optional["PcCountPair"] = None,
+        end: Optional["PcCountPair"] = None,
+        **kwargs
+    ):
         self._start = start
         self._end = end
+
+        if self._start is None:
+            if start_pc is None or start_pc_count is None:
+                raise ValueError(
+                    "start_pc and start_pc_count must be provided"
+                )
+            self._start = PcCountPair(int(start_pc, 16), int(start_pc_count))
+
+        if self._end is None:
+            if end_pc is None or end_pc_count is None:
+                raise ValueError("end_pc and end_pc_count must be provided")
+            self._end = PcCountPair(int(end_pc, 16), int(end_pc_count))
+
         self._manager = PcCountTrackerManager()
         self._manager.targets = self.get_targets()
 
diff --git a/src/python/gem5/resources/resource.py b/src/python/gem5/resources/resource.py
index bccf0a1b87..eb3ca78fe9 100644
--- a/src/python/gem5/resources/resource.py
+++ b/src/python/gem5/resources/resource.py
@@ -57,6 +57,7 @@ from .client import (
 )
 from .client_api.client_query import ClientQuery
 from .downloader import get_resource
+from .elfie import ELFieInfo
 from .looppoint import (
     LooppointCsvLoader,
     LooppointJsonLoader,
@@ -1411,4 +1412,5 @@ _get_resource_json_type_map = {
     "looppoint-json": LooppointJsonResource,
     "suite": SuiteResource,
     "workload": WorkloadResource,
+    "elfie-info": ELFieInfo,
 }