SimObject.py revision 12035:7b8e1b36875d
12SN/A# Copyright (c) 2017 ARM Limited 21762SN/A# All rights reserved. 37534Ssteve.reinhardt@amd.com# 42SN/A# The license below extends only to copyright in the software and shall 52SN/A# not be construed as granting a license to any other intellectual 62SN/A# property including but not limited to intellectual property relating 72SN/A# to a hardware implementation of the functionality of the software 82SN/A# licensed hereunder. You may use the software subject to the license 92SN/A# terms below provided that you ensure that this notice is replicated 102SN/A# unmodified and in its entirety in all distributions of the software, 112SN/A# modified or unmodified, in source code or in binary form. 122SN/A# 132SN/A# Copyright (c) 2004-2006 The Regents of The University of Michigan 142SN/A# Copyright (c) 2010-20013 Advanced Micro Devices, Inc. 152SN/A# Copyright (c) 2013 Mark D. Hill and David A. Wood 162SN/A# All rights reserved. 172SN/A# 182SN/A# Redistribution and use in source and binary forms, with or without 192SN/A# modification, are permitted provided that the following conditions are 202SN/A# met: redistributions of source code must retain the above copyright 212SN/A# notice, this list of conditions and the following disclaimer; 222SN/A# redistributions in binary form must reproduce the above copyright 232SN/A# notice, this list of conditions and the following disclaimer in the 242SN/A# documentation and/or other materials provided with the distribution; 252SN/A# neither the name of the copyright holders nor the names of its 262SN/A# contributors may be used to endorse or promote products derived from 272SN/A# this software without specific prior written permission. 282665Ssaidi@eecs.umich.edu# 292665Ssaidi@eecs.umich.edu# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 302665Ssaidi@eecs.umich.edu# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 312SN/A# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 322SN/A# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 332SN/A# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 342SN/A# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 352SN/A# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 362SN/A# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 372SN/A# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 382SN/A# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 392SN/A# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 405491Sgblack@eecs.umich.edu# 415491Sgblack@eecs.umich.edu# Authors: Steve Reinhardt 422SN/A# Nathan Binkert 435491Sgblack@eecs.umich.edu# Andreas Hansson 442SN/A# Andreas Sandberg 452SN/A 468737Skoansin.tan@gmail.comimport sys 474762Snate@binkert.orgfrom types import FunctionType, MethodType, ModuleType 489342SAndreas.Sandberg@arm.comfrom functools import wraps 499356Snilay@cs.wisc.eduimport inspect 5056SN/A 512SN/Aimport m5 522797Sktlim@umich.edufrom m5.util import * 532797Sktlim@umich.edufrom m5.util.pybind import * 542609SN/A 559196SAndreas.Sandberg@arm.com# Have to import params up top since Param is referenced on initial 562SN/A# load (when SimObject class references Param to create a class 572SN/A# variable, the 'name' param)... 582SN/Afrom m5.params import * 599196SAndreas.Sandberg@arm.com# There are a few things we need that aren't in params.__all__ since 609196SAndreas.Sandberg@arm.com# normal users don't need them 619196SAndreas.Sandberg@arm.comfrom m5.params import ParamDesc, VectorParamDesc, \ 629196SAndreas.Sandberg@arm.com isNullPointer, SimObjectVector, Port 639196SAndreas.Sandberg@arm.com 649196SAndreas.Sandberg@arm.comfrom m5.proxy import * 659196SAndreas.Sandberg@arm.comfrom m5.proxy import isproxy 669196SAndreas.Sandberg@arm.com 679196SAndreas.Sandberg@arm.com##################################################################### 689196SAndreas.Sandberg@arm.com# 699196SAndreas.Sandberg@arm.com# M5 Python Configuration Utility 709196SAndreas.Sandberg@arm.com# 719196SAndreas.Sandberg@arm.com# The basic idea is to write simple Python programs that build Python 729196SAndreas.Sandberg@arm.com# objects corresponding to M5 SimObjects for the desired simulation 739196SAndreas.Sandberg@arm.com# configuration. For now, the Python emits a .ini file that can be 749196SAndreas.Sandberg@arm.com# parsed by M5. In the future, some tighter integration between M5 759196SAndreas.Sandberg@arm.com# and the Python interpreter may allow bypassing the .ini file. 769342SAndreas.Sandberg@arm.com# 779196SAndreas.Sandberg@arm.com# Each SimObject class in M5 is represented by a Python class with the 789196SAndreas.Sandberg@arm.com# same name. The Python inheritance tree mirrors the M5 C++ tree 799196SAndreas.Sandberg@arm.com# (e.g., SimpleCPU derives from BaseCPU in both cases, and all 809196SAndreas.Sandberg@arm.com# SimObjects inherit from a single SimObject base class). To specify 819196SAndreas.Sandberg@arm.com# an instance of an M5 SimObject in a configuration, the user simply 829196SAndreas.Sandberg@arm.com# instantiates the corresponding Python object. The parameters for 839196SAndreas.Sandberg@arm.com# that SimObject are given by assigning to attributes of the Python 842SN/A# object, either using keyword assignment in the constructor or in 859342SAndreas.Sandberg@arm.com# separate assignment statements. For example: 862SN/A# 872SN/A# cache = BaseCache(size='64KB') 882SN/A# cache.hit_latency = 3 892SN/A# cache.assoc = 8 909196SAndreas.Sandberg@arm.com# 912SN/A# The magic lies in the mapping of the Python attributes for SimObject 922SN/A# classes to the actual SimObject parameter specifications. This 934762Snate@binkert.org# allows parameter validity checking in the Python code. Continuing 949196SAndreas.Sandberg@arm.com# the example above, the statements "cache.blurfl=3" or 954762Snate@binkert.org# "cache.assoc='hello'" would both result in runtime errors in Python, 964762Snate@binkert.org# since the BaseCache object has no 'blurfl' parameter and the 'assoc' 972SN/A# parameter requires an integer, respectively. This magic is done 984762Snate@binkert.org# primarily by overriding the special __setattr__ method that controls 994762Snate@binkert.org# assignment to object attributes. 1004762Snate@binkert.org# 1012SN/A# Once a set of Python objects have been instantiated in a hierarchy, 1022SN/A# calling 'instantiate(obj)' (where obj is the root of the hierarchy) 1035034Smilesck@eecs.umich.edu# will generate a .ini file. 1045034Smilesck@eecs.umich.edu# 1051553SN/A##################################################################### 106265SN/A 1077532Ssteve.reinhardt@amd.com# list of all SimObject classes 1087532Ssteve.reinhardt@amd.comallClasses = {} 1097532Ssteve.reinhardt@amd.com 1107532Ssteve.reinhardt@amd.com# dict to look up SimObjects based on path 1117532Ssteve.reinhardt@amd.cominstanceDict = {} 1127532Ssteve.reinhardt@amd.com 113465SN/A# Did any of the SimObjects lack a header file? 114465SN/AnoCxxHeader = False 1157532Ssteve.reinhardt@amd.com 1167532Ssteve.reinhardt@amd.comdef public_value(key, value): 1177532Ssteve.reinhardt@amd.com return key.startswith('_') or \ 1187532Ssteve.reinhardt@amd.com isinstance(value, (FunctionType, MethodType, ModuleType, 1197532Ssteve.reinhardt@amd.com classmethod, type)) 1207532Ssteve.reinhardt@amd.com 1217532Ssteve.reinhardt@amd.comdef createCxxConfigDirectoryEntryFile(code, name, simobj, is_header): 1227532Ssteve.reinhardt@amd.com entry_class = 'CxxConfigDirectoryEntry_%s' % name 1239196SAndreas.Sandberg@arm.com param_class = '%sCxxConfigParams' % name 1249196SAndreas.Sandberg@arm.com 1257532Ssteve.reinhardt@amd.com code('#include "params/%s.hh"' % name) 1267532Ssteve.reinhardt@amd.com 1277532Ssteve.reinhardt@amd.com if not is_header: 1287532Ssteve.reinhardt@amd.com for param in simobj._params.values(): 1297532Ssteve.reinhardt@amd.com if isSimObjectClass(param.ptype): 1307532Ssteve.reinhardt@amd.com code('#include "%s"' % param.ptype._value_dict['cxx_header']) 1317532Ssteve.reinhardt@amd.com code('#include "params/%s.hh"' % param.ptype.__name__) 1327532Ssteve.reinhardt@amd.com else: 1337532Ssteve.reinhardt@amd.com param.ptype.cxx_ini_predecls(code) 1347532Ssteve.reinhardt@amd.com 1359196SAndreas.Sandberg@arm.com if is_header: 1369196SAndreas.Sandberg@arm.com member_prefix = '' 1379196SAndreas.Sandberg@arm.com end_of_decl = ';' 1382SN/A code('#include "sim/cxx_config.hh"') 1399196SAndreas.Sandberg@arm.com code() 1409196SAndreas.Sandberg@arm.com code('class ${param_class} : public CxxConfigParams,' 1419196SAndreas.Sandberg@arm.com ' public ${name}Params') 1429196SAndreas.Sandberg@arm.com code('{') 143330SN/A code(' private:') 1442SN/A code.indent() 1457532Ssteve.reinhardt@amd.com code('class DirectoryEntry : public CxxConfigDirectoryEntry') 1467532Ssteve.reinhardt@amd.com code('{') 1477532Ssteve.reinhardt@amd.com code(' public:') 1487823Ssteve.reinhardt@amd.com code.indent() 1497532Ssteve.reinhardt@amd.com code('DirectoryEntry();'); 1507532Ssteve.reinhardt@amd.com code() 1517492Ssteve.reinhardt@amd.com code('CxxConfigParams *makeParamsObject() const') 152330SN/A code('{ return new ${param_class}; }') 1539196SAndreas.Sandberg@arm.com code.dedent() 1549342SAndreas.Sandberg@arm.com code('};') 1559342SAndreas.Sandberg@arm.com code() 1569342SAndreas.Sandberg@arm.com code.dedent() 1579342SAndreas.Sandberg@arm.com code(' public:') 1589342SAndreas.Sandberg@arm.com code.indent() 1599342SAndreas.Sandberg@arm.com else: 1609342SAndreas.Sandberg@arm.com member_prefix = '%s::' % param_class 1619196SAndreas.Sandberg@arm.com end_of_decl = '' 1629196SAndreas.Sandberg@arm.com code('#include "%s"' % simobj._value_dict['cxx_header']) 1639196SAndreas.Sandberg@arm.com code('#include "base/str.hh"') 164938SN/A code('#include "cxx_config/${name}.hh"') 1651031SN/A 1661031SN/A if simobj._ports.values() != []: 1671031SN/A code('#include "mem/mem_object.hh"') 1681031SN/A code('#include "mem/port.hh"') 1691031SN/A 1701031SN/A code() 1715314Sstever@gmail.com code('${member_prefix}DirectoryEntry::DirectoryEntry()'); 1725314Sstever@gmail.com code('{') 1735315Sstever@gmail.com 1745314Sstever@gmail.com def cxx_bool(b): 1755314Sstever@gmail.com return 'true' if b else 'false' 1765314Sstever@gmail.com 1772SN/A code.indent() 1782SN/A for param in simobj._params.values(): 1799554Sandreas.hansson@arm.com is_vector = isinstance(param, m5.params.VectorParamDesc) 1809554Sandreas.hansson@arm.com is_simobj = issubclass(param.ptype, m5.SimObject.SimObject) 1819554Sandreas.hansson@arm.com 1829554Sandreas.hansson@arm.com code('parameters["%s"] = new ParamDesc("%s", %s, %s);' % 1832SN/A (param.name, param.name, cxx_bool(is_vector), 184 cxx_bool(is_simobj))); 185 186 for port in simobj._ports.values(): 187 is_vector = isinstance(port, m5.params.VectorPort) 188 is_master = port.role == 'MASTER' 189 190 code('ports["%s"] = new PortDesc("%s", %s, %s);' % 191 (port.name, port.name, cxx_bool(is_vector), 192 cxx_bool(is_master))) 193 194 code.dedent() 195 code('}') 196 code() 197 198 code('bool ${member_prefix}setSimObject(const std::string &name,') 199 code(' SimObject *simObject)${end_of_decl}') 200 201 if not is_header: 202 code('{') 203 code.indent() 204 code('bool ret = true;') 205 code() 206 code('if (false) {') 207 for param in simobj._params.values(): 208 is_vector = isinstance(param, m5.params.VectorParamDesc) 209 is_simobj = issubclass(param.ptype, m5.SimObject.SimObject) 210 211 if is_simobj and not is_vector: 212 code('} else if (name == "${{param.name}}") {') 213 code.indent() 214 code('this->${{param.name}} = ' 215 'dynamic_cast<${{param.ptype.cxx_type}}>(simObject);') 216 code('if (simObject && !this->${{param.name}})') 217 code(' ret = false;') 218 code.dedent() 219 code('} else {') 220 code(' ret = false;') 221 code('}') 222 code() 223 code('return ret;') 224 code.dedent() 225 code('}') 226 227 code() 228 code('bool ${member_prefix}setSimObjectVector(' 229 'const std::string &name,') 230 code(' const std::vector<SimObject *> &simObjects)${end_of_decl}') 231 232 if not is_header: 233 code('{') 234 code.indent() 235 code('bool ret = true;') 236 code() 237 code('if (false) {') 238 for param in simobj._params.values(): 239 is_vector = isinstance(param, m5.params.VectorParamDesc) 240 is_simobj = issubclass(param.ptype, m5.SimObject.SimObject) 241 242 if is_simobj and is_vector: 243 code('} else if (name == "${{param.name}}") {') 244 code.indent() 245 code('this->${{param.name}}.clear();') 246 code('for (auto i = simObjects.begin(); ' 247 'ret && i != simObjects.end(); i ++)') 248 code('{') 249 code.indent() 250 code('${{param.ptype.cxx_type}} object = ' 251 'dynamic_cast<${{param.ptype.cxx_type}}>(*i);') 252 code('if (*i && !object)') 253 code(' ret = false;') 254 code('else') 255 code(' this->${{param.name}}.push_back(object);') 256 code.dedent() 257 code('}') 258 code.dedent() 259 code('} else {') 260 code(' ret = false;') 261 code('}') 262 code() 263 code('return ret;') 264 code.dedent() 265 code('}') 266 267 code() 268 code('void ${member_prefix}setName(const std::string &name_)' 269 '${end_of_decl}') 270 271 if not is_header: 272 code('{') 273 code.indent() 274 code('this->name = name_;') 275 code.dedent() 276 code('}') 277 278 if is_header: 279 code('const std::string &${member_prefix}getName()') 280 code('{ return this->name; }') 281 282 code() 283 code('bool ${member_prefix}setParam(const std::string &name,') 284 code(' const std::string &value, const Flags flags)${end_of_decl}') 285 286 if not is_header: 287 code('{') 288 code.indent() 289 code('bool ret = true;') 290 code() 291 code('if (false) {') 292 for param in simobj._params.values(): 293 is_vector = isinstance(param, m5.params.VectorParamDesc) 294 is_simobj = issubclass(param.ptype, m5.SimObject.SimObject) 295 296 if not is_simobj and not is_vector: 297 code('} else if (name == "${{param.name}}") {') 298 code.indent() 299 param.ptype.cxx_ini_parse(code, 300 'value', 'this->%s' % param.name, 'ret =') 301 code.dedent() 302 code('} else {') 303 code(' ret = false;') 304 code('}') 305 code() 306 code('return ret;') 307 code.dedent() 308 code('}') 309 310 code() 311 code('bool ${member_prefix}setParamVector(' 312 'const std::string &name,') 313 code(' const std::vector<std::string> &values,') 314 code(' const Flags flags)${end_of_decl}') 315 316 if not is_header: 317 code('{') 318 code.indent() 319 code('bool ret = true;') 320 code() 321 code('if (false) {') 322 for param in simobj._params.values(): 323 is_vector = isinstance(param, m5.params.VectorParamDesc) 324 is_simobj = issubclass(param.ptype, m5.SimObject.SimObject) 325 326 if not is_simobj and is_vector: 327 code('} else if (name == "${{param.name}}") {') 328 code.indent() 329 code('${{param.name}}.clear();') 330 code('for (auto i = values.begin(); ' 331 'ret && i != values.end(); i ++)') 332 code('{') 333 code.indent() 334 code('${{param.ptype.cxx_type}} elem;') 335 param.ptype.cxx_ini_parse(code, 336 '*i', 'elem', 'ret =') 337 code('if (ret)') 338 code(' this->${{param.name}}.push_back(elem);') 339 code.dedent() 340 code('}') 341 code.dedent() 342 code('} else {') 343 code(' ret = false;') 344 code('}') 345 code() 346 code('return ret;') 347 code.dedent() 348 code('}') 349 350 code() 351 code('bool ${member_prefix}setPortConnectionCount(' 352 'const std::string &name,') 353 code(' unsigned int count)${end_of_decl}') 354 355 if not is_header: 356 code('{') 357 code.indent() 358 code('bool ret = true;') 359 code() 360 code('if (false)') 361 code(' ;') 362 for port in simobj._ports.values(): 363 code('else if (name == "${{port.name}}")') 364 code(' this->port_${{port.name}}_connection_count = count;') 365 code('else') 366 code(' ret = false;') 367 code() 368 code('return ret;') 369 code.dedent() 370 code('}') 371 372 code() 373 code('SimObject *${member_prefix}simObjectCreate()${end_of_decl}') 374 375 if not is_header: 376 code('{') 377 if hasattr(simobj, 'abstract') and simobj.abstract: 378 code(' return NULL;') 379 else: 380 code(' return this->create();') 381 code('}') 382 383 if is_header: 384 code() 385 code('static CxxConfigDirectoryEntry' 386 ' *${member_prefix}makeDirectoryEntry()') 387 code('{ return new DirectoryEntry; }') 388 389 if is_header: 390 code.dedent() 391 code('};') 392 393# The metaclass for SimObject. This class controls how new classes 394# that derive from SimObject are instantiated, and provides inherited 395# class behavior (just like a class controls how instances of that 396# class are instantiated, and provides inherited instance behavior). 397class MetaSimObject(type): 398 # Attributes that can be set only at initialization time 399 init_keywords = { 400 'abstract' : bool, 401 'cxx_class' : str, 402 'cxx_type' : str, 403 'cxx_header' : str, 404 'type' : str, 405 'cxx_bases' : list, 406 'cxx_exports' : list, 407 'cxx_param_exports' : list, 408 } 409 # Attributes that can be set any time 410 keywords = { 'check' : FunctionType } 411 412 # __new__ is called before __init__, and is where the statements 413 # in the body of the class definition get loaded into the class's 414 # __dict__. We intercept this to filter out parameter & port assignments 415 # and only allow "private" attributes to be passed to the base 416 # __new__ (starting with underscore). 417 def __new__(mcls, name, bases, dict): 418 assert name not in allClasses, "SimObject %s already present" % name 419 420 # Copy "private" attributes, functions, and classes to the 421 # official dict. Everything else goes in _init_dict to be 422 # filtered in __init__. 423 cls_dict = {} 424 value_dict = {} 425 cxx_exports = [] 426 for key,val in dict.items(): 427 try: 428 cxx_exports.append(getattr(val, "__pybind")) 429 except AttributeError: 430 pass 431 432 if public_value(key, val): 433 cls_dict[key] = val 434 else: 435 # must be a param/port setting 436 value_dict[key] = val 437 if 'abstract' not in value_dict: 438 value_dict['abstract'] = False 439 if 'cxx_bases' not in value_dict: 440 value_dict['cxx_bases'] = [] 441 if 'cxx_exports' not in value_dict: 442 value_dict['cxx_exports'] = cxx_exports 443 else: 444 value_dict['cxx_exports'] += cxx_exports 445 if 'cxx_param_exports' not in value_dict: 446 value_dict['cxx_param_exports'] = [] 447 cls_dict['_value_dict'] = value_dict 448 cls = super(MetaSimObject, mcls).__new__(mcls, name, bases, cls_dict) 449 if 'type' in value_dict: 450 allClasses[name] = cls 451 return cls 452 453 # subclass initialization 454 def __init__(cls, name, bases, dict): 455 # calls type.__init__()... I think that's a no-op, but leave 456 # it here just in case it's not. 457 super(MetaSimObject, cls).__init__(name, bases, dict) 458 459 # initialize required attributes 460 461 # class-only attributes 462 cls._params = multidict() # param descriptions 463 cls._ports = multidict() # port descriptions 464 465 # class or instance attributes 466 cls._values = multidict() # param values 467 cls._hr_values = multidict() # human readable param values 468 cls._children = multidict() # SimObject children 469 cls._port_refs = multidict() # port ref objects 470 cls._instantiated = False # really instantiated, cloned, or subclassed 471 472 # We don't support multiple inheritance of sim objects. If you want 473 # to, you must fix multidict to deal with it properly. Non sim-objects 474 # are ok, though 475 bTotal = 0 476 for c in bases: 477 if isinstance(c, MetaSimObject): 478 bTotal += 1 479 if bTotal > 1: 480 raise TypeError, \ 481 "SimObjects do not support multiple inheritance" 482 483 base = bases[0] 484 485 # Set up general inheritance via multidicts. A subclass will 486 # inherit all its settings from the base class. The only time 487 # the following is not true is when we define the SimObject 488 # class itself (in which case the multidicts have no parent). 489 if isinstance(base, MetaSimObject): 490 cls._base = base 491 cls._params.parent = base._params 492 cls._ports.parent = base._ports 493 cls._values.parent = base._values 494 cls._hr_values.parent = base._hr_values 495 cls._children.parent = base._children 496 cls._port_refs.parent = base._port_refs 497 # mark base as having been subclassed 498 base._instantiated = True 499 else: 500 cls._base = None 501 502 # default keyword values 503 if 'type' in cls._value_dict: 504 if 'cxx_class' not in cls._value_dict: 505 cls._value_dict['cxx_class'] = cls._value_dict['type'] 506 507 cls._value_dict['cxx_type'] = '%s *' % cls._value_dict['cxx_class'] 508 509 if 'cxx_header' not in cls._value_dict: 510 global noCxxHeader 511 noCxxHeader = True 512 warn("No header file specified for SimObject: %s", name) 513 514 # Now process the _value_dict items. They could be defining 515 # new (or overriding existing) parameters or ports, setting 516 # class keywords (e.g., 'abstract'), or setting parameter 517 # values or port bindings. The first 3 can only be set when 518 # the class is defined, so we handle them here. The others 519 # can be set later too, so just emulate that by calling 520 # setattr(). 521 for key,val in cls._value_dict.items(): 522 # param descriptions 523 if isinstance(val, ParamDesc): 524 cls._new_param(key, val) 525 526 # port objects 527 elif isinstance(val, Port): 528 cls._new_port(key, val) 529 530 # init-time-only keywords 531 elif cls.init_keywords.has_key(key): 532 cls._set_keyword(key, val, cls.init_keywords[key]) 533 534 # default: use normal path (ends up in __setattr__) 535 else: 536 setattr(cls, key, val) 537 538 def _set_keyword(cls, keyword, val, kwtype): 539 if not isinstance(val, kwtype): 540 raise TypeError, 'keyword %s has bad type %s (expecting %s)' % \ 541 (keyword, type(val), kwtype) 542 if isinstance(val, FunctionType): 543 val = classmethod(val) 544 type.__setattr__(cls, keyword, val) 545 546 def _new_param(cls, name, pdesc): 547 # each param desc should be uniquely assigned to one variable 548 assert(not hasattr(pdesc, 'name')) 549 pdesc.name = name 550 cls._params[name] = pdesc 551 if hasattr(pdesc, 'default'): 552 cls._set_param(name, pdesc.default, pdesc) 553 554 def _set_param(cls, name, value, param): 555 assert(param.name == name) 556 try: 557 hr_value = value 558 value = param.convert(value) 559 except Exception, e: 560 msg = "%s\nError setting param %s.%s to %s\n" % \ 561 (e, cls.__name__, name, value) 562 e.args = (msg, ) 563 raise 564 cls._values[name] = value 565 # if param value is a SimObject, make it a child too, so that 566 # it gets cloned properly when the class is instantiated 567 if isSimObjectOrVector(value) and not value.has_parent(): 568 cls._add_cls_child(name, value) 569 # update human-readable values of the param if it has a literal 570 # value and is not an object or proxy. 571 if not (isSimObjectOrVector(value) or\ 572 isinstance(value, m5.proxy.BaseProxy)): 573 cls._hr_values[name] = hr_value 574 575 def _add_cls_child(cls, name, child): 576 # It's a little funky to have a class as a parent, but these 577 # objects should never be instantiated (only cloned, which 578 # clears the parent pointer), and this makes it clear that the 579 # object is not an orphan and can provide better error 580 # messages. 581 child.set_parent(cls, name) 582 cls._children[name] = child 583 584 def _new_port(cls, name, port): 585 # each port should be uniquely assigned to one variable 586 assert(not hasattr(port, 'name')) 587 port.name = name 588 cls._ports[name] = port 589 590 # same as _get_port_ref, effectively, but for classes 591 def _cls_get_port_ref(cls, attr): 592 # Return reference that can be assigned to another port 593 # via __setattr__. There is only ever one reference 594 # object per port, but we create them lazily here. 595 ref = cls._port_refs.get(attr) 596 if not ref: 597 ref = cls._ports[attr].makeRef(cls) 598 cls._port_refs[attr] = ref 599 return ref 600 601 # Set attribute (called on foo.attr = value when foo is an 602 # instance of class cls). 603 def __setattr__(cls, attr, value): 604 # normal processing for private attributes 605 if public_value(attr, value): 606 type.__setattr__(cls, attr, value) 607 return 608 609 if cls.keywords.has_key(attr): 610 cls._set_keyword(attr, value, cls.keywords[attr]) 611 return 612 613 if cls._ports.has_key(attr): 614 cls._cls_get_port_ref(attr).connect(value) 615 return 616 617 if isSimObjectOrSequence(value) and cls._instantiated: 618 raise RuntimeError, \ 619 "cannot set SimObject parameter '%s' after\n" \ 620 " class %s has been instantiated or subclassed" \ 621 % (attr, cls.__name__) 622 623 # check for param 624 param = cls._params.get(attr) 625 if param: 626 cls._set_param(attr, value, param) 627 return 628 629 if isSimObjectOrSequence(value): 630 # If RHS is a SimObject, it's an implicit child assignment. 631 cls._add_cls_child(attr, coerceSimObjectOrVector(value)) 632 return 633 634 # no valid assignment... raise exception 635 raise AttributeError, \ 636 "Class %s has no parameter \'%s\'" % (cls.__name__, attr) 637 638 def __getattr__(cls, attr): 639 if attr == 'cxx_class_path': 640 return cls.cxx_class.split('::') 641 642 if attr == 'cxx_class_name': 643 return cls.cxx_class_path[-1] 644 645 if attr == 'cxx_namespaces': 646 return cls.cxx_class_path[:-1] 647 648 if cls._values.has_key(attr): 649 return cls._values[attr] 650 651 if cls._children.has_key(attr): 652 return cls._children[attr] 653 654 raise AttributeError, \ 655 "object '%s' has no attribute '%s'" % (cls.__name__, attr) 656 657 def __str__(cls): 658 return cls.__name__ 659 660 # See ParamValue.cxx_predecls for description. 661 def cxx_predecls(cls, code): 662 code('#include "params/$cls.hh"') 663 664 def pybind_predecls(cls, code): 665 code('#include "${{cls.cxx_header}}"') 666 667 def pybind_decl(cls, code): 668 class_path = cls.cxx_class.split('::') 669 namespaces, classname = class_path[:-1], class_path[-1] 670 py_class_name = '_COLONS_'.join(class_path) if namespaces else \ 671 classname; 672 673 # The 'local' attribute restricts us to the params declared in 674 # the object itself, not including inherited params (which 675 # will also be inherited from the base class's param struct 676 # here). Sort the params based on their key 677 params = map(lambda (k, v): v, sorted(cls._params.local.items())) 678 ports = cls._ports.local 679 680 code('''#include "pybind11/pybind11.h" 681#include "pybind11/stl.h" 682 683#include "params/$cls.hh" 684#include "python/pybind11/core.hh" 685#include "sim/init.hh" 686#include "sim/sim_object.hh" 687 688#include "${{cls.cxx_header}}" 689 690''') 691 692 for param in params: 693 param.pybind_predecls(code) 694 695 code('''namespace py = pybind11; 696 697static void 698module_init(py::module &m_internal) 699{ 700 py::module m = m_internal.def_submodule("param_${cls}"); 701''') 702 code.indent() 703 if cls._base: 704 code('py::class_<${cls}Params, ${{cls._base.type}}Params>(m, ' \ 705 '"${cls}Params")') 706 else: 707 code('py::class_<${cls}Params>(m, "${cls}Params")') 708 709 code.indent() 710 if not hasattr(cls, 'abstract') or not cls.abstract: 711 code('.def(py::init<>())') 712 code('.def("create", &${cls}Params::create)') 713 714 param_exports = cls.cxx_param_exports + [ 715 PyBindProperty(k) 716 for k, v in sorted(cls._params.local.items()) 717 ] + [ 718 PyBindProperty("port_%s_connection_count" % port.name) 719 for port in ports.itervalues() 720 ] 721 for exp in param_exports: 722 exp.export(code, "%sParams" % cls) 723 724 code(';') 725 code() 726 code.dedent() 727 728 bases = [ cls._base.cxx_class ] + cls.cxx_bases if cls._base else \ 729 cls.cxx_bases 730 if bases: 731 base_str = ", ".join(bases) 732 code('py::class_<${{cls.cxx_class}}, ${base_str}>(m, ' \ 733 '"${py_class_name}")') 734 else: 735 code('py::class_<${{cls.cxx_class}}>(m, "${py_class_name}")') 736 code.indent() 737 for exp in cls.cxx_exports: 738 exp.export(code, cls.cxx_class) 739 code(';') 740 code.dedent() 741 code() 742 code.dedent() 743 code('}') 744 code() 745 code('static EmbeddedPyBind embed_obj("${0}", module_init, "${1}");', 746 cls, cls._base.type if cls._base else "") 747 748 749 # Generate the C++ declaration (.hh file) for this SimObject's 750 # param struct. Called from src/SConscript. 751 def cxx_param_decl(cls, code): 752 # The 'local' attribute restricts us to the params declared in 753 # the object itself, not including inherited params (which 754 # will also be inherited from the base class's param struct 755 # here). Sort the params based on their key 756 params = map(lambda (k, v): v, sorted(cls._params.local.items())) 757 ports = cls._ports.local 758 try: 759 ptypes = [p.ptype for p in params] 760 except: 761 print cls, p, p.ptype_str 762 print params 763 raise 764 765 class_path = cls._value_dict['cxx_class'].split('::') 766 767 code('''\ 768#ifndef __PARAMS__${cls}__ 769#define __PARAMS__${cls}__ 770 771''') 772 773 774 # The base SimObject has a couple of params that get 775 # automatically set from Python without being declared through 776 # the normal Param mechanism; we slip them in here (needed 777 # predecls now, actual declarations below) 778 if cls == SimObject: 779 code('''#include <string>''') 780 781 # A forward class declaration is sufficient since we are just 782 # declaring a pointer. 783 for ns in class_path[:-1]: 784 code('namespace $ns {') 785 code('class $0;', class_path[-1]) 786 for ns in reversed(class_path[:-1]): 787 code('} // namespace $ns') 788 code() 789 790 for param in params: 791 param.cxx_predecls(code) 792 for port in ports.itervalues(): 793 port.cxx_predecls(code) 794 code() 795 796 if cls._base: 797 code('#include "params/${{cls._base.type}}.hh"') 798 code() 799 800 for ptype in ptypes: 801 if issubclass(ptype, Enum): 802 code('#include "enums/${{ptype.__name__}}.hh"') 803 code() 804 805 # now generate the actual param struct 806 code("struct ${cls}Params") 807 if cls._base: 808 code(" : public ${{cls._base.type}}Params") 809 code("{") 810 if not hasattr(cls, 'abstract') or not cls.abstract: 811 if 'type' in cls.__dict__: 812 code(" ${{cls.cxx_type}} create();") 813 814 code.indent() 815 if cls == SimObject: 816 code(''' 817 SimObjectParams() {} 818 virtual ~SimObjectParams() {} 819 820 std::string name; 821 ''') 822 823 for param in params: 824 param.cxx_decl(code) 825 for port in ports.itervalues(): 826 port.cxx_decl(code) 827 828 code.dedent() 829 code('};') 830 831 code() 832 code('#endif // __PARAMS__${cls}__') 833 return code 834 835 # Generate the C++ declaration/definition files for this SimObject's 836 # param struct to allow C++ initialisation 837 def cxx_config_param_file(cls, code, is_header): 838 createCxxConfigDirectoryEntryFile(code, cls.__name__, cls, is_header) 839 return code 840 841# This *temporary* definition is required to support calls from the 842# SimObject class definition to the MetaSimObject methods (in 843# particular _set_param, which gets called for parameters with default 844# values defined on the SimObject class itself). It will get 845# overridden by the permanent definition (which requires that 846# SimObject be defined) lower in this file. 847def isSimObjectOrVector(value): 848 return False 849 850def cxxMethod(*args, **kwargs): 851 """Decorator to export C++ functions to Python""" 852 853 def decorate(func): 854 name = func.func_name 855 override = kwargs.get("override", False) 856 cxx_name = kwargs.get("cxx_name", name) 857 858 args, varargs, keywords, defaults = inspect.getargspec(func) 859 if varargs or keywords: 860 raise ValueError("Wrapped methods must not contain variable " \ 861 "arguments") 862 863 # Create tuples of (argument, default) 864 if defaults: 865 args = args[:-len(defaults)] + zip(args[-len(defaults):], defaults) 866 # Don't include self in the argument list to PyBind 867 args = args[1:] 868 869 870 @wraps(func) 871 def cxx_call(self, *args, **kwargs): 872 ccobj = self.getCCObject() 873 return getattr(ccobj, name)(*args, **kwargs) 874 875 @wraps(func) 876 def py_call(self, *args, **kwargs): 877 return self.func(*args, **kwargs) 878 879 f = py_call if override else cxx_call 880 f.__pybind = PyBindMethod(name, cxx_name=cxx_name, args=args) 881 882 return f 883 884 if len(args) == 0: 885 return decorate 886 elif len(args) == 1 and len(kwargs) == 0: 887 return decorate(*args) 888 else: 889 raise TypeError("One argument and no kwargs, or only kwargs expected") 890 891# This class holds information about each simobject parameter 892# that should be displayed on the command line for use in the 893# configuration system. 894class ParamInfo(object): 895 def __init__(self, type, desc, type_str, example, default_val, access_str): 896 self.type = type 897 self.desc = desc 898 self.type_str = type_str 899 self.example_str = example 900 self.default_val = default_val 901 # The string representation used to access this param through python. 902 # The method to access this parameter presented on the command line may 903 # be different, so this needs to be stored for later use. 904 self.access_str = access_str 905 self.created = True 906 907 # Make it so we can only set attributes at initialization time 908 # and effectively make this a const object. 909 def __setattr__(self, name, value): 910 if not "created" in self.__dict__: 911 self.__dict__[name] = value 912 913# The SimObject class is the root of the special hierarchy. Most of 914# the code in this class deals with the configuration hierarchy itself 915# (parent/child node relationships). 916class SimObject(object): 917 # Specify metaclass. Any class inheriting from SimObject will 918 # get this metaclass. 919 __metaclass__ = MetaSimObject 920 type = 'SimObject' 921 abstract = True 922 923 cxx_header = "sim/sim_object.hh" 924 cxx_bases = [ "Drainable", "Serializable" ] 925 eventq_index = Param.UInt32(Parent.eventq_index, "Event Queue Index") 926 927 cxx_exports = [ 928 PyBindMethod("init"), 929 PyBindMethod("initState"), 930 PyBindMethod("memInvalidate"), 931 PyBindMethod("memWriteback"), 932 PyBindMethod("regStats"), 933 PyBindMethod("resetStats"), 934 PyBindMethod("regProbePoints"), 935 PyBindMethod("regProbeListeners"), 936 PyBindMethod("startup"), 937 ] 938 939 cxx_param_exports = [ 940 PyBindProperty("name"), 941 ] 942 943 @cxxMethod 944 def loadState(self, cp): 945 """Load SimObject state from a checkpoint""" 946 pass 947 948 # Returns a dict of all the option strings that can be 949 # generated as command line options for this simobject instance 950 # by tracing all reachable params in the top level instance and 951 # any children it contains. 952 def enumerateParams(self, flags_dict = {}, 953 cmd_line_str = "", access_str = ""): 954 if hasattr(self, "_paramEnumed"): 955 print "Cycle detected enumerating params" 956 else: 957 self._paramEnumed = True 958 # Scan the children first to pick up all the objects in this SimObj 959 for keys in self._children: 960 child = self._children[keys] 961 next_cmdline_str = cmd_line_str + keys 962 next_access_str = access_str + keys 963 if not isSimObjectVector(child): 964 next_cmdline_str = next_cmdline_str + "." 965 next_access_str = next_access_str + "." 966 flags_dict = child.enumerateParams(flags_dict, 967 next_cmdline_str, 968 next_access_str) 969 970 # Go through the simple params in the simobject in this level 971 # of the simobject hierarchy and save information about the 972 # parameter to be used for generating and processing command line 973 # options to the simulator to set these parameters. 974 for keys,values in self._params.items(): 975 if values.isCmdLineSettable(): 976 type_str = '' 977 ex_str = values.example_str() 978 ptype = None 979 if isinstance(values, VectorParamDesc): 980 type_str = 'Vector_%s' % values.ptype_str 981 ptype = values 982 else: 983 type_str = '%s' % values.ptype_str 984 ptype = values.ptype 985 986 if keys in self._hr_values\ 987 and keys in self._values\ 988 and not isinstance(self._values[keys], 989 m5.proxy.BaseProxy): 990 cmd_str = cmd_line_str + keys 991 acc_str = access_str + keys 992 flags_dict[cmd_str] = ParamInfo(ptype, 993 self._params[keys].desc, type_str, ex_str, 994 values.pretty_print(self._hr_values[keys]), 995 acc_str) 996 elif not keys in self._hr_values\ 997 and not keys in self._values: 998 # Empty param 999 cmd_str = cmd_line_str + keys 1000 acc_str = access_str + keys 1001 flags_dict[cmd_str] = ParamInfo(ptype, 1002 self._params[keys].desc, 1003 type_str, ex_str, '', acc_str) 1004 1005 return flags_dict 1006 1007 # Initialize new instance. For objects with SimObject-valued 1008 # children, we need to recursively clone the classes represented 1009 # by those param values as well in a consistent "deep copy"-style 1010 # fashion. That is, we want to make sure that each instance is 1011 # cloned only once, and that if there are multiple references to 1012 # the same original object, we end up with the corresponding 1013 # cloned references all pointing to the same cloned instance. 1014 def __init__(self, **kwargs): 1015 ancestor = kwargs.get('_ancestor') 1016 memo_dict = kwargs.get('_memo') 1017 if memo_dict is None: 1018 # prepare to memoize any recursively instantiated objects 1019 memo_dict = {} 1020 elif ancestor: 1021 # memoize me now to avoid problems with recursive calls 1022 memo_dict[ancestor] = self 1023 1024 if not ancestor: 1025 ancestor = self.__class__ 1026 ancestor._instantiated = True 1027 1028 # initialize required attributes 1029 self._parent = None 1030 self._name = None 1031 self._ccObject = None # pointer to C++ object 1032 self._ccParams = None 1033 self._instantiated = False # really "cloned" 1034 1035 # Clone children specified at class level. No need for a 1036 # multidict here since we will be cloning everything. 1037 # Do children before parameter values so that children that 1038 # are also param values get cloned properly. 1039 self._children = {} 1040 for key,val in ancestor._children.iteritems(): 1041 self.add_child(key, val(_memo=memo_dict)) 1042 1043 # Inherit parameter values from class using multidict so 1044 # individual value settings can be overridden but we still 1045 # inherit late changes to non-overridden class values. 1046 self._values = multidict(ancestor._values) 1047 self._hr_values = multidict(ancestor._hr_values) 1048 # clone SimObject-valued parameters 1049 for key,val in ancestor._values.iteritems(): 1050 val = tryAsSimObjectOrVector(val) 1051 if val is not None: 1052 self._values[key] = val(_memo=memo_dict) 1053 1054 # clone port references. no need to use a multidict here 1055 # since we will be creating new references for all ports. 1056 self._port_refs = {} 1057 for key,val in ancestor._port_refs.iteritems(): 1058 self._port_refs[key] = val.clone(self, memo_dict) 1059 # apply attribute assignments from keyword args, if any 1060 for key,val in kwargs.iteritems(): 1061 setattr(self, key, val) 1062 1063 # "Clone" the current instance by creating another instance of 1064 # this instance's class, but that inherits its parameter values 1065 # and port mappings from the current instance. If we're in a 1066 # "deep copy" recursive clone, check the _memo dict to see if 1067 # we've already cloned this instance. 1068 def __call__(self, **kwargs): 1069 memo_dict = kwargs.get('_memo') 1070 if memo_dict is None: 1071 # no memo_dict: must be top-level clone operation. 1072 # this is only allowed at the root of a hierarchy 1073 if self._parent: 1074 raise RuntimeError, "attempt to clone object %s " \ 1075 "not at the root of a tree (parent = %s)" \ 1076 % (self, self._parent) 1077 # create a new dict and use that. 1078 memo_dict = {} 1079 kwargs['_memo'] = memo_dict 1080 elif memo_dict.has_key(self): 1081 # clone already done & memoized 1082 return memo_dict[self] 1083 return self.__class__(_ancestor = self, **kwargs) 1084 1085 def _get_port_ref(self, attr): 1086 # Return reference that can be assigned to another port 1087 # via __setattr__. There is only ever one reference 1088 # object per port, but we create them lazily here. 1089 ref = self._port_refs.get(attr) 1090 if ref == None: 1091 ref = self._ports[attr].makeRef(self) 1092 self._port_refs[attr] = ref 1093 return ref 1094 1095 def __getattr__(self, attr): 1096 if self._ports.has_key(attr): 1097 return self._get_port_ref(attr) 1098 1099 if self._values.has_key(attr): 1100 return self._values[attr] 1101 1102 if self._children.has_key(attr): 1103 return self._children[attr] 1104 1105 # If the attribute exists on the C++ object, transparently 1106 # forward the reference there. This is typically used for 1107 # methods exported to Python (e.g., init(), and startup()) 1108 if self._ccObject and hasattr(self._ccObject, attr): 1109 return getattr(self._ccObject, attr) 1110 1111 err_string = "object '%s' has no attribute '%s'" \ 1112 % (self.__class__.__name__, attr) 1113 1114 if not self._ccObject: 1115 err_string += "\n (C++ object is not yet constructed," \ 1116 " so wrapped C++ methods are unavailable.)" 1117 1118 raise AttributeError, err_string 1119 1120 # Set attribute (called on foo.attr = value when foo is an 1121 # instance of class cls). 1122 def __setattr__(self, attr, value): 1123 # normal processing for private attributes 1124 if attr.startswith('_'): 1125 object.__setattr__(self, attr, value) 1126 return 1127 1128 if self._ports.has_key(attr): 1129 # set up port connection 1130 self._get_port_ref(attr).connect(value) 1131 return 1132 1133 param = self._params.get(attr) 1134 if param: 1135 try: 1136 hr_value = value 1137 value = param.convert(value) 1138 except Exception, e: 1139 msg = "%s\nError setting param %s.%s to %s\n" % \ 1140 (e, self.__class__.__name__, attr, value) 1141 e.args = (msg, ) 1142 raise 1143 self._values[attr] = value 1144 # implicitly parent unparented objects assigned as params 1145 if isSimObjectOrVector(value) and not value.has_parent(): 1146 self.add_child(attr, value) 1147 # set the human-readable value dict if this is a param 1148 # with a literal value and is not being set as an object 1149 # or proxy. 1150 if not (isSimObjectOrVector(value) or\ 1151 isinstance(value, m5.proxy.BaseProxy)): 1152 self._hr_values[attr] = hr_value 1153 1154 return 1155 1156 # if RHS is a SimObject, it's an implicit child assignment 1157 if isSimObjectOrSequence(value): 1158 self.add_child(attr, value) 1159 return 1160 1161 # no valid assignment... raise exception 1162 raise AttributeError, "Class %s has no parameter %s" \ 1163 % (self.__class__.__name__, attr) 1164 1165 1166 # this hack allows tacking a '[0]' onto parameters that may or may 1167 # not be vectors, and always getting the first element (e.g. cpus) 1168 def __getitem__(self, key): 1169 if key == 0: 1170 return self 1171 raise IndexError, "Non-zero index '%s' to SimObject" % key 1172 1173 # this hack allows us to iterate over a SimObject that may 1174 # not be a vector, so we can call a loop over it and get just one 1175 # element. 1176 def __len__(self): 1177 return 1 1178 1179 # Also implemented by SimObjectVector 1180 def clear_parent(self, old_parent): 1181 assert self._parent is old_parent 1182 self._parent = None 1183 1184 # Also implemented by SimObjectVector 1185 def set_parent(self, parent, name): 1186 self._parent = parent 1187 self._name = name 1188 1189 # Return parent object of this SimObject, not implemented by 1190 # SimObjectVector because the elements in a SimObjectVector may not share 1191 # the same parent 1192 def get_parent(self): 1193 return self._parent 1194 1195 # Also implemented by SimObjectVector 1196 def get_name(self): 1197 return self._name 1198 1199 # Also implemented by SimObjectVector 1200 def has_parent(self): 1201 return self._parent is not None 1202 1203 # clear out child with given name. This code is not likely to be exercised. 1204 # See comment in add_child. 1205 def clear_child(self, name): 1206 child = self._children[name] 1207 child.clear_parent(self) 1208 del self._children[name] 1209 1210 # Add a new child to this object. 1211 def add_child(self, name, child): 1212 child = coerceSimObjectOrVector(child) 1213 if child.has_parent(): 1214 warn("add_child('%s'): child '%s' already has parent", name, 1215 child.get_name()) 1216 if self._children.has_key(name): 1217 # This code path had an undiscovered bug that would make it fail 1218 # at runtime. It had been here for a long time and was only 1219 # exposed by a buggy script. Changes here will probably not be 1220 # exercised without specialized testing. 1221 self.clear_child(name) 1222 child.set_parent(self, name) 1223 self._children[name] = child 1224 1225 # Take SimObject-valued parameters that haven't been explicitly 1226 # assigned as children and make them children of the object that 1227 # they were assigned to as a parameter value. This guarantees 1228 # that when we instantiate all the parameter objects we're still 1229 # inside the configuration hierarchy. 1230 def adoptOrphanParams(self): 1231 for key,val in self._values.iteritems(): 1232 if not isSimObjectVector(val) and isSimObjectSequence(val): 1233 # need to convert raw SimObject sequences to 1234 # SimObjectVector class so we can call has_parent() 1235 val = SimObjectVector(val) 1236 self._values[key] = val 1237 if isSimObjectOrVector(val) and not val.has_parent(): 1238 warn("%s adopting orphan SimObject param '%s'", self, key) 1239 self.add_child(key, val) 1240 1241 def path(self): 1242 if not self._parent: 1243 return '<orphan %s>' % self.__class__ 1244 elif isinstance(self._parent, MetaSimObject): 1245 return str(self.__class__) 1246 1247 ppath = self._parent.path() 1248 if ppath == 'root': 1249 return self._name 1250 return ppath + "." + self._name 1251 1252 def __str__(self): 1253 return self.path() 1254 1255 def config_value(self): 1256 return self.path() 1257 1258 def ini_str(self): 1259 return self.path() 1260 1261 def find_any(self, ptype): 1262 if isinstance(self, ptype): 1263 return self, True 1264 1265 found_obj = None 1266 for child in self._children.itervalues(): 1267 visited = False 1268 if hasattr(child, '_visited'): 1269 visited = getattr(child, '_visited') 1270 1271 if isinstance(child, ptype) and not visited: 1272 if found_obj != None and child != found_obj: 1273 raise AttributeError, \ 1274 'parent.any matched more than one: %s %s' % \ 1275 (found_obj.path, child.path) 1276 found_obj = child 1277 # search param space 1278 for pname,pdesc in self._params.iteritems(): 1279 if issubclass(pdesc.ptype, ptype): 1280 match_obj = self._values[pname] 1281 if found_obj != None and found_obj != match_obj: 1282 raise AttributeError, \ 1283 'parent.any matched more than one: %s and %s' % \ 1284 (found_obj.path, match_obj.path) 1285 found_obj = match_obj 1286 return found_obj, found_obj != None 1287 1288 def find_all(self, ptype): 1289 all = {} 1290 # search children 1291 for child in self._children.itervalues(): 1292 # a child could be a list, so ensure we visit each item 1293 if isinstance(child, list): 1294 children = child 1295 else: 1296 children = [child] 1297 1298 for child in children: 1299 if isinstance(child, ptype) and not isproxy(child) and \ 1300 not isNullPointer(child): 1301 all[child] = True 1302 if isSimObject(child): 1303 # also add results from the child itself 1304 child_all, done = child.find_all(ptype) 1305 all.update(dict(zip(child_all, [done] * len(child_all)))) 1306 # search param space 1307 for pname,pdesc in self._params.iteritems(): 1308 if issubclass(pdesc.ptype, ptype): 1309 match_obj = self._values[pname] 1310 if not isproxy(match_obj) and not isNullPointer(match_obj): 1311 all[match_obj] = True 1312 # Also make sure to sort the keys based on the objects' path to 1313 # ensure that the order is the same on all hosts 1314 return sorted(all.keys(), key = lambda o: o.path()), True 1315 1316 def unproxy(self, base): 1317 return self 1318 1319 def unproxyParams(self): 1320 for param in self._params.iterkeys(): 1321 value = self._values.get(param) 1322 if value != None and isproxy(value): 1323 try: 1324 value = value.unproxy(self) 1325 except: 1326 print "Error in unproxying param '%s' of %s" % \ 1327 (param, self.path()) 1328 raise 1329 setattr(self, param, value) 1330 1331 # Unproxy ports in sorted order so that 'append' operations on 1332 # vector ports are done in a deterministic fashion. 1333 port_names = self._ports.keys() 1334 port_names.sort() 1335 for port_name in port_names: 1336 port = self._port_refs.get(port_name) 1337 if port != None: 1338 port.unproxy(self) 1339 1340 def print_ini(self, ini_file): 1341 print >>ini_file, '[' + self.path() + ']' # .ini section header 1342 1343 instanceDict[self.path()] = self 1344 1345 if hasattr(self, 'type'): 1346 print >>ini_file, 'type=%s' % self.type 1347 1348 if len(self._children.keys()): 1349 print >>ini_file, 'children=%s' % \ 1350 ' '.join(self._children[n].get_name() \ 1351 for n in sorted(self._children.keys())) 1352 1353 for param in sorted(self._params.keys()): 1354 value = self._values.get(param) 1355 if value != None: 1356 print >>ini_file, '%s=%s' % (param, 1357 self._values[param].ini_str()) 1358 1359 for port_name in sorted(self._ports.keys()): 1360 port = self._port_refs.get(port_name, None) 1361 if port != None: 1362 print >>ini_file, '%s=%s' % (port_name, port.ini_str()) 1363 1364 print >>ini_file # blank line between objects 1365 1366 # generate a tree of dictionaries expressing all the parameters in the 1367 # instantiated system for use by scripts that want to do power, thermal 1368 # visualization, and other similar tasks 1369 def get_config_as_dict(self): 1370 d = attrdict() 1371 if hasattr(self, 'type'): 1372 d.type = self.type 1373 if hasattr(self, 'cxx_class'): 1374 d.cxx_class = self.cxx_class 1375 # Add the name and path of this object to be able to link to 1376 # the stats 1377 d.name = self.get_name() 1378 d.path = self.path() 1379 1380 for param in sorted(self._params.keys()): 1381 value = self._values.get(param) 1382 if value != None: 1383 d[param] = value.config_value() 1384 1385 for n in sorted(self._children.keys()): 1386 child = self._children[n] 1387 # Use the name of the attribute (and not get_name()) as 1388 # the key in the JSON dictionary to capture the hierarchy 1389 # in the Python code that assembled this system 1390 d[n] = child.get_config_as_dict() 1391 1392 for port_name in sorted(self._ports.keys()): 1393 port = self._port_refs.get(port_name, None) 1394 if port != None: 1395 # Represent each port with a dictionary containing the 1396 # prominent attributes 1397 d[port_name] = port.get_config_as_dict() 1398 1399 return d 1400 1401 def getCCParams(self): 1402 if self._ccParams: 1403 return self._ccParams 1404 1405 cc_params_struct = getattr(m5.internal.params, '%sParams' % self.type) 1406 cc_params = cc_params_struct() 1407 cc_params.name = str(self) 1408 1409 param_names = self._params.keys() 1410 param_names.sort() 1411 for param in param_names: 1412 value = self._values.get(param) 1413 if value is None: 1414 fatal("%s.%s without default or user set value", 1415 self.path(), param) 1416 1417 value = value.getValue() 1418 if isinstance(self._params[param], VectorParamDesc): 1419 assert isinstance(value, list) 1420 vec = getattr(cc_params, param) 1421 assert not len(vec) 1422 # Some types are exposed as opaque types. They support 1423 # the append operation unlike the automatically 1424 # wrapped types. 1425 if isinstance(vec, list): 1426 setattr(cc_params, param, list(value)) 1427 else: 1428 for v in value: 1429 getattr(cc_params, param).append(v) 1430 else: 1431 setattr(cc_params, param, value) 1432 1433 port_names = self._ports.keys() 1434 port_names.sort() 1435 for port_name in port_names: 1436 port = self._port_refs.get(port_name, None) 1437 if port != None: 1438 port_count = len(port) 1439 else: 1440 port_count = 0 1441 setattr(cc_params, 'port_' + port_name + '_connection_count', 1442 port_count) 1443 self._ccParams = cc_params 1444 return self._ccParams 1445 1446 # Get C++ object corresponding to this object, calling C++ if 1447 # necessary to construct it. Does *not* recursively create 1448 # children. 1449 def getCCObject(self): 1450 if not self._ccObject: 1451 # Make sure this object is in the configuration hierarchy 1452 if not self._parent and not isRoot(self): 1453 raise RuntimeError, "Attempt to instantiate orphan node" 1454 # Cycles in the configuration hierarchy are not supported. This 1455 # will catch the resulting recursion and stop. 1456 self._ccObject = -1 1457 if not self.abstract: 1458 params = self.getCCParams() 1459 self._ccObject = params.create() 1460 elif self._ccObject == -1: 1461 raise RuntimeError, "%s: Cycle found in configuration hierarchy." \ 1462 % self.path() 1463 return self._ccObject 1464 1465 def descendants(self): 1466 yield self 1467 # The order of the dict is implementation dependent, so sort 1468 # it based on the key (name) to ensure the order is the same 1469 # on all hosts 1470 for (name, child) in sorted(self._children.iteritems()): 1471 for obj in child.descendants(): 1472 yield obj 1473 1474 # Call C++ to create C++ object corresponding to this object 1475 def createCCObject(self): 1476 self.getCCParams() 1477 self.getCCObject() # force creation 1478 1479 def getValue(self): 1480 return self.getCCObject() 1481 1482 # Create C++ port connections corresponding to the connections in 1483 # _port_refs 1484 def connectPorts(self): 1485 # Sort the ports based on their attribute name to ensure the 1486 # order is the same on all hosts 1487 for (attr, portRef) in sorted(self._port_refs.iteritems()): 1488 portRef.ccConnect() 1489 1490# Function to provide to C++ so it can look up instances based on paths 1491def resolveSimObject(name): 1492 obj = instanceDict[name] 1493 return obj.getCCObject() 1494 1495def isSimObject(value): 1496 return isinstance(value, SimObject) 1497 1498def isSimObjectClass(value): 1499 return issubclass(value, SimObject) 1500 1501def isSimObjectVector(value): 1502 return isinstance(value, SimObjectVector) 1503 1504def isSimObjectSequence(value): 1505 if not isinstance(value, (list, tuple)) or len(value) == 0: 1506 return False 1507 1508 for val in value: 1509 if not isNullPointer(val) and not isSimObject(val): 1510 return False 1511 1512 return True 1513 1514def isSimObjectOrSequence(value): 1515 return isSimObject(value) or isSimObjectSequence(value) 1516 1517def isRoot(obj): 1518 from m5.objects import Root 1519 return obj and obj is Root.getInstance() 1520 1521def isSimObjectOrVector(value): 1522 return isSimObject(value) or isSimObjectVector(value) 1523 1524def tryAsSimObjectOrVector(value): 1525 if isSimObjectOrVector(value): 1526 return value 1527 if isSimObjectSequence(value): 1528 return SimObjectVector(value) 1529 return None 1530 1531def coerceSimObjectOrVector(value): 1532 value = tryAsSimObjectOrVector(value) 1533 if value is None: 1534 raise TypeError, "SimObject or SimObjectVector expected" 1535 return value 1536 1537baseClasses = allClasses.copy() 1538baseInstances = instanceDict.copy() 1539 1540def clear(): 1541 global allClasses, instanceDict, noCxxHeader 1542 1543 allClasses = baseClasses.copy() 1544 instanceDict = baseInstances.copy() 1545 noCxxHeader = False 1546 1547# __all__ defines the list of symbols that get exported when 1548# 'from config import *' is invoked. Try to keep this reasonably 1549# short to avoid polluting other namespaces. 1550__all__ = [ 1551 'SimObject', 1552 'cxxMethod', 1553 'PyBindMethod', 1554 'PyBindProperty', 1555] 1556