mosesdecoder/moses/MockHypothesis.h

/***********************************************************************
Moses - factored phrase-based language decoder
Copyright (C) 2010 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
***********************************************************************/

#ifndef _MOCK_HYPOTHESIS_
#define _MOCK_HYPOTHESIS_

#include <memory>
#include <vector>

#include "moses/FF/UnknownWordPenaltyProducer.h"
#include "moses/FF/DistortionScoreProducer.h"
#include "moses/FF/WordPenaltyProducer.h"
#include "Hypothesis.h"
#include "Manager.h"
#include "TranslationOption.h"

namespace MosesTest
{

//
// Construct a hypothesis with arbitrary source and target phrase
// sequences. Useful for testing feature functions.
//

typedef std::pair<size_t,size_t> Alignment; //(first,last) in source

class MockHypothesisGuard
{
public:
  /** Creates a phrase-based hypothesis.
  */
  MockHypothesisGuard(
    const std::string& sourceSentence,
    const std::vector<Alignment>& alignments,
    const std::vector<std::string>& targetSegments);
  Moses::Hypothesis* operator*() const {
    return m_hypothesis;
  }

  /** Destroy the hypothesis chain */
  ~MockHypothesisGuard();

private:
  Moses::TranslationOption m_initialTransOpt;
  Moses::Sentence m_sentence;
  Moses::WordPenaltyProducer m_wp;
  Moses::UnknownWordPenaltyProducer m_uwp;
  Moses::DistortionScoreProducer m_dist;
  Moses::Manager m_manager;
  Moses::Hypothesis* m_hypothesis;
  std::vector<Moses::TargetPhrase> m_targetPhrases;
  std::vector<Moses::TranslationOption*> m_toptions;
};

class HypothesisFixture
{
public:
  HypothesisFixture();
  const Moses::Hypothesis* empty() {
    return **m_empty;
  }
  const Moses::Hypothesis* partial() {
    return **m_partial;
  }
  const Moses::Hypothesis* full() {
    return **m_full;
  }

private:
  std::auto_ptr<MockHypothesisGuard> m_empty;
  std::auto_ptr<MockHypothesisGuard> m_partial;
  std::auto_ptr<MockHypothesisGuard> m_full;
};


}

#endif