From b2fc6c70434674d74551c3a6c01ffb3233499312 Mon Sep 17 00:00:00 2001 From: Willem Jan Palenstijn Date: Mon, 1 Jul 2013 22:34:11 +0000 Subject: Update version to 1.3 --- src/CudaSartAlgorithm.cpp | 136 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 136 insertions(+) create mode 100644 src/CudaSartAlgorithm.cpp (limited to 'src/CudaSartAlgorithm.cpp') diff --git a/src/CudaSartAlgorithm.cpp b/src/CudaSartAlgorithm.cpp new file mode 100644 index 0000000..610793f --- /dev/null +++ b/src/CudaSartAlgorithm.cpp @@ -0,0 +1,136 @@ +/* +----------------------------------------------------------------------- +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 . + +----------------------------------------------------------------------- +$Id$ +*/ + +#ifdef ASTRA_CUDA + +#include "astra/CudaSartAlgorithm.h" + +#include "../cuda/2d/sart.h" + +using namespace std; + +namespace astra { + +// type of the algorithm, needed to register with CAlgorithmFactory +std::string CCudaSartAlgorithm::type = "SART_CUDA"; + +//---------------------------------------------------------------------------------------- +// Constructor +CCudaSartAlgorithm::CCudaSartAlgorithm() +{ + m_bIsInitialized = false; + CCudaReconstructionAlgorithm2D::_clear(); +} + +//---------------------------------------------------------------------------------------- +// Destructor +CCudaSartAlgorithm::~CCudaSartAlgorithm() +{ + // The actual work is done by ~CCudaReconstructionAlgorithm2D +} + +//--------------------------------------------------------------------------------------- +// Initialize - Config +bool CCudaSartAlgorithm::initialize(const Config& _cfg) +{ + ASTRA_ASSERT(_cfg.self); + ConfigStackCheck CC("CudaSartAlgorithm", this, _cfg); + + m_bIsInitialized = CCudaReconstructionAlgorithm2D::initialize(_cfg); + + if (!m_bIsInitialized) + return false; + + astraCUDA::SART *sart = new astraCUDA::SART(); + + m_pAlgo = sart; + m_bAlgoInit = false; + + // projection order + int projectionCount = m_pSinogram->getGeometry()->getProjectionAngleCount(); + int* projectionOrder = NULL; + string projOrder = _cfg.self->getOption("ProjectionOrder", "random"); + CC.markOptionParsed("ProjectionOrder"); + if (projOrder == "sequential") { + projectionOrder = new int[projectionCount]; + for (int i = 0; i < projectionCount; i++) { + projectionOrder[i] = i; + } + sart->setProjectionOrder(projectionOrder, projectionCount); + delete[] projectionOrder; + } else if (projOrder == "random") { + projectionOrder = new int[projectionCount]; + for (int i = 0; i < projectionCount; i++) { + projectionOrder[i] = i; + } + for (int i = 0; i < projectionCount-1; i++) { + int k = (rand() % (projectionCount - i)); + int t = projectionOrder[i]; + projectionOrder[i] = projectionOrder[i + k]; + projectionOrder[i + k] = t; + } + sart->setProjectionOrder(projectionOrder, projectionCount); + delete[] projectionOrder; + } else if (projOrder == "custom") { + vector projOrderList = _cfg.self->getOptionNumericalArray("ProjectionOrderList"); + projectionOrder = new int[projOrderList.size()]; + for (int i = 0; i < projOrderList.size(); i++) { + projectionOrder[i] = static_cast(projOrderList[i]); + } + sart->setProjectionOrder(projectionOrder, projectionCount); + delete[] projectionOrder; + CC.markOptionParsed("ProjectionOrderList"); + } + + + + return true; +} + +//--------------------------------------------------------------------------------------- +// Initialize - C++ +bool CCudaSartAlgorithm::initialize(CProjector2D* _pProjector, + CFloat32ProjectionData2D* _pSinogram, + CFloat32VolumeData2D* _pReconstruction, + int _iGPUindex, int _iDetectorSuperSampling) +{ + m_bIsInitialized = CCudaReconstructionAlgorithm2D::initialize(_pProjector, _pSinogram, _pReconstruction, _iGPUindex, _iDetectorSuperSampling, 1); + + if (!m_bIsInitialized) + return false; + + m_pAlgo = new astraCUDA::SART(); + m_bAlgoInit = false; + + return true; +} + + +} // namespace astra + +#endif // ASTRA_CUDA -- cgit v1.2.3