mention_detector.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
//

//
//   Author: Jordi Turmo (turmo@lsi.upc.edu)
//


#ifndef MENTION_DETECTOR_H
#define MENTION_DETECTOR_H

#include <map>
#include <set>
#include <algorithm>

#include "freeling/windll.h"
#include "freeling/morfo/language.h"

namespace freeling {


  class WINDLL mention_detector {

  private:
    typedef enum {NONE, REL_CLAUSE, ESSENTIAL, PRE_NON_ESSENTIAL, NON_ESSENTIAL} subordinateType;

    std::map<std::wstring, freeling::regexp> _Labels;
    std::set<std::wstring> _No_heads;
    bool _Reduce_mentions;
    std::wstring _Language;

    void candidates(int, paragraph::const_iterator, int&, sentence::const_iterator &, int&, parse_tree::const_iterator, std::vector<mention>&, subordinateType&) const;

    void discard_mentions(std::vector<mention>&) const;

    void add_mention(const mention&, std::vector<mention>&, int&) const;
    void mark_initial_mentions_from(int, std::vector<mention>&) const;

  public:
    mention_detector(const std::wstring &, const std::wstring &lang=L"");
    ~mention_detector();
  
    std::map<std::wstring, freeling::regexp>& get_config_labels();

    void detect(const document &, std::vector<mention> &);
    
  };

} // namespace

#endif