# ----------------------------------------------------------------------- # Copyright: 2010-2016, iMinds-Vision Lab, University of Antwerp # 2013-2016, CWI, Amsterdam # # Contact: astra@uantwerpen.be # 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 . # # ----------------------------------------------------------------------- # # distutils: language = c++ # distutils: libraries = astra import six from .PyIncludes cimport * cimport utils from .utils import wrap_from_bytes cimport PyProjector2DFactory from .PyProjector2DFactory cimport CProjector2DFactory cimport PyProjector2DManager from .PyProjector2DManager cimport CProjector2DManager cimport PyXMLDocument from .PyXMLDocument cimport XMLDocument cimport PyMatrixManager from .PyMatrixManager cimport CMatrixManager cdef CProjector2DManager * manProj = PyProjector2DManager.getSingletonPtr() cdef CMatrixManager * manM = PyMatrixManager.getSingletonPtr() include "config.pxi" IF HAVE_CUDA: cdef extern from *: CCudaProjector2D* dynamic_cast_cuda_projector "dynamic_cast" (CProjector2D*) def create(config): cdef Config * cfg = utils.dictToConfig(six.b('Projector2D'), config) cdef CProjector2D * proj proj = PyProjector2DFactory.getSingletonPtr().create(cfg.self.getAttribute(six.b('type'))) if proj == NULL: del cfg raise Exception("Unknown Projector2D.") if not proj.initialize(cfg[0]): del cfg del proj raise Exception("Unable to initialize Projector2D.") del cfg return manProj.store(proj) def delete(ids): try: for i in ids: manProj.remove(i) except TypeError: manProj.remove(ids) def clear(): manProj.clear() def info(): six.print_(wrap_from_bytes(manProj.info())) cdef CProjector2D * getObject(i) except NULL: cdef CProjector2D * proj = manProj.get(i) if proj == NULL: raise Exception("Projector not initialized.") if not proj.isInitialized(): raise Exception("Projector not initialized.") return proj def projection_geometry(i): cdef CProjector2D * proj = getObject(i) cdef Config * cfg = proj.getProjectionGeometry().getConfiguration() dct = utils.configToDict(cfg) del cfg return dct def volume_geometry(i): cdef CProjector2D * proj = getObject(i) cdef Config * cfg = proj.getVolumeGeometry().getConfiguration() dct = utils.configToDict(cfg) del cfg return dct def weights_single_ray(i, projection_index, detector_index): raise Exception("Not yet implemented") def weights_projection(i, projection_index): raise Exception("Not yet implemented") def splat(i, row, col): raise Exception("Not yet implemented") def is_cuda(i): cdef CProjector2D * proj = getObject(i) IF HAVE_CUDA==True: cdef CCudaProjector2D * cudaproj = NULL cudaproj = dynamic_cast_cuda_projector(proj) if cudaproj==NULL: return False else: return True ELSE: return False def matrix(i): cdef CProjector2D * proj = getObject(i) cdef CSparseMatrix * mat = proj.getMatrix() if mat == NULL: del mat raise Exception("Data object not found") if not mat.isInitialized(): del mat raise Exception("Data object not initialized properly.") return manM.store(mat)