Update version to 1.3

1 files changed, 217 insertions, 0 deletions

diff --git a/src/Projector2D.cpp b/src/Projector2D.cpp
new file mode 100644
index 0000000..e67cd0c
--- /dev/null
+++ b/src/Projector2D.cpp
@@ -0,0 +1,217 @@
+/*
+-----------------------------------------------------------------------
+Copyright 2012 iMinds-Vision Lab, University of Antwerp
+
+Contact: astra@ua.ac.be
+Website: http://astra.ua.ac.be
+
+
+This file is part of the
+All Scale Tomographic Reconstruction Antwerp Toolbox ("ASTRA Toolbox").
+
+The ASTRA Toolbox is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+The ASTRA Toolbox 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 General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with the ASTRA Toolbox. If not, see <http://www.gnu.org/licenses/>.
+
+-----------------------------------------------------------------------
+$Id$
+*/
+
+#include "astra/Projector2D.h"
+
+#include "astra/FanFlatProjectionGeometry2D.h"
+#include "astra/FanFlatVecProjectionGeometry2D.h"
+#include "astra/SparseMatrixProjectionGeometry2D.h"
+#include "astra/SparseMatrix.h"
+
+
+namespace astra
+{
+
+//----------------------------------------------------------------------------------------
+// constructor
+CProjector2D::CProjector2D() : configCheckData(0)
+{
+
+	m_bIsInitialized = false;
+}
+
+//----------------------------------------------------------------------------------------
+// constructor
+CProjector2D::CProjector2D(CProjectionGeometry2D* _pProjectionGeometry, CVolumeGeometry2D* _pVolumeGeometry) : configCheckData(0)
+{
+	m_pProjectionGeometry = _pProjectionGeometry->clone();
+	m_pVolumeGeometry = _pVolumeGeometry->clone();
+	m_bIsInitialized = true;
+}
+
+//----------------------------------------------------------------------------------------
+// destructor
+CProjector2D::~CProjector2D()
+{
+	clear();
+}
+
+//---------------------------------------------------------------------------------------
+// Clear - Constructors
+void CProjector2D::_clear()
+{
+	m_pProjectionGeometry = NULL;
+	m_pVolumeGeometry = NULL;
+	m_bIsInitialized = false;
+}
+
+//---------------------------------------------------------------------------------------
+// Clear - Public
+void CProjector2D::clear()
+{
+	if (m_pProjectionGeometry) {
+		delete m_pProjectionGeometry;
+		m_pProjectionGeometry = NULL;
+	}
+	if (m_pVolumeGeometry) {
+		delete m_pVolumeGeometry;
+		m_pVolumeGeometry = NULL;
+	}
+	m_bIsInitialized = false;
+}
+
+//---------------------------------------------------------------------------------------
+// Check
+bool CProjector2D::_check()
+{
+	// check pointers
+	ASTRA_CONFIG_CHECK(m_pProjectionGeometry, "Projector2D", "Invalid Projection Geometry Object.");
+	ASTRA_CONFIG_CHECK(m_pVolumeGeometry, "Projector2D", "Invalid Volume Geometry Object.");
+
+	// check initializations
+	ASTRA_CONFIG_CHECK(m_pProjectionGeometry->isInitialized(), "Projector2D", "Projection Geometry Object Not Initialized.");
+	ASTRA_CONFIG_CHECK(m_pVolumeGeometry->isInitialized(), "Projector2D", "Volume Geometry Object Not Initialized.");
+
+	// success
+	return true;
+}
+
+//---------------------------------------------------------------------------------------
+// Initialize, use a Config object
+bool CProjector2D::initialize(const Config& _cfg)
+{
+	ASTRA_ASSERT(_cfg.self);
+	ConfigStackCheck<CProjector2D> CC("Projector2D", this, _cfg);
+
+	// if already initialized, clear first
+	if (m_bIsInitialized) {
+		clear();
+	}
+
+	// required: ProjectionGeometry
+	XMLNode* node = _cfg.self->getSingleNode("ProjectionGeometry");
+	ASTRA_CONFIG_CHECK(node, "Projector2D", "No ProjectionGeometry tag specified.");
+
+	// FIXME: Change how the base class is created. (This is duplicated
+	// in astra_mex_data2d.cpp.)
+	std::string type = node->getAttribute("type");
+	if (type == "sparse_matrix") {
+		m_pProjectionGeometry = new CSparseMatrixProjectionGeometry2D();
+		m_pProjectionGeometry->initialize(Config(node));
+	} else if (type == "fanflat") {
+		CFanFlatProjectionGeometry2D* pFanFlatProjectionGeometry = new CFanFlatProjectionGeometry2D();
+		pFanFlatProjectionGeometry->initialize(Config(node));
+		m_pProjectionGeometry = pFanFlatProjectionGeometry;
+	} else if (type == "fanflat_vec") {
+		CFanFlatVecProjectionGeometry2D* pFanFlatVecProjectionGeometry = new CFanFlatVecProjectionGeometry2D();
+		pFanFlatVecProjectionGeometry->initialize(Config(node));
+		m_pProjectionGeometry = pFanFlatVecProjectionGeometry;
+	} else {
+		m_pProjectionGeometry = new CParallelProjectionGeometry2D();
+		m_pProjectionGeometry->initialize(Config(node));
+	}
+	// "node" is deleted by the temp Config(node) objects
+	ASTRA_CONFIG_CHECK(m_pProjectionGeometry->isInitialized(), "Projector2D", "ProjectionGeometry not initialized.");	
+	CC.markNodeParsed("ProjectionGeometry");
+
+
+	// required: VolumeGeometry
+	node = _cfg.self->getSingleNode("VolumeGeometry");
+	ASTRA_CONFIG_CHECK(node, "Projector2D", "No VolumeGeometry tag specified.");
+	m_pVolumeGeometry = new CVolumeGeometry2D();
+	m_pVolumeGeometry->initialize(Config(node));
+	// "node" is deleted by the temp Config(node) object
+	ASTRA_CONFIG_CHECK(m_pVolumeGeometry->isInitialized(), "Projector2D", "VolumeGeometry not initialized.");
+	CC.markNodeParsed("VolumeGeometry");
+
+	return true;
+}
+
+//----------------------------------------------------------------------------------------
+// weights of each detector in a projection angle
+void CProjector2D::computeProjectionRayWeights(int _iProjection, SPixelWeight* _pfWeightedPixels, int* _piRayStoredPixelCount)
+{
+	int iPixelBufferSize = getProjectionWeightsCount(_iProjection);
+	
+	int iDetector;
+	for(iDetector = m_pProjectionGeometry->getDetectorCount()-1; iDetector >= 0; --iDetector) {
+		computeSingleRayWeights(_iProjection,									// projector index
+								iDetector,										// detector index
+								&_pfWeightedPixels[iDetector*iPixelBufferSize],	// pixel buffer
+								iPixelBufferSize,								// pixel buffer size
+								_piRayStoredPixelCount[iDetector]);				// stored pixel count
+	}
+
+}
+
+//----------------------------------------------------------------------------------------
+// explicit projection matrix
+CSparseMatrix* CProjector2D::getMatrix()
+{
+	unsigned int iProjectionCount = m_pProjectionGeometry->getProjectionAngleCount();
+	unsigned int iDetectorCount = m_pProjectionGeometry->getDetectorCount();
+	unsigned int iRayCount = iProjectionCount * iDetectorCount;
+	unsigned int iVolumeSize = m_pVolumeGeometry->getGridTotCount();
+	unsigned long lSize = 0;
+	unsigned int iMaxRayLength = 0;
+	for (unsigned int i = 0; i < iProjectionCount; ++i) {
+		unsigned int iRayLength = getProjectionWeightsCount(i);
+		lSize += iDetectorCount * iRayLength;
+		if (iRayLength > iMaxRayLength)
+			iMaxRayLength = iRayLength;
+	}
+	CSparseMatrix* pMatrix = new CSparseMatrix(iRayCount, iVolumeSize, lSize);
+
+	if (!pMatrix || !pMatrix->isInitialized()) {
+		delete pMatrix;
+		return 0;
+	}
+
+	SPixelWeight* pEntries = new SPixelWeight[iMaxRayLength];
+	unsigned long lMatrixIndex = 0;
+	for (unsigned int iRay = 0; iRay < iRayCount; ++iRay) {
+		pMatrix->m_plRowStarts[iRay] = lMatrixIndex;
+		int iPixelCount;
+		int iProjIndex, iDetIndex;
+		m_pProjectionGeometry->indexToAngleDetectorIndex(iRay, iProjIndex, iDetIndex);
+		computeSingleRayWeights(iProjIndex, iDetIndex, pEntries, iMaxRayLength, iPixelCount);
+		
+		for (int i = 0; i < iPixelCount; ++i) {
+			pMatrix->m_piColIndices[lMatrixIndex] = pEntries[i].m_iIndex;
+			pMatrix->m_pfValues[lMatrixIndex] = pEntries[i].m_fWeight;
+			++lMatrixIndex;
+		}
+
+	}
+	pMatrix->m_plRowStarts[iRayCount] = lMatrixIndex;
+	
+	delete[] pEntries;
+	return pMatrix;
+}
+
+} // end namespace