mosesdecoder/misc/queryLexicalTable.cpp

#include <iostream>
#include <string>

#include "Phrase.h"
#include "FactorCollection.h"
#include "Timer.h"
#include "InputFileStream.h"
#include "LexicalReorderingTable.h"

using namespace Moses;

Timer timer;

void printHelp(){
  std::cerr << "Usage:\n"
	"options: \n"
	"\t-table file   -- input table file name\n"
	"\t-f     string -- f query phrase\n"
	"\t-e     string -- e query phrase\n"
	"\t-c     string -- context query phrase\n"
	"\n"; 
}

std::ostream& operator<<(std::ostream& o, Scores s){
  for(int i = 0; i < s.size(); ++i){
	o << s[i] << "  ";
  }
  //o << std::endln;
  return o;
};

int main(int argc, char** argv){
  std::cerr << "queryLexicalTable v0.2 by Konrad Rawlik\n";
  std::string inFilePath;
  std::string outFilePath("out");
  bool cache = false;
  std::string query_e, query_f, query_c;
  bool use_context = false;
  bool use_e       = false;
  if(1 >= argc){
	printHelp();
	return 1;
  }
  for(int i = 1; i < argc; ++i){
    std::string arg(argv[i]);
    if("-table" == arg && i+1 < argc){
	  //std::cerr << "Table is " << argv[i];
      ++i;
      inFilePath = argv[i];
    } else if("-f" == arg && i+1 < argc){
	  ++i;
	  //std::cerr << "F is " << argv[i];
	  query_f = argv[i];
    } else if("-e" == arg && i+1 < argc){
	  ++i;
	  query_e = argv[i];
	  use_e = true;
    } else if("-c" == arg){
	  if(i+1 < argc && '-' != argv[i+1][0]){
		++i;
		query_c = argv[i];
		use_context = true;
	  } else {
		use_context = false;
	  }
	} else if("-cache" == arg){
	  ++i;
	  cache = true; 
    } else {
      //somethings wrong... print help
	  printHelp();
      return 1;
    }
  }

  FactorList f_mask;
  FactorList e_mask;
  FactorList c_mask;
  f_mask.push_back(0);
  if(use_e){
	e_mask.push_back(0);
  }
  if(use_context){
	c_mask.push_back(0);
  }
  Phrase e(Output),f(Input),c(Output);
  e.CreateFromString(e_mask, query_e, "|");
  f.CreateFromString(f_mask, query_f, "|");
  c.CreateFromString(c_mask,  query_c,"|");
  LexicalReorderingTable* table;
  if(FileExists(inFilePath+".binlexr.idx")){
	std::cerr << "Loading binary table...\n";
	table = new LexicalReorderingTableTree(inFilePath, f_mask, e_mask, c_mask);
  } else {
	std::cerr << "Loading ordinary table...\n";
	table = new LexicalReorderingTableMemory(inFilePath, f_mask, e_mask, c_mask);
  }
  //table->DbgDump(&std::cerr);
  if(cache){
	std::cerr << "Caching for f\n";
	table->InitializeForInputPhrase(f);
  }
  std::cerr << "Querying: f='" << f.GetStringRep(f_mask) << "' e='" << e.GetStringRep(e_mask) << "' c='" << c.GetStringRep(c_mask) << "'\n";
  std::cerr << table->GetScore(f,e,c) << "\n";
  //table->DbgDump(&std::cerr);
  delete table;
  return 0;
}