# Copyright (c) 1999-2008 Mark D. Hill and David A. Wood # Copyright (c) 2009 The Hewlett-Packard Development Company # All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions are # met: redistributions of source code must retain the above copyright # notice, this list of conditions and the following disclaimer; # redistributions in binary form must reproduce the above copyright # notice, this list of conditions and the following disclaimer in the # documentation and/or other materials provided with the distribution; # neither the name of the copyright holders nor the names of its # contributors may be used to endorse or promote products derived from # this software without specific prior written permission. # # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. from m5.util import code_formatter, orderdict from slicc.symbols.Symbol import Symbol from slicc.symbols.Var import Var import slicc.generate.html as html class StateMachine(Symbol): def __init__(self, symtab, ident, location, pairs, config_parameters): super(StateMachine, self).__init__(symtab, ident, location, pairs) self.table = None self.config_parameters = config_parameters for param in config_parameters: var = Var(symtab, param.name, location, param.type_ast.type, "m_%s" % param.name, {}, self) self.symtab.registerSym(param.name, var) self.states = orderdict() self.events = orderdict() self.actions = orderdict() self.transitions = [] self.in_ports = [] self.functions = [] self.objects = [] self.message_buffer_names = [] def __repr__(self): return "[StateMachine: %s]" % self.ident def addState(self, state): assert self.table is None self.states[state.ident] = state def addEvent(self, event): assert self.table is None self.events[event.ident] = event def addAction(self, action): assert self.table is None # Check for duplicate action for other in self.actions.itervalues(): if action.ident == other.ident: action.warning("Duplicate action definition: %s" % action.ident) action.error("Duplicate action definition: %s" % action.ident) if action.short == other.short: other.warning("Duplicate action shorthand: %s" % other.ident) other.warning(" shorthand = %s" % other.short) action.warning("Duplicate action shorthand: %s" % action.ident) action.error(" shorthand = %s" % action.short) self.actions[action.ident] = action def addTransition(self, trans): assert self.table is None self.transitions.append(trans) def addInPort(self, var): self.in_ports.append(var) def addFunc(self, func): # register func in the symbol table self.symtab.registerSym(str(func), func) self.functions.append(func) def addObject(self, obj): self.objects.append(obj) # Needs to be called before accessing the table def buildTable(self): assert self.table is None table = {} for trans in self.transitions: # Track which actions we touch so we know if we use them # all -- really this should be done for all symbols as # part of the symbol table, then only trigger it for # Actions, States, Events, etc. for action in trans.actions: action.used = True index = (trans.state, trans.event) if index in table: table[index].warning("Duplicate transition: %s" % table[index]) trans.error("Duplicate transition: %s" % trans) table[index] = trans # Look at all actions to make sure we used them all for action in self.actions.itervalues(): if not action.used: error_msg = "Unused action: %s" % action.ident if "desc" in action: error_msg += ", " + action.desc action.warning(error_msg) self.table = table def writeCodeFiles(self, path): self.printControllerHH(path) self.printControllerCC(path) self.printCSwitch(path) self.printCWakeup(path) self.printProfilerCC(path) self.printProfilerHH(path) for func in self.functions: func.writeCodeFiles(path) def printControllerHH(self, path): '''Output the method declarations for the class declaration''' code = code_formatter() ident = self.ident c_ident = "%s_Controller" % self.ident self.message_buffer_names = [] code(''' /** \\file $ident.hh * * Auto generated C++ code started by $__file__:$__line__ * Created by slicc definition of Module "${{self.short}}" */ #ifndef ${ident}_CONTROLLER_H #define ${ident}_CONTROLLER_H #include "mem/ruby/common/Global.hh" #include "mem/ruby/common/Consumer.hh" #include "mem/ruby/slicc_interface/AbstractController.hh" #include "mem/protocol/TransitionResult.hh" #include "mem/protocol/Types.hh" #include "mem/protocol/${ident}_Profiler.hh" ''') seen_types = set() for var in self.objects: if var.type.ident not in seen_types and not var.type.isPrimitive: code('#include "mem/protocol/${{var.type.c_ident}}.hh"') seen_types.add(var.type.ident) # for adding information to the protocol debug trace code(''' extern stringstream ${ident}_transitionComment; class $c_ident : public AbstractController { #ifdef CHECK_COHERENCE #endif /* CHECK_COHERENCE */ public: $c_ident(const string & name); static int getNumControllers(); void init(Network* net_ptr, const vector & argv); MessageBuffer* getMandatoryQueue() const; const int & getVersion() const; const string toString() const; const string getName() const; const MachineType getMachineType() const; void print(ostream& out) const; void printConfig(ostream& out) const; void wakeup(); void set_atomic(Address addr); void started_writes(); void clear_atomic(); void printStats(ostream& out) const { s_profiler.dumpStats(out); } void clearStats() { s_profiler.clearStats(); } private: ''') code.indent() # added by SS for param in self.config_parameters: code('int m_${{param.ident}};') if self.ident == "L1Cache": code(''' int servicing_atomic; bool started_receiving_writes; Address locked_read_request1; Address locked_read_request2; Address locked_read_request3; Address locked_read_request4; int read_counter; ''') code(''' int m_number_of_TBEs; TransitionResult doTransition(${ident}_Event event, ${ident}_State state, const Address& addr); // in ${ident}_Transitions.cc TransitionResult doTransitionWorker(${ident}_Event event, ${ident}_State state, ${ident}_State& next_state, const Address& addr); // in ${ident}_Transitions.cc string m_name; int m_transitions_per_cycle; int m_buffer_size; int m_recycle_latency; map< string, string > m_cfg; NodeID m_version; Network* m_net_ptr; MachineID m_machineID; ${ident}_Profiler s_profiler; static int m_num_controllers; // Internal functions ''') for func in self.functions: proto = func.prototype if proto: code('$proto') code(''' // Actions ''') for action in self.actions.itervalues(): code('/** \\brief ${{action.desc}} */') code('void ${{action.ident}}(const Address& addr);') # the controller internal variables code(''' // Object ''') for var in self.objects: th = var.get("template_hack", "") code('${{var.type.c_ident}}$th* m_${{var.c_ident}}_ptr;') if var.type.ident == "MessageBuffer": self.message_buffer_names.append("m_%s_ptr" % var.c_ident) code.dedent() code('};') code('#endif // ${ident}_CONTROLLER_H') code.write(path, '%s.hh' % c_ident) def printControllerCC(self, path): '''Output the actions for performing the actions''' code = code_formatter() ident = self.ident c_ident = "%s_Controller" % self.ident code(''' /** \\file $ident.cc * * Auto generated C++ code started by $__file__:$__line__ * Created by slicc definition of Module "${{self.short}}" */ #include "mem/ruby/common/Global.hh" #include "mem/ruby/slicc_interface/RubySlicc_includes.hh" #include "mem/protocol/${ident}_Controller.hh" #include "mem/protocol/${ident}_State.hh" #include "mem/protocol/${ident}_Event.hh" #include "mem/protocol/Types.hh" #include "mem/ruby/system/System.hh" ''') # include object classes seen_types = set() for var in self.objects: if var.type.ident not in seen_types and not var.type.isPrimitive: code('#include "mem/protocol/${{var.type.c_ident}}.hh"') seen_types.add(var.type.ident) code(''' int $c_ident::m_num_controllers = 0; stringstream ${ident}_transitionComment; #define APPEND_TRANSITION_COMMENT(str) (${ident}_transitionComment << str) /** \\brief constructor */ $c_ident::$c_ident(const string &name) : m_name(name) { ''') code.indent() if self.ident == "L1Cache": code(''' servicing_atomic = 0; started_receiving_writes = false; locked_read_request1 = Address(-1); locked_read_request2 = Address(-1); locked_read_request3 = Address(-1); locked_read_request4 = Address(-1); read_counter = 0; ''') code('m_num_controllers++;') for var in self.objects: if var.ident.find("mandatoryQueue") >= 0: code('m_${{var.c_ident}}_ptr = new ${{var.type.c_ident}}();') code.dedent() code(''' } void $c_ident::init(Network *net_ptr, const vector &argv) { for (size_t i = 0; i < argv.size(); i += 2) { if (argv[i] == "version") m_version = atoi(argv[i+1].c_str()); else if (argv[i] == "transitions_per_cycle") m_transitions_per_cycle = atoi(argv[i+1].c_str()); else if (argv[i] == "buffer_size") m_buffer_size = atoi(argv[i+1].c_str()); else if (argv[i] == "recycle_latency") m_recycle_latency = atoi(argv[i+1].c_str()); else if (argv[i] == "number_of_TBEs") m_number_of_TBEs = atoi(argv[i+1].c_str()); ''') code.indent() code.indent() for param in self.config_parameters: code('else if (argv[i] == "${{param.name}}")') if param.type_ast.type.ident == "int": code(' m_${{param.name}} = atoi(argv[i+1].c_str());') elif param.type_ast.type.ident == "bool": code(' m_${{param.name}} = string_to_bool(argv[i+1]);') else: self.error("only int and bool parameters are "\ "currently supported") code.dedent() code.dedent() code(''' } m_net_ptr = net_ptr; m_machineID.type = MachineType_${ident}; m_machineID.num = m_version; for (size_t i = 0; i < argv.size(); i += 2) { if (argv[i] != "version") m_cfg[argv[i]] = argv[i+1]; } // Objects s_profiler.setVersion(m_version); ''') code.indent() for var in self.objects: vtype = var.type vid = "m_%s_ptr" % var.c_ident if "network" not in var: # Not a network port object if "primitive" in vtype: code('$vid = new ${{vtype.c_ident}};') if "default" in var: code('(*$vid) = ${{var["default"]}};') else: # Normal Object # added by SS if "factory" in var: code('$vid = ${{var["factory"]}};') elif var.ident.find("mandatoryQueue") < 0: th = var.get("template_hack", "") expr = "%s = new %s%s" % (vid, vtype.c_ident, th) args = "" if "non_obj" not in vtype and not vtype.isEnumeration: if expr.find("TBETable") >= 0: args = "m_number_of_TBEs" else: args = var.get("constructor_hack", "") args = "(%s)" % args code('$expr$args;') else: code(';') code('assert($vid != NULL);') if "default" in var: code('(*$vid) = ${{var["default"]}}; // Object default') elif "default" in vtype: code('(*$vid) = ${{vtype["default"]}}; // Type ${{vtype.ident}} default') # Set ordering if "ordered" in var and "trigger_queue" not in var: # A buffer code('$vid->setOrdering(${{var["ordered"]}});') # Set randomization if "random" in var: # A buffer code('$vid->setRandomization(${{var["random"]}});') # Set Priority if vtype.isBuffer and \ "rank" in var and "trigger_queue" not in var: code('$vid->setPriority(${{var["rank"]}});') else: # Network port object network = var["network"] ordered = var["ordered"] vnet = var["virtual_network"] assert var.machine is not None code(''' $vid = m_net_ptr->get${network}NetQueue(m_version+MachineType_base_number(string_to_MachineType("${{var.machine.ident}}")), $ordered, $vnet); ''') code('assert($vid != NULL);') # Set ordering if "ordered" in var: # A buffer code('$vid->setOrdering(${{var["ordered"]}});') # Set randomization if "random" in var: # A buffer code('$vid->setRandomization(${{var["random"]}})') # Set Priority if "rank" in var: code('$vid->setPriority(${{var["rank"]}})') # Set buffer size if vtype.isBuffer: code(''' if (m_buffer_size > 0) { $vid->setSize(m_buffer_size); } ''') # set description (may be overriden later by port def) code('$vid->setDescription("[Version " + int_to_string(m_version) + ", ${ident}, name=${{var.c_ident}}]");') # Set the queue consumers code.insert_newline() for port in self.in_ports: code('${{port.code}}.setConsumer(this);') # Set the queue descriptions code.insert_newline() for port in self.in_ports: code('${{port.code}}.setDescription("[Version " + int_to_string(m_version) + ", $ident, $port]");') # Initialize the transition profiling code.insert_newline() for trans in self.transitions: # Figure out if we stall stall = False for action in trans.actions: if action.ident == "z_stall": stall = True # Only possible if it is not a 'z' case if not stall: state = "%s_State_%s" % (self.ident, trans.state.ident) event = "%s_Event_%s" % (self.ident, trans.event.ident) code('s_profiler.possibleTransition($state, $event);') # added by SS to initialize recycle_latency of message buffers for buf in self.message_buffer_names: code("$buf->setRecycleLatency(m_recycle_latency);") code.dedent() code('}') has_mandatory_q = False for port in self.in_ports: if port.code.find("mandatoryQueue_ptr") >= 0: has_mandatory_q = True if has_mandatory_q: mq_ident = "m_%s_mandatoryQueue_ptr" % self.ident else: mq_ident = "NULL" code(''' int $c_ident::getNumControllers() { return m_num_controllers; } MessageBuffer* $c_ident::getMandatoryQueue() const { return $mq_ident; } const int & $c_ident::getVersion() const{ return m_version; } const string $c_ident::toString() const{ return "$c_ident"; } const string $c_ident::getName() const{ return m_name; } const MachineType $c_ident::getMachineType() const{ return MachineType_${ident}; } void $c_ident::print(ostream& out) const { out << "[$c_ident " << m_version << "]"; } void $c_ident::printConfig(ostream& out) const { out << "$c_ident config: " << m_name << endl; out << " version: " << m_version << endl; for (map::const_iterator it = m_cfg.begin(); it != m_cfg.end(); it++) { out << " " << (*it).first << ": " << (*it).second << endl; } } // Actions ''') for action in self.actions.itervalues(): if "c_code" not in action: continue code(''' /** \\brief ${{action.desc}} */ void $c_ident::${{action.ident}}(const Address& addr) { DEBUG_MSG(GENERATED_COMP, HighPrio, "executing"); ${{action["c_code"]}} } ''') code.write(path, "%s.cc" % c_ident) def printCWakeup(self, path): '''Output the wakeup loop for the events''' code = code_formatter() ident = self.ident code(''' // Auto generated C++ code started by $__file__:$__line__ // ${ident}: ${{self.short}} #include "mem/ruby/common/Global.hh" #include "mem/ruby/slicc_interface/RubySlicc_includes.hh" #include "mem/protocol/${ident}_Controller.hh" #include "mem/protocol/${ident}_State.hh" #include "mem/protocol/${ident}_Event.hh" #include "mem/protocol/Types.hh" #include "mem/ruby/system/System.hh" void ${ident}_Controller::wakeup() { int counter = 0; while (true) { // Some cases will put us into an infinite loop without this limit assert(counter <= m_transitions_per_cycle); if (counter == m_transitions_per_cycle) { g_system_ptr->getProfiler()->controllerBusy(m_machineID); // Count how often we\'re fully utilized g_eventQueue_ptr->scheduleEvent(this, 1); // Wakeup in another cycle and try again break; } ''') code.indent() code.indent() # InPorts # # Find the position of the mandatory queue in the vector so # that we can print it out first mandatory_q = None if self.ident == "L1Cache": for i,port in enumerate(self.in_ports): assert "c_code_in_port" in port if str(port).find("mandatoryQueue_in") >= 0: assert mandatory_q is None mandatory_q = port assert mandatory_q is not None # print out the mandatory queue here port = mandatory_q code('// ${ident}InPort $port') output = port["c_code_in_port"] pos = output.find("TransitionResult result = doTransition((L1Cache_mandatory_request_type_to_event(((*in_msg_ptr)).m_Type)), L1Cache_getState(addr), addr);") assert pos >= 0 atomics_string = ''' if ((((*in_msg_ptr)).m_Type) == CacheRequestType_ATOMIC) { if (servicing_atomic == 0) { if (locked_read_request1 == Address(-1)) { assert(read_counter == 0); locked_read_request1 = addr; assert(read_counter == 0); read_counter++; } else if (addr == locked_read_request1) { ; // do nothing } else { assert(0); // should never be here if servicing one request at a time } } else if (!started_receiving_writes) { if (servicing_atomic == 1) { if (locked_read_request2 == Address(-1)) { assert(locked_read_request1 != Address(-1)); assert(read_counter == 1); locked_read_request2 = addr; assert(read_counter == 1); read_counter++; } else if (addr == locked_read_request2) { ; // do nothing } else { assert(0); // should never be here if servicing one request at a time } } else if (servicing_atomic == 2) { if (locked_read_request3 == Address(-1)) { assert(locked_read_request1 != Address(-1)); assert(locked_read_request2 != Address(-1)); assert(read_counter == 1); locked_read_request3 = addr; assert(read_counter == 2); read_counter++; } else if (addr == locked_read_request3) { ; // do nothing } else { assert(0); // should never be here if servicing one request at a time } } else if (servicing_atomic == 3) { if (locked_read_request4 == Address(-1)) { assert(locked_read_request1 != Address(-1)); assert(locked_read_request2 != Address(-1)); assert(locked_read_request3 != Address(-1)); assert(read_counter == 1); locked_read_request4 = addr; assert(read_counter == 3); read_counter++; } else if (addr == locked_read_request4) { ; // do nothing } else { assert(0); // should never be here if servicing one request at a time } } else { assert(0); } } } else { if (servicing_atomic > 0) { // reset servicing_atomic = 0; read_counter = 0; started_receiving_writes = false; locked_read_request1 = Address(-1); locked_read_request2 = Address(-1); locked_read_request3 = Address(-1); locked_read_request4 = Address(-1); } } ''' output = output[:pos] + atomics_string + output[pos:] code('$output') for port in self.in_ports: # don't print out mandatory queue twice if port == mandatory_q: continue if ident == "L1Cache": if str(port).find("forwardRequestNetwork_in") >= 0: code(''' bool postpone = false; if ((((*m_L1Cache_forwardToCache_ptr)).isReady())) { const RequestMsg* in_msg_ptr; in_msg_ptr = dynamic_cast(((*m_L1Cache_forwardToCache_ptr)).peek()); if ((((servicing_atomic == 1) && (locked_read_request1 == ((*in_msg_ptr)).m_Address)) || ((servicing_atomic == 2) && (locked_read_request1 == ((*in_msg_ptr)).m_Address || locked_read_request2 == ((*in_msg_ptr)).m_Address)) || ((servicing_atomic == 3) && (locked_read_request1 == ((*in_msg_ptr)).m_Address || locked_read_request2 == ((*in_msg_ptr)).m_Address || locked_read_request3 == ((*in_msg_ptr)).m_Address)) || ((servicing_atomic == 4) && (locked_read_request1 == ((*in_msg_ptr)).m_Address || locked_read_request2 == ((*in_msg_ptr)).m_Address || locked_read_request3 == ((*in_msg_ptr)).m_Address || locked_read_request1 == ((*in_msg_ptr)).m_Address)))) { postpone = true; } } if (!postpone) { ''') code.indent() code('// ${ident}InPort $port') code('${{port["c_code_in_port"]}}') code.dedent() if ident == "L1Cache": if str(port).find("forwardRequestNetwork_in") >= 0: code.dedent() code('}') code.indent() code('') code.dedent() code.dedent() code(''' break; // If we got this far, we have nothing left todo } } ''') if self.ident == "L1Cache": code(''' void ${ident}_Controller::set_atomic(Address addr) { servicing_atomic++; } void ${ident}_Controller::started_writes() { started_receiving_writes = true; } void ${ident}_Controller::clear_atomic() { assert(servicing_atomic > 0); read_counter--; servicing_atomic--; if (read_counter == 0) { servicing_atomic = 0; started_receiving_writes = false; locked_read_request1 = Address(-1); locked_read_request2 = Address(-1); locked_read_request3 = Address(-1); locked_read_request4 = Address(-1); } } ''') else: code(''' void ${ident}_Controller::started_writes() { assert(0); } void ${ident}_Controller::set_atomic(Address addr) { assert(0); } void ${ident}_Controller::clear_atomic() { assert(0); } ''') code.write(path, "%s_Wakeup.cc" % self.ident) def printCSwitch(self, path): '''Output switch statement for transition table''' code = code_formatter() ident = self.ident code(''' // Auto generated C++ code started by $__file__:$__line__ // ${ident}: ${{self.short}} #include "mem/ruby/common/Global.hh" #include "mem/protocol/${ident}_Controller.hh" #include "mem/protocol/${ident}_State.hh" #include "mem/protocol/${ident}_Event.hh" #include "mem/protocol/Types.hh" #include "mem/ruby/system/System.hh" #define HASH_FUN(state, event) ((int(state)*${ident}_Event_NUM)+int(event)) #define GET_TRANSITION_COMMENT() (${ident}_transitionComment.str()) #define CLEAR_TRANSITION_COMMENT() (${ident}_transitionComment.str("")) TransitionResult ${ident}_Controller::doTransition(${ident}_Event event, ${ident}_State state, const Address& addr ) { ${ident}_State next_state = state; DEBUG_NEWLINE(GENERATED_COMP, MedPrio); DEBUG_MSG(GENERATED_COMP, MedPrio, *this); DEBUG_EXPR(GENERATED_COMP, MedPrio, g_eventQueue_ptr->getTime()); DEBUG_EXPR(GENERATED_COMP, MedPrio,state); DEBUG_EXPR(GENERATED_COMP, MedPrio,event); DEBUG_EXPR(GENERATED_COMP, MedPrio,addr); TransitionResult result = doTransitionWorker(event, state, next_state, addr); if (result == TransitionResult_Valid) { DEBUG_EXPR(GENERATED_COMP, MedPrio, next_state); DEBUG_NEWLINE(GENERATED_COMP, MedPrio); s_profiler.countTransition(state, event); if (Debug::getProtocolTrace()) { g_system_ptr->getProfiler()->profileTransition("${ident}", m_version, addr, ${ident}_State_to_string(state), ${ident}_Event_to_string(event), ${ident}_State_to_string(next_state), GET_TRANSITION_COMMENT()); } CLEAR_TRANSITION_COMMENT(); ${ident}_setState(addr, next_state); } else if (result == TransitionResult_ResourceStall) { if (Debug::getProtocolTrace()) { g_system_ptr->getProfiler()->profileTransition("${ident}", m_version, addr, ${ident}_State_to_string(state), ${ident}_Event_to_string(event), ${ident}_State_to_string(next_state), "Resource Stall"); } } else if (result == TransitionResult_ProtocolStall) { DEBUG_MSG(GENERATED_COMP, HighPrio, "stalling"); DEBUG_NEWLINE(GENERATED_COMP, MedPrio); if (Debug::getProtocolTrace()) { g_system_ptr->getProfiler()->profileTransition("${ident}", m_version, addr, ${ident}_State_to_string(state), ${ident}_Event_to_string(event), ${ident}_State_to_string(next_state), "Protocol Stall"); } } return result; } TransitionResult ${ident}_Controller::doTransitionWorker(${ident}_Event event, ${ident}_State state, ${ident}_State& next_state, const Address& addr ) { switch(HASH_FUN(state, event)) { ''') # This map will allow suppress generating duplicate code cases = orderdict() for trans in self.transitions: case_string = "%s_State_%s, %s_Event_%s" % \ (self.ident, trans.state.ident, self.ident, trans.event.ident) case = code_formatter() # Only set next_state if it changes if trans.state != trans.nextState: ns_ident = trans.nextState.ident case('next_state = ${ident}_State_${ns_ident};') actions = trans.actions # Check for resources case_sorter = [] res = trans.resources for key,val in res.iteritems(): if key.type.ident != "DNUCAStopTable": val = ''' if (!%s.areNSlotsAvailable(%s)) { return TransitionResult_ResourceStall; } ''' % (key.code, val) case_sorter.append(val) # Emit the code sequences in a sorted order. This makes the # output deterministic (without this the output order can vary # since Map's keys() on a vector of pointers is not deterministic for c in sorted(case_sorter): case("$c") # Figure out if we stall stall = False for action in actions: if action.ident == "z_stall": stall = True break if stall: case('return TransitionResult_ProtocolStall;') else: for action in actions: case('${{action.ident}}(addr);') case('return TransitionResult_Valid;') case = str(case) # Look to see if this transition code is unique. if case not in cases: cases[case] = [] cases[case].append(case_string) # Walk through all of the unique code blocks and spit out the # corresponding case statement elements for case,transitions in cases.iteritems(): # Iterative over all the multiple transitions that share # the same code for trans in transitions: code(' case HASH_FUN($trans):') code(' {') code(' $case') code(' }') code(''' default: WARN_EXPR(m_version); WARN_EXPR(g_eventQueue_ptr->getTime()); WARN_EXPR(addr); WARN_EXPR(event); WARN_EXPR(state); ERROR_MSG(\"Invalid transition\"); } return TransitionResult_Valid; } ''') code.write(path, "%s_Transitions.cc" % self.ident) def printProfilerHH(self, path): code = code_formatter() ident = self.ident code(''' // Auto generated C++ code started by $__file__:$__line__ // ${ident}: ${{self.short}} #ifndef ${ident}_PROFILER_H #define ${ident}_PROFILER_H #include "mem/ruby/common/Global.hh" #include "mem/protocol/${ident}_State.hh" #include "mem/protocol/${ident}_Event.hh" class ${ident}_Profiler { public: ${ident}_Profiler(); void setVersion(int version); void countTransition(${ident}_State state, ${ident}_Event event); void possibleTransition(${ident}_State state, ${ident}_Event event); void dumpStats(ostream& out) const; void clearStats(); private: int m_counters[${ident}_State_NUM][${ident}_Event_NUM]; int m_event_counters[${ident}_Event_NUM]; bool m_possible[${ident}_State_NUM][${ident}_Event_NUM]; int m_version; }; #endif // ${ident}_PROFILER_H ''') code.write(path, "%s_Profiler.hh" % self.ident) def printProfilerCC(self, path): code = code_formatter() ident = self.ident code(''' // Auto generated C++ code started by $__file__:$__line__ // ${ident}: ${{self.short}} #include "mem/protocol/${ident}_Profiler.hh" ${ident}_Profiler::${ident}_Profiler() { for (int state = 0; state < ${ident}_State_NUM; state++) { for (int event = 0; event < ${ident}_Event_NUM; event++) { m_possible[state][event] = false; m_counters[state][event] = 0; } } for (int event = 0; event < ${ident}_Event_NUM; event++) { m_event_counters[event] = 0; } } void ${ident}_Profiler::setVersion(int version) { m_version = version; } void ${ident}_Profiler::clearStats() { for (int state = 0; state < ${ident}_State_NUM; state++) { for (int event = 0; event < ${ident}_Event_NUM; event++) { m_counters[state][event] = 0; } } for (int event = 0; event < ${ident}_Event_NUM; event++) { m_event_counters[event] = 0; } } void ${ident}_Profiler::countTransition(${ident}_State state, ${ident}_Event event) { assert(m_possible[state][event]); m_counters[state][event]++; m_event_counters[event]++; } void ${ident}_Profiler::possibleTransition(${ident}_State state, ${ident}_Event event) { m_possible[state][event] = true; } void ${ident}_Profiler::dumpStats(ostream& out) const { out << " --- ${ident} " << m_version << " ---" << endl; out << " - Event Counts -" << endl; for (int event = 0; event < ${ident}_Event_NUM; event++) { int count = m_event_counters[event]; out << (${ident}_Event) event << " " << count << endl; } out << endl; out << " - Transitions -" << endl; for (int state = 0; state < ${ident}_State_NUM; state++) { for (int event = 0; event < ${ident}_Event_NUM; event++) { if (m_possible[state][event]) { int count = m_counters[state][event]; out << (${ident}_State) state << " " << (${ident}_Event) event << " " << count; if (count == 0) { out << " <-- "; } out << endl; } } out << endl; } } ''') code.write(path, "%s_Profiler.cc" % self.ident) # ************************** # ******* HTML Files ******* # ************************** def frameRef(self, click_href, click_target, over_href, over_target_num, text): code = code_formatter(fix_newlines=False) code("""${{html.formatShorthand(text)}}""") return str(code) def writeHTMLFiles(self, path): # Create table with no row hilighted self.printHTMLTransitions(path, None) # Generate transition tables for state in self.states.itervalues(): self.printHTMLTransitions(path, state) # Generate action descriptions for action in self.actions.itervalues(): name = "%s_action_%s.html" % (self.ident, action.ident) code = html.createSymbol(action, "Action") code.write(path, name) # Generate state descriptions for state in self.states.itervalues(): name = "%s_State_%s.html" % (self.ident, state.ident) code = html.createSymbol(state, "State") code.write(path, name) # Generate event descriptions for event in self.events.itervalues(): name = "%s_Event_%s.html" % (self.ident, event.ident) code = html.createSymbol(event, "Event") code.write(path, name) def printHTMLTransitions(self, path, active_state): code = code_formatter() code('''

${{html.formatShorthand(self.short)}}: ''') code.indent() for i,machine in enumerate(self.symtab.getAllType(StateMachine)): mid = machine.ident if i != 0: extra = " - " else: extra = "" if machine == self: code('$extra$mid') else: code('$extra$mid') code.dedent() code("""

""") for event in self.events.itervalues(): href = "%s_Event_%s.html" % (self.ident, event.ident) ref = self.frameRef(href, "Status", href, "1", event.short) code('') code('') # -- Body of table for state in self.states.itervalues(): # -- Each row if state == active_state: color = "yellow" else: color = "white" click = "%s_table_%s.html" % (self.ident, state.ident) over = "%s_State_%s.html" % (self.ident, state.ident) text = html.formatShorthand(state.short) ref = self.frameRef(click, "Table", over, "1", state.short) code(''' ''') # -- One column for each event for event in self.events.itervalues(): trans = self.table.get((state,event), None) if trans is None: # This is the no transition case if state == active_state: color = "#C0C000" else: color = "lightgrey" code('') continue next = trans.nextState stall_action = False # -- Get the actions for action in trans.actions: if action.ident == "z_stall" or \ action.ident == "zz_recycleMandatoryQueue": stall_action = True # -- Print out "actions/next-state" if stall_action: if state == active_state: color = "#C0C000" else: color = "lightgrey" elif active_state and next.ident == active_state.ident: color = "aqua" elif state == active_state: color = "yellow" else: color = "white" fix = code.nofix() code('\n") code.fix(fix) # -- Each row if state == active_state: color = "yellow" else: color = "white" click = "%s_table_%s.html" % (self.ident, state.ident) over = "%s_State_%s.html" % (self.ident, state.ident) ref = self.frameRef(click, "Table", over, "1", state.short) code(''' ''') code(''' ''') for event in self.events.itervalues(): href = "%s_Event_%s.html" % (self.ident, event.ident) ref = self.frameRef(href, "Status", href, "1", event.short) code('') code('''
$ref
$ref ') for action in trans.actions: href = "%s_action_%s.html" % (self.ident, action.ident) ref = self.frameRef(href, "Status", href, "1", action.short) code(' $ref\n') if next != state: if trans.actions: code('/') click = "%s_table_%s.html" % (self.ident, next.ident) over = "%s_State_%s.html" % (self.ident, next.ident) ref = self.frameRef(click, "Table", over, "1", next.short) code("$ref") code("$ref
$ref
''') if active_state: name = "%s_table_%s.html" % (self.ident, active_state.ident) else: name = "%s_table.html" % self.ident code.write(path, name) __all__ = [ "StateMachine" ]