nuclear@0: #include <stdio.h>
nuclear@2: #include <string.h>
nuclear@2: #include <math.h>
nuclear@2: #include <errno.h>
nuclear@0: #include <assert.h>
nuclear@0: #include "ocl.h"
nuclear@0: 
nuclear@2: struct RendInfo {
nuclear@2: 	int xsz, ysz;
nuclear@2: 	int num_sph;
nuclear@2: 	int max_iter;
nuclear@2: } __attribute__((packed));
nuclear@2: 
nuclear@1: struct Sphere {
nuclear@1: 	cl_float4 pos;
nuclear@1: 	cl_float radius;
nuclear@1: 
nuclear@1: 	cl_float4 color;
nuclear@1: } __attribute__((packed));
nuclear@1: 
nuclear@1: struct Ray {
nuclear@1: 	cl_float4 origin, dir;
nuclear@1: } __attribute__((packed));
nuclear@1: 
nuclear@1: 
nuclear@2: Ray get_primary_ray(int x, int y, int w, int h, float vfov_deg);
nuclear@2: bool write_ppm(const char *fname, float *fb, int xsz, int ysz);
nuclear@2: 
nuclear@0: int main()
nuclear@0: {
nuclear@2: 	const int xsz = 800;
nuclear@2: 	const int ysz = 600;
nuclear@0: 
nuclear@2: 	Sphere sphlist[] = {
nuclear@2: 		{{0, 0, 10, 1}, 1.0, {1, 0, 0, 1}}
nuclear@2: 	};
nuclear@2: 	RendInfo rinf = {xsz, ysz, sizeof sphlist / sizeof *sphlist, 6};
nuclear@2: 	Ray *prim_rays = new Ray[xsz * ysz];
nuclear@2: 	float *fb = new float[xsz * ysz * 4];
nuclear@2: 
nuclear@2: 	/* calculate primary rays */
nuclear@2: 	for(int i=0; i<ysz; i++) {
nuclear@2: 		for(int j=0; j<xsz; j++) {
nuclear@2: 			prim_rays[i * xsz + j] = get_primary_ray(j, i, xsz, ysz, 45.0);
nuclear@2: 		}
nuclear@0: 	}
nuclear@0: 
nuclear@2: 	/* setup opencl */
nuclear@2: 	CLProgram prog("render");
nuclear@2: 	if(!prog.load("rt.cl")) {
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 
nuclear@2: 	prog.set_arg_buffer(0, ARG_WR, xsz * ysz * 4 * sizeof(float), fb);
nuclear@2: 	prog.set_arg_buffer(1, ARG_RD, sizeof rinf, &rinf);
nuclear@2: 	prog.set_arg_buffer(2, ARG_RD, sizeof sphlist, sphlist);
nuclear@2: 	prog.set_arg_buffer(3, ARG_RD, xsz * ysz * sizeof *prim_rays, prim_rays);
nuclear@2: 
nuclear@2: 	if(!prog.run(1, xsz * ysz)) {
nuclear@0: 		return 1;
nuclear@0: 	}
nuclear@0: 
nuclear@2: 	CLMemBuffer *mbuf = prog.get_arg_buffer(0);
nuclear@0: 	map_mem_buffer(mbuf, MAP_RD);
nuclear@2: 	if(!write_ppm("out.ppm", fb, xsz, ysz)) {
nuclear@2: 		return 1;
nuclear@2: 	}
nuclear@2: 	unmap_mem_buffer(mbuf);
nuclear@0: 
nuclear@2: 	delete [] fb;
nuclear@2: 	delete [] prim_rays;
nuclear@0: 
nuclear@0: 	return 0;
nuclear@0: }
nuclear@2: 
nuclear@2: Ray get_primary_ray(int x, int y, int w, int h, float vfov_deg)
nuclear@2: {
nuclear@2: 	float vfov = M_PI * vfov_deg / 180.0;
nuclear@2: 	float aspect = (float)w / (float)h;
nuclear@2: 
nuclear@2: 	float ysz = 2.0;
nuclear@2: 	float xsz = aspect * ysz;
nuclear@2: 
nuclear@2: 	float px = ((float)x / (float)w) * xsz - xsz / 2.0;
nuclear@2: 	float py = 1.0 - ((float)y / (float)h) * ysz;
nuclear@2: 	float pz = 1.0 / tan(0.5 * vfov);
nuclear@2: 
nuclear@2: 	pz *= 1000.0;
nuclear@2: 
nuclear@2: 	Ray ray = {{0, 0, 0, 1}, {px, py, pz, 1}};
nuclear@2: 	return ray;
nuclear@2: }
nuclear@2: 
nuclear@2: bool write_ppm(const char *fname, float *fb, int xsz, int ysz)
nuclear@2: {
nuclear@2: 	FILE *fp;
nuclear@2: 
nuclear@2: 	if(!(fp = fopen(fname, "wb"))) {
nuclear@2: 		fprintf(stderr, "write_ppm: failed to open file %s for writing: %s\n", fname, strerror(errno));
nuclear@2: 		return false;
nuclear@2: 	}
nuclear@2: 	fprintf(fp, "P6\n%d %d\n255\n", xsz, ysz);
nuclear@2: 
nuclear@2: 	for(int i=0; i<xsz * ysz * 4; i++) {
nuclear@2: 		if(i % 4 == 3) continue;
nuclear@2: 
nuclear@2: 		unsigned char c = (unsigned char)(fb[i] * 255.0);
nuclear@2: 		fputc(c, fp);
nuclear@2: 	}
nuclear@2: 	fclose(fp);
nuclear@2: 	return true;
nuclear@2: }