TransformMatrix< T > Class Template Reference

#include <cuda_xmipp_utils.h>

Public Member Functions
	TransformMatrix ()
	TransformMatrix (myStreamHandle &myStream, size_t _Ndim, size_t _Xdim=3, size_t _Ydim=3, size_t _Zdim=1)
template<typename T1 >
void	resize (const TransformMatrix< T1 > &array, myStreamHandle &myStream)
void	resize (myStreamHandle &myStream, size_t _Ndim, size_t _Xdim=3, size_t _Ydim=3, size_t _Zdim=1)
bool	isEmpty ()
void	clear ()
	~TransformMatrix ()
void	initialize (myStreamHandle &myStream)
void	setTranslation (float *posX, float *posY, float *d_out_max, myStreamHandle &myStream)
void	setRotation (float *ang, myStreamHandle &myStream)
void	copyMatrix (TransformMatrix< float > &lastMatrix, myStreamHandle &myStream)
void	copyMatrixToCpu (myStreamHandle &myStream)
void	copyOneMatrixToCpu (float *&matrixCpu, int idxCpu, int idxGpu, myStreamHandle &myStream)

Public Attributes
size_t	Xdim
size_t	Ydim
size_t	Zdim
size_t	Ndim
size_t	yxdim
size_t	zyxdim
size_t	nzyxdim
T *	d_data
T *	h_data

Detailed Description

template<typename T>
class TransformMatrix< T >

Definition at line 96 of file cuda_xmipp_utils.h.

Constructor & Destructor Documentation

TransformMatrix() [1/2]

template<typename T>

TransformMatrix< T >::TransformMatrix ( )

inline

Definition at line 103 of file cuda_xmipp_utils.h.

    {
        Xdim=Ydim=Zdim=Ndim=yxdim=zyxdim=nzyxdim=0;
        d_data=nullptr;
        h_data=nullptr;
    }

TransformMatrix() [2/2]

template<typename T>

TransformMatrix< T >::TransformMatrix ( myStreamHandle &  myStream, size_t  _Ndim, size_t  _Xdim = 3, size_t  _Ydim = 3, size_t  _Zdim = 1  )

inline

Definition at line 110 of file cuda_xmipp_utils.h.

    {
        Xdim=Ydim=Zdim=Ndim=yxdim=zyxdim=nzyxdim=0;
        d_data=NULL;
        h_data=NULL;
        resize(myStream,_Ndim, _Xdim, _Ydim, _Zdim);
    }

~TransformMatrix()

template<typename T>

TransformMatrix< T >::~TransformMatrix ( )

inline

Definition at line 134 of file cuda_xmipp_utils.h.

    {
        clear();
    }

Member Function Documentation

clear()

template<typename T >

template void TransformMatrix< T >::clear

(

)

Definition at line 183 of file cuda_xmipp_utils.cpp.

{
    if (d_data!=NULL)
        gpuFree((void*) d_data);
    if (h_data!=NULL)
        gpuErrchk(cudaFreeHost(h_data));
    Xdim=Ydim=Zdim=Ndim=yxdim=zyxdim=nzyxdim=0;
    d_data=NULL;
    h_data=NULL;
}

copyMatrix()

template<typename T>

void TransformMatrix< T >::copyMatrix ( TransformMatrix< float > &  lastMatrix, myStreamHandle &  myStream  )

inline

Definition at line 158 of file cuda_xmipp_utils.h.

    {
        if (lastMatrix.isEmpty())
            lastMatrix.resize(myStream, Ndim, 3, 3, 1);

        gpuCopyFromGPUToGPUStream(d_data, lastMatrix.d_data, nzyxdim*sizeof(float), myStream);
    }

copyMatrixToCpu()

template<typename T>

void TransformMatrix< T >::copyMatrixToCpu ( myStreamHandle &  myStream )

inline

Definition at line 166 of file cuda_xmipp_utils.h.

    {
        gpuCopyFromGPUToCPUStream(d_data, h_data, nzyxdim*sizeof(float), myStream);
    }

copyOneMatrixToCpu()

template<typename T>

void TransformMatrix< T >::copyOneMatrixToCpu ( float *&  matrixCpu, int  idxCpu, int  idxGpu, myStreamHandle &  myStream  )

inline

Definition at line 171 of file cuda_xmipp_utils.h.

    {
        gpuCopyFromGPUToCPUStream(&d_data[9*idxGpu], &matrixCpu[9*idxCpu], 9*sizeof(float), myStream);
    }

initialize()

template<typename T>

void TransformMatrix< T >::initialize ( myStreamHandle &  myStream )

inline

Definition at line 139 of file cuda_xmipp_utils.h.

    {
        initializeIdentity(d_data, h_data, Ndim, myStream);
    }

isEmpty()

template<typename T>

bool TransformMatrix< T >::isEmpty ( )

inline

Definition at line 127 of file cuda_xmipp_utils.h.

    {
        return d_data==NULL && h_data==NULL;
    }

resize() [1/2]

template<typename T>

template<typename T1 >

void TransformMatrix< T >::resize ( const TransformMatrix< T1 > &  array, myStreamHandle &  myStream  )

inline

Definition at line 119 of file cuda_xmipp_utils.h.

    {

        resize(myStream, array.Ndim, array.Xdim, array.Ydim, array.Zdim);
    }

resize() [2/2]

template<typename T >

template void TransformMatrix< T >::resize	(	myStreamHandle &	myStream,
		size_t	_Ndim,
		size_t	_Xdim = 3,
		size_t	_Ydim = 3,
		size_t	_Zdim = 1
	)

Definition at line 21 of file cuda_xmipp_utils.cpp.

                                                                {
    if (_Xdim*_Ydim*_Zdim*_Ndim==nzyxdim)
        return;

    clear();

    Xdim=_Xdim;
    Ydim=_Ydim;
    Zdim=_Zdim;
    Ndim=_Ndim;
    yxdim=_Ydim*_Xdim;
    zyxdim=yxdim*_Zdim;
    nzyxdim=zyxdim*_Ndim;
    gpuErrchk(cudaMalloc(&d_data,nzyxdim*sizeof(T)));
    gpuErrchk(cudaMallocHost(&h_data,nzyxdim*sizeof(T)));
    initializeIdentity(d_data, h_data, Ndim, myStream);
}

setRotation()

template<typename T>

void TransformMatrix< T >::setRotation ( float *  ang, myStreamHandle &  myStream  )

inline

Definition at line 151 of file cuda_xmipp_utils.h.

    {
        /*for(int i=0; i<Ndim; i++)
            setRotationMatrix(d_data, -ang[i], i);*/
        setRotationMatrix(d_data, ang, Ndim, myStream);
    }

setTranslation()

template<typename T>

void TransformMatrix< T >::setTranslation ( float *  posX, float *  posY, float *  d_out_max, myStreamHandle &  myStream  )

inline

Definition at line 144 of file cuda_xmipp_utils.h.

    {
        /*for(int i=0; i<Ndim; i++)
            setTranslationMatrix(d_data, -posX[i], -posY[i], i);*/
        setTranslationMatrix(d_data, posX, posY, Ndim, myStream);
    }

Member Data Documentation

d_data

template<typename T>

T* TransformMatrix< T >::d_data

Definition at line 100 of file cuda_xmipp_utils.h.

h_data

template<typename T>

T* TransformMatrix< T >::h_data

Definition at line 101 of file cuda_xmipp_utils.h.

Ndim

template<typename T>

size_t TransformMatrix< T >::Ndim

Definition at line 99 of file cuda_xmipp_utils.h.

nzyxdim

template<typename T>

size_t TransformMatrix< T >::nzyxdim

Definition at line 99 of file cuda_xmipp_utils.h.

Xdim

template<typename T>

size_t TransformMatrix< T >::Xdim

Definition at line 99 of file cuda_xmipp_utils.h.

Ydim

template<typename T>

size_t TransformMatrix< T >::Ydim

Definition at line 99 of file cuda_xmipp_utils.h.

yxdim

template<typename T>

size_t TransformMatrix< T >::yxdim

Definition at line 99 of file cuda_xmipp_utils.h.

Zdim

template<typename T>

size_t TransformMatrix< T >::Zdim

Definition at line 99 of file cuda_xmipp_utils.h.

zyxdim

template<typename T>

size_t TransformMatrix< T >::zyxdim

Definition at line 99 of file cuda_xmipp_utils.h.

---

The documentation for this class was generated from the following files:

• xmipp/libraries/reconstruction_cuda/cuda_xmipp_utils.h
• xmipp/libraries/reconstruction_cuda/cuda_xmipp_utils.cpp