mosesdecoder/OnDiskPt/OnDiskWrapper.h

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;
};
}