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@4
|
11 static int init_anm_track_vec3(struct anm_track_vec3 *v3t);
|
nuclear@4
|
12 static void destroy_anm_track_vec3(struct anm_track_vec3 *v3t);
|
nuclear@4
|
13 static void set_v3value(struct anm_track_vec3 *v3t, anm_time_t tm, vec3_t v);
|
nuclear@4
|
14 static vec3_t get_v3value(struct anm_track_vec3 *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@4
|
55 if(init_anm_track_vec3(&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@4
|
67 if(init_anm_track_vec3(&em->dir) == -1)
|
nuclear@3
|
68 goto err;
|
nuclear@4
|
69 if(init_anm_track_vec3(&em->dir_range) == -1)
|
nuclear@3
|
70 goto err;
|
nuclear@4
|
71 if(init_anm_track_vec3(&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@4
|
99 destroy_anm_track_vec3(&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@4
|
104 destroy_anm_track_vec3(&em->dir);
|
nuclear@4
|
105 destroy_anm_track_vec3(&em->dir_range);
|
nuclear@4
|
106 destroy_anm_track_vec3(&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 /* --- update and render --- */
|
nuclear@0
|
213
|
nuclear@0
|
214 void psys_update(struct psys_emitter *em, float tm)
|
nuclear@0
|
215 {
|
nuclear@2
|
216 float dt, spawn_dt, spawn_tm;
|
nuclear@0
|
217 int i, spawn_count;
|
nuclear@0
|
218 struct psys_particle *p, pdummy;
|
nuclear@0
|
219 anm_time_t atm;
|
nuclear@0
|
220
|
nuclear@0
|
221 assert(em->spawn && em->update);
|
nuclear@0
|
222
|
nuclear@0
|
223 atm = ANM_SEC2TM(tm);
|
nuclear@0
|
224
|
nuclear@1
|
225 em->cur_rate = anm_get_value(&em->rate, atm);
|
nuclear@0
|
226 dt = tm - em->last_update;
|
nuclear@0
|
227
|
nuclear@0
|
228 /* how many particles to spawn for this interval ? */
|
nuclear@2
|
229 em->spawn_acc += em->cur_rate * dt;
|
nuclear@2
|
230 if(em->spawn_acc >= 1.0) {
|
nuclear@2
|
231 spawn_count = em->spawn_acc;
|
nuclear@2
|
232 em->spawn_acc = fmod(em->spawn_acc, 1.0);
|
nuclear@2
|
233 } else {
|
nuclear@2
|
234 spawn_count = 0;
|
nuclear@2
|
235 }
|
nuclear@0
|
236
|
nuclear@0
|
237 em->cur_dir = get_v3value(&em->dir, atm);
|
nuclear@0
|
238 em->cur_life = anm_get_value(&em->life, atm);
|
nuclear@1
|
239 em->cur_grav = get_v3value(&em->grav, atm);
|
nuclear@0
|
240
|
nuclear@0
|
241 spawn_dt = dt / (float)spawn_count;
|
nuclear@2
|
242 spawn_tm = em->last_update;
|
nuclear@0
|
243 for(i=0; i<spawn_count; i++) {
|
nuclear@0
|
244 /* update emitter position for this spawning */
|
nuclear@2
|
245 em->cur_pos = anm_get_position(&em->prs, ANM_SEC2TM(spawn_tm));
|
nuclear@0
|
246
|
nuclear@0
|
247 if(!(p = palloc())) {
|
nuclear@0
|
248 return;
|
nuclear@0
|
249 }
|
nuclear@0
|
250 if(em->spawn(em, p, em->spawn_cls) == -1) {
|
nuclear@0
|
251 pfree(p);
|
nuclear@0
|
252 }
|
nuclear@2
|
253 spawn_tm += spawn_dt;
|
nuclear@0
|
254 }
|
nuclear@0
|
255
|
nuclear@0
|
256 /* update all particles */
|
nuclear@0
|
257 p = em->plist;
|
nuclear@0
|
258 while(p) {
|
nuclear@1
|
259 em->update(em, p, tm, dt, em->upd_cls);
|
nuclear@0
|
260 p = p->next;
|
nuclear@0
|
261 }
|
nuclear@0
|
262
|
nuclear@0
|
263 /* cleanup dead particles */
|
nuclear@0
|
264 pdummy.next = em->plist;
|
nuclear@0
|
265 p = &pdummy;
|
nuclear@0
|
266 while(p->next) {
|
nuclear@0
|
267 if(p->next->life <= 0) {
|
nuclear@0
|
268 struct psys_particle *tmp = p->next;
|
nuclear@0
|
269 p->next = p->next->next;
|
nuclear@0
|
270 pfree(tmp);
|
nuclear@0
|
271 } else {
|
nuclear@0
|
272 p = p->next;
|
nuclear@0
|
273 }
|
nuclear@0
|
274 }
|
nuclear@0
|
275 em->plist = pdummy.next;
|
nuclear@2
|
276
|
nuclear@2
|
277 em->last_update = tm;
|
nuclear@0
|
278 }
|
nuclear@0
|
279
|
nuclear@0
|
280 void psys_draw(struct psys_emitter *em)
|
nuclear@0
|
281 {
|
nuclear@0
|
282 struct psys_particle *p;
|
nuclear@0
|
283
|
nuclear@0
|
284 if(em->draw_start) {
|
nuclear@1
|
285 em->draw_start(em, em->draw_cls);
|
nuclear@0
|
286 }
|
nuclear@0
|
287
|
nuclear@0
|
288 p = em->plist;
|
nuclear@0
|
289 while(p) {
|
nuclear@1
|
290 em->draw(em, p, em->draw_cls);
|
nuclear@0
|
291 p = p->next;
|
nuclear@0
|
292 }
|
nuclear@0
|
293
|
nuclear@0
|
294 if(em->draw_end) {
|
nuclear@1
|
295 em->draw_end(em, em->draw_cls);
|
nuclear@0
|
296 }
|
nuclear@0
|
297 }
|
nuclear@0
|
298
|
nuclear@0
|
299 static int spawn(struct psys_emitter *em, struct psys_particle *p, void *cls)
|
nuclear@0
|
300 {
|
nuclear@3
|
301 p->pos = random_vec3(em->cur_pos, em->cur_pos_range);
|
nuclear@3
|
302 p->vel = random_vec3(em->cur_dir, em->dir_range);
|
nuclear@3
|
303 p->size = random_val(em->cur_size, em->cur_size_range);
|
nuclear@0
|
304 p->life = em->cur_life;
|
nuclear@0
|
305
|
nuclear@0
|
306 psys_add_particle(em, p);
|
nuclear@0
|
307 return 0;
|
nuclear@0
|
308 }
|
nuclear@0
|
309
|
nuclear@0
|
310 static void update_particle(struct psys_emitter *em, struct psys_particle *p, float tm, float dt, void *cls)
|
nuclear@0
|
311 {
|
nuclear@1
|
312 vec3_t accel;
|
nuclear@0
|
313
|
nuclear@1
|
314 accel.x = em->cur_grav.x - p->vel.x * em->drag;
|
nuclear@1
|
315 accel.y = em->cur_grav.y - p->vel.y * em->drag;
|
nuclear@1
|
316 accel.z = em->cur_grav.z - p->vel.z * em->drag;
|
nuclear@1
|
317
|
nuclear@1
|
318 p->vel.x += accel.x * dt;
|
nuclear@1
|
319 p->vel.y += accel.y * dt;
|
nuclear@1
|
320 p->vel.z += accel.z * dt;
|
nuclear@0
|
321
|
nuclear@0
|
322 p->pos.x += p->vel.x * dt;
|
nuclear@0
|
323 p->pos.y += p->vel.y * dt;
|
nuclear@0
|
324 p->pos.z += p->vel.z * dt;
|
nuclear@2
|
325
|
nuclear@2
|
326 p->life -= dt;
|
nuclear@0
|
327 }
|
nuclear@0
|
328
|
nuclear@4
|
329 /* --- anm_track_vec3 helper --- */
|
nuclear@0
|
330
|
nuclear@4
|
331 int init_anm_track_vec3(struct anm_track_vec3 *v3t)
|
nuclear@0
|
332 {
|
nuclear@0
|
333 if(anm_init_track(&v3t->x) == -1) {
|
nuclear@0
|
334 return -1;
|
nuclear@0
|
335 }
|
nuclear@0
|
336 if(anm_init_track(&v3t->y) == -1) {
|
nuclear@0
|
337 anm_destroy_track(&v3t->x);
|
nuclear@0
|
338 return -1;
|
nuclear@0
|
339 }
|
nuclear@0
|
340 if(anm_init_track(&v3t->z) == -1) {
|
nuclear@0
|
341 anm_destroy_track(&v3t->x);
|
nuclear@0
|
342 anm_destroy_track(&v3t->y);
|
nuclear@0
|
343 return -1;
|
nuclear@0
|
344 }
|
nuclear@0
|
345 return 0;
|
nuclear@0
|
346 }
|
nuclear@0
|
347
|
nuclear@4
|
348 static void destroy_anm_track_vec3(struct anm_track_vec3 *v3t)
|
nuclear@0
|
349 {
|
nuclear@0
|
350 anm_destroy_track(&v3t->x);
|
nuclear@0
|
351 anm_destroy_track(&v3t->y);
|
nuclear@0
|
352 anm_destroy_track(&v3t->z);
|
nuclear@0
|
353 }
|
nuclear@0
|
354
|
nuclear@4
|
355 static void set_v3value(struct anm_track_vec3 *v3t, anm_time_t tm, vec3_t v)
|
nuclear@0
|
356 {
|
nuclear@0
|
357 anm_set_value(&v3t->x, tm, v.x);
|
nuclear@0
|
358 anm_set_value(&v3t->y, tm, v.y);
|
nuclear@0
|
359 anm_set_value(&v3t->z, tm, v.z);
|
nuclear@0
|
360 }
|
nuclear@0
|
361
|
nuclear@4
|
362 static vec3_t get_v3value(struct anm_track_vec3 *v3t, anm_time_t tm)
|
nuclear@0
|
363 {
|
nuclear@0
|
364 vec3_t v;
|
nuclear@0
|
365 v.x = anm_get_value(&v3t->x, tm);
|
nuclear@0
|
366 v.y = anm_get_value(&v3t->y, tm);
|
nuclear@0
|
367 v.z = anm_get_value(&v3t->z, tm);
|
nuclear@0
|
368 return v;
|
nuclear@0
|
369 }
|
nuclear@0
|
370
|
nuclear@0
|
371 /* --- particle allocation pool --- */
|
nuclear@0
|
372
|
nuclear@0
|
373 static struct psys_particle *palloc(void)
|
nuclear@0
|
374 {
|
nuclear@0
|
375 struct psys_particle *p;
|
nuclear@0
|
376
|
nuclear@0
|
377 pthread_mutex_lock(&pool_lock);
|
nuclear@0
|
378 if(ppool) {
|
nuclear@0
|
379 p = ppool;
|
nuclear@0
|
380 ppool = ppool->next;
|
nuclear@0
|
381 ppool_size--;
|
nuclear@0
|
382 } else {
|
nuclear@0
|
383 p = malloc(sizeof *p);
|
nuclear@0
|
384 }
|
nuclear@0
|
385 pthread_mutex_unlock(&pool_lock);
|
nuclear@0
|
386
|
nuclear@0
|
387 if(p) {
|
nuclear@0
|
388 memset(p, 0, sizeof *p);
|
nuclear@1
|
389 /*reset_pattr(&p->attr);*/
|
nuclear@0
|
390 }
|
nuclear@0
|
391 return p;
|
nuclear@0
|
392 }
|
nuclear@0
|
393
|
nuclear@0
|
394 static void pfree(struct psys_particle *p)
|
nuclear@0
|
395 {
|
nuclear@0
|
396 pthread_mutex_lock(&pool_lock);
|
nuclear@0
|
397 p->next = ppool;
|
nuclear@0
|
398 ppool = p;
|
nuclear@0
|
399 ppool_size++;
|
nuclear@0
|
400 pthread_mutex_unlock(&pool_lock);
|
nuclear@0
|
401 }
|
nuclear@3
|
402
|
nuclear@3
|
403 static float random_val(float x, float range)
|
nuclear@3
|
404 {
|
nuclear@3
|
405 return x + range * (float)rand() / (float)RAND_MAX - 0.5 * range;
|
nuclear@3
|
406 }
|
nuclear@3
|
407
|
nuclear@3
|
408 static vec3_t random_vec3(vec3_t v, vec3_t range)
|
nuclear@3
|
409 {
|
nuclear@3
|
410 vec3_t res;
|
nuclear@3
|
411 res.x = random_val(v.x, range.x);
|
nuclear@3
|
412 res.y = random_val(v.y, range.y);
|
nuclear@3
|
413 res.z = random_val(v.z, range.z);
|
nuclear@3
|
414 return res;
|
nuclear@3
|
415 }
|