#include "ctx_aut.hh"

CtxAut::CtxAut(Options *opts, RctSys *parent_rctsys)
{
  setOptions(opts);
  this->parent_rctsys = parent_rctsys;
}

bool CtxAut::hasState(std::string name)
{
  if (states_names.find(name) == states_names.end()) {
    return false;
  }
  else {
    return true;
  }
}

State CtxAut::getStateID(std::string name)
{
  if (!hasState(name)) {
    FERROR("No such state: " << name);
  }

  return states_names[name];
}

std::string CtxAut::getStateName(State state_id)
{
  assert(state_id < states_ids.size());
  return states_ids[state_id];
}

void CtxAut::addState(std::string name)
{
  if (!hasState(name)) {
    State new_state_id = states_ids.size();

    VERB_L2("Adding state: " << name << " index=" << new_state_id);

    states_ids.push_back(name);
    states_names[name] = new_state_id;

    if (!init_state_defined) {
      setInitState(name);
    }
  }
}

void CtxAut::setInitState(std::string name)
{
  VERB_L2("Setting initial CA state: " << name);
  State state_id = getStateID(name);
  VERB_L2("Got initial CA state ID: index=" << state_id);
  init_state_id = state_id;
  init_state_defined = true;
}

State CtxAut::getInitState(void)
{
  assert(init_state_defined);
  return init_state_id;
}

void CtxAut::printAutomaton(void)
{
  showStates();
  showTransitions();
}

void CtxAut::showStates(void)
{
  cout << "# Context Automaton States:" << endl;
  cout << " = Init state: " << getStateName(init_state_id) << endl;

  for (const auto &s : states_ids) {
    cout << " * " << s << endl;
  }
}

void CtxAut::pushContextEntity(Entity entity_id)
{
  tmpEntities.insert(entity_id);
}

void CtxAut::addTransition(std::string srcStateName, std::string dstStateName)
{
  VERB_L3("Saving transition");

  CtxAutTransition new_transition;

  new_transition.src_state = getStateID(srcStateName);
  new_transition.ctx = tmpEntities;
  tmpEntities.clear();
  new_transition.dst_state = getStateID(dstStateName);
  transitions.push_back(new_transition);
}

void CtxAut::showTransitions(void)
{
  cout << "# Context Automaton Transitions:" << endl;

  for (const auto &t : transitions) {
    cout << " * [" << getStateName(t.src_state) << " -> " << getStateName(
           t.dst_state)
         << "]: {" << parent_rctsys->entitiesToStr(t.ctx) << "}" << endl;
  }
}

/** EOF **/