File size: 2,282 Bytes
158b61b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
#pragma once
#include <iostream>
#include <vector>
#include "../Vector.h"
namespace Moses2
{
class System;
class Sentence;
class Bitmap;
class MemPool;
#define NOT_A_ZONE 999999999
class ReorderingConstraint
{
protected:
// const size_t m_size; /**< number of words in sentence */
size_t m_size; /**< number of words in sentence */
bool *m_wall; /**< flag for each word if it is a wall */
//size_t *m_wall; /**< flag for each word if it is a wall */
size_t *m_localWall; /**< flag for each word if it is a local wall */
Vector< std::pair<size_t,size_t> > m_zone; /** zones that limit reordering */
bool m_active; /**< flag indicating, if there are any active constraints */
int m_max_distortion;
MemPool &m_pool;
ReorderingConstraint(const ReorderingConstraint &); // do not implement
public:
//! create ReorderingConstraint of length size and initialise to zero
ReorderingConstraint(MemPool &pool)
: m_wall(NULL)
, m_localWall(NULL)
, m_active(false)
, m_pool(pool)
, m_zone(pool)
{}
//! destructer
~ReorderingConstraint();
//! allocate memory for memory for a sentence of a given size
void InitializeWalls(size_t size, int max_distortion);
//! changes walls in zones into local walls
void FinalizeWalls();
//! set value at a particular position
void SetWall( size_t pos, bool value );
//! whether a word has been translated at a particular position
bool GetWall(size_t pos) const {
return m_wall[pos];
}
//! whether a word has been translated at a particular position
bool GetLocalWall(size_t pos, size_t zone ) const {
return (m_localWall[pos] == zone);
}
//! set a zone
void SetZone( size_t startPos, size_t endPos );
//! returns the vector of zones
Vector< std::pair< size_t,size_t> > & GetZones() {
return m_zone;
}
//! set the reordering walls based on punctuation in the sentence
void SetMonotoneAtPunctuation( const Sentence & sentence );
//! check if all constraints are fulfilled -> all find
bool Check( const Bitmap &bitmap, size_t start, size_t end ) const;
//! checks if reordering constraints will be enforced
bool IsActive() const {
return m_active;
}
std::ostream &Debug(std::ostream &out, const System &system) const;
};
}
|