fsm_load.hxx

// fsm_load.hxx: this file is part of the Vaucanson project.
//
// Vaucanson, a generic library for finite state machines.
//
// Copyright (C) 2001, 2002, 2004, 2005, 2006, 2008 The Vaucanson Group.
//
// This program 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.
//
// The complete GNU General Public Licence Notice can be found as the
// `COPYING' file in the root directory.
//
// The Vaucanson Group consists of people listed in the `AUTHORS' file.
//
#ifndef VCSN_TOOLS_FSM_LOAD_HXX
# define VCSN_TOOLS_FSM_LOAD_HXX

# include <vaucanson/tools/fsm_load.hh>
# include <map>
# include <set>
# include <string>
# include <vaucanson/automata/concept/handlers.hh>
# include <vaucanson/misc/usual_macros.hh>
# include <vaucanson/automata/concept/automata_base.hh>

namespace vcsn {

  namespace tools {

    /*---------.
    | fsm_load |
    `---------*/
    enum data_e
    {
      final,
      transition
    };

    struct line_data
    {
        data_e          type;
        unsigned int    from;
        unsigned int    to;
        std::string     letter;
        float           weight;
    };

    std::pair<std::string, std::string>
    next_token(std::string line)
    {
      std::string token;
      std::string::iterator i = line.begin();
      while ((i != line.end()) && ((*i == '\t')
                                   || (*i == ' ') || (*i == '\0')))
        ++i;
      for (;i != line.end();++i)
      {
        if ((*i == '\t') || (*i == ' ')
            || (*i == '\n') || (*i == '\0'))
          break;
        else
          token.push_back(*i);
      }
      if (i != line.end())
      {
        ++i;
        return std::make_pair(token,
                              std::string(line,
                                          (unsigned)(i - line.begin()),
                                          (unsigned)(line.end() - i + 1)));
      }
      else
        return std::make_pair(token, std::string());
    }

    template <typename St, typename AutoType_>
    void fsm_load(St& in, AutoType_& a)
    {
      AUTOMATON_TYPES(AutoType_);
      AUTOMATON_FREEMONOID_TYPES(AutoType_);
      // read everything and build the alphabet.
      alphabet_t                alpha;
      unsigned                  nb = 0;
      std::vector<line_data>    stock;
      std::string               line;
      std::pair<std::string,    std::string> tmp;
      std::vector<std::string>  tokens;

      while (!in.eof())
      {
        tokens.clear();
        ++nb;
        stock.resize(nb);
        getline(in, line);
        while (true)
        {
          tmp = next_token(line);
          line = tmp.second;
          if (tmp.first.length() != 0)
            tokens.push_back(tmp.first);
          if (line.length() == 0)
            break;
        }
        if (tokens.size() == 0)
        {
          nb--;
          stock.resize(nb);
          break;
        }
        if (tokens.size() == 1)
        {
          stock[nb-1].type = final;
          stock[nb-1].from = atoi(tokens[0].c_str());
          stock[nb-1].weight = 0.;
        }
        else if (tokens.size() == 2)
        {
          stock[nb-1].type = final;
          stock[nb-1].from = atoi(tokens[0].c_str());
          stock[nb-1].weight = atof(tokens[1].c_str());
        }
        else if (tokens.size() == 4)
        {
          stock[nb-1].type   = transition;
          stock[nb-1].from   = atoi(tokens[0].c_str());
          stock[nb-1].to     = atoi(tokens[1].c_str());
          stock[nb-1].letter = tokens[2][0];
          stock[nb-1].weight = atof(tokens[3].c_str());
          alpha.insert(stock[nb-1].letter);
        }
      }
      // construct the automaton.
      monoid_t    monoid(alpha);
      semiring_t   semiring;
      series_set_t        series(semiring, monoid);
      automata_set_t aset(series);
      automaton_t automaton(aset);
      std::map<int, hstate_t> to_h;

      for_all_const_(std::vector<line_data>, i, stock)
      {
        if (i->type == transition)
        {
          if (to_h.find(i->from) == to_h.end())
            to_h[i->from] = automaton.add_state();
          if (to_h.find(i->to) == to_h.end())
            to_h[i->to] = automaton.add_state();
          if (i == stock.begin())
            automaton.set_initial(to_h[i->from]);
          // FIXME: please be generic w.r.t spontaneous transition.
          if (i->letter == "1")
            automaton.add_spontaneous(to_h[i->from],
                                      to_h[i->to]);
          else
            automaton.add_letter_transition(to_h[i->from],
                                            to_h[i->to],
                                            vcsn::algebra::letter_traits<letter_t>::literal_to_letter(i->letter).second);
        }
        else if (i->type == final)
        {
          if (to_h.find(i->from) == to_h.end())
            to_h[i->from] = automaton.add_state();
          automaton.set_final(to_h[i->from]);
        }
      }
      a = automaton;
    }

  } // tools

} // vcsn

#endif // ! VCSN_TOOLS_FSM_LOAD_HXX