src/python/m5/event.py

   1 # Copyright (c) 2017 ARM Limited
   2 # All rights reserved.
   3 #
   4 # The license below extends only to copyright in the software and shall
   5 # not be construed as granting a license to any other intellectual
   6 # property including but not limited to intellectual property relating
   7 # to a hardware implementation of the functionality of the software
   8 # licensed hereunder.  You may use the software subject to the license
   9 # terms below provided that you ensure that this notice is replicated
  10 # unmodified and in its entirety in all distributions of the software,
  11 # modified or unmodified, in source code or in binary form.
  12 #
  13 # Copyright (c) 2006 The Regents of The University of Michigan
  14 # Copyright (c) 2013 Advanced Micro Devices, Inc.
  15 # Copyright (c) 2013 Mark D. Hill and David A. Wood
  16 # All rights reserved.
  17 #
  18 # Redistribution and use in source and binary forms, with or without
  19 # modification, are permitted provided that the following conditions are
  20 # met: redistributions of source code must retain the above copyright
  21 # notice, this list of conditions and the following disclaimer;
  22 # redistributions in binary form must reproduce the above copyright
  23 # notice, this list of conditions and the following disclaimer in the
  24 # documentation and/or other materials provided with the distribution;
  25 # neither the name of the copyright holders nor the names of its
  26 # contributors may be used to endorse or promote products derived from
  27 # this software without specific prior written permission.
  28 #
  29 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  30 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  31 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  32 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  33 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  34 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  35 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  36 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  37 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  38 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  39 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  40 #
  41 # Authors: Nathan Binkert
  42
  43 import m5
  44 import _m5.event
  45
  46 from _m5.event import GlobalSimLoopExitEvent as SimExit
  47 from _m5.event import Event, getEventQueue, setEventQueue
  48
  49 mainq = None
  50
  51 class EventWrapper(Event):
  52     """Helper class to wrap callable objects in an Event base class"""
  53
  54     def __init__(self, func, **kwargs):
  55         super(EventWrapper, self).__init__(**kwargs)
  56
  57         if not callable(func):
  58             raise RuntimeError("Can't wrap '%s', object is not callable" % \
  59                                str(func))
  60
  61         self._func = func
  62
  63     def __call__(self):
  64         self._func()
  65
  66     def __str__(self):
  67         return "EventWrapper(%s)" % (str(self._func), )
  68
  69
  70 class ProgressEvent(Event):
  71     def __init__(self, eventq, period):
  72         super(ProgressEvent, self).__init__()
  73         self.period = int(period)
  74         self.eventq = eventq
  75         self.eventq.schedule(self, m5.curTick() + self.period)
  76
  77     def __call__(self):
  78         print "Progress! Time now %fs" % (m5.curTick()/1e12)
  79         self.eventq.schedule(self, m5.curTick() + self.period)
  80
  81
  82 def create(func, priority=Event.Default_Pri):
  83     """Create an Event from a function"""
  84
  85     return EventWrapper(func, priority=priority)
  86
  87 __all__ = [ 'Event', 'EventWrapper', 'ProgressEvent', 'SimExit',
  88             'mainq', 'create' ]