summaryrefslogtreecommitdiffstats
path: root/python/astra/PyIncludes.pxd
blob: cca99f86230e4f197beaf8adb715a2dd3d1100f2 (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
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
# -----------------------------------------------------------------------
# Copyright: 2010-2021, imec Vision Lab, University of Antwerp
#            2013-2021, 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/>.
#
# -----------------------------------------------------------------------

from libcpp cimport bool
from libcpp.string cimport string
from .PyXMLDocument cimport XMLNode

include "config.pxi"

cdef extern from "astra/Globals.h" namespace "astra":
    ctypedef float float32
    ctypedef double float64
    ctypedef unsigned short int uint16
    ctypedef signed short int sint16
    ctypedef unsigned char uchar8
    ctypedef signed char schar8
    ctypedef int int32
    ctypedef short int int16

cdef extern from "astra/Config.h" namespace "astra":
    cdef cppclass Config:
        Config()
        void initialize(string rootname)
        XMLNode self

cdef extern from "astra/VolumeGeometry2D.h" namespace "astra":
    cdef cppclass CVolumeGeometry2D:
        bool initialize(Config)
        int getGridColCount()
        int getGridRowCount()
        int getGridTotCount()
        float32 getWindowLengthX()
        float32 getWindowLengthY()
        float32 getWindowArea()
        float32 getPixelLengthX()
        float32 getPixelLengthY()
        float32 getPixelArea()
        float32 getWindowMinX()
        float32 getWindowMinY()
        float32 getWindowMaxX()
        float32 getWindowMaxY()
        Config* getConfiguration()
        bool isEqual(CVolumeGeometry2D*)

cdef extern from "astra/Float32Data2D.h" namespace "astra":
    cdef cppclass CFloat32CustomMemory:
        pass

cdef extern from "astra/Float32VolumeData2D.h" namespace "astra":
    cdef cppclass CFloat32VolumeData2D:
        CFloat32VolumeData2D(CVolumeGeometry2D*)
        CFloat32VolumeData2D(CVolumeGeometry2D*, CFloat32CustomMemory*)
        CVolumeGeometry2D * getGeometry()
        int getWidth()
        int getHeight()
        void changeGeometry(CVolumeGeometry2D*)
        Config* getConfiguration()



cdef extern from "astra/ProjectionGeometry2D.h" namespace "astra":
    cdef cppclass CProjectionGeometry2D:
        CProjectionGeometry2D()
        bool initialize(Config)
        int getDetectorCount()
        int getProjectionAngleCount()
        bool isOfType(string)
        float32 getProjectionAngle(int)
        float32 getDetectorWidth()
        Config* getConfiguration()
        bool isEqual(CProjectionGeometry2D*)

cdef extern from "astra/Float32Data2D.h" namespace "astra::CFloat32Data2D":
    cdef enum TWOEDataType "astra::CFloat32Data2D::EDataType":
        TWOPROJECTION "astra::CFloat32Data2D::PROJECTION"
        TWOVOLUME "astra::CFloat32Data2D::VOLUME"

cdef extern from "astra/Float32Data3D.h" namespace "astra::CFloat32Data3D":
    cdef enum THREEEDataType "astra::CFloat32Data3D::EDataType":
        THREEPROJECTION "astra::CFloat32Data3D::PROJECTION"
        THREEVOLUME "astra::CFloat32Data3D::VOLUME"


cdef extern from "astra/Float32Data2D.h" namespace "astra":
    cdef cppclass CFloat32Data2D:
        bool isInitialized()
        int getSize()
        float32 *getData()
        float32 **getData2D()
        int getWidth()
        int getHeight()
        TWOEDataType getType()




cdef extern from "astra/SparseMatrixProjectionGeometry2D.h" namespace "astra":
    cdef cppclass CSparseMatrixProjectionGeometry2D:
        CSparseMatrixProjectionGeometry2D()

cdef extern from "astra/FanFlatProjectionGeometry2D.h" namespace "astra":
    cdef cppclass CFanFlatProjectionGeometry2D:
        CFanFlatProjectionGeometry2D()

cdef extern from "astra/FanFlatVecProjectionGeometry2D.h" namespace "astra":
    cdef cppclass CFanFlatVecProjectionGeometry2D:
        CFanFlatVecProjectionGeometry2D()

cdef extern from "astra/ParallelVecProjectionGeometry2D.h" namespace "astra":
    cdef cppclass CParallelVecProjectionGeometry2D:
        CParallelVecProjectionGeometry2D()

cdef extern from "astra/ParallelProjectionGeometry2D.h" namespace "astra":
    cdef cppclass CParallelProjectionGeometry2D:
        CParallelProjectionGeometry2D()


cdef extern from "astra/Float32ProjectionData2D.h" namespace "astra":
    cdef cppclass CFloat32ProjectionData2D:
        CFloat32ProjectionData2D(CProjectionGeometry2D*)
        CFloat32ProjectionData2D(CProjectionGeometry2D*, CFloat32CustomMemory*)
        CProjectionGeometry2D * getGeometry()
        void changeGeometry(CProjectionGeometry2D*)
        int getDetectorCount()
        int getAngleCount()

cdef extern from "astra/Algorithm.h" namespace "astra":
    cdef cppclass CAlgorithm:
        bool initialize(Config)
        void run(int) nogil
        bool isInitialized()

cdef extern from "astra/ReconstructionAlgorithm2D.h" namespace "astra":
    cdef cppclass CReconstructionAlgorithm2D:
        bool getResidualNorm(float32&)

cdef extern from "astra/ReconstructionAlgorithm3D.h" namespace "astra":
    cdef cppclass CReconstructionAlgorithm3D:
        bool getResidualNorm(float32&)

cdef extern from "astra/Projector2D.h" namespace "astra":
    cdef cppclass CProjector2D:
        bool isInitialized()
        bool initialize(Config)
        CProjectionGeometry2D* getProjectionGeometry()
        CVolumeGeometry2D* getVolumeGeometry()
        CSparseMatrix* getMatrix()

cdef extern from "astra/Projector3D.h" namespace "astra":
    cdef cppclass CProjector3D:
        bool isInitialized()
        bool initialize(Config)
        CProjectionGeometry3D* getProjectionGeometry()
        CVolumeGeometry3D* getVolumeGeometry()

IF HAVE_CUDA==True:
    cdef extern from "astra/CudaProjector3D.h" namespace "astra":
        cdef cppclass CCudaProjector3D

    cdef extern from "astra/CudaProjector2D.h" namespace "astra":
        cdef cppclass CCudaProjector2D

    cdef extern from "astra/Float32Data3DGPU.h" namespace "astraCUDA3d":
        cdef cppclass MemHandle3D:
            pass

    cdef extern from "astra/Float32Data3DGPU.h" namespace "astraCUDA3d":
        cdef MemHandle3D wrapHandle(float *D_ptr, unsigned int x, unsigned int y, unsigned int z, unsigned int pitch)


cdef extern from "astra/SparseMatrix.h" namespace "astra":
    cdef cppclass CSparseMatrix:
        CSparseMatrix(unsigned int,unsigned int,unsigned long)
        unsigned int m_iWidth
        unsigned int m_iHeight
        unsigned long m_lSize
        bool isInitialized()
        float32* m_pfValues
        unsigned int* m_piColIndices
        unsigned long* m_plRowStarts

cdef extern from "astra/Float32Data3D.h" namespace "astra":
    cdef cppclass CFloat32Data3D:
        bool isInitialized()
        int getSize()
        int getWidth()
        int getHeight()
        int getDepth()


cdef extern from "astra/Float32Data3DMemory.h" namespace "astra":
    cdef cppclass CFloat32Data3DMemory(CFloat32Data3D):
        CFloat32Data3DMemory()
        void updateStatistics()
        float32 *getData()
        THREEEDataType getType()


cdef extern from "astra/VolumeGeometry3D.h" namespace "astra":
    cdef cppclass CVolumeGeometry3D:
        CVolumeGeometry3D()
        bool initialize(Config)
        Config * getConfiguration()
        int getGridColCount()
        int getGridRowCount()
        int getGridSliceCount()

cdef extern from "astra/ProjectionGeometry3D.h" namespace "astra":
    cdef cppclass CProjectionGeometry3D:
        CProjectionGeometry3D()
        bool initialize(Config)
        Config * getConfiguration()
        int getProjectionCount()
        int getDetectorColCount()
        int getDetectorRowCount()

cdef extern from "astra/Float32VolumeData3D.h" namespace "astra":
    cdef cppclass CFloat32VolumeData3D(CFloat32Data3D):
        pass

cdef extern from "astra/Float32ProjectionData3D.h" namespace "astra":
    cdef cppclass CFloat32ProjectionData3D(CFloat32Data3D):
        pass


cdef extern from "astra/Float32VolumeData3DMemory.h" namespace "astra":
    cdef cppclass CFloat32VolumeData3DMemory(CFloat32VolumeData3D):
        CFloat32VolumeData3DMemory(CVolumeGeometry3D*)
        CFloat32VolumeData3DMemory(CVolumeGeometry3D*, CFloat32CustomMemory*)
        CVolumeGeometry3D* getGeometry()
        void changeGeometry(CVolumeGeometry3D*)
        int getRowCount()
        int getColCount()
        int getSliceCount()
        bool isInitialized()

cdef extern from "astra/ParallelProjectionGeometry3D.h" namespace "astra":
    cdef cppclass CParallelProjectionGeometry3D:
        CParallelProjectionGeometry3D()

cdef extern from "astra/ParallelVecProjectionGeometry3D.h" namespace "astra":
    cdef cppclass CParallelVecProjectionGeometry3D:
        CParallelVecProjectionGeometry3D()

cdef extern from "astra/ConeProjectionGeometry3D.h" namespace "astra":
    cdef cppclass CConeProjectionGeometry3D:
        CConeProjectionGeometry3D()
        bool initialize(Config)

cdef extern from "astra/ConeVecProjectionGeometry3D.h" namespace "astra":
    cdef cppclass CConeVecProjectionGeometry3D:
        CConeVecProjectionGeometry3D()

cdef extern from "astra/Float32ProjectionData3DMemory.h" namespace "astra":
    cdef cppclass CFloat32ProjectionData3DMemory(CFloat32ProjectionData3D):
        CFloat32ProjectionData3DMemory(CProjectionGeometry3D*)
        CFloat32ProjectionData3DMemory(CConeProjectionGeometry3D*)
        CFloat32ProjectionData3DMemory(CProjectionGeometry3D*, CFloat32CustomMemory*)
        CFloat32ProjectionData3DMemory(CConeProjectionGeometry3D*, CFloat32CustomMemory*)
        CProjectionGeometry3D* getGeometry()
        void changeGeometry(CProjectionGeometry3D*)
        int getDetectorColCount()
        int getDetectorRowCount()
        int getAngleCount()
        bool isInitialized()

IF HAVE_CUDA==True:
    cdef extern from "astra/Float32VolumeData3DGPU.h" namespace "astra":
        cdef cppclass CFloat32VolumeData3DGPU(CFloat32VolumeData3D):
            CFloat32VolumeData3DGPU(CVolumeGeometry3D*, MemHandle3D)
            CVolumeGeometry3D* getGeometry()
            void changeGeometry(CVolumeGeometry3D*)
            int getRowCount()
            int getColCount()
            int getSliceCount()
            bool isInitialized()

    cdef extern from "astra/Float32ProjectionData3DGPU.h" namespace "astra":
        cdef cppclass CFloat32ProjectionData3DGPU(CFloat32ProjectionData3D):
            CFloat32ProjectionData3DGPU(CProjectionGeometry3D*, MemHandle3D)
            CProjectionGeometry3D* getGeometry()
            void changeGeometry(CProjectionGeometry3D*)
            int getRowCount()
            int getColCount()
            int getSliceCount()
            bool isInitialized()