mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-12-27 22:14:57 +03:00
354 lines
12 KiB
C++
354 lines
12 KiB
C++
// vim:tabstop=2
|
|
|
|
/***********************************************************************
|
|
Moses - factored phrase-based language decoder
|
|
Copyright (C) 2006 University of Edinburgh
|
|
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Lesser General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2.1 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
|
|
Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public
|
|
License along with this library; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
|
***********************************************************************/
|
|
|
|
#include "moses/TranslationModel/PhraseDictionary.h"
|
|
#include "moses/TranslationModel/PhraseDictionaryTreeAdaptor.h"
|
|
#include "moses/TranslationModel/RuleTable/PhraseDictionarySCFG.h"
|
|
#include "moses/TranslationModel/RuleTable/PhraseDictionaryOnDisk.h"
|
|
#include "moses/TranslationModel/RuleTable/PhraseDictionaryALSuffixArray.h"
|
|
#include "moses/TranslationModel/RuleTable/PhraseDictionaryFuzzyMatch.h"
|
|
|
|
#ifndef WIN32
|
|
#include "moses/TranslationModel/PhraseDictionaryDynSuffixArray.h"
|
|
#include "moses/TranslationModel/CompactPT/PhraseDictionaryCompact.h"
|
|
#endif
|
|
#include "moses/TranslationModel/RuleTable/UTrie.h"
|
|
|
|
#include "moses/StaticData.h"
|
|
#include "moses/InputType.h"
|
|
#include "moses/TranslationOption.h"
|
|
#include "moses/UserMessage.h"
|
|
|
|
using namespace std;
|
|
|
|
namespace Moses
|
|
{
|
|
|
|
const TargetPhraseCollection *PhraseDictionary::
|
|
GetTargetPhraseCollection(InputType const& src,WordsRange const& range) const
|
|
{
|
|
return GetTargetPhraseCollection(src.GetSubString(range));
|
|
}
|
|
|
|
PhraseDictionaryFeature::PhraseDictionaryFeature(const std::string &line)
|
|
:DecodeFeature("PhraseModel", line)
|
|
,m_tableLimit(20) // TODO default?
|
|
{
|
|
for (size_t i = 0; i < m_args.size(); ++i) {
|
|
const vector<string> &args = m_args[i];
|
|
|
|
if (args[0] == "implementation") {
|
|
m_implementation = (PhraseTableImplementation) Scan<size_t>(args[1]);
|
|
}
|
|
else if (args[0] == "input-factor") {
|
|
m_input =Tokenize<FactorType>(args[1]);
|
|
}
|
|
else if (args[0] == "output-factor") {
|
|
m_output =Tokenize<FactorType>(args[1]);
|
|
}
|
|
else if (args[0] == "num-input-features") {
|
|
m_numInputScores = Scan<unsigned>(args[1]);
|
|
}
|
|
else if (args[0] == "path") {
|
|
m_filePath = args[1];
|
|
}
|
|
else if (args[0] == "table-limit") {
|
|
m_tableLimit = Scan<size_t>(args[1]);
|
|
}
|
|
else if (args[0] == "target-path") {
|
|
m_targetFile = args[1];
|
|
}
|
|
else if (args[0] == "alignment-path") {
|
|
m_alignmentsFile = args[1];
|
|
}
|
|
|
|
else {
|
|
UserMessage::Add("Unknown argument " + args[0]);
|
|
abort();
|
|
}
|
|
} // for (size_t i = 0; i < toks.size(); ++i) {
|
|
|
|
}
|
|
|
|
PhraseDictionaryFeature::PhraseDictionaryFeature
|
|
(PhraseTableImplementation implementation
|
|
, size_t numScoreComponent
|
|
, unsigned numInputScores
|
|
, const std::vector<FactorType> &input
|
|
, const std::vector<FactorType> &output
|
|
, const std::string &filePath
|
|
, size_t tableLimit
|
|
, const std::string &targetFile // default param
|
|
, const std::string &alignmentsFile) // default param
|
|
:DecodeFeature("PhraseModel",numScoreComponent,input,output, "PhraseModel"),
|
|
m_numInputScores(numInputScores),
|
|
m_filePath(filePath),
|
|
m_tableLimit(tableLimit),
|
|
m_implementation(implementation),
|
|
m_targetFile(targetFile),
|
|
m_alignmentsFile(alignmentsFile)
|
|
{
|
|
if (implementation == Memory || implementation == SCFG || implementation == SuffixArray ||
|
|
implementation==Compact || implementation==FuzzyMatch ) {
|
|
m_useThreadSafePhraseDictionary = true;
|
|
} else {
|
|
m_useThreadSafePhraseDictionary = false;
|
|
}
|
|
}
|
|
|
|
PhraseDictionary* PhraseDictionaryFeature::LoadPhraseTable(const TranslationSystem* system)
|
|
{
|
|
const StaticData& staticData = StaticData::Instance();
|
|
std::vector<float> weightT = staticData.GetWeights(this);
|
|
|
|
if (m_implementation == Memory) {
|
|
// memory phrase table
|
|
VERBOSE(2,"using standard phrase tables" << std::endl);
|
|
if (!FileExists(m_filePath) && FileExists(m_filePath + ".gz")) {
|
|
m_filePath += ".gz";
|
|
VERBOSE(2,"Using gzipped file" << std::endl);
|
|
}
|
|
if (staticData.GetInputType() != SentenceInput) {
|
|
UserMessage::Add("Must use binary phrase table for this input type");
|
|
CHECK(false);
|
|
}
|
|
|
|
PhraseDictionaryMemory* pdm = new PhraseDictionaryMemory(GetNumScoreComponents(),this);
|
|
bool ret = pdm->Load(GetInput(), GetOutput()
|
|
, m_filePath
|
|
, weightT
|
|
, m_tableLimit
|
|
, staticData.GetLMList()
|
|
, staticData.GetWeightWordPenalty());
|
|
CHECK(ret);
|
|
return pdm;
|
|
} else if (m_implementation == Binary) {
|
|
PhraseDictionaryTreeAdaptor* pdta = new PhraseDictionaryTreeAdaptor(GetNumScoreComponents(), m_numInputScores,this);
|
|
bool ret = pdta->Load( GetInput()
|
|
, GetOutput()
|
|
, m_filePath
|
|
, weightT
|
|
, m_tableLimit
|
|
, staticData.GetLMList()
|
|
, staticData.GetWeightWordPenalty());
|
|
CHECK(ret);
|
|
return pdta;
|
|
} else if (m_implementation == SCFG || m_implementation == Hiero) {
|
|
// memory phrase table
|
|
if (m_implementation == Hiero) {
|
|
VERBOSE(2,"using Hiero format phrase tables" << std::endl);
|
|
} else {
|
|
VERBOSE(2,"using Moses-formatted SCFG phrase tables" << std::endl);
|
|
}
|
|
if (!FileExists(m_filePath) && FileExists(m_filePath + ".gz")) {
|
|
m_filePath += ".gz";
|
|
VERBOSE(2,"Using gzipped file" << std::endl);
|
|
}
|
|
|
|
RuleTableTrie *dict;
|
|
if (staticData.GetParsingAlgorithm() == ParseScope3) {
|
|
dict = new RuleTableUTrie(GetNumScoreComponents(), this);
|
|
} else {
|
|
dict = new PhraseDictionarySCFG(GetNumScoreComponents(), this);
|
|
}
|
|
bool ret = dict->Load(GetInput()
|
|
, GetOutput()
|
|
, m_filePath
|
|
, weightT
|
|
, m_tableLimit
|
|
, staticData.GetLMList()
|
|
, staticData.GetWordPenaltyProducer());
|
|
CHECK(ret);
|
|
return dict;
|
|
} else if (m_implementation == ALSuffixArray) {
|
|
// memory phrase table
|
|
VERBOSE(2,"using Hiero format phrase tables" << std::endl);
|
|
if (!FileExists(m_filePath) && FileExists(m_filePath + ".gz")) {
|
|
m_filePath += ".gz";
|
|
VERBOSE(2,"Using gzipped file" << std::endl);
|
|
}
|
|
|
|
PhraseDictionaryALSuffixArray* pdm = new PhraseDictionaryALSuffixArray(GetNumScoreComponents(),this);
|
|
bool ret = pdm->Load(GetInput()
|
|
, GetOutput()
|
|
, m_filePath
|
|
, weightT
|
|
, m_tableLimit
|
|
, staticData.GetLMList()
|
|
, staticData.GetWordPenaltyProducer());
|
|
CHECK(ret);
|
|
return pdm;
|
|
} else if (m_implementation == OnDisk) {
|
|
|
|
PhraseDictionaryOnDisk* pdta = new PhraseDictionaryOnDisk(GetNumScoreComponents(), this);
|
|
bool ret = pdta->Load(GetInput()
|
|
, GetOutput()
|
|
, m_filePath
|
|
, weightT
|
|
, m_tableLimit
|
|
, staticData.GetLMList()
|
|
, staticData.GetWordPenaltyProducer());
|
|
CHECK(ret);
|
|
return pdta;
|
|
} else if (m_implementation == SuffixArray) {
|
|
#ifndef WIN32
|
|
PhraseDictionaryDynSuffixArray *pd = new PhraseDictionaryDynSuffixArray(GetNumScoreComponents(), this);
|
|
if(!(pd->Load(
|
|
GetInput()
|
|
,GetOutput()
|
|
,m_filePath
|
|
,m_targetFile
|
|
,m_alignmentsFile
|
|
,weightT, m_tableLimit
|
|
,staticData.GetLMList()
|
|
,staticData.GetWeightWordPenalty()))) {
|
|
std::cerr << "FAILED TO LOAD\n" << endl;
|
|
delete pd;
|
|
pd = NULL;
|
|
}
|
|
std::cerr << "Suffix array phrase table loaded" << std::endl;
|
|
return pd;
|
|
#else
|
|
CHECK(false);
|
|
#endif
|
|
} else if (m_implementation == FuzzyMatch) {
|
|
|
|
PhraseDictionaryFuzzyMatch *dict = new PhraseDictionaryFuzzyMatch(GetNumScoreComponents(), this);
|
|
|
|
bool ret = dict->Load(GetInput()
|
|
, GetOutput()
|
|
, m_filePath
|
|
, weightT
|
|
, m_tableLimit
|
|
, staticData.GetLMList()
|
|
, staticData.GetWordPenaltyProducer());
|
|
CHECK(ret);
|
|
|
|
return dict;
|
|
} else if (m_implementation == Compact) {
|
|
#ifndef WIN32
|
|
VERBOSE(2,"Using compact phrase table" << std::endl);
|
|
|
|
PhraseDictionaryCompact* pd = new PhraseDictionaryCompact(GetNumScoreComponents(), m_implementation, this);
|
|
bool ret = pd->Load(GetInput(), GetOutput()
|
|
, m_filePath
|
|
, weightT
|
|
, m_tableLimit
|
|
, staticData.GetLMList()
|
|
, staticData.GetWeightWordPenalty());
|
|
CHECK(ret);
|
|
return pd;
|
|
#else
|
|
CHECK(false);
|
|
#endif
|
|
}
|
|
else {
|
|
std::cerr << "Unknown phrase table type " << m_implementation << endl;
|
|
CHECK(false);
|
|
}
|
|
}
|
|
|
|
void PhraseDictionaryFeature::InitDictionary(const TranslationSystem* system)
|
|
{
|
|
//Thread-safe phrase dictionaries get loaded now
|
|
if (m_useThreadSafePhraseDictionary && !m_threadSafePhraseDictionary.get()) {
|
|
IFVERBOSE(1)
|
|
PrintUserTime("Start loading phrase table from " + m_filePath);
|
|
m_threadSafePhraseDictionary.reset(LoadPhraseTable(system));
|
|
IFVERBOSE(1)
|
|
PrintUserTime("Finished loading phrase tables");
|
|
}
|
|
//Other types will be lazy loaded
|
|
}
|
|
|
|
const PhraseDictionary* PhraseDictionaryFeature::GetDictionary() const
|
|
{
|
|
PhraseDictionary* dict;
|
|
if (m_useThreadSafePhraseDictionary) {
|
|
dict = m_threadSafePhraseDictionary.get();
|
|
} else {
|
|
dict = m_threadUnsafePhraseDictionary.get();
|
|
}
|
|
CHECK(dict);
|
|
return dict;
|
|
}
|
|
|
|
PhraseDictionary* PhraseDictionaryFeature::GetDictionary()
|
|
{
|
|
PhraseDictionary* dict;
|
|
if (m_useThreadSafePhraseDictionary) {
|
|
dict = m_threadSafePhraseDictionary.get();
|
|
} else {
|
|
dict = m_threadUnsafePhraseDictionary.get();
|
|
}
|
|
CHECK(dict);
|
|
return dict;
|
|
}
|
|
|
|
|
|
PhraseDictionaryFeature::~PhraseDictionaryFeature()
|
|
{}
|
|
|
|
bool PhraseDictionaryFeature::ComputeValueInTranslationOption() const
|
|
{
|
|
return true;
|
|
}
|
|
|
|
const PhraseDictionaryFeature* PhraseDictionary::GetFeature() const
|
|
{
|
|
return m_feature;
|
|
}
|
|
|
|
void PhraseDictionaryFeature::InitializeForInput(const InputType& source)
|
|
{
|
|
PhraseDictionary* dict;
|
|
if (m_useThreadSafePhraseDictionary) {
|
|
//thread safe dictionary should already be loaded
|
|
dict = m_threadSafePhraseDictionary.get();
|
|
} else {
|
|
//thread-unsafe dictionary may need to be loaded if this is a new thread.
|
|
if (!m_threadUnsafePhraseDictionary.get()) {
|
|
m_threadUnsafePhraseDictionary.reset(LoadPhraseTable(NULL));
|
|
}
|
|
dict = m_threadUnsafePhraseDictionary.get();
|
|
}
|
|
CHECK(dict);
|
|
dict->InitializeForInput(source);
|
|
|
|
}
|
|
|
|
void PhraseDictionaryFeature::CleanUpAfterSentenceProcessing(const InputType& source)
|
|
{
|
|
PhraseDictionary* dict;
|
|
if (m_useThreadSafePhraseDictionary) {
|
|
//thread safe dictionary should already be loaded
|
|
dict = m_threadSafePhraseDictionary.get();
|
|
} else {
|
|
dict = m_threadUnsafePhraseDictionary.get();
|
|
}
|
|
CHECK(dict);
|
|
dict->CleanUpAfterSentenceProcessing(source);
|
|
|
|
}
|
|
|
|
}
|
|
|