libpsys

annotate src/psys.c @ 3:133094e2f5a5

find changesets by author, revision, files, or words in the commit message
reorganizing
author John Tsiombikas <nuclear@mutantstargoat.com>
date Mon, 26 Sep 2011 18:20:11 +0300
parents 6e5342a2529a
children 0fe624ffcb4f
rev   line source
nuclear@0 1 #include <stdlib.h>
nuclear@0 2 #include <math.h>
nuclear@0 3 #include <assert.h>
nuclear@0 4 #include <pthread.h>
nuclear@0 5 #include <vmath.h>
nuclear@0 6 #include "psys_impl.h"
nuclear@0 7
nuclear@0 8 static int spawn(struct psys_emitter *em, struct psys_particle *p, void *cls);
nuclear@0 9 static void update_particle(struct psys_emitter *em, struct psys_particle *p, float tm, float dt, void *cls);
nuclear@0 10
nuclear@0 11 static int init_v3track(struct v3track *v3t);
nuclear@0 12 static void destroy_v3track(struct v3track *v3t);
nuclear@0 13 static void set_v3value(struct v3track *v3t, anm_time_t tm, vec3_t v);
nuclear@0 14 static vec3_t get_v3value(struct v3track *v3t, anm_time_t tm);
nuclear@0 15
nuclear@3 16 static float random_val(float x, float range);
nuclear@3 17 static vec3_t random_vec3(vec3_t v, vec3_t range);
nuclear@3 18
nuclear@0 19 /* particle pool */
nuclear@0 20 static struct psys_particle *ppool;
nuclear@0 21 static int ppool_size;
nuclear@0 22 static pthread_mutex_t pool_lock = PTHREAD_MUTEX_INITIALIZER;
nuclear@0 23
nuclear@0 24 static struct psys_particle *palloc(void);
nuclear@0 25 static void pfree(struct psys_particle *p);
nuclear@0 26
nuclear@0 27 /* --- constructors and shit --- */
nuclear@0 28
nuclear@0 29 struct psys_emitter *psys_create(void)
nuclear@0 30 {
nuclear@0 31 struct psys_emitter *em;
nuclear@0 32
nuclear@0 33 if(!(em = malloc(sizeof *em))) {
nuclear@0 34 return 0;
nuclear@0 35 }
nuclear@0 36 if(psys_init(em) == -1) {
nuclear@0 37 free(em);
nuclear@0 38 return 0;
nuclear@0 39 }
nuclear@0 40 return em;
nuclear@0 41 }
nuclear@0 42
nuclear@0 43 void psys_free(struct psys_emitter *em)
nuclear@0 44 {
nuclear@0 45 psys_destroy(em);
nuclear@0 46 free(em);
nuclear@0 47 }
nuclear@0 48
nuclear@0 49 int psys_init(struct psys_emitter *em)
nuclear@0 50 {
nuclear@0 51 memset(em, 0, sizeof *em);
nuclear@0 52
nuclear@3 53 if(anm_init_node(&em->prs) == -1)
nuclear@3 54 goto err;
nuclear@3 55 if(init_v3track(&em->pos_range) == -1)
nuclear@3 56 goto err;
nuclear@3 57 if(anm_init_track(&em->rate) == -1)
nuclear@3 58 goto err;
nuclear@3 59 if(anm_init_track(&em->life) == -1)
nuclear@3 60 goto err;
nuclear@3 61 if(anm_init_track(&em->life_range) == -1)
nuclear@3 62 goto err;
nuclear@3 63 if(anm_init_track(&em->size) == -1)
nuclear@3 64 goto err;
nuclear@3 65 if(anm_init_track(&em->size_range) == -1)
nuclear@3 66 goto err;
nuclear@3 67 if(init_v3track(&em->dir) == -1)
nuclear@3 68 goto err;
nuclear@3 69 if(init_v3track(&em->dir_range) == -1)
nuclear@3 70 goto err;
nuclear@3 71 if(init_v3track(&em->grav) == -1)
nuclear@3 72 goto err;
nuclear@0 73
nuclear@0 74 em->spawn = spawn;
nuclear@0 75 em->update = update_particle;
nuclear@0 76
nuclear@0 77 em->draw = psys_gl_draw;
nuclear@0 78 em->draw_start = psys_gl_draw_start;
nuclear@0 79 em->draw_end = psys_gl_draw_end;
nuclear@0 80
nuclear@0 81 return 0;
nuclear@3 82 err:
nuclear@3 83 psys_destroy(em);
nuclear@3 84 return -1;
nuclear@0 85 }
nuclear@0 86
nuclear@0 87 void psys_destroy(struct psys_emitter *em)
nuclear@0 88 {
nuclear@0 89 struct psys_particle *part;
nuclear@0 90
nuclear@0 91 part = em->plist;
nuclear@0 92 while(part) {
nuclear@0 93 struct psys_particle *tmp = part;
nuclear@0 94 part = part->next;
nuclear@0 95 pfree(tmp);
nuclear@0 96 }
nuclear@0 97
nuclear@0 98 anm_destroy_node(&em->prs);
nuclear@3 99 destroy_v3track(&em->pos_range);
nuclear@0 100 anm_destroy_track(&em->rate);
nuclear@3 101 anm_destroy_track(&em->life);
nuclear@3 102 anm_destroy_track(&em->size);
nuclear@3 103 anm_destroy_track(&em->size_range);
nuclear@0 104 destroy_v3track(&em->dir);
nuclear@3 105 destroy_v3track(&em->dir_range);
nuclear@3 106 destroy_v3track(&em->grav);
nuclear@0 107 }
nuclear@0 108
nuclear@2 109 void psys_set_texture(struct psys_emitter *em, unsigned int tex)
nuclear@2 110 {
nuclear@2 111 em->tex = tex;
nuclear@2 112 }
nuclear@2 113
nuclear@3 114 void psys_set_pos(struct psys_emitter *em, vec3_t pos, vec3_t range, float tm)
nuclear@0 115 {
nuclear@0 116 anm_set_position(&em->prs, pos, ANM_SEC2TM(tm));
nuclear@3 117 set_v3value(&em->pos_range, ANM_SEC2TM(tm), range);
nuclear@0 118 }
nuclear@0 119
nuclear@0 120 void psys_set_rot(struct psys_emitter *em, quat_t rot, float tm)
nuclear@0 121 {
nuclear@0 122 anm_set_rotation(&em->prs, rot, ANM_SEC2TM(tm));
nuclear@0 123 }
nuclear@0 124
nuclear@0 125 void psys_set_pivot(struct psys_emitter *em, vec3_t pivot)
nuclear@0 126 {
nuclear@0 127 anm_set_pivot(&em->prs, pivot);
nuclear@0 128 }
nuclear@0 129
nuclear@0 130 void psys_set_rate(struct psys_emitter *em, float rate, float tm)
nuclear@0 131 {
nuclear@0 132 anm_set_value(&em->rate, ANM_SEC2TM(tm), rate);
nuclear@0 133 }
nuclear@0 134
nuclear@3 135 void psys_set_life(struct psys_emitter *em, float life, float range, float tm)
nuclear@1 136 {
nuclear@1 137 anm_set_value(&em->life, ANM_SEC2TM(tm), life);
nuclear@3 138 anm_set_value(&em->life_range, ANM_SEC2TM(tm), range);
nuclear@1 139 }
nuclear@1 140
nuclear@3 141 void psys_set_size(struct psys_emitter *em, float size, float range, float tm)
nuclear@3 142 {
nuclear@3 143 anm_set_value(&em->size, ANM_SEC2TM(tm), size);
nuclear@3 144 anm_set_value(&em->size_range, ANM_SEC2TM(tm), range);
nuclear@3 145 }
nuclear@3 146
nuclear@3 147 void psys_set_dir(struct psys_emitter *em, vec3_t dir, vec3_t range, float tm)
nuclear@0 148 {
nuclear@0 149 set_v3value(&em->dir, ANM_SEC2TM(tm), dir);
nuclear@3 150 set_v3value(&em->dir_range, ANM_SEC2TM(tm), range);
nuclear@0 151 }
nuclear@0 152
nuclear@1 153 void psys_set_grav(struct psys_emitter *em, vec3_t grav, float tm)
nuclear@1 154 {
nuclear@1 155 set_v3value(&em->grav, ANM_SEC2TM(tm), grav);
nuclear@1 156 }
nuclear@1 157
nuclear@0 158
nuclear@0 159 void psys_clear_collision_planes(struct psys_emitter *em)
nuclear@0 160 {
nuclear@0 161 struct col_plane *plane;
nuclear@0 162
nuclear@0 163 plane = em->planes;
nuclear@0 164 while(plane) {
nuclear@0 165 struct col_plane *tmp = plane;
nuclear@0 166 plane = plane->next;
nuclear@1 167 free(tmp);
nuclear@0 168 }
nuclear@0 169 }
nuclear@0 170
nuclear@0 171 int psys_add_collision_plane(struct psys_emitter *em, plane_t plane, float elast)
nuclear@0 172 {
nuclear@0 173 struct col_plane *node;
nuclear@0 174
nuclear@0 175 if(!(node = malloc(sizeof *node))) {
nuclear@0 176 return -1;
nuclear@0 177 }
nuclear@0 178 node->p = plane;
nuclear@0 179 node->elasticity = elast;
nuclear@0 180 node->next = em->planes;
nuclear@0 181 em->planes = node;
nuclear@0 182 return 0;
nuclear@0 183 }
nuclear@0 184
nuclear@0 185 void psys_add_particle(struct psys_emitter *em, struct psys_particle *p)
nuclear@0 186 {
nuclear@0 187 p->next = em->plist;
nuclear@0 188 em->plist = p;
nuclear@0 189 }
nuclear@0 190
nuclear@0 191 void psys_spawn_func(struct psys_emitter *em, psys_spawn_func_t func, void *cls)
nuclear@0 192 {
nuclear@0 193 em->spawn = func;
nuclear@0 194 em->spawn_cls = cls;
nuclear@0 195 }
nuclear@0 196
nuclear@0 197 void psys_update_func(struct psys_emitter *em, psys_update_func_t func, void *cls)
nuclear@0 198 {
nuclear@0 199 em->update = func;
nuclear@0 200 em->upd_cls = cls;
nuclear@0 201 }
nuclear@0 202
nuclear@0 203 void psys_draw_func(struct psys_emitter *em, psys_draw_func_t draw,
nuclear@0 204 psys_draw_start_func_t start, psys_draw_end_func_t end, void *cls)
nuclear@0 205 {
nuclear@0 206 em->draw = draw;
nuclear@0 207 em->draw_start = start;
nuclear@0 208 em->draw_end = end;
nuclear@0 209 em->draw_cls = cls;
nuclear@0 210 }
nuclear@0 211
nuclear@0 212 /* --- query current state --- */
nuclear@2 213 unsigned int psys_get_texture(struct psys_emitter *em)
nuclear@2 214 {
nuclear@2 215 return em->tex;
nuclear@2 216 }
nuclear@2 217
nuclear@0 218 vec3_t psys_get_pos(struct psys_emitter *em)
nuclear@0 219 {
nuclear@0 220 return em->cur_pos;
nuclear@0 221 }
nuclear@0 222
nuclear@0 223 quat_t psys_get_rot(struct psys_emitter *em)
nuclear@0 224 {
nuclear@0 225 return em->cur_rot;
nuclear@0 226 }
nuclear@0 227
nuclear@0 228 float psys_get_rate(struct psys_emitter *em)
nuclear@0 229 {
nuclear@0 230 return em->cur_rate;
nuclear@0 231 }
nuclear@0 232
nuclear@0 233 float psys_get_life(struct psys_emitter *em)
nuclear@0 234 {
nuclear@0 235 return em->cur_life;
nuclear@0 236 }
nuclear@0 237
nuclear@0 238 vec3_t psys_get_dir(struct psys_emitter *em)
nuclear@0 239 {
nuclear@0 240 return em->cur_dir;
nuclear@0 241 }
nuclear@0 242
nuclear@1 243 vec3_t psys_get_grav(struct psys_emitter *em)
nuclear@1 244 {
nuclear@1 245 return em->cur_grav;
nuclear@1 246 }
nuclear@0 247
nuclear@0 248 /* --- update and render --- */
nuclear@0 249
nuclear@0 250 void psys_update(struct psys_emitter *em, float tm)
nuclear@0 251 {
nuclear@2 252 float dt, spawn_dt, spawn_tm;
nuclear@0 253 int i, spawn_count;
nuclear@0 254 struct psys_particle *p, pdummy;
nuclear@0 255 anm_time_t atm;
nuclear@0 256
nuclear@0 257 assert(em->spawn && em->update);
nuclear@0 258
nuclear@0 259 atm = ANM_SEC2TM(tm);
nuclear@0 260
nuclear@1 261 em->cur_rate = anm_get_value(&em->rate, atm);
nuclear@0 262 dt = tm - em->last_update;
nuclear@0 263
nuclear@0 264 /* how many particles to spawn for this interval ? */
nuclear@2 265 em->spawn_acc += em->cur_rate * dt;
nuclear@2 266 if(em->spawn_acc >= 1.0) {
nuclear@2 267 spawn_count = em->spawn_acc;
nuclear@2 268 em->spawn_acc = fmod(em->spawn_acc, 1.0);
nuclear@2 269 } else {
nuclear@2 270 spawn_count = 0;
nuclear@2 271 }
nuclear@0 272
nuclear@0 273 em->cur_dir = get_v3value(&em->dir, atm);
nuclear@0 274 em->cur_life = anm_get_value(&em->life, atm);
nuclear@1 275 em->cur_grav = get_v3value(&em->grav, atm);
nuclear@0 276
nuclear@0 277 spawn_dt = dt / (float)spawn_count;
nuclear@2 278 spawn_tm = em->last_update;
nuclear@0 279 for(i=0; i<spawn_count; i++) {
nuclear@0 280 /* update emitter position for this spawning */
nuclear@2 281 em->cur_pos = anm_get_position(&em->prs, ANM_SEC2TM(spawn_tm));
nuclear@0 282
nuclear@0 283 if(!(p = palloc())) {
nuclear@0 284 return;
nuclear@0 285 }
nuclear@0 286 if(em->spawn(em, p, em->spawn_cls) == -1) {
nuclear@0 287 pfree(p);
nuclear@0 288 }
nuclear@2 289 spawn_tm += spawn_dt;
nuclear@0 290 }
nuclear@0 291
nuclear@0 292 /* update all particles */
nuclear@0 293 p = em->plist;
nuclear@0 294 while(p) {
nuclear@1 295 em->update(em, p, tm, dt, em->upd_cls);
nuclear@0 296 p = p->next;
nuclear@0 297 }
nuclear@0 298
nuclear@0 299 /* cleanup dead particles */
nuclear@0 300 pdummy.next = em->plist;
nuclear@0 301 p = &pdummy;
nuclear@0 302 while(p->next) {
nuclear@0 303 if(p->next->life <= 0) {
nuclear@0 304 struct psys_particle *tmp = p->next;
nuclear@0 305 p->next = p->next->next;
nuclear@0 306 pfree(tmp);
nuclear@0 307 } else {
nuclear@0 308 p = p->next;
nuclear@0 309 }
nuclear@0 310 }
nuclear@0 311 em->plist = pdummy.next;
nuclear@2 312
nuclear@2 313 em->last_update = tm;
nuclear@0 314 }
nuclear@0 315
nuclear@0 316 void psys_draw(struct psys_emitter *em)
nuclear@0 317 {
nuclear@0 318 struct psys_particle *p;
nuclear@0 319
nuclear@0 320 if(em->draw_start) {
nuclear@1 321 em->draw_start(em, em->draw_cls);
nuclear@0 322 }
nuclear@0 323
nuclear@0 324 p = em->plist;
nuclear@0 325 while(p) {
nuclear@1 326 em->draw(em, p, em->draw_cls);
nuclear@0 327 p = p->next;
nuclear@0 328 }
nuclear@0 329
nuclear@0 330 if(em->draw_end) {
nuclear@1 331 em->draw_end(em, em->draw_cls);
nuclear@0 332 }
nuclear@0 333 }
nuclear@0 334
nuclear@0 335 static int spawn(struct psys_emitter *em, struct psys_particle *p, void *cls)
nuclear@0 336 {
nuclear@3 337 p->pos = random_vec3(em->cur_pos, em->cur_pos_range);
nuclear@3 338 p->vel = random_vec3(em->cur_dir, em->dir_range);
nuclear@3 339 p->size = random_val(em->cur_size, em->cur_size_range);
nuclear@0 340 p->life = em->cur_life;
nuclear@0 341
nuclear@0 342 psys_add_particle(em, p);
nuclear@0 343 return 0;
nuclear@0 344 }
nuclear@0 345
nuclear@0 346 static void update_particle(struct psys_emitter *em, struct psys_particle *p, float tm, float dt, void *cls)
nuclear@0 347 {
nuclear@1 348 vec3_t accel;
nuclear@0 349
nuclear@1 350 accel.x = em->cur_grav.x - p->vel.x * em->drag;
nuclear@1 351 accel.y = em->cur_grav.y - p->vel.y * em->drag;
nuclear@1 352 accel.z = em->cur_grav.z - p->vel.z * em->drag;
nuclear@1 353
nuclear@1 354 p->vel.x += accel.x * dt;
nuclear@1 355 p->vel.y += accel.y * dt;
nuclear@1 356 p->vel.z += accel.z * dt;
nuclear@0 357
nuclear@0 358 p->pos.x += p->vel.x * dt;
nuclear@0 359 p->pos.y += p->vel.y * dt;
nuclear@0 360 p->pos.z += p->vel.z * dt;
nuclear@2 361
nuclear@2 362 p->life -= dt;
nuclear@0 363 }
nuclear@0 364
nuclear@0 365 /* --- v3track helper --- */
nuclear@0 366
nuclear@0 367 int init_v3track(struct v3track *v3t)
nuclear@0 368 {
nuclear@0 369 if(anm_init_track(&v3t->x) == -1) {
nuclear@0 370 return -1;
nuclear@0 371 }
nuclear@0 372 if(anm_init_track(&v3t->y) == -1) {
nuclear@0 373 anm_destroy_track(&v3t->x);
nuclear@0 374 return -1;
nuclear@0 375 }
nuclear@0 376 if(anm_init_track(&v3t->z) == -1) {
nuclear@0 377 anm_destroy_track(&v3t->x);
nuclear@0 378 anm_destroy_track(&v3t->y);
nuclear@0 379 return -1;
nuclear@0 380 }
nuclear@0 381 return 0;
nuclear@0 382 }
nuclear@0 383
nuclear@0 384 static void destroy_v3track(struct v3track *v3t)
nuclear@0 385 {
nuclear@0 386 anm_destroy_track(&v3t->x);
nuclear@0 387 anm_destroy_track(&v3t->y);
nuclear@0 388 anm_destroy_track(&v3t->z);
nuclear@0 389 }
nuclear@0 390
nuclear@0 391 static void set_v3value(struct v3track *v3t, anm_time_t tm, vec3_t v)
nuclear@0 392 {
nuclear@0 393 anm_set_value(&v3t->x, tm, v.x);
nuclear@0 394 anm_set_value(&v3t->y, tm, v.y);
nuclear@0 395 anm_set_value(&v3t->z, tm, v.z);
nuclear@0 396 }
nuclear@0 397
nuclear@0 398 static vec3_t get_v3value(struct v3track *v3t, anm_time_t tm)
nuclear@0 399 {
nuclear@0 400 vec3_t v;
nuclear@0 401 v.x = anm_get_value(&v3t->x, tm);
nuclear@0 402 v.y = anm_get_value(&v3t->y, tm);
nuclear@0 403 v.z = anm_get_value(&v3t->z, tm);
nuclear@0 404 return v;
nuclear@0 405 }
nuclear@0 406
nuclear@0 407 /* --- particle allocation pool --- */
nuclear@0 408
nuclear@0 409 static struct psys_particle *palloc(void)
nuclear@0 410 {
nuclear@0 411 struct psys_particle *p;
nuclear@0 412
nuclear@0 413 pthread_mutex_lock(&pool_lock);
nuclear@0 414 if(ppool) {
nuclear@0 415 p = ppool;
nuclear@0 416 ppool = ppool->next;
nuclear@0 417 ppool_size--;
nuclear@0 418 } else {
nuclear@0 419 p = malloc(sizeof *p);
nuclear@0 420 }
nuclear@0 421 pthread_mutex_unlock(&pool_lock);
nuclear@0 422
nuclear@0 423 if(p) {
nuclear@0 424 memset(p, 0, sizeof *p);
nuclear@1 425 /*reset_pattr(&p->attr);*/
nuclear@0 426 }
nuclear@0 427 return p;
nuclear@0 428 }
nuclear@0 429
nuclear@0 430 static void pfree(struct psys_particle *p)
nuclear@0 431 {
nuclear@0 432 pthread_mutex_lock(&pool_lock);
nuclear@0 433 p->next = ppool;
nuclear@0 434 ppool = p;
nuclear@0 435 ppool_size++;
nuclear@0 436 pthread_mutex_unlock(&pool_lock);
nuclear@0 437 }
nuclear@3 438
nuclear@3 439 static float random_val(float x, float range)
nuclear@3 440 {
nuclear@3 441 return x + range * (float)rand() / (float)RAND_MAX - 0.5 * range;
nuclear@3 442 }
nuclear@3 443
nuclear@3 444 static vec3_t random_vec3(vec3_t v, vec3_t range)
nuclear@3 445 {
nuclear@3 446 vec3_t res;
nuclear@3 447 res.x = random_val(v.x, range.x);
nuclear@3 448 res.y = random_val(v.y, range.y);
nuclear@3 449 res.z = random_val(v.z, range.z);
nuclear@3 450 return res;
nuclear@3 451 }