djairoh/cuda-based-raytrace
1
0
Fork 0
mirror of https://github.com/MartinOpat/cuda-based-raytrace.git synced 2025-06-07 02:13:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

added shader

Browse Source
This commit is contained in:
Djairo Hougee
2025-01-05 22:48:09 +01:00
parent 126ef4ace8
commit edf4d9fd60
17 changed files with 257 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/illumination/FrameBuffer.cu
View File

@@ -2,7 +2,9 @@
#include "linalg/linalg.h"
__host__ FrameBuffer::FrameBuffer(unsigned int w, unsigned int h) : w(w), h(h) {}
__host__ FrameBuffer::FrameBuffer(unsigned int w, unsigned int h) : w(w), h(h) {
this->buffer_size = w*h*4;
}
__device__ void FrameBuffer::writePixel(int x, int y, float r, float g, float b) {

2
src/illumination/FrameBuffer.h
View File

@@ -8,7 +8,7 @@
class FrameBuffer {
public:
uint32_t* buffer;
unsigned int* buffer;
std::size_t buffer_size;
unsigned int w;
unsigned int h;

20
src/illumination/Raycaster.cu
View File

@@ -10,6 +10,7 @@
#include "objs/sphere.h"
// TODO: instead of IMAGEWIDTH and IMAGEHEIGHT this should reflect the windowSize;
__global__ void raycastKernel(float* volumeData, FrameBuffer framebuffer) {
int px = blockIdx.x * blockDim.x + threadIdx.x;
int py = blockIdx.y * blockDim.y + threadIdx.y;
@@ -41,7 +42,7 @@ __global__ void raycastKernel(float* volumeData, FrameBuffer framebuffer) {
auto intersectAxis = [&](float start, float dirVal) {
if (fabsf(dirVal) < epsilon) {
if (start < 0.f || start > 1.f) {
tNear = 1e9f;
tNear = 1e9f;
tFar = -1e9f;
}
} else {
@@ -127,12 +128,13 @@ __global__ void raycastKernel(float* volumeData, FrameBuffer framebuffer) {
}
Raycaster::Raycaster(cudaGraphicsResource_t resources, int w, int h) {
Raycaster::Raycaster(cudaGraphicsResource_t resources, int w, int h, float* data) {
this->resources = resources;
this->w = h;
this->w = h;
this->fb = new FrameBuffer(w, h);
this->data = data;
// camera_info = CameraInfo(Vec3(0.0f, 0.0f, 0.0f), Vec3(0.0f, 0.0f, 0.0f), 90.0f, (float) w, (float) h);
// d_camera = thrust::device_new<Camera*>();
@@ -154,15 +156,14 @@ Raycaster::Raycaster(cudaGraphicsResource_t resources, int w, int h) {
void Raycaster::render() {
int res = cudaGraphicsMapresources(1, this->resources);
int res = cudaGraphicsMapResources(1, &this->resources);
if (res) {
std::cout << "CUDA error while mapping graphic resource: " << res;
cudaDeviceReset();
exit(1);
}
// check_cuda_errors(cudaGraphicsResourceGetMappedPointer((void**)&(frame_buffer->device_ptr), &(frame_buffer->buffer_size), resources));
res = cudaGraphicsResourceGetMappedPointer((void**)(this->fb->buffer), &this->fb->buffer_size, this->resources);
res = cudaGraphicsResourceGetMappedPointer((void**)&(this->fb->buffer), &(this->fb->buffer_size), resources);
if (res) {
std::cout << "CUDA error while fetching resource pointer: " << res;
cudaDeviceReset();
@@ -177,13 +178,12 @@ void Raycaster::render() {
dim3 threads(tx, ty);
// TODO: pass camera info at some point
// TODO: pass float volume data.
// frame buffer is implicitly copied to the device each frame
raycastKernel<<<blocks, threads>>> (nullptr, this->fb);
raycastKernel<<<blocks, threads>>> (this->data, *this->fb);
res = cudaGetLastError();
if (res) {
std::cout << "CUDA error while raycasting: " << res;
cudaError_t err = cudaGetLastError();
if (err) {
std::cout << "CUDA error while raycasting: " << cudaGetErrorString(err);
cudaDeviceReset();
exit(1);
}

26
src/illumination/Raycaster.h
View File

@@ -12,22 +12,22 @@ __global__ void raycastKernel(float* volumeData, unsigned char* framebuffer);
struct Raycaster {
// thrust::device_ptr<Camera*> d_camera;
// CameraInfo camera_info;
// thrust::device_ptr<Camera*> d_camera;
// CameraInfo camera_info;
cudaGraphicsResource_t resources;
FrameBuffer* fb;
cudaGraphicsResource_t resources;
FrameBuffer* fb;
float* data;
int w;
int h;
int w;
int h;
Raycaster() {};
Raycaster(cudaGraphicsResource_t resources, int nx, int ny);
// ~Raycaster();
Raycaster(cudaGraphicsResource_t resources, int nx, int ny, float* data);
// ~Raycaster();
void set_camera(Vec3 position, Vec3 forward, Vec3 up);
void render();
void resize(int nx, int ny);
// void raycastKernel(float* volumeData, unsigned char* framebuffer); // TODO: proper framebuffer class
void set_camera(Vec3 position, Vec3 forward, Vec3 up);
void render();
void resize(int nx, int ny);
};
#endif // RAYCASTER_H