Paris7/work-stealing-scheduler
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

YAY

Browse source
This commit is contained in:
Mylloon 2024-04-24 14:41:42 +02:00
parent 2ae7a47667
commit 9978f49c86
Signed by: Anri
GPG key ID: A82D63DFF8D1317F
1 changed files with 36 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

61
src/mandelbrot.c
View file

@ -11,6 +11,7 @@
#define WIDTH 3840 #define WIDTH 3840
#define HEIGHT 2160 #define HEIGHT 2160
#define ITERATIONS 1000 #define ITERATIONS 1000
#define CHUNK_SIZE 32
#define SCALE (WIDTH / 4.0) #define SCALE (WIDTH / 4.0)
#define DX (WIDTH / 2) #define DX (WIDTH / 2)
@ -18,11 +19,12 @@
struct mandelbrot_args { struct mandelbrot_args {
unsigned int *image; unsigned int *image;
int x, y; int start_x, start_y, end_x, end_y;
}; };
struct mandelbrot_args * struct mandelbrot_args *
new_mandelbrot_args(unsigned int *image, int x, int y) new_mandelbrot_args(unsigned int *image, int start_x, int start_y, int end_x,
int end_y)
{ {
struct mandelbrot_args *args; struct mandelbrot_args *args;
@ -32,8 +34,10 @@ new_mandelbrot_args(unsigned int *image, int x, int y)
} }
args->image = image; args->image = image;
args->x = x; args->start_x = start_x;
args->y = y; args->start_y = start_y;
args->end_x = end_x;
args->end_y = end_y;
return args; return args;
} }
@ -94,35 +98,41 @@ pixel(unsigned int *image, int x, int y)
image[y * WIDTH + x] = rgb; image[y * WIDTH + x] = rgb;
} }
void
draw_pixel(void *closure, struct scheduler *s)
{
struct mandelbrot_args *args = (struct mandelbrot_args *)closure;
unsigned int *image = args->image;
int x = args->x;
int y = args->y;
free(closure);
(void)s; // pas de nouvelle tâche dans le scheduler
pixel(image, x, y);
}
void void
draw(void *closure, struct scheduler *s) draw(void *closure, struct scheduler *s)
{ {
struct mandelbrot_args *args = (struct mandelbrot_args *)closure; struct mandelbrot_args *args = (struct mandelbrot_args *)closure;
unsigned int *image = args->image; unsigned int *image = args->image;
unsigned int start_x = args->start_x;
unsigned int start_y = args->start_y;
unsigned int end_x = args->end_x;
unsigned int end_y = args->end_y;
free(closure); free(closure);
for(int y = 0; y < HEIGHT; y++) { if((end_x - start_x) < CHUNK_SIZE && (end_y - start_y) < CHUNK_SIZE) {
for(int x = 0; x < WIDTH; x++) { // Si le morceau est petit alors on dessine
int rc = for(int y = start_y; y < end_y; y++) {
sched_spawn(draw_pixel, new_mandelbrot_args(image, x, y), s); for(int x = start_x; x < end_x; x++) {
assert(rc >= 0); pixel(image, x, y);
}
} }
} else {
// Sinon on recoupe le morceau
int mid_x = (start_x + end_x) / 2;
int mid_y = (start_y + end_y) / 2;
int rc1 = sched_spawn(
draw, new_mandelbrot_args(image, start_x, start_y, mid_x, mid_y),
s);
int rc2 = sched_spawn(
draw, new_mandelbrot_args(image, mid_x, start_y, end_x, mid_y), s);
int rc3 = sched_spawn(
draw, new_mandelbrot_args(image, start_x, mid_y, mid_x, end_y), s);
int rc4 = sched_spawn(
draw, new_mandelbrot_args(image, mid_x, mid_y, end_x, end_y), s);
assert(rc1 >= 0 && rc2 >= 0 && rc3 >= 0 && rc4 >= 0);
} }
} }
@ -155,7 +165,8 @@ benchmark_mandelbrot(int serial, int nthreads)
if(serial) { if(serial) {
draw_serial(image); draw_serial(image);
} else { } else {
rc = sched_init(nthreads, size, draw, new_mandelbrot_args(image, 0, 0)); rc = sched_init(nthreads, size, draw,
new_mandelbrot_args(image, 0, 0, WIDTH, HEIGHT));
assert(rc >= 0); assert(rc >= 0);
} }