[gem5-dev] changeset in gem5: Config: Add support for a Self.all proxy object

[Ali Saidi]

changeset b8c3c20d0385 in /z/repo/gem5
details: http://repo.gem5.org/gem5?cmd=changeset;node=b8c3c20d0385
description:
        Config: Add support for a Self.all proxy object

diffstat:

 src/python/m5/SimObject.py |  15 +++++++++++++++
 src/python/m5/params.py    |   5 ++++-
 src/python/m5/proxy.py     |  11 +++++++++++
 3 files changed, 30 insertions(+), 1 deletions(-)

diffs (68 lines):

diff -r 510879e6bea2 -r b8c3c20d0385 src/python/m5/SimObject.py
--- a/src/python/m5/SimObject.py        Sun Jul 10 12:56:08 2011 -0500
+++ b/src/python/m5/SimObject.py        Sun Jul 10 12:56:08 2011 -0500
@@ -746,6 +746,21 @@
                 found_obj = match_obj
         return found_obj, found_obj != None
 
+    def find_all(self, ptype):
+        all = {}
+        # search children
+        for child in self._children.itervalues():
+            if isinstance(child, ptype) and not isproxy(child) and \
+               not isNullPointer(child):
+                all[child] = True
+        # search param space
+        for pname,pdesc in self._params.iteritems():
+            if issubclass(pdesc.ptype, ptype):
+                match_obj = self._values[pname]
+                if not isproxy(match_obj) and not isNullPointer(match_obj):
+                    all[match_obj] = True
+        return all.keys(), True
+
     def unproxy(self, base):
         return self
 
diff -r 510879e6bea2 -r b8c3c20d0385 src/python/m5/params.py
--- a/src/python/m5/params.py   Sun Jul 10 12:56:08 2011 -0500
+++ b/src/python/m5/params.py   Sun Jul 10 12:56:08 2011 -0500
@@ -184,7 +184,10 @@
         return [ v.getValue() for v in self ]
 
     def unproxy(self, base):
-        return [v.unproxy(base) for v in self]
+        if len(self) == 1 and isinstance(self[0], AllProxy):
+            return self[0].unproxy(base)
+        else:
+             return [v.unproxy(base) for v in self]
 
 class SimObjectVector(VectorParamValue):
     # support clone operation
diff -r 510879e6bea2 -r b8c3c20d0385 src/python/m5/proxy.py
--- a/src/python/m5/proxy.py    Sun Jul 10 12:56:08 2011 -0500
+++ b/src/python/m5/proxy.py    Sun Jul 10 12:56:08 2011 -0500
@@ -184,6 +184,13 @@
     def path(self):
         return 'any'
 
+class AllProxy(BaseProxy):
+    def find(self, obj):
+        return obj.find_all(self._pdesc.ptype)
+
+    def path(self):
+        return 'all'
+
 def isproxy(obj):
     if isinstance(obj, (BaseProxy, params.EthernetAddr)):
         return True
@@ -201,6 +208,10 @@
     def __getattr__(self, attr):
         if attr == 'any':
             return AnyProxy(self.search_self, self.search_up)
+        elif attr == 'all':
+            if self.search_up:
+                assert("Parant.all is not supported")
+            return AllProxy(self.search_self, self.search_up)
         else:
             return AttrProxy(self.search_self, self.search_up, attr)
 
_______________________________________________
gem5-dev mailing list
gem5-dev@gem5.org
http://m5sim.org/mailman/listinfo/gem5-dev