# -----------------------------------------------------------------------
# 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)