mosesdecoder/biconcor/PhrasePairCollection.h
2013-04-05 11:26:00 +01:00

47 lines
1.0 KiB
C++

#pragma once
#include <vector>
#include <string>
class Alignment;
class PhrasePair;
class SuffixArray;
class TargetCorpus;
class Mismatch;
class PhrasePairCollection
{
public:
typedef unsigned int INDEX;
private:
SuffixArray *m_suffixArray;
TargetCorpus *m_targetCorpus;
Alignment *m_alignment;
std::vector<std::vector<PhrasePair*> > m_collection;
std::vector< Mismatch* > m_mismatch, m_unaligned;
int m_size;
int m_max_lookup;
int m_max_translation;
int m_max_example;
// No copying allowed.
PhrasePairCollection(const PhrasePairCollection&);
void operator=(const PhrasePairCollection&);
public:
PhrasePairCollection ( SuffixArray *, TargetCorpus *, Alignment *, int, int );
~PhrasePairCollection ();
int GetCollection( const std::vector<std::string >& sourceString );
void Print(bool pretty) const;
void PrintHTML() const;
};
// sorting helper
struct CompareBySize {
bool operator()(const std::vector<PhrasePair*>& a, const std::vector<PhrasePair*>& b ) const {
return a.size() > b.size();
}
};