Skip to content

Commit 8997761

Browse files
author
Tsing
committedApr 10, 2017
Start working GPU
1 parent 50b90c7 commit 8997761

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed
 
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
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**)&params_d_[CAMERA_MAT], sizeof(float)*9));
19+
gpuErrchk( cudaMalloc((void**)&params_d_[DISTOR_MAT], sizeof(float)*5));
20+
gpuErrchk( cudaMalloc((void**)&params_d_[ROT_MAT], sizeof(float)*9));
21+
gpuErrchk( cudaMalloc((void**)&params_d_[TRANS_MAT], sizeof(float)*3));
22+
gpuErrchk( cudaMalloc((void**)&camTransMat_d_, sizeof(float)*16));
23+
};
24+
void computeShadowsAndThresholds() override;
25+
26+
};
27+
}

0 commit comments

Comments
 (0)
Please sign in to comment.