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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#include "gtools.h"
#include "geng.h"
#include <ucontext.h>
#include <stdint.h>
#include <stdbool.h>
// TODO: only on macos
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
void
printgraph(graph *g, int n)
{
set *gi;
for (int i = 0; i < n; ++i)
{
gi = GRAPHROW(g, 1, i);
for (int j = 0; j < n; ++j)
{
int q = ISELEMENT(gi, j);
printf("%i ", q);
}
printf("\n");
}
printf("\n");
}
// TODO: probably don't need to name this function with macro
void
OUTPROC(__attribute__((unused)) FILE *outfile,
graph *g, int n, struct geng_iterator *iter)
{
iter->cur = g;
iter->graph_size = n;
// TODO: add support for generating graphs in batches (for speed)
#if 0
static int i = 1;
i++;
if (i % 100 == 0)
{
swapcontext(&geng_worker, &geng_user);
i = 0;
}
#else
swapcontext(&iter->geng_worker, &iter->geng_user);
#endif
}
// TODO: probably don't need to name this function with macro
// TODO: move to geng.h maybe?
extern int
GENG_MAIN(int argc, char *argv[]);
// geng_iterator_init gives ownership of iterator memory to caller
void
geng_iterator_create(struct geng_iterator **iterator_ptr,
size_t graph_size,
size_t batch_capacity)
{
// TODO: malloc geng_iterator batch
*iterator_ptr = malloc(sizeof(struct geng_iterator));
struct geng_iterator *iterator = *iterator_ptr;
iterator->iteration_done = false;
// TODO: add support for more arguments
int geng_argc = 3;
char **geng_argv;
geng_argv = malloc((geng_argc + 1) * sizeof(char *));
geng_argv[0] = "geng";
geng_argv[1] = "-q";
char n_str[20];
snprintf(n_str, sizeof(n_str), "%zu", graph_size);
geng_argv[2] = n_str;
geng_argv[3] = NULL;
uint32_t p_argv[2];
p_argv[0] = (uint32_t) (((size_t) &geng_argv) & ((1llu << 32) - 1llu));
p_argv[1] = ((size_t) &geng_argv) >> 32;
uint32_t p_iter[2];
p_iter[0] = (uint32_t) (((size_t) &iterator) & ((1llu << 32) - 1llu));
p_iter[1] = ((size_t) &iterator) >> 32;
getcontext(&iterator->geng_user);
iterator->geng_worker = iterator->geng_user;
iterator->geng_worker.uc_stack.ss_sp = iterator->geng_stack;
iterator->geng_worker.uc_stack.ss_size = sizeof(iterator->geng_stack);
iterator->geng_worker.uc_link = &iterator->geng_user;
makecontext(
&iterator->geng_worker, (void (*) (void)) GENG_MAIN,
5, geng_argc, p_argv[0], p_argv[1], p_iter[0], p_iter[1]
);
swapcontext(&iterator->geng_user, &iterator->geng_worker);
free(geng_argv);
}
bool
geng_iterator_next(struct geng_iterator *iter, graph *g)
{
if (iter->iteration_done) return false;
else
{
memcpy(g, iter->cur, sizeof(set) * iter->graph_size);
swapcontext(&iter->geng_user, &iter->geng_worker);
if (!iter->iteration_done && iter->generation_done)
iter->iteration_done = true;
return true;
}
}
// TODO: only on macos
#pragma GCC diagnostic pop
|
