module.hh revision 13079:e7e261dd975b
1/* 2 * Copyright 2018 Google, Inc. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions are 6 * met: redistributions of source code must retain the above copyright 7 * notice, this list of conditions and the following disclaimer; 8 * redistributions in binary form must reproduce the above copyright 9 * notice, this list of conditions and the following disclaimer in the 10 * documentation and/or other materials provided with the distribution; 11 * neither the name of the copyright holders nor the names of its 12 * contributors may be used to endorse or promote products derived from 13 * this software without specific prior written permission. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 16 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 17 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 18 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 19 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 20 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 21 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 25 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 26 * 27 * Authors: Gabe Black 28 */ 29 30#ifndef __SYSTEMC_CORE_MODULE_HH__ 31#define __SYSTEMC_CORE_MODULE_HH__ 32 33#include <cassert> 34#include <list> 35#include <map> 36#include <sstream> 37#include <string> 38#include <vector> 39 40#include "systemc/core/object.hh" 41#include "systemc/ext/core/sc_module.hh" 42 43namespace sc_core 44{ 45 46class sc_port_base; 47class sc_export_base; 48 49} // namespace sc_core 50 51namespace sc_gem5 52{ 53 54class UniqueNameGen 55{ 56 private: 57 std::map<std::string, int> counts; 58 std::string buf; 59 60 public: 61 const char * 62 gen(std::string seed) 63 { 64 std::ostringstream os; 65 os << seed << "_" << counts[seed]++; 66 buf = os.str(); 67 return buf.c_str(); 68 } 69}; 70 71class Module 72{ 73 private: 74 const char *_name; 75 sc_core::sc_module *_sc_mod; 76 Object *_obj; 77 78 UniqueNameGen nameGen; 79 80 public: 81 82 Module(const char *name); 83 ~Module(); 84 85 void finish(Object *this_obj); 86 87 const char *name() const { return _name; } 88 89 sc_core::sc_module * 90 sc_mod() const 91 { 92 assert(_sc_mod); 93 return _sc_mod; 94 } 95 96 void 97 sc_mod(sc_core::sc_module *sc_mod) 98 { 99 assert(!_sc_mod); 100 _sc_mod = sc_mod; 101 } 102 103 Object * 104 obj() 105 { 106 assert(_obj); 107 return _obj; 108 } 109 110 void pop(); 111 112 const char *uniqueName(const char *seed) { return nameGen.gen(seed); } 113 114 std::vector<::sc_core::sc_port_base *> ports; 115 std::vector<::sc_core::sc_export_base *> exports; 116}; 117 118Module *currentModule(); 119Module *newModuleChecked(); 120Module *newModule(); 121 122void callbackModule(Module *m); 123Module *callbackModule(); 124 125extern std::list<Module *> allModules; 126 127} // namespace sc_gem5 128 129#endif //__SYSTEMC_CORE_MODULE_HH__ 130