summaryrefslogtreecommitdiffstats
path: root/cuda/2d/algo.h
blob: 2ce929cf47879f51500b7ad8f052087db62099a5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
/*
-----------------------------------------------------------------------
Copyright: 2010-2018, imec Vision Lab, University of Antwerp
           2014-2018, CWI, Amsterdam

Contact: astra@astra-toolbox.com
Website: http://www.astra-toolbox.com/

This file is part of the 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/>.

-----------------------------------------------------------------------
*/

#ifndef _CUDA_ALGO_H
#define _CUDA_ALGO_H

#include "astra/Globals.h"
#include "dims.h"

namespace astra {

class CParallelProjectionGeometry2D;
class CParallelVecProjectionGeometry2D;
class CFanFlatProjectionGeometry2D;
class CFanFlatVecProjectionGeometry2D;
class CVolumeGeometry2D;
class CProjectionGeometry2D;

}

namespace astraCUDA {

class _AstraExport ReconAlgo {
public:
	ReconAlgo();
	virtual ~ReconAlgo();

	bool setGPUIndex(int iGPUIndex);

	bool setGeometry(const astra::CVolumeGeometry2D* pVolGeom,
	                 const astra::CProjectionGeometry2D* pProjGeom);

	bool setSuperSampling(int raysPerDet, int raysPerPixelDim);

	void signalAbort() { shouldAbort = true; }

	virtual bool enableVolumeMask();
	virtual bool enableSinogramMask();

	// init should be called after setting all geometry
	virtual bool init() = 0;

	// setVolumeMask should be called after init and before iterate,
	// but only if enableVolumeMask was called before init.
	// It may be called again after iterate.
	bool setVolumeMask(float* D_maskData, unsigned int maskPitch);

	// setSinogramMask should be called after init and before iterate,
	// but only if enableSinogramMask was called before init.
	// It may be called again after iterate.
	bool setSinogramMask(float* D_smaskData, unsigned int smaskPitch);


	// setBuffers should be called after init and before iterate.
	// It may be called again after iterate.
	virtual bool setBuffers(float* D_volumeData, unsigned int volumePitch,
	                        float* D_projData, unsigned int projPitch);


	// instead of calling setBuffers, you can also call allocateBuffers
	// to let ReconAlgo manage its own GPU memory
	virtual bool allocateBuffers();

	// copy data to GPU. This must be called after allocateBuffers.
	// pfSinogram, pfReconstruction, pfVolMask, pfSinoMask are the
	// sinogram, reconstruction, volume mask and sinogram mask in system RAM,
	// respectively. The corresponding pitch variables give the pitches
	// of these buffers, measured in floats.
	// The sinogram is multiplied by fSinogramScale after uploading it.
	virtual bool copyDataToGPU(const float* pfSinogram, unsigned int iSinogramPitch, float fSinogramScale,
	                           const float* pfReconstruction, unsigned int iReconstructionPitch,
	                           const float* pfVolMask, unsigned int iVolMaskPitch,
	                           const float* pfSinoMask, unsigned int iSinoMaskPitch);



	// set Min/Max constraints. They may be called at any time, and will affect
	// any iterate() calls afterwards.
	virtual bool setMinConstraint(float fMin);
	virtual bool setMaxConstraint(float fMax);


	// iterate should be called after init and setBuffers.
	// It may be called multiple times.
	virtual bool iterate(unsigned int iterations) = 0;

	// Compute the norm of the difference of the FP of the current
	// reconstruction and the sinogram. (This performs one FP.)
	// It can be called after iterate.
	virtual float computeDiffNorm() = 0;
	// TODO: computeDiffNorm shouldn't be virtual, but for it to be
	// implemented in ReconAlgo, it needs a way to get a suitable
	// temporary sinogram buffer.

	bool getReconstruction(float* pfReconstruction,
                           unsigned int iReconstructionPitch) const;



protected:
	void reset();

	bool callFP(float* D_volumeData, unsigned int volumePitch,
	            float* D_projData, unsigned int projPitch,
	            float outputScale);
	bool callBP(float* D_volumeData, unsigned int volumePitch,
	            float* D_projData, unsigned int projPitch,
	            float outputScale);


	SDimensions dims;
	SParProjection* parProjs;
	SFanProjection* fanProjs;
	float fOutputScale;

	volatile bool shouldAbort;

	bool freeGPUMemory;

	// Input/output
	float* D_sinoData;
	unsigned int sinoPitch;

	float* D_volumeData;
	unsigned int volumePitch;

	// Masks
	bool useVolumeMask;
	bool useSinogramMask;

	float* D_maskData;
	unsigned int maskPitch;
	float* D_smaskData;
	unsigned int smaskPitch;

	// Min/max
	bool useMinConstraint;
	bool useMaxConstraint;
	float fMinConstraint;
	float fMaxConstraint;


};


}

#endif