ner_module.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 _NER_MODULE
#define _NER_MODULE

#include <set>
#include "freeling/windll.h"
#include "freeling/morfo/language.h"
#include "freeling/morfo/automat.h"

namespace freeling {


  class ner_status : public automat_status {
  public:
    bool initialNoun;
  };


  class WINDLL ner_module : public automat<ner_status> {
  
  protected: 
    unsigned int Title_length;
    // the same for title-formatted multiwords, e.g. "Strangers In Paradise")
    unsigned int AllCaps_Title_length;

    std::wstring NE_tag;
    
    bool splitNPs;

  public:
    // constructor
    ner_module(const std::wstring &);
    // destructor
    virtual ~ner_module() {};

    virtual int ComputeToken(int, sentence::iterator &, sentence &) const;
    virtual void ResetActions(ner_status *) const;
    virtual void StateActions(int, int, int, sentence::const_iterator,ner_status *) const;

    virtual void SetMultiwordAnalysis(sentence::iterator, int, const ner_status *) const;

    bool ValidMultiWord(const word &,ner_status *) const;
    sentence::iterator BuildMultiword(sentence &, sentence::iterator, sentence::iterator, int, bool&, ner_status *) const;

  };

} // namespace

#endif