summaryrefslogtreecommitdiffstats
path: root/src/Projector2D.cpp
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <WillemJan.Palenstijn@uantwerpen.be>2013-07-01 22:34:11 +0000
committerwpalenst <WillemJan.Palenstijn@uantwerpen.be>2013-07-01 22:34:11 +0000
commitb2fc6c70434674d74551c3a6c01ffb3233499312 (patch)
treeb17f080ebc504ab85ebb7c3d89f917fd87ce9e00 /src/Projector2D.cpp
downloadastra-b2fc6c70434674d74551c3a6c01ffb3233499312.tar.gz
astra-b2fc6c70434674d74551c3a6c01ffb3233499312.tar.bz2
astra-b2fc6c70434674d74551c3a6c01ffb3233499312.tar.xz
astra-b2fc6c70434674d74551c3a6c01ffb3233499312.zip
Update version to 1.3
Diffstat (limited to 'src/Projector2D.cpp')
-rw-r--r--src/Projector2D.cpp217
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