foma_FSM.h

Go to the documentation of this file.

//
//    FreeLing - Open Source Language Analyzers
//
//    Copyright (C) 2014   TALP Research Center
//                         Universitat Politecnica de Catalunya
//
//    This library is free software; you can redistribute it and/or
//    modify it under the terms of the GNU Affero General Public
//    License as published by the Free Software Foundation; either
//    version 3 of the License, or (at your option) any later version.
//
//    This library 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
//    Affero General Public License for more details.
//
//    You should have received a copy of the GNU Affero General Public
//    License along with this library; if not, write to the Free Software
//    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
//    contact: Lluis Padro (padro@lsi.upc.es)
//             TALP Research Center
//             despatx C6.212 - Campus Nord UPC
//             08034 Barcelona.  SPAIN
//

#ifndef _FOMA_FSM_H
#define _FOMA_FSM_H

#include <sstream>
#include <string>

#ifndef _Bool
typedef bool _Bool;
#endif


#include "foma/fomalib.h"


namespace freeling {


  class WINDLL foma_FSM {
  private:
    struct fsm *fsa;
    struct apply_med_handle *h_fsa;

    struct fsm* load_dictionary_file(const std::wstring &fname) const;
    struct fsm* load_dictionary_buffer(std::wistream &buff) const;
    void load_cost_matrix(const std::wstring &mcost);
    void create_compound_FSA(const std::list<std::wstring> &joins);

    // Auxiliary for constructors: initialize FSM for minimum edit distance searches
    void init_MED();
    void complete_alphabet(const std::wstring &);
    // Auxiliary for constructor: Update FSM alphabet by composing it with Sigma* 
    void update_FSM_alphabet(const std::set<std::wstring> &);

  public:
    foma_FSM(const std::wstring &, const std::wstring &mcost=L"", 
             const std::list<std::wstring> &joins=std::list<std::wstring>());
    foma_FSM(std::wistream &, const std::wstring &mcost=L"", 
             const std::list<std::wstring> &joins=std::list<std::wstring>());
    ~foma_FSM();

    void get_similar_words(const std::wstring &, std::list<std::pair<std::wstring,int> > &) const;    
    void set_cutoff_threshold(int);
    void set_num_matches(int);
    void set_basic_operation_cost(int);
    void set_operation_cost(const std::wstring &, const std::wstring &, int);
    std::set<std::wstring> get_alphabet();
  };

}

#endif