// Copyright (C) 2011, 2012 Laboratoire de Recherche et Developpement
// de l'Epita (LRDE)
//
// 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.


#include "kripkeprint.hh"
#include "kripkeexplicit.hh"
#include "tgba/bddprint.hh"
#include "misc/escape.hh"
#include "tgbaalgos/reachiter.hh"
#include <iostream>
#include <sstream>

namespace spot
{
  namespace
  {

    class kripke_printer : public tgba_reachable_iterator_breadth_first
    {
    public:
      kripke_printer(const kripke* a, std::ostream& os)
	: tgba_reachable_iterator_breadth_first(a), os_(os)
      {
      }

      void process_state(const state* s, int, tgba_succ_iterator* si)
      {
	const bdd_dict* d = aut_->get_dict();
	os_ << "\"";
	escape_str(os_, aut_->format_state(s));
	os_ << "\", \"";
	const kripke* automata = down_cast<const kripke*> (aut_);
	assert(automata);
	escape_str(os_, bdd_format_formula(d,
					   automata->state_condition(s)));

	os_ << "\",";
	for (si->first(); !si->done(); si->next())
	  {
	    state* dest = si->current_state();
	    os_ << " \"";
	    escape_str(os_, aut_->format_state(dest));
	    os_ << "\"";
	}
	os_ << ";\n";
      }

    private:
      std::ostream& os_;
    };

    class kripke_printer_renumbered :
      public tgba_reachable_iterator_breadth_first
    {
    public:
      kripke_printer_renumbered(const kripke* a, std::ostream& os)
	: tgba_reachable_iterator_breadth_first(a), os_(os),
	  notfirst(false)
      {
      }

      void finish_state()
      {
	os_ << lastsuccs.str() << ";\n";
	lastsuccs.str("");
      }

      void process_state(const state* s, int in_s, tgba_succ_iterator*)
      {
	if (notfirst)
	  finish_state();
	else
	  notfirst = true;

	const bdd_dict* d = aut_->get_dict();
	std::string cur = aut_->format_state(s);
	os_ << "S" << in_s << ", \"";
	const kripke* automata = down_cast<const kripke*>(aut_);
	assert(automata);
	escape_str(os_, bdd_format_formula(d,
					   automata->state_condition(s)));
	os_ << "\",";
      }

      void
      process_link(const state*, int, const state*, int d,
		   const tgba_succ_iterator*)
      {
	lastsuccs << " S" << d;
      }

      void
      end()
      {
	finish_state();
      }

    private:
      std::ostream& os_;
      std::ostringstream lastsuccs;
      bool notfirst;
    };

  }

  std::ostream&
  kripke_save_reachable(std::ostream& os, const kripke* k)
  {
    kripke_printer p(k, os);
    p.run();
    return os;
  }

  std::ostream&
  kripke_save_reachable_renumbered(std::ostream& os, const kripke* k)
  {
    kripke_printer_renumbered p(k, os);
    p.run();
    return os;
  }


} // End namespace Spot