initial commit

2026-09-17 23:39:08 +09:00 · 2026-08-07 15:56:42 +09:00
commit 91ad25aca9
1012 changed files with 159314 additions and 0 deletions
@@ -0,0 +1,121 @@
+/*
+-----------------------------------------------------------------------
+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