coref_fex.h

Go to the documentation of this file.

//
//    FreeLing - Open Source Language Analyzers
//
//    Copyright (C) 2004   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 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
//    General Public License for more details.
//
//    You should have received a copy of the GNU 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
//

//    Feature extractor for coreference resolution

#ifndef CORE_FEX_H
#define CORE_FEX_H

#include <list>
#include <fstream>
#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <cctype>

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

// -- Feature group codes
#define COREFEX_DIST            0x00000001
#define COREFEX_IPRON           0x00000002
#define COREFEX_JPRON           0x00000004
#define COREFEX_IPRONM          0x00000008
#define COREFEX_JPRONM          0x00000010
#define COREFEX_STRMATCH        0x00000020
#define COREFEX_DEFNP           0x00000040
#define COREFEX_DEMNP           0x00000080
#define COREFEX_NUMBER          0x00000100
#define COREFEX_GENDER          0x00000200
#define COREFEX_SEMCLASS        0x00000400
#define COREFEX_PROPNAME        0x00000800
#define COREFEX_ALIAS           0x00001000
#define COREFEX_APPOS           0x00002000
#define COREFEX_QUOTES          0x00004000
#define COREFEX_THIRDP          0x00010000
//
#define COREFEX_ALL             0xFFFFFFFF


class mention {
 public:
  int sent;
  int numde;
  int posbegin;
  int posend;
  parse_tree::iterator ptree;
  std::vector<std::wstring> tokens;
  std::vector<std::wstring> tags;
};



class coref_fex {
 private:
  semanticDB *semdb;
  unsigned int active_features;

  wchar_t extract_number(const std::wstring &);
  wchar_t extract_gender(const std::wstring &);
  std::wstring extract_semclass(const std::wstring &, const std::wstring &);
  word& get_head_word(parse_tree::iterator);
  bool check_tag(const mention &, int, const std::wstring &);
  bool check_word(const std::wstring &, const std::wstring &);
  bool check_acronim(const mention &, const mention &);
  bool check_prefix(const mention &, const mention &);
  bool check_sufix(const mention &, const mention &);
  bool check_order(const mention &, const mention &);

  int get_dist(const mention &, const mention &); 
  int get_numdedist(const mention &, const mention &);
  int get_dedist(const mention &, const mention &);
  int get_i_pronoun(const mention &, const mention &);
  int get_j_pronoun(const mention &, const mention &);
  int get_i_pronoun_p(const mention &, const mention &);
  int get_j_pronoun_p(const mention &, const mention &);
  int get_i_pronoun_d(const mention &, const mention &);
  int get_j_pronoun_d(const mention &, const mention &);
  int get_i_pronoun_x(const mention &, const mention &);
  int get_j_pronoun_x(const mention &, const mention &);
  int get_i_pronoun_i(const mention &, const mention &);
  int get_j_pronoun_i(const mention &, const mention &);
  int get_i_pronoun_t(const mention &, const mention &);
  int get_j_pronoun_t(const mention &, const mention &);
  int get_i_pronoun_r(const mention &, const mention &);
  int get_j_pronoun_r(const mention &, const mention &);
  int get_i_pronoun_e(const mention &, const mention &);
  int get_j_pronoun_e(const mention &, const mention &);
  int get_str_match(const mention &, const mention &);
  int get_def_np(const mention &, const mention &);
  int get_dem_np(const mention &, const mention &);
  int get_number(const mention &, const mention &);
  int get_gender(const mention &, const mention &);
  int get_semclass(const mention &, const mention &);
  int get_proper_noun_i(const mention &, const mention &);
  int get_proper_noun_j(const mention &, const mention &);
  int get_alias_acro(const mention &, const mention &);
  int get_alias_prefix(const mention &, const mention &);
  int get_alias_sufix(const mention &, const mention &);
  int get_alias_order(const mention &, const mention &);
  int get_appositive(const mention &, const mention &);
  int get_i_inquotes(const mention &, const mention &);
  int get_j_inquotes(const mention &, const mention &);
  int get_i_inparenthesis(const mention &, const mention &);
  int get_j_inparenthesis(const mention &, const mention &);
  int get_i_thirdperson(const mention &, const mention &);
  int get_j_thirdperson(const mention &, const mention &);
  
  inline void put_feature(int, std::vector<int> &);

 public:
  coref_fex(const unsigned int, const std::wstring&);
  ~coref_fex();

  void extract(const mention &, const mention &, std::vector<int> &);
};
#endif