turbopool.c


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

#include <u.h>
#include <libc.h>
#include <thread.h>
#include <draw.h>
#include <memdraw.h>
#include <geometry.h>
#include "libobj/obj.h"
#include "graphics.h"
#include "internal.h"

static void
_turboproc(void *arg)
{
	Turbopool *p;
	Turbotask *t;

	threadsetname("turboproc");

	p = arg;

	while((t = recvp(p->subq)) != nil){
		t->fn(t->arg);
		free(t);
	}
}

Turbopool *
mkturbopool(ulong nprocs)
{
	Turbopool *p;

	p = emalloc(sizeof *p);
	memset(p, 0, sizeof *p);
	p->nprocs = nprocs;
	p->subq = chancreate(sizeof(void*), nprocs);
	while(nprocs--)
		proccreate(_turboproc, p, mainstacksize);
	return p;
}

void
turbopoolexec(Turbopool *p, void (*fn)(void*), void *arg)
{
	Turbotask *t;

	t = emalloc(sizeof *t);
	t->fn = fn;
	t->arg = arg;

	sendp(p->subq, t);
}

void
rmturbopool(Turbopool *p)
{
	while(p->nprocs--)
		sendp(p->subq, nil);
	chanfree(p->subq);
	free(p);
}