1 files changed, 86 insertions, 0 deletions
diff --git a/moses/src/SquareMatrix.h b/moses/src/SquareMatrix.h
new file mode 100644
index 000000000..2935c02de
--- /dev/null
+++ b/moses/src/SquareMatrix.h
@@ -0,0 +1,86 @@
+// $Id: SquareMatrix.h 1964 2008-12-20 20:22:35Z phkoehn $
+
+/***********************************************************************
+Moses - factored phrase-based language decoder
+Copyright (C) 2006 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
+***********************************************************************/
+
+#pragma once
+
+#include <iostream>
+#include "TypeDef.h"
+#include "Util.h"
+#include "WordsBitmap.h"
+
+namespace Moses
+{
+
+//! A square array of floats to store future costs
+class SquareMatrix
+{
+	friend std::ostream& operator<<(std::ostream &out, const SquareMatrix &matrix);
+protected:
+	const size_t m_size; /**< length of the square (sentence length) */
+	float *m_array; /**< two-dimensional array to store floats */
+
+	SquareMatrix(); // not implemented
+	SquareMatrix(const SquareMatrix &copy); // not implemented
+	
+public:
+	SquareMatrix(size_t size)
+	:m_size(size)
+	{
+		m_array = (float*) malloc(sizeof(float) * size * size);
+	}
+	~SquareMatrix()
+	{
+		free(m_array);
+	}
+	/** Returns length of the square: typically the sentence length */
+	inline size_t GetSize() const
+	{
+		return m_size;
+	}
+	/** Get a future cost score for a span */
+	inline float GetScore(size_t startPos, size_t endPos) const
+	{
+		return m_array[startPos * m_size + endPos];
+	}
+	/** Set a future cost score for a span */
+	inline void SetScore(size_t startPos, size_t endPos, float value)
+	{
+		m_array[startPos * m_size + endPos] = value;
+	}
+	float CalcFutureScore( WordsBitmap const& ) const;
+	float CalcFutureScore( WordsBitmap const&, size_t startPos, size_t endPos ) const;
+	
+	TO_STRING();
+};
+
+inline std::ostream& operator<<(std::ostream &out, const SquareMatrix &matrix)
+{
+	for (size_t endPos = 0 ; endPos < matrix.GetSize() ; endPos++)
+	{
+		for (size_t startPos = 0 ; startPos < matrix.GetSize() ; startPos++)
+			out << matrix.GetScore(startPos, endPos) << " ";
+		out << std::endl;
+	}
+	
+	return out;
+}
+
+}