erebus

annotate liberebus/src/erebus.cc @ 9:d38e13d6063c

find changesets by author, revision, files, or words in the commit message
foo
author John Tsiombikas <nuclear@member.fsf.org>
date Sat, 24 May 2014 17:22:53 +0300
parents e2d9bf168a41
children 506e114b7ca2
rev   line source
nuclear@2 1 #include <string.h>
nuclear@2 2 #include <limits.h>
nuclear@2 3 #include <chrono>
nuclear@2 4 #include <random>
nuclear@2 5 #include "erebus.h"
nuclear@2 6 #include "vmath/vector.h"
nuclear@2 7 #include "image.h"
nuclear@4 8 #include "scene.h"
nuclear@4 9 #include "geomobj.h"
nuclear@5 10 #include "rt.h"
nuclear@2 11
nuclear@8 12 #define INF_SAMPLES (INT_MAX / 2)
nuclear@8 13
nuclear@2 14 using namespace std::chrono;
nuclear@2 15
nuclear@2 16 struct Rect {
nuclear@2 17 int x, y, width, height;
nuclear@2 18
nuclear@2 19 bool operator ==(const Rect &r) { return memcmp(this, &r, sizeof r) == 0; }
nuclear@2 20 bool operator !=(const Rect &r) { return memcmp(this, &r, sizeof r) != 0; }
nuclear@2 21 };
nuclear@2 22
nuclear@2 23 #define INVALID_RECT Rect{0, 0, 0, 0}
nuclear@2 24
nuclear@2 25 struct erebus {
nuclear@4 26 Scene *scn;
nuclear@4 27
nuclear@2 28 Image<float> fbimg;
nuclear@8 29 Image<float> accum; // sample accumulator per pixel
nuclear@2 30 Vector4 options[ERB_NUM_OPTIONS];
nuclear@2 31
nuclear@2 32 // render state
nuclear@2 33 long cur_time;
nuclear@2 34 int cur_pixel_x, cur_pixel_y;
nuclear@2 35 Rect cur_rect;
nuclear@8 36 int cur_sample;
nuclear@9 37
nuclear@9 38 // interactive input
nuclear@9 39 std::vector<bool> keystate;
nuclear@9 40 std::vector<bool> bnstate;
nuclear@9 41 int mouse_pos[2];
nuclear@2 42 };
nuclear@2 43
nuclear@8 44 static void render_pixel(struct erebus *ctx, int x, int y, int sample);
nuclear@2 45
nuclear@2 46 static std::mt19937 rnd_gen;
nuclear@2 47
nuclear@2 48 extern "C" {
nuclear@2 49
nuclear@2 50 struct erebus *erb_init(void)
nuclear@2 51 {
nuclear@2 52 struct erebus *ctx;
nuclear@2 53 try {
nuclear@2 54 ctx = new struct erebus;
nuclear@2 55 }
nuclear@2 56 catch(...) {
nuclear@2 57 return 0;
nuclear@2 58 }
nuclear@2 59
nuclear@4 60 ctx->scn = 0;
nuclear@2 61 ctx->cur_time = 0;
nuclear@2 62 ctx->cur_rect = INVALID_RECT;
nuclear@8 63
nuclear@8 64 ctx->options[ERB_OPT_MAX_SAMPLES].x = (float)INF_SAMPLES;
nuclear@2 65 return ctx;
nuclear@2 66 }
nuclear@2 67
nuclear@2 68 void erb_destroy(struct erebus *ctx)
nuclear@2 69 {
nuclear@2 70 delete ctx;
nuclear@2 71 }
nuclear@2 72
nuclear@2 73 void erb_setopti(struct erebus *ctx, enum erb_option opt, int val)
nuclear@2 74 {
nuclear@2 75 ctx->options[opt].x = val;
nuclear@2 76 }
nuclear@2 77 void erb_setoptf(struct erebus *ctx, enum erb_option opt, float val)
nuclear@2 78 {
nuclear@2 79 ctx->options[opt].x = val;
nuclear@2 80 }
nuclear@2 81 void erb_setoptfv(struct erebus *ctx, enum erb_option opt, float *vec)
nuclear@2 82 {
nuclear@2 83 for(int i=0; i<4; i++) {
nuclear@2 84 ctx->options[opt][i] = vec[i];
nuclear@2 85 }
nuclear@2 86 }
nuclear@2 87
nuclear@2 88 int erb_getopti(struct erebus *ctx, enum erb_option opt)
nuclear@2 89 {
nuclear@2 90 return ctx->options[opt].x;
nuclear@2 91 }
nuclear@2 92 float erb_getoptf(struct erebus *ctx, enum erb_option opt)
nuclear@2 93 {
nuclear@2 94 return ctx->options[opt].x;
nuclear@2 95 }
nuclear@2 96 float *erb_getoptfv(struct erebus *ctx, enum erb_option opt)
nuclear@2 97 {
nuclear@2 98 return &ctx->options[opt].x;
nuclear@2 99 }
nuclear@2 100
nuclear@2 101 float *erb_get_framebuffer(struct erebus *ctx)
nuclear@2 102 {
nuclear@2 103 return ctx->fbimg.get_pixels();
nuclear@2 104 }
nuclear@2 105
nuclear@2 106 void erb_begin_frame(struct erebus *ctx, long ms)
nuclear@2 107 {
nuclear@8 108 printf("starting new frame...\n");
nuclear@2 109 ctx->cur_time = ms;
nuclear@4 110
nuclear@4 111 int xsz = ctx->options[ERB_OPT_WIDTH].x;
nuclear@4 112 int ysz = ctx->options[ERB_OPT_HEIGHT].x;
nuclear@4 113
nuclear@4 114 ctx->fbimg.create(xsz, ysz);
nuclear@8 115 ctx->accum.create(xsz, ysz);
nuclear@2 116 }
nuclear@2 117
nuclear@2 118 int erb_render(struct erebus *ctx, long timeout)
nuclear@2 119 {
nuclear@2 120 return erb_render_rect(ctx, 0, 0, ctx->fbimg.get_width(), ctx->fbimg.get_height(), timeout);
nuclear@2 121 }
nuclear@2 122
nuclear@2 123 int erb_render_rect(struct erebus *ctx, int x, int y, int width, int height, long timeout)
nuclear@2 124 {
nuclear@2 125 if(!width || !height) return -1;
nuclear@2 126
nuclear@2 127 Rect rect{x, y, width, height};
nuclear@2 128 if(ctx->cur_rect != rect) {
nuclear@8 129 // starting a new rendering apparently
nuclear@2 130 ctx->cur_rect = rect;
nuclear@2 131 ctx->cur_pixel_x = x;
nuclear@2 132 ctx->cur_pixel_y = y;
nuclear@8 133 ctx->cur_sample = 0;
nuclear@2 134 }
nuclear@2 135
nuclear@4 136 ctx->scn->update();
nuclear@4 137
nuclear@8 138 int max_samples = ctx->options[ERB_OPT_MAX_SAMPLES].x;
nuclear@8 139
nuclear@2 140 if(timeout > 0) {
nuclear@2 141 auto start_time = steady_clock::now();
nuclear@2 142 while(duration_cast<milliseconds>(steady_clock::now() - start_time).count() < timeout) {
nuclear@8 143 render_pixel(ctx, ctx->cur_pixel_x, ctx->cur_pixel_y, ctx->cur_sample);
nuclear@2 144
nuclear@2 145 if(++ctx->cur_pixel_x >= ctx->cur_rect.width) {
nuclear@8 146 ctx->cur_pixel_x = ctx->cur_rect.x;
nuclear@2 147 if(++ctx->cur_pixel_y >= ctx->cur_rect.height) {
nuclear@8 148 ctx->cur_pixel_y = ctx->cur_rect.y;
nuclear@8 149 if(++ctx->cur_sample >= max_samples) {
nuclear@8 150 ctx->cur_rect = INVALID_RECT;
nuclear@8 151 return 0;
nuclear@8 152 }
nuclear@2 153 }
nuclear@2 154 }
nuclear@2 155 }
nuclear@2 156 return 1;
nuclear@2 157 }
nuclear@2 158
nuclear@8 159 if(ctx->options[ERB_OPT_MAX_SAMPLES].x == (float)INF_SAMPLES) {
nuclear@8 160 max_samples = 128;
nuclear@8 161 }
nuclear@8 162
nuclear@2 163 for(int i=0; i<height; i++) {
nuclear@2 164 for(int j=0; j<width; j++) {
nuclear@8 165 for(int k=0; k<max_samples; k++) {
nuclear@8 166 render_pixel(ctx, j, i, k);
nuclear@8 167 }
nuclear@2 168 }
nuclear@2 169 }
nuclear@2 170 return 0;
nuclear@2 171 }
nuclear@2 172
nuclear@2 173 int erb_get_progress(struct erebus *ctx)
nuclear@2 174 {
nuclear@2 175 return 0; // TODO
nuclear@2 176 }
nuclear@2 177
nuclear@2 178 int erb_load_scene(struct erebus *ctx, const char *fname)
nuclear@2 179 {
nuclear@4 180 delete ctx->scn;
nuclear@4 181 ctx->scn = new Scene;
nuclear@2 182
nuclear@4 183 // XXX for now just create a test scene here
nuclear@4 184 Sphere *sph = new Sphere;
nuclear@8 185 sph->mtl.set_attrib("albedo", Color(1.0, 0.3, 0.2));
nuclear@4 186 SceneNode *sph_node = new SceneNode(sph);
nuclear@4 187 ctx->scn->add_object(sph);
nuclear@4 188 ctx->scn->add_node(sph_node);
nuclear@4 189
nuclear@8 190 sph = new Sphere;
nuclear@8 191 sph->mtl.set_attrib("albedo", Color(0.3, 0.4, 1.0));
nuclear@8 192 sph_node = new SceneNode(sph);
nuclear@8 193 sph_node->set_position(Vector3(0, -3.0, 0));
nuclear@8 194 //sph_node->set_scaling(Vector3(4.0, 4.0, 4.0) * 0.3);
nuclear@8 195 ctx->scn->add_object(sph);
nuclear@8 196 ctx->scn->add_node(sph_node);
nuclear@8 197
nuclear@8 198 Sphere *lt = new Sphere;
nuclear@8 199 lt->mtl.set_attrib("emissive", Color(10, 10, 10));
nuclear@8 200 SceneNode *lt_node = new SceneNode(lt);
nuclear@8 201 lt_node->set_position(Vector3(-15, 15, -10));
nuclear@8 202 lt_node->set_scaling(Vector3(5, 5, 5));
nuclear@8 203 ctx->scn->add_object(lt);
nuclear@8 204 ctx->scn->add_node(lt_node);
nuclear@8 205
nuclear@8 206 TargetCamera *cam = new TargetCamera(Vector3(0, 4, -8), Vector3(0, 0, 0));
nuclear@4 207 //ctx->scn->add_object(cam);
nuclear@4 208 ctx->scn->use_camera(cam);
nuclear@4 209
nuclear@4 210 return 0;
nuclear@2 211 }
nuclear@2 212
nuclear@9 213 bool erb_input_keyboard(struct erebus *ctx, int key, bool pressed)
nuclear@9 214 {
nuclear@9 215 if(!ctx) return false;
nuclear@9 216 if(ctx->keystate.size() <= key) ctx->keystate.resize(key < 256 ? 256 : key + 1);
nuclear@9 217
nuclear@9 218 ctx->keystate[key] = pressed;
nuclear@9 219 return false;
nuclear@9 220 }
nuclear@9 221
nuclear@9 222 bool erb_input_mouse_button(struct erebus *ctx, int bn, bool pressed, int x, int y)
nuclear@9 223 {
nuclear@9 224 if(!ctx) return false;
nuclear@9 225 if(ctx->bnstate.size() <= bn) ctx->bnstate.resize(bn < 32 ? 32 : bn + 1);
nuclear@9 226
nuclear@9 227 ctx->bnstate[bn] = pressed;
nuclear@9 228 ctx->mouse_pos[0] = x;
nuclear@9 229 ctx->mouse_pos[1] = y;
nuclear@9 230 }
nuclear@9 231
nuclear@9 232 bool erb_input_mouse_motion(struct erebus *ctx, int x, int y);
nuclear@9 233 bool erb_input_6dof_button(struct erebus *ctx, int bn, bool pressed);
nuclear@9 234 bool erb_input_6dof_motion(struct erebus *ctx, float x, float y, float z);
nuclear@9 235
nuclear@9 236
nuclear@2 237 } // extern "C"
nuclear@2 238
nuclear@2 239 float randf(float low, float high)
nuclear@2 240 {
nuclear@2 241 std::uniform_real_distribution<float> unirnd(low, high);
nuclear@2 242 return unirnd(rnd_gen);
nuclear@2 243 }
nuclear@2 244
nuclear@8 245 static void render_pixel(struct erebus *ctx, int x, int y, int sample)
nuclear@2 246 {
nuclear@5 247 Camera *cam = ctx->scn->get_active_camera();
nuclear@5 248 if(!cam) return;
nuclear@5 249
nuclear@5 250 int xsz = ctx->fbimg.get_width();
nuclear@5 251 int ysz = ctx->fbimg.get_height();
nuclear@8 252 int offs = (y * xsz + x) * 4;
nuclear@5 253
nuclear@8 254 float *pix = ctx->fbimg.get_pixels() + offs;
nuclear@8 255 float *accum = ctx->accum.get_pixels() + offs;
nuclear@8 256
nuclear@8 257 Ray ray = cam->get_primary_ray(x, y, xsz, ysz, sample);
nuclear@8 258 Color c = ray_trace(ray, ctx->scn, 0);
nuclear@8 259 accum[0] += c.x;
nuclear@8 260 accum[1] += c.y;
nuclear@8 261 accum[2] += c.z;
nuclear@8 262 accum[3] += c.w;
nuclear@8 263
nuclear@8 264 float inv_samples = 1.0f / (float)(sample + 1);
nuclear@8 265 pix[0] = accum[0] * inv_samples;
nuclear@8 266 pix[1] = accum[1] * inv_samples;
nuclear@8 267 pix[2] = accum[2] * inv_samples;
nuclear@8 268 pix[3] = accum[3] * inv_samples;
nuclear@2 269 }