File tree 1 file changed +27
-0
lines changed
src/lib/ReconstructorCUDA
1 file changed +27
-0
lines changed Original file line number Diff line number Diff line change
1
+ #pragma once
2
+ #include < core/ImageFileProcessor.h>
3
+ #include " DynamicBits.cuh"
4
+ #include < string>
5
+
6
+ namespace SLS {
7
+ class ImageFileProcessorCUDA : public ImageFileProcessor {
8
+ protected:
9
+ Dynamic_Bitset_Array *maskGPU_; // Sorry can't init before reading the images
10
+
11
+ // Init configurations in device
12
+ std::array<float *, PARAM_COUNT> params_d_;
13
+ float * camTransMat_d_;
14
+ public:
15
+ ImageFileProcessorCUDA ()=delete ;
16
+ ImageFileProcessorCUDA (const std::string& cName):
17
+ ImageFileProcessor (cName), maskGPU_(nullptr ){
18
+ gpuErrchk ( cudaMalloc ((void **)¶ms_d_[CAMERA_MAT], sizeof (float )*9 ));
19
+ gpuErrchk ( cudaMalloc ((void **)¶ms_d_[DISTOR_MAT], sizeof (float )*5 ));
20
+ gpuErrchk ( cudaMalloc ((void **)¶ms_d_[ROT_MAT], sizeof (float )*9 ));
21
+ gpuErrchk ( cudaMalloc ((void **)¶ms_d_[TRANS_MAT], sizeof (float )*3 ));
22
+ gpuErrchk ( cudaMalloc ((void **)&camTransMat_d_, sizeof (float )*16 ));
23
+ };
24
+ void computeShadowsAndThresholds () override ;
25
+
26
+ };
27
+ }
You can’t perform that action at this time.
0 commit comments