Enhancements to python config proxy class.
authorSteve Reinhardt <stever@eecs.umich.edu>
Wed, 16 Mar 2005 05:40:48 +0000 (00:40 -0500)
committerSteve Reinhardt <stever@eecs.umich.edu>
Wed, 16 Mar 2005 05:40:48 +0000 (00:40 -0500)
python/m5/config.py:
    - Enhanced Proxy class now supports subscripting, e.g.,
    parent.cpu[0] or even parent.cpu[0].icache.

    - Proxy also supports multiplication (e.g., parent.cycle * 3),
    though this feature has not been tested.

    - Subscript 0 works even on non-lists, so you can safely say
    cpu[0] and get the first cpu even if there's only one.

    - Changed name of proxy object from 'Super' to 'parent', and
    changed "wild card" notation from plain 'Super' to 'parent.any'.
python/m5/objects/AlphaConsole.mpy:
python/m5/objects/BaseCPU.mpy:
python/m5/objects/BaseSystem.mpy:
python/m5/objects/Device.mpy:
python/m5/objects/Ethernet.mpy:
python/m5/objects/Ide.mpy:
python/m5/objects/IntrControl.mpy:
python/m5/objects/Pci.mpy:
python/m5/objects/PhysicalMemory.mpy:
python/m5/objects/Platform.mpy:
python/m5/objects/SimConsole.mpy:
python/m5/objects/SimpleDisk.mpy:
python/m5/objects/Tsunami.mpy:
python/m5/objects/Uart.mpy:
    Change 'Super.foo' to 'parent.foo' (and 'Super' to 'parent.any').

--HG--
extra : convert_revision : f996d0a3366d5e3e60ae5973691148c3d7cd497d

15 files changed:
python/m5/config.py
python/m5/objects/AlphaConsole.mpy
python/m5/objects/BaseCPU.mpy
python/m5/objects/BaseSystem.mpy
python/m5/objects/Device.mpy
python/m5/objects/Ethernet.mpy
python/m5/objects/Ide.mpy
python/m5/objects/IntrControl.mpy
python/m5/objects/Pci.mpy
python/m5/objects/PhysicalMemory.mpy
python/m5/objects/Platform.mpy
python/m5/objects/SimConsole.mpy
python/m5/objects/SimpleDisk.mpy
python/m5/objects/Tsunami.mpy
python/m5/objects/Uart.mpy

index 182acf393e82b651fbb690c6da6263392b8a012a..e6ad5a0ba5a1edb60b55c8283642b4bf1c32d438 100644 (file)
@@ -139,25 +139,90 @@ class Singleton(type):
 #####################################################################
 
 class Proxy(object):
-    def __init__(self, path = ()):
+    def __init__(self, path):
         self._object = None
-        self._path = path
+        if path == 'any':
+            self._path = None
+        else:
+            # path is a list of (attr,index) tuples
+            self._path = [(path,None)]
+        self._index = None
+        self._multiplier = None
 
     def __getattr__(self, attr):
-        return Proxy(self._path + (attr, ))
+        if attr == '__bases__':
+            return super(Proxy, self).__getattr__(self, attr)
+        self._path.append((attr,None))
+        return self
 
     def __setattr__(self, attr, value):
         if not attr.startswith('_'):
             raise AttributeError, 'cannot set attribute %s' % attr
         super(Proxy, self).__setattr__(attr, value)
 
-    def _convert(self):
-        obj = self._object
-        for attr in self._path:
-            obj = obj.__getattribute__(attr)
+    # support indexing on proxies (e.g., parent.cpu[0])
+    def __getitem__(self, key):
+        if not isinstance(key, int):
+            raise TypeError, "Proxy object requires integer index"
+        if self._path == None:
+            raise IndexError, "Index applied to 'any' proxy"
+        # replace index portion of last path element with new index
+        self._path[-1] = (self._path[-1][0], key)
+        return self
+
+    # support multiplying proxies by constants
+    def __mul__(self, other):
+        if not isinstance(other, int):
+            raise TypeError, "Proxy multiplier must be integer"
+        if self._multiplier == None:
+            self._multiplier = other
+        else:
+            # support chained multipliers
+            self._multiplier *= other
+        return self
+
+    def _mulcheck(self, result):
+        if self._multiplier == None:
+            return result
+        if not isinstance(result, int):
+            raise TypeError, "Proxy with multiplier resolves to " \
+                  "non-integer value"
+        return result * self._multiplier
+
+    def unproxy(self, base, ptype):
+        obj = base
+        done = False
+        while not done:
+            if obj is None:
+                raise AttributeError, \
+                      'Parent of %s type %s not found at path %s' \
+                      % (base.name, ptype, self._path)
+            found, done = obj.find(ptype, self._path)
+            if isinstance(found, Proxy):
+                done = False
+            obj = obj.parent
+
+        return self._mulcheck(found)
+
+    def getindex(obj, index):
+        if index == None:
+            return obj
+        try:
+            obj = obj[index]
+        except TypeError:
+            if index != 0:
+                raise
+            # if index is 0 and item is not subscriptable, just
+            # use item itself (so cpu[0] works on uniprocessors)
         return obj
+    getindex = staticmethod(getindex)
+
+class ProxyFactory(object):
+    def __getattr__(self, attr):
+        return Proxy(attr)
 
-Super = Proxy()
+# global object for handling parent.foo proxies
+parent = ProxyFactory()
 
 def isSubClass(value, cls):
     try:
@@ -643,50 +708,40 @@ class Node(object):
                 if issubclass(child.realtype, realtype):
                     if obj is not None:
                         raise AttributeError, \
-                              'Super matched more than one: %s %s' % \
+                              'parent.any matched more than one: %s %s' % \
                               (obj.path, child.path)
                     obj = child
             return obj, obj is not None
 
         try:
             obj = self
-            for node in path[:-1]:
-                obj = obj.child_names[node]
+            for (node,index) in path[:-1]:
+                if obj.child_names.has_key(node):
+                    obj = obj.child_names[node]
+                else:
+                    obj = obj.top_child_names[node]
+                obj = Proxy.getindex(obj, index)
 
-            last = path[-1]
+            (last,index) = path[-1]
             if obj.child_names.has_key(last):
                 value = obj.child_names[last]
-                if issubclass(value.realtype, realtype):
-                    return value, True
+                return Proxy.getindex(value, index), True
+            elif obj.top_child_names.has_key(last):
+                value = obj.top_child_names[last]
+                return Proxy.getindex(value, index), True
             elif obj.param_names.has_key(last):
                 value = obj.param_names[last]
                 realtype._convert(value.value)
-                return value.value, True
+                return Proxy.getindex(value.value, index), True
         except KeyError:
             pass
 
         return None, False
 
-    def unproxy(self, ptype, value):
-        if not isinstance(value, Proxy):
-            return value
-
-        if value is None:
-            raise AttributeError, 'Error while fixing up %s' % self.path
-
-        obj = self
-        done = False
-        while not done:
-            if obj is None:
-                raise AttributeError, \
-                      'Parent of %s type %s not found at path %s' \
-                      % (self.name, ptype, value._path)
-            found, done = obj.find(ptype, value._path)
-            if isinstance(found, Proxy):
-                done = False
-            obj = obj.parent
-
-        return found
+    def unproxy(self, param, ptype):
+        if not isinstance(param, Proxy):
+            return param
+        return param.unproxy(self, ptype)
 
     def fixup(self):
         self.all[self.path] = self
@@ -697,9 +752,9 @@ class Node(object):
 
             try:
                 if isinstance(pval, (list, tuple)):
-                    param.value = [ self.unproxy(ptype, pv) for pv in pval ]
+                    param.value = [ self.unproxy(pv, ptype) for pv in pval ]
                 else:
-                    param.value = self.unproxy(ptype, pval)
+                    param.value = self.unproxy(pval, ptype)
             except:
                 print 'Error while fixing up %s:%s' % (self.path, param.name)
                 raise
@@ -1337,7 +1392,7 @@ class SimObject(ConfigNode, ParamType):
 # 'from config import *' is invoked.  Try to keep this reasonably
 # short to avoid polluting other namespaces.
 __all__ = ['ConfigNode', 'SimObject', 'ParamContext', 'Param', 'VectorParam',
-           'Super', 'Enum',
+           'parent', 'Enum',
            'Int', 'Unsigned', 'Int8', 'UInt8', 'Int16', 'UInt16',
            'Int32', 'UInt32', 'Int64', 'UInt64',
            'Counter', 'Addr', 'Tick', 'Percent',
index 79918a01e273c36cbcbf2e46d2937cc55c3d459a..63aea5b7d6e70935840f273b971179706f4efec7 100644 (file)
@@ -2,8 +2,8 @@ from Device import PioDevice
 
 simobj AlphaConsole(PioDevice):
     type = 'AlphaConsole'
-    cpu = Param.BaseCPU(Super, "Processor")
+    cpu = Param.BaseCPU(parent.any, "Processor")
     disk = Param.SimpleDisk("Simple Disk")
     num_cpus = Param.Int(1, "Number of CPUs")
-    sim_console = Param.SimConsole(Super, "The Simulator Console")
-    system = Param.BaseSystem(Super, "system object")
+    sim_console = Param.SimConsole(parent.any, "The Simulator Console")
+    system = Param.BaseSystem(parent.any, "system object")
index 5d8305d888a4480df476b80f0b5c05be0df34007..d84e30e5344e3499ca46993b3d1180928a8d8b5d 100644 (file)
@@ -8,7 +8,7 @@ simobj BaseCPU(SimObject):
         dtb = Param.AlphaDTB("Data TLB")
         itb = Param.AlphaITB("Instruction TLB")
         mem = Param.FunctionalMemory("memory")
-        system = Param.BaseSystem(Super, "system object")
+        system = Param.BaseSystem(parent.any, "system object")
     else:
         workload = VectorParam.Process("processes to run")
 
index 1cbdf4e99ac3f9bfd2d2b96dfdb2af28826c6b4a..450b6a58e1cc4e8ddfb0617fb849f113bfc50d5e 100644 (file)
@@ -1,8 +1,8 @@
 simobj BaseSystem(SimObject):
     type = 'BaseSystem'
     abstract = True
-    memctrl = Param.MemoryController(Super, "memory controller")
-    physmem = Param.PhysicalMemory(Super, "phsyical memory")
+    memctrl = Param.MemoryController(parent.any, "memory controller")
+    physmem = Param.PhysicalMemory(parent.any, "phsyical memory")
     kernel = Param.String("file that contains the kernel code")
     console = Param.String("file that contains the console code")
     pal = Param.String("file that contains palcode")
index 47f8db1cbe790115acbdc4d8b07418b27cbb60fd..a0d02a647300dfa1ea59a6365e27f540da61be26 100644 (file)
@@ -14,7 +14,7 @@ simobj FooPioDevice(FunctionalMemory):
     type = 'PioDevice'
     abstract = True
     addr = Param.Addr("Device Address")
-    mmu = Param.MemoryController(Super, "Memory Controller")
+    mmu = Param.MemoryController(parent.any, "Memory Controller")
     io_bus = Param.Bus(NULL, "The IO Bus to attach to")
     pio_latency = Param.Tick(1, "Programmed IO latency in bus cycles")
 
@@ -25,7 +25,7 @@ simobj FooDmaDevice(FooPioDevice):
 simobj PioDevice(FooPioDevice):
     type = 'PioDevice'
     abstract = True
-    platform = Param.Platform(Super, "Platform")
+    platform = Param.Platform(parent.any, "Platform")
 
 simobj DmaDevice(PioDevice):
     type = 'DmaDevice'
index 088df4b9328d44222ae2ca03ad35fba5a0421708..cd251f36da09903c0b59758446df4224804e1dcc 100644 (file)
@@ -49,8 +49,8 @@ simobj EtherDev(DmaDevice):
 
     intr_delay = Param.Tick(0, "Interrupt Delay in microseconds")
     payload_bus = Param.Bus(NULL, "The IO Bus to attach to for payload")
-    physmem = Param.PhysicalMemory(Super, "Physical Memory")
-    tlaser = Param.Turbolaser(Super, "Turbolaser")
+    physmem = Param.PhysicalMemory(parent.any, "Physical Memory")
+    tlaser = Param.Turbolaser(parent.any, "Turbolaser")
 
 simobj NSGigE(PciDevice):
     type = 'NSGigE'
@@ -73,7 +73,7 @@ simobj NSGigE(PciDevice):
 
     intr_delay = Param.Tick(0, "Interrupt Delay in microseconds")
     payload_bus = Param.Bus(NULL, "The IO Bus to attach to for payload")
-    physmem = Param.PhysicalMemory(Super, "Physical Memory")
+    physmem = Param.PhysicalMemory(parent.any, "Physical Memory")
 
 simobj EtherDevInt(EtherInt):
     type = 'EtherDevInt'
index ce760ad96bbaf7656151c7831774653bfa41aecf..786109efa4758cef2e9737d10283fbb88e63b48b 100644 (file)
@@ -7,7 +7,7 @@ simobj IdeDisk(SimObject):
     delay = Param.Tick(1, "Fixed disk delay in microseconds")
     driveID = Param.IdeID('master', "Drive ID")
     image = Param.DiskImage("Disk image")
-    physmem = Param.PhysicalMemory(Super, "Physical memory")
+    physmem = Param.PhysicalMemory(parent.any, "Physical memory")
 
 simobj IdeController(PciDevice):
     type = 'IdeController'
index 1ef5a17ee3d999f14771ba4029fb92a7f42fa65d..144be0fd46b36d93ac37aede5e2715bd74efcb7d 100644 (file)
@@ -1,3 +1,3 @@
 simobj IntrControl(SimObject):
     type = 'IntrControl'
-    cpu = Param.BaseCPU(Super, "the cpu")
+    cpu = Param.BaseCPU(parent.any, "the cpu")
index f7c6674f74446f71b358c246a19c778287ec4857..b9b3e5a956da45f6f132c0ec796d657e7cc9817d 100644 (file)
@@ -47,5 +47,5 @@ simobj PciDevice(DmaDevice):
     pci_bus = Param.Int("PCI bus")
     pci_dev = Param.Int("PCI device number")
     pci_func = Param.Int("PCI function code")
-    configdata = Param.PciConfigData(Super, "PCI Config data")
-    configspace = Param.PciConfigAll(Super, "PCI Configspace")
+    configdata = Param.PciConfigData(parent.any, "PCI Config data")
+    configspace = Param.PciConfigAll(parent.any, "PCI Configspace")
index d1e4ad4b401942cccac1f7d93d2fc88c44ec2a12..e6df2a1614e0708ec2ee47b423eed97e80b7f9d4 100644 (file)
@@ -4,4 +4,4 @@ simobj PhysicalMemory(FunctionalMemory):
     type = 'PhysicalMemory'
     range = Param.AddrRange("Device Address")
     file = Param.String('', "memory mapped file")
-    mmu = Param.MemoryController(Super, "Memory Controller")
+    mmu = Param.MemoryController(parent.any, "Memory Controller")
index d0510eaf873c0ce68d770e298271a30516ae6754..a71ab3b770d392a9f1ffe4f03e72e3d0e350fd26 100644 (file)
@@ -2,4 +2,4 @@ simobj Platform(SimObject):
     type = 'Platform'
     abstract = True
     interrupt_frequency = Param.Tick(1200, "frequency of interrupts")
-    intrctrl = Param.IntrControl(Super, "interrupt controller")
+    intrctrl = Param.IntrControl(parent.any, "interrupt controller")
index fb74f177502298a05e93f208940768681807322d..3588a949d8966189c4918bc4ed0c4ae2d3f14886 100644 (file)
@@ -5,7 +5,7 @@ simobj ConsoleListener(SimObject):
 simobj SimConsole(SimObject):
     type = 'SimConsole'
     append_name = Param.Bool(True, "append name() to filename")
-    intr_control = Param.IntrControl(Super, "interrupt controller")
+    intr_control = Param.IntrControl(parent.any, "interrupt controller")
     listener = Param.ConsoleListener("console listener")
     number = Param.Int(0, "console number")
     output = Param.String('console', "file to dump output to")
index c4dd5435bf3e5ee92f1b983667d04547078315f0..b616fb3d128e09d04907201a25acfd3e57886152 100644 (file)
@@ -1,4 +1,4 @@
 simobj SimpleDisk(SimObject):
     type = 'SimpleDisk'
     disk = Param.DiskImage("Disk Image")
-    physmem = Param.PhysicalMemory(Super, "Physical Memory")
+    physmem = Param.PhysicalMemory(parent.any, "Physical Memory")
index cfe23977e040c593ddade959ad9e3c55d970543e..a8471cee23319a7e4bddbff28a0993a5c4ad0e26 100644 (file)
@@ -4,12 +4,12 @@ from Platform import Platform
 simobj Tsunami(Platform):
     type = 'Tsunami'
     pciconfig = Param.PciConfigAll("PCI configuration")
-    system = Param.BaseSystem(Super, "system")
+    system = Param.BaseSystem(parent.any, "system")
     interrupt_frequency = Param.Int(1024, "frequency of interrupts")
 
 simobj TsunamiCChip(FooPioDevice):
     type = 'TsunamiCChip'
-    tsunami = Param.Tsunami(Super, "Tsunami")
+    tsunami = Param.Tsunami(parent.any, "Tsunami")
 
 simobj TsunamiFake(FooPioDevice):
     type = 'TsunamiFake'
@@ -18,8 +18,8 @@ simobj TsunamiIO(FooPioDevice):
     type = 'TsunamiIO'
     time = Param.UInt64(1136073600,
         "System time to use (0 for actual time, default is 1/1/06)")
-    tsunami = Param.Tsunami(Super, "Tsunami")
+    tsunami = Param.Tsunami(parent.any, "Tsunami")
 
 simobj TsunamiPChip(FooPioDevice):
     type = 'TsunamiPChip'
-    tsunami = Param.Tsunami(Super, "Tsunami")
+    tsunami = Param.Tsunami(parent.any, "Tsunami")
index 76ee8805f8569ae30f8f4fbdbb7cc74044de5713..5a6c25f8ec74d63d81665363e8dac96aaf1f8586 100644 (file)
@@ -2,5 +2,5 @@ from Device import PioDevice
 
 simobj Uart(PioDevice):
     type = 'Uart'
-    console = Param.SimConsole(Super, "The console")
+    console = Param.SimConsole(parent.any, "The console")
     size = Param.Addr(0x8, "Device size")