mirror of
https://github.com/c-sooyoung/fold_slice.git
synced 2026-09-17 22:49:08 +09:00
122 lines
4.3 KiB
C++
122 lines
4.3 KiB
C++
/*
|
|
-----------------------------------------------------------------------
|
|
Copyright: 2010-2015, iMinds-Vision Lab, University of Antwerp
|
|
2014-2015, CWI, Amsterdam
|
|
|
|
Contact: astra@uantwerpen.be
|
|
Website: http://sf.net/projects/astra-toolbox
|
|
|
|
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/>.
|
|
|
|
-----------------------------------------------------------------------
|
|
$Id$
|
|
*/
|
|
|
|
|
|
#include <cuda.h>
|
|
#include <driver_types.h>
|
|
|
|
#ifdef _MSC_VER
|
|
|
|
#ifdef DLL_EXPORTS
|
|
#define _AstraExport __declspec(dllexport)
|
|
#define EXPIMP_TEMPLATE
|
|
#else
|
|
#define _AstraExport __declspec(dllimport)
|
|
#define EXPIMP_TEMPLATE extern
|
|
#endif
|
|
|
|
#else
|
|
|
|
#define _AstraExport
|
|
|
|
#endif
|
|
|
|
//#include "dims.h"
|
|
|
|
#ifndef M_PI
|
|
#define M_PI 3.14159265358979323846
|
|
#endif
|
|
|
|
#define ASTRA_CUDA_ASSERT(err) do { if (err != cudaSuccess) { astraCUDA3d::reportCudaError(err); assert(err == cudaSuccess); } } while(0)
|
|
|
|
|
|
#ifndef _CUDA_UTIL3D_H
|
|
#define _CUDA_UTIL3D_H
|
|
|
|
#include <cuda.h>
|
|
#include "dims3d.h"
|
|
|
|
#ifndef M_PI
|
|
#define M_PI 3.14159265358979323846
|
|
#endif
|
|
//#include "../2d/util.h"
|
|
|
|
|
|
|
|
|
|
namespace astraCUDA3d {
|
|
|
|
typedef texture<float, 3, cudaReadModeElementType> texture3D;
|
|
|
|
cudaPitchedPtr allocateVolumeData(const SDimensions3D& dims);
|
|
cudaPitchedPtr allocateProjectionData(const SDimensions3D& dims);
|
|
bool zeroVolumeData(cudaPitchedPtr& D_data, const SDimensions3D& dims);
|
|
bool zeroProjectionData(cudaPitchedPtr& D_data, const SDimensions3D& dims);
|
|
bool copyVolumeToDevice(const float* data, cudaPitchedPtr& D_data, const SDimensions3D& dims, unsigned int pitch = 0);
|
|
bool copyProjectionsToDevice(const float* data, cudaPitchedPtr& D_data, const SDimensions3D& dims, unsigned int pitch = 0);
|
|
bool copyVolumeFromDevice(float* data, const cudaPitchedPtr& D_data, const SDimensions3D& dims, unsigned int pitch = 0);
|
|
bool copyProjectionsFromDevice(float* data, const cudaPitchedPtr& D_data, const SDimensions3D& dims, unsigned int pitch = 0);
|
|
bool duplicateVolumeData(cudaPitchedPtr& D_dest, const cudaPitchedPtr& D_src, const SDimensions3D& dims);
|
|
bool duplicateProjectionData(cudaPitchedPtr& D_dest, const cudaPitchedPtr& D_src, const SDimensions3D& dims);
|
|
|
|
bool transferVolumeToArray_1D(cudaPitchedPtr D_volumeData, cudaArray* array, const SDimensions3D& dims);
|
|
|
|
bool transferProjectionsToArray(cudaPitchedPtr D_projData, cudaArray* array, const SDimensions3D& dims);
|
|
bool transferVolumeToArray(cudaPitchedPtr D_volumeData, cudaArray* array, const SDimensions3D& dims);
|
|
bool zeroProjectionArray(cudaArray* array, const SDimensions3D& dims);
|
|
bool zeroVolumeArray(cudaArray* array, const SDimensions3D& dims);
|
|
cudaArray* allocateProjectionArray(const SDimensions3D& dims);
|
|
cudaArray* allocateVolumeArray(const SDimensions3D& dims);
|
|
cudaArray* transferDeformationToArray(const mxGPUArray * m_img);
|
|
bool bindDataTexture(const cudaArray* array, texture3D & Texture, cudaTextureAddressMode bordermode, bool normalized);
|
|
|
|
//float dotProduct3D(cudaPitchedPtr data, unsigned int x, unsigned int y, unsigned int z);
|
|
|
|
int calcNextPowerOfTwo(int _iValue);
|
|
|
|
|
|
bool cudaTextForceKernelsCompletion();
|
|
void reportCudaError(cudaError_t err);
|
|
|
|
double toc(double tstart);
|
|
double tic();
|
|
int checkLastError(char * msg);
|
|
void printFreeMemory();
|
|
|
|
int dumpArray(char* filename, int width, int height, float *buffer);
|
|
int dumpCudaArray(cudaPitchedPtr projData, int syart, int end, char * filename);
|
|
int writeImage(char * fname, int w, int h, float * data);
|
|
int writeBMPImage(char * fname, int w, int h, float ** red, float ** green, float ** blue);
|
|
int writeImageCudaArray(cudaPitchedPtr Data, int start, int end, char * filename);
|
|
|
|
|
|
|
|
}
|
|
|
|
#endif
|