summaryrefslogtreecommitdiffstats
path: root/cuda/3d/algo3d.cu
diff options
context:
space:
mode:
Diffstat (limited to 'cuda/3d/algo3d.cu')
-rw-r--r--cuda/3d/algo3d.cu108
1 files changed, 108 insertions, 0 deletions
diff --git a/cuda/3d/algo3d.cu b/cuda/3d/algo3d.cu
new file mode 100644
index 0000000..20e7381
--- /dev/null
+++ b/cuda/3d/algo3d.cu
@@ -0,0 +1,108 @@
+/*
+-----------------------------------------------------------------------
+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 <cassert>
+
+#include "algo3d.h"
+#include "cone_fp.h"
+#include "cone_bp.h"
+#include "par3d_fp.h"
+#include "par3d_bp.h"
+
+namespace astraCUDA3d {
+
+ReconAlgo3D::ReconAlgo3D()
+{
+ coneProjs = 0;
+ par3DProjs = 0;
+ shouldAbort = false;
+}
+
+ReconAlgo3D::~ReconAlgo3D()
+{
+ reset();
+}
+
+void ReconAlgo3D::reset()
+{
+ delete[] coneProjs;
+ coneProjs = 0;
+ delete[] par3DProjs;
+ par3DProjs = 0;
+ shouldAbort = false;
+}
+
+bool ReconAlgo3D::setConeGeometry(const SDimensions3D& _dims, const SConeProjection* _angles)
+{
+ dims = _dims;
+
+ coneProjs = new SConeProjection[dims.iProjAngles];
+ par3DProjs = 0;
+
+ memcpy(coneProjs, _angles, sizeof(coneProjs[0]) * dims.iProjAngles);
+
+ return true;
+}
+
+bool ReconAlgo3D::setPar3DGeometry(const SDimensions3D& _dims, const SPar3DProjection* _angles)
+{
+ dims = _dims;
+
+ par3DProjs = new SPar3DProjection[dims.iProjAngles];
+ coneProjs = 0;
+
+ memcpy(par3DProjs, _angles, sizeof(par3DProjs[0]) * dims.iProjAngles);
+
+ return true;
+}
+
+
+bool ReconAlgo3D::callFP(cudaPitchedPtr& D_volumeData,
+ cudaPitchedPtr& D_projData,
+ float outputScale)
+{
+ if (coneProjs) {
+ return ConeFP(D_volumeData, D_projData, dims, coneProjs, outputScale);
+ } else {
+ return Par3DFP(D_volumeData, D_projData, dims, par3DProjs, outputScale);
+ }
+}
+
+bool ReconAlgo3D::callBP(cudaPitchedPtr& D_volumeData,
+ cudaPitchedPtr& D_projData)
+{
+ if (coneProjs) {
+ return ConeBP(D_volumeData, D_projData, dims, coneProjs);
+ } else {
+ return Par3DBP(D_volumeData, D_projData, dims, par3DProjs);
+ }
+}
+
+
+
+}