RIOT-OS/RIOT
RIOT-OS/RIOT
1
0
Fork 0
mirror of https://github.com/RIOT-OS/RIOT.git synced 2025-01-18 12:52:44 +01:00
Code Releases Activity

tools/zep_dispatch: topogen: add support for generating grid topology

Browse Source
This commit is contained in:
Benjamin Valentin 2023-02-15 16:51:53 +01:00
parent c9f7090240
commit b4efc378a0
1 changed files with 34 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
dist/tools/zep_dispatch/topogen.c vendored
View File

@ -27,6 +27,7 @@ struct world {
unsigned h; unsigned h;
unsigned num_nodes; unsigned num_nodes;
struct node *nodes; struct node *nodes;
bool grid;
}; };
static unsigned random_range(unsigned lower, unsigned upper) static unsigned random_range(unsigned lower, unsigned upper)
@ -35,13 +36,31 @@ static unsigned random_range(unsigned lower, unsigned upper)
return lower + rand() % range; return lower + rand() % range;
} }
static struct node *node_generate(struct node *n, const struct world *w, static struct node *node_generate(struct node *node, const struct world *w,
unsigned range) unsigned range, unsigned idx)
{ {
n->x = random_range(0, w->w); if (w->grid) {
n->y = random_range(0, w->h); float width = w->w;
n->r = range; float height = w->h;
return n; float num = w->num_nodes;
/* https://math.stackexchange.com/a/1039514 */
float n_x = sqrtf(num * width/height
+ powf(width - height, 2)/(4 * height*height))
- (width - height)/2;
float n_y = num / n_x;
unsigned step_x = width / n_x;
unsigned step_y = height / n_y;
node->x = (idx * step_x + step_x / 2) % w->w;
node->y = ((idx * step_x + step_x / 2) / w->w) * step_y + step_y / 2;
} else {
node->x = random_range(0, w->w);
node->y = random_range(0, w->h);
}
node->r = range;
return node;
} }
static double node_distance(const struct node *a, const struct node *b) static double node_distance(const struct node *a, const struct node *b)
@ -82,7 +101,7 @@ static void world_gen(struct world *w, unsigned num_nodes,
w->nodes = calloc(num_nodes, sizeof(*w->nodes)); w->nodes = calloc(num_nodes, sizeof(*w->nodes));
for (unsigned i = 0; i < num_nodes; ++i) { for (unsigned i = 0; i < num_nodes; ++i) {
node_generate(&w->nodes[i], w, random_range(range - var, range + var)); node_generate(&w->nodes[i], w, random_range(range - var, range + var), i);
node_name(&w->nodes[i], i); node_name(&w->nodes[i], i);
} }
} }
@ -205,13 +224,17 @@ int main(int argc, char** argv)
unsigned var = 0; unsigned var = 0;
unsigned num = 10; unsigned num = 10;
bool binary = false; bool binary = false;
bool grid = false;
char c; char c;
while ((c = getopt(argc, argv, "s:w:h:r:v:n:b")) != -1) { while ((c = getopt(argc, argv, "s:w:h:r:v:n:bg")) != -1) {
switch (c) { switch (c) {
case 'b': case 'b':
binary = true; binary = true;
break; break;
case 'g':
grid = true;
break;
case 's': case 's':
seed = atoi(optarg); seed = atoi(optarg);
break; break;
@ -238,7 +261,9 @@ int main(int argc, char** argv)
srand(seed); srand(seed);
struct world w; struct world w = {
.grid = grid,
};
world_gen(&w, num, width, height, range, var); world_gen(&w, num, width, height, range, var);
printf("# seed = %u\n", seed); printf("# seed = %u\n", seed);