wiki-cased / tools /mosesdecoder /moses /ChartCellCollection.h

Upload folder using huggingface_hub

41f6dd8 verified almost 2 years ago

3.24 kB

// $Id$
// vim:tabstop=2
/***********************************************************************
 Moses - factored phrase-based language decoder
 Copyright (C) 2010 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
 ***********************************************************************/
#pragma once

#include <boost/ptr_container/ptr_vector.hpp>
#include "InputType.h"
#include "ChartCell.h"
#include "Range.h"
#include "InputPath.h"

namespace Moses
{
class InputType;
class ChartManager;
class ChartParser;

class ChartCellCollectionBase
{
public:
  template <class Factory> ChartCellCollectionBase(const InputType &input,
      const Factory &factory,
      const ChartParser &parser)
    :m_cells(input.GetSize()) {

    size_t size = input.GetSize();
    for (size_t startPos = 0; startPos < size; ++startPos) {
      std::vector<ChartCellBase*> &inner = m_cells[startPos];
      inner.reserve(size - startPos);
      for (size_t endPos = startPos; endPos < size; ++endPos) {
        inner.push_back(factory(startPos, endPos));
      }
      /* Hack: ChartCellLabel shouldn't need to know its span, but the parser
       * gets it from there :-(.  The span is actually stored as a reference,
       * which needs to point somewhere, so I have it refer to the ChartCell.
       */
      const Range &range = inner[0]->GetCoverage();

      m_source.push_back(new ChartCellLabel(range, input.GetWord(startPos)));
    }
  }

  virtual ~ChartCellCollectionBase();


  const ChartCellBase &GetBase(const Range &coverage) const {
    return *m_cells[coverage.GetStartPos()][coverage.GetEndPos() - coverage.GetStartPos()];
  }

  ChartCellBase &MutableBase(const Range &coverage) {
    return *m_cells[coverage.GetStartPos()][coverage.GetEndPos() - coverage.GetStartPos()];
  }


  const ChartCellLabel &GetSourceWordLabel(size_t at) const {
    return m_source[at];
  }

private:
  std::vector<std::vector<ChartCellBase*> > m_cells;

  boost::ptr_vector<ChartCellLabel> m_source;

};

/** Hold all the chart cells for 1 input sentence. A variable of this type is held by the ChartManager
 */
class ChartCellCollection : public ChartCellCollectionBase
{
public:
  ChartCellCollection(const InputType &input, ChartManager &manager);

  //! get a chart cell for a particular range
  ChartCell &Get(const Range &coverage) {
    return static_cast<ChartCell&>(MutableBase(coverage));
  }

  //! get a chart cell for a particular range
  const ChartCell &Get(const Range &coverage) const {
    return static_cast<const ChartCell&>(GetBase(coverage));
  }
};

}

	// $Id$
	// vim:tabstop=2
	/***********************************************************************
	Moses - factored phrase-based language decoder
	Copyright (C) 2010 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
	***********************************************************************/
	#pragma once

	#include <boost/ptr_container/ptr_vector.hpp>
	#include "InputType.h"
	#include "ChartCell.h"
	#include "Range.h"
	#include "InputPath.h"

	namespace Moses
	{
	class InputType;
	class ChartManager;
	class ChartParser;

	class ChartCellCollectionBase
	{
	public:
	template <class Factory> ChartCellCollectionBase(const InputType &input,
	const Factory &factory,
	const ChartParser &parser)
	:m_cells(input.GetSize()) {

	size_t size = input.GetSize();
	for (size_t startPos = 0; startPos < size; ++startPos) {
	std::vector<ChartCellBase*> &inner = m_cells[startPos];
	inner.reserve(size - startPos);
	for (size_t endPos = startPos; endPos < size; ++endPos) {
	inner.push_back(factory(startPos, endPos));
	}
	/* Hack: ChartCellLabel shouldn't need to know its span, but the parser
	* gets it from there :-(. The span is actually stored as a reference,
	* which needs to point somewhere, so I have it refer to the ChartCell.
	*/
	const Range &range = inner[0]->GetCoverage();

	m_source.push_back(new ChartCellLabel(range, input.GetWord(startPos)));
	}
	}

	virtual ~ChartCellCollectionBase();


	const ChartCellBase &GetBase(const Range &coverage) const {
	return *m_cells[coverage.GetStartPos()][coverage.GetEndPos() - coverage.GetStartPos()];
	}

	ChartCellBase &MutableBase(const Range &coverage) {
	return *m_cells[coverage.GetStartPos()][coverage.GetEndPos() - coverage.GetStartPos()];
	}


	const ChartCellLabel &GetSourceWordLabel(size_t at) const {
	return m_source[at];
	}

	private:
	std::vector<std::vector<ChartCellBase*> > m_cells;

	boost::ptr_vector<ChartCellLabel> m_source;

	};

	/** Hold all the chart cells for 1 input sentence. A variable of this type is held by the ChartManager
	*/
	class ChartCellCollection : public ChartCellCollectionBase
	{
	public:
	ChartCellCollection(const InputType &input, ChartManager &manager);

	//! get a chart cell for a particular range
	ChartCell &Get(const Range &coverage) {
	return static_cast<ChartCell&>(MutableBase(coverage));
	}

	//! get a chart cell for a particular range
	const ChartCell &Get(const Range &coverage) const {
	return static_cast<const ChartCell&>(GetBase(coverage));
	}
	};

	}