mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-11-09 16:04:41 +03:00
116 lines
3.1 KiB
C++
116 lines
3.1 KiB
C++
#pragma once
|
|
// $Id$
|
|
/***********************************************************************
|
|
Moses - factored phrase-based, hierarchical and syntactic language decoder
|
|
Copyright (C) 2009 Hieu Hoang
|
|
|
|
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 <string>
|
|
#include <fstream>
|
|
#include "Vocab.h"
|
|
#include "PhraseNode.h"
|
|
#include "moses/Word.h"
|
|
|
|
namespace OnDiskPt
|
|
{
|
|
const float DEFAULT_COUNT = 66666;
|
|
|
|
/** Global class with misc information need to create and use the on-disk rule table.
|
|
* 1 object of this class should be instantiated per rule table.
|
|
* Currently only hierarchical/syntax models use this, but can & should be used with pb models too
|
|
*/
|
|
class OnDiskWrapper
|
|
{
|
|
protected:
|
|
Vocab m_vocab;
|
|
std::string m_filePath;
|
|
int m_numSourceFactors, m_numTargetFactors, m_numScores;
|
|
std::fstream m_fileMisc, m_fileVocab, m_fileSource, m_fileTarget, m_fileTargetInd, m_fileTargetColl;
|
|
|
|
size_t m_defaultNodeSize;
|
|
PhraseNode *m_rootSourceNode;
|
|
|
|
std::map<std::string, uint64_t> m_miscInfo;
|
|
|
|
void SaveMisc();
|
|
bool OpenForLoad(const std::string &filePath);
|
|
bool LoadMisc();
|
|
|
|
public:
|
|
static int VERSION_NUM;
|
|
|
|
OnDiskWrapper();
|
|
~OnDiskWrapper();
|
|
|
|
void BeginLoad(const std::string &filePath);
|
|
|
|
void BeginSave(const std::string &filePath
|
|
, int numSourceFactors, int numTargetFactors, int numScores);
|
|
void EndSave();
|
|
|
|
Vocab &GetVocab() {
|
|
return m_vocab;
|
|
}
|
|
const Vocab &GetVocab() const {
|
|
return m_vocab;
|
|
}
|
|
|
|
size_t GetSourceWordSize() const;
|
|
size_t GetTargetWordSize() const;
|
|
|
|
std::fstream &GetFileSource() {
|
|
return m_fileSource;
|
|
}
|
|
std::fstream &GetFileTargetInd() {
|
|
return m_fileTargetInd;
|
|
}
|
|
std::fstream &GetFileTargetColl() {
|
|
return m_fileTargetColl;
|
|
}
|
|
std::fstream &GetFileVocab() {
|
|
return m_fileVocab;
|
|
}
|
|
|
|
size_t GetNumSourceFactors() const {
|
|
return m_numSourceFactors;
|
|
}
|
|
size_t GetNumTargetFactors() const {
|
|
return m_numTargetFactors;
|
|
}
|
|
|
|
size_t GetNumScores() const {
|
|
return m_numScores;
|
|
}
|
|
size_t GetNumCounts() const {
|
|
return 1;
|
|
}
|
|
|
|
PhraseNode &GetRootSourceNode() {
|
|
return *m_rootSourceNode;
|
|
}
|
|
const PhraseNode &GetRootSourceNode() const {
|
|
return *m_rootSourceNode;
|
|
}
|
|
|
|
uint64_t GetMisc(const std::string &key) const;
|
|
|
|
Word *ConvertFromMoses(const std::vector<Moses::FactorType> &factorsVec
|
|
, const Moses::Word &origWord) const;
|
|
|
|
};
|
|
|
|
}
|