tgbaexplicit.hh

// -*- encoding: utf-8 -*-
// Copyright (C) 2009, 2010, 2011, 2012 Laboratoire de Recherche et
// Développement de l'Epita.
// Copyright (C) 2003, 2004, 2006 Laboratoire d'Informatique de Paris
// 6 (LIP6), département Systèmes Répartis Coopératifs (SRC),
// Université Pierre et Marie Curie.
//
// This file is part of Spot, a model checking library.
//
// Spot is free software; you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// Spot is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
// or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public
// License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Spot; see the file COPYING.  If not, write to the Free
// Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
// 02111-1307, USA.

#ifndef SPOT_TGBA_TGBAEXPLICIT_HH
# define SPOT_TGBA_TGBAEXPLICIT_HH

#include <sstream>
#include <list>

#include "tgba.hh"
#include "tgba/formula2bdd.hh"
#include "misc/hash.hh"
#include "misc/bddop.hh"
#include "ltlast/formula.hh"
#include "ltlvisit/tostring.hh"

namespace spot
{
  ///misc
  template<typename T>
  struct destroy_key
  {
    void destroy(T t)
    {
      (void) t;
    }
  };

  template<>
  struct destroy_key<const ltl::formula*>
  {
    void destroy(const ltl::formula* t)
    {
      t->destroy();
    }
  };

  class sba: public tgba
  {
  public:
    /// A State in an SBA (State-based Buchi Automaton) is accepting
    /// iff all outgoing transitions are accepting.
    virtual bool is_accepting(const spot::state* s) const = 0;
  };

  /// States used by spot::explicit_graph implementation
  /// \ingroup tgba_representation
  template<typename Label, typename label_hash>
  class state_explicit: public spot::state
  {
  public:
    state_explicit()
    {
    }

    state_explicit(const Label& l):
      label_(l)
    {
    }

    virtual ~state_explicit()
    {
    }

    void destroy() const
    {
    }

    typedef Label Label_t;
    typedef label_hash label_hash_t;

    struct transition
    {
      bdd condition;
      bdd acceptance_conditions;
      const state_explicit<Label, label_hash>* dest;
    };

    typedef std::list<transition> transitions_t;
    transitions_t successors;

    const Label& label() const
    {
      return label_;
    }

    bool empty() const
    {
      return successors.empty();
    }

    virtual int compare(const state* other) const
    {
      const state_explicit<Label, label_hash>* s =
	down_cast<const state_explicit<Label, label_hash>*>(other);
      assert (s);

      // Do not simply return "o - this", it might not fit in an int.
      if (s < this)
	return -1;
      if (s > this)
	return 1;
      return 0;
    }

    virtual size_t hash() const
    {
      return
	reinterpret_cast<const char*>(this) - static_cast<const char*>(0);
    }

    virtual state_explicit<Label, label_hash>*
    clone() const
    {
      return const_cast<state_explicit<Label, label_hash>*>(this);
    }

  protected:
    Label label_;
  };

  ///states labeled by an int
  /// \ingroup tgba_representation
  class state_explicit_number:
    public state_explicit<int, identity_hash<int> >
  {
  public:
    state_explicit_number()
      : state_explicit<int, identity_hash<int> >()
    {
    }

    state_explicit_number(int label)
      : state_explicit<int, identity_hash<int> >(label)
    {
    }

    virtual void destroy()
    {
    }

    static const int default_val;
    static std::string to_string(int l)
    {
      std::stringstream ss;
      ss << l;
      return ss.str();
    }
  };

  /// States labeled by a string
  /// \ingroup tgba_representation
  class state_explicit_string:
    public state_explicit<std::string, string_hash>
  {
  public:
    state_explicit_string():
      state_explicit<std::string, string_hash>()
    {
    }

    state_explicit_string(const std::string& label)
      : state_explicit<std::string, string_hash>(label)
    {
    }

    virtual void destroy()
    {
    }

    static const std::string default_val;
    static std::string to_string(const std::string& l)
    {
      return l;
    }
  };

  /// States labeled by a formula
  /// \ingroup tgba_representation
  class state_explicit_formula:
    public state_explicit<const ltl::formula*, ltl::formula_ptr_hash>
  {
  public:
    state_explicit_formula():
      state_explicit<const ltl::formula*, ltl::formula_ptr_hash>()
    {
    }

    state_explicit_formula(const ltl::formula* label)
      : state_explicit<const ltl::formula*, ltl::formula_ptr_hash>(label)
    {
    }

    virtual void destroy()
    {
    }

    static const ltl::formula* default_val;
    static std::string to_string(const ltl::formula* l)
    {
      return ltl::to_string(l);
    }
  };

  /// Successor iterators used by spot::tgba_explicit.
  /// \ingroup tgba_representation
  template<typename State>
  class tgba_explicit_succ_iterator: public tgba_succ_iterator
  {
  public:
    tgba_explicit_succ_iterator(const State* start,
				bdd all_acc)
      : start_(start),
	all_acceptance_conditions_(all_acc)
    {
    }

    virtual void first()
    {
      it_ = start_->successors.begin();
    }

    virtual void next()
    {
      ++it_;
    }

    virtual bool done() const
    {
      return it_ == start_->successors.end();
    }

    virtual State* current_state() const
    {
      assert(!done());

      //ugly but I can't see any other wayout
      const State* res = down_cast<const State*>(it_->dest);
      assert(res);

      return
	const_cast<State*>(res);
    }

    virtual bdd current_condition() const
    {
      assert(!done());
      return it_->condition;
    }

    virtual bdd current_acceptance_conditions() const
    {
      assert(!done());
      return it_->acceptance_conditions & all_acceptance_conditions_;
    }

    typename State::transitions_t::const_iterator
    get_iterator() const
    {
      return it_;
    }

  private:
    const State* start_;
    typename State::transitions_t::const_iterator it_;
    bdd all_acceptance_conditions_;
  };

  /// Graph implementation for automata representation
  /// \ingroup tgba_representation
  template<typename State, typename Type>
  class explicit_graph: public Type
  {
  protected:
    typedef Sgi::hash_map<typename State::Label_t, State,
			  typename State::label_hash_t> ls_map;
    typedef Sgi::hash_map<const State*, typename State::Label_t,
			  ptr_hash<State> > sl_map;

  public:
    typedef typename State::transition transition;

    explicit_graph(bdd_dict* dict)
      : ls_(),
	sl_(),
	init_(0),
	dict_(dict),
	all_acceptance_conditions_(bddfalse),
	all_acceptance_conditions_computed_(false),
	neg_acceptance_conditions_(bddtrue)
    {
    }

    State* add_default_init()
    {
      return add_state(State::default_val);
    }

    transition*
    create_transition(State* source, const State* dest)
    {
      transition t;

      t.dest = dest;
      t.condition = bddtrue;
      t.acceptance_conditions = bddfalse;

      typename State::transitions_t::iterator i =
	source->successors.insert(source->successors.end(), t);

      return &*i;
    }

    transition*
    create_transition(const typename State::Label_t& source,
		      const typename State::Label_t& dest)
    {
      // It's important that the source be created before the
      // destination, so the first source encountered becomes the
      // default initial state.
      State* s = add_state(source);
      return create_transition(s, add_state(dest));
    }

    transition*
    get_transition(const tgba_explicit_succ_iterator<State>* si)
    {
      return const_cast<transition*>(&(*(si->get_iterator())));
    }

    void add_condition(transition* t, const ltl::formula* f)
    {
      t->condition &= formula_to_bdd(f, dict_, this);
      f->destroy();
    }

    /// This assumes that all variables in \a f are known from dict.
    void add_conditions(transition* t, bdd f)
    {
      dict_->register_propositions(f, this);
      t->condition &= f;
    }

    bool has_acceptance_condition(const ltl::formula* f) const
    {
      return dict_->is_registered_acceptance_variable(f, this);
    }

    //old tgba explicit labelled interface
    bool has_state(const typename State::Label_t& name)
    {
      return ls_.find(name) != ls_.end();
    }

    const typename State::Label_t& get_label(const State* s) const
    {
      typename sl_map::const_iterator i = sl_.find(s);
      assert(i != sl_.end());
      return i->second;
    }

    const typename State::Label_t& get_label(const spot::state* s) const
    {
      const State* se = down_cast<const State*>(s);
      assert(se);
      return get_label(se);
    }

    transition*
    create_trainsition(const typename State::Label_t& source,
		       const typename State::Label_t& dest)
    {
      // It's important that the source be created before the
      // destination, so the first source encountered becomes the
      // default initial state.
      State* s = add_state(source);
      return create_transition(s, add_state(dest));
    }

    void
    complement_all_acceptance_conditions()
    {
      bdd all = this->all_acceptance_conditions();
      typename ls_map::iterator i;
      for (i = ls_.begin(); i != ls_.end(); ++i)
	{
	  typename State::transitions_t::iterator i2;
	  for (i2 = i->second.successors.begin();
	       i2 != i->second.successors.end(); ++i2)
	    i2->acceptance_conditions = all - i2->acceptance_conditions;
	}
    }

    void
    merge_transitions()
    {
      typename ls_map::iterator i;
      for (i = ls_.begin(); i != ls_.end(); ++i)
      {
	typename State::transitions_t::iterator t1;
	for (t1 = i->second.successors.begin();
	     t1 != i->second.successors.end(); ++t1)
	{
	  bdd acc = t1->acceptance_conditions;
	  const state_explicit<typename State::Label_t,
			       typename State::label_hash_t>* dest = t1->dest;

	  // Find another transition with the same destination and
	  // acceptance conditions.
	  typename State::transitions_t::iterator t2 = t1;
	  ++t2;
	  while (t2 != i->second.successors.end())
	  {
	    typename State::transitions_t::iterator t2copy = t2++;
	    if (t2copy->acceptance_conditions == acc && t2copy->dest == dest)
	    {
	      t1->condition |= t2copy->condition;
	      i->second.successors.erase(t2copy);
	    }
	  }
	}
      }
    }

    /// Return the state_explicit for \a name, creating the state if
    /// it does not exist.
    State* add_state(const typename State::Label_t& name)
    {
      typename ls_map::iterator i = ls_.find(name);
      if (i == ls_.end())
	{
	  State s(name);
	  ls_[name] = s;
	  sl_[&ls_[name]] = name;

	  // The first state we add is the inititial state.
	  // It can also be overridden with set_init_state().
	  if (!init_)
	    init_ = &ls_[name];

	  return &(ls_[name]);
	}
      return &(i->second);
    }

    State*
    set_init_state(const typename State::Label_t& state)
    {
      State* s = add_state(state);
      init_ = s;
      return s;
    }

    // tgba interface
    virtual ~explicit_graph()
    {
      typename ls_map::iterator i = ls_.begin();

      while (i != ls_.end())
      {
	typename State::Label_t s = i->first;

	// Do not erase the same state twice(Because of possible aliases).
	if (sl_.erase(&(i->second)))
	  i->second.destroy();

	++i;
	destroy_key<typename State::Label_t> dest;
	dest.destroy(s);
      }

      this->dict_->unregister_all_my_variables(this);
      // These have already been destroyed by subclasses.
      // Prevent destroying by tgba::~tgba.
      this->last_support_conditions_input_ = 0;
      this->last_support_variables_input_ = 0;
    }

    virtual State* get_init_state() const
    {
      if (!init_)
	const_cast<explicit_graph<State, Type>*>(this)->add_default_init();

      return init_;
    }

    virtual tgba_explicit_succ_iterator<State>*
    succ_iter(const spot::state* state,
	      const spot::state* global_state = 0,
	      const tgba* global_automaton = 0) const
    {
      const State* s = down_cast<const State*>(state);
      assert(s);

      (void) global_state;
      (void) global_automaton;

      return
	new tgba_explicit_succ_iterator<State>(s,
					       this
					       ->all_acceptance_conditions());
    }

    //no need?
    virtual std::string format_state(const spot::state* state) const
    {
      const State* se = down_cast<const State*>(state);
      assert(se);
      typename sl_map::const_iterator i = sl_.find(se);
      assert(i != sl_.end());
      return State::to_string(i->second);
    }

    /// old implementation in tgba_explicit_string
    /// Create an alias for a state.  Any reference to \a alias_name
    /// will act as a reference to \a real_name.
    void add_state_alias(const typename State::Label_t& alias,
			 const typename State::Label_t& real)
    {
      ls_[alias] = *(add_state(real));
    }


    /// \brief Copy the acceptance conditions of a tgba.
    ///
    /// If used, this function should be called before creating any
    /// transition.
    void copy_acceptance_conditions_of(const tgba *a)
    {
      assert(neg_acceptance_conditions_ == bddtrue);
      assert(all_acceptance_conditions_computed_ == false);
      bdd f = a->neg_acceptance_conditions();
      this->dict_->register_acceptance_variables(f, this);
      neg_acceptance_conditions_ = f;
    }


    /// Acceptance conditions handlingx
    void set_acceptance_conditions(bdd acc)
    {
      assert(neg_acceptance_conditions_ == bddtrue);
      assert(all_acceptance_conditions_computed_ == false);

      this->dict_->register_acceptance_variables(bdd_support(acc), this);
      neg_acceptance_conditions_ = compute_neg_acceptance_conditions(acc);
      all_acceptance_conditions_computed_ = true;
      all_acceptance_conditions_ = acc;
    }

    void add_acceptance_condition(transition* t, const ltl::formula* f)
    {
      bdd c = get_acceptance_condition(f);
      t->acceptance_conditions |= c;
    }

    /// This assumes that all acceptance conditions in \a f are known from
    /// dict.
    void add_acceptance_conditions(transition* t, bdd f)
    {
      bdd sup = bdd_support(f);
      this->dict_->register_acceptance_variables(sup, this);
      while (sup != bddtrue)
	{
	  neg_acceptance_conditions_ &= bdd_nithvar(bdd_var(sup));
	  sup = bdd_high(sup);
	}
      t->acceptance_conditions |= f;
    }

    virtual bdd all_acceptance_conditions() const
    {
      if (!all_acceptance_conditions_computed_)
      {
	all_acceptance_conditions_ =
	  compute_all_acceptance_conditions(neg_acceptance_conditions_);
	all_acceptance_conditions_computed_ = true;
      }
      return all_acceptance_conditions_;
    }

    virtual bdd_dict* get_dict() const
    {
      return this->dict_;
    }

    virtual bdd neg_acceptance_conditions() const
    {
      return neg_acceptance_conditions_;
    }

    void
    declare_acceptance_condition(const ltl::formula* f)
    {
      int v = this->dict_->register_acceptance_variable(f, this);
      f->destroy();
      bdd neg = bdd_nithvar(v);
      neg_acceptance_conditions_ &= neg;

      // Append neg to all acceptance conditions.
      typename explicit_graph<State, tgba>::ls_map::iterator i;
      for (i = this->ls_.begin(); i != this->ls_.end(); ++i)
	{
	  typename State::transitions_t::iterator i2;
	  for (i2 = i->second.successors.begin();
	       i2 != i->second.successors.end(); ++i2)
	    i2->acceptance_conditions &= neg;
	}

      all_acceptance_conditions_computed_ = false;
    }

  protected:

    bdd get_acceptance_condition(const ltl::formula* f)
    {
      bdd_dict::fv_map::iterator i = this->dict_->acc_map.find(f);
      assert(this->has_acceptance_condition(f));
      f->destroy();
      bdd v = bdd_ithvar(i->second);
      v &= bdd_exist(neg_acceptance_conditions_, v);
      return v;
    }

    virtual bdd compute_support_conditions(const spot::state* in) const
    {
      const State* s = down_cast<const State*>(in);
      assert(s);
      const typename State::transitions_t& st = s->successors;

      bdd res = bddfalse;

      typename State::transitions_t::const_iterator i;
      for (i = st.begin(); i != st.end(); ++i)
	res |= i->condition;

      return res;
    }

    virtual bdd compute_support_variables(const spot::state* in) const
    {
      const State* s = down_cast<const State*>(in);
      assert(s);
      const typename State::transitions_t& st = s->successors;

      bdd res = bddtrue;

      typename State::transitions_t::const_iterator i;
      for (i = st.begin(); i != st.end(); ++i)
	res &= bdd_support(i->condition);

      return res;
    }

    ls_map ls_;
    sl_map sl_;
    State* init_;

    bdd_dict* dict_;

    mutable bdd all_acceptance_conditions_;
    mutable bool all_acceptance_conditions_computed_;
    bdd neg_acceptance_conditions_;
  };

  template <typename State>
  class tgba_explicit: public explicit_graph<State, tgba>
  {
  public:
    typedef typename State::transition transition;
    typedef State state;

    tgba_explicit(bdd_dict* dict):
      explicit_graph<State,tgba>(dict)
    {
    }

    virtual ~tgba_explicit()
    {
    }

  private:
    // Disallow copy.
    tgba_explicit(const tgba_explicit<State>& other);
    tgba_explicit& operator=(const tgba_explicit& other);
  };

  template <typename State>
  class sba_explicit: public explicit_graph<State, sba>
  {
  public:
    typedef typename State::transition transition;
    typedef State state;

    sba_explicit(bdd_dict* dict):
      explicit_graph<State, sba>(dict)
    {
    }

    virtual ~sba_explicit()
    {
    }

    virtual bool is_accepting(const spot::state* s) const
    {
      bdd acc = bddtrue;
      bool transition = false;

      tgba_explicit_succ_iterator<State>* it = this->succ_iter(s);
      for (it->first(); !it->done() && acc != bddfalse; it->next())
      {
	transition = true;
	acc &= it->current_acceptance_conditions();
      }

      delete it;

      return acc != bddfalse && transition;
    }

  private:
    // Disallow copy.
    sba_explicit(const sba_explicit<State>& other);
    sba_explicit& operator=(const sba_explicit& other);
  };


  /// Typedefs for tgba
  typedef tgba_explicit<state_explicit_string> tgba_explicit_string;
  typedef tgba_explicit<state_explicit_formula> tgba_explicit_formula;
  typedef tgba_explicit<state_explicit_number> tgba_explicit_number;

  /// Typedefs for sba
  typedef sba_explicit<state_explicit_string> sba_explicit_string;
  typedef sba_explicit<state_explicit_formula> sba_explicit_formula;
  typedef sba_explicit<state_explicit_number> sba_explicit_number;
}

#endif // SPOT_TGBA_TGBAEXPLICIT_HH