lol

main.c

@@ -1,40 +1,205 @@
+#include "algorithms.h"
+#include "benchmark.h"
 #include "structs.h"
 #include "render.h"
 #include <SDL2/SDL.h>
+#include <stdlib.h>
 
 #define WINDOW_WIDTH 800
 #define WINDOW_HEIGHT 600
 
-int main(int argc, char *argv[])
+static void print_usage(const char *prog)
 {
-	// Figure 1
-	graph_t *g = generate_graph(20, 1, 0.04, 122);
+	fprintf(stderr,
+		"Usage:\n"
+		"  %s -fig1|-fig2|-fig3|-fig4|-fig5\n"
+		"  %s -b\n"
+		"  %s <n> <p> <q> <clique|louvain|cbla>\n\n"
+		"Options:\n"
+		"  -fig1    preset: n=20,  p=0.9,  q=0.25\n"
+		"  -fig2    preset: n=50,  p=0.7,  q=0.15\n"
+		"  -fig3    preset: n=100, p=0.5,  q=0.05\n"
+		"  -fig4    preset: n=200, p=0.2,  q=0.02\n"
+		"  -fig5    preset: n=500, p=0.01, q=0.00025\n"
+		"  -b       run benchmark\n"
+		"  n        number of nodes\n"
+		"  p        intra-group edge probability\n"
+		"  q        inter-group edge probability\n"
+		"  algo     clique | louvain | cbla\n",
+		prog, prog, prog);
+}
 
-	SDL_Init(SDL_INIT_VIDEO);
+typedef struct {
+	int n;
+	double p;
+	double q;
+	int seed;
+	int k;
+	VISUALIZATION_TYPE algo;
+} run_config_t;
 
-	SDL_Window *window = SDL_CreateWindow(
-		"Graph Render", SDL_WINDOWPOS_CENTERED, SDL_WINDOWPOS_CENTERED,
-		WINDOW_WIDTH, WINDOW_HEIGHT, SDL_WINDOW_SHOWN);
-	SDL_Renderer *renderer =
-		SDL_CreateRenderer(window, -1, SDL_RENDERER_ACCELERATED);
+static int parse_algo(const char *s, VISUALIZATION_TYPE *out)
+{
+	if (strcmp(s, "clique") == 0) {
+		*out = CLIQUE;
+		return 1;
+	} else if (strcmp(s, "louvain") == 0) {
+		*out = LOUVAIN;
+		return 1;
+	} else if (strcmp(s, "cbla") == 0) {
+		*out = CBLA;
+		return 1;
+	}
+	return 0;
+}
 
-	// Figure 2
-	render_graph(renderer, g, LOUVAIN);
+static int parse_fig(const char *arg, run_config_t *cfg)
+{
+	if (strcmp(arg, "-fig1") == 0) {
+		cfg->n = 20;
+		cfg->p = 1.0;
+		cfg->q = 0.04;
+		cfg->k = 3;
+		cfg->seed = 122;
+		cfg->algo = CLIQUE;
+		return 1;
+	} else if (strcmp(arg, "-fig2") == 0) {
+		cfg->n = 20;
+		cfg->q = 1.0;
+		cfg->p = 0.04;
+		cfg->k = 3;
+		cfg->seed = 122;
+		cfg->algo = CLIQUE;
+		return 1;
+	} else if (strcmp(arg, "-fig3") == 0) {
+		cfg->n = 20;
+		cfg->p = 0.04;
+		cfg->q = 0.04;
+		cfg->k = 3;
+		cfg->seed = 122;
+		cfg->algo = CLIQUE;
+		return 1;
+	} else if (strcmp(arg, "-fig4") == 0) {
+		cfg->n = 20;
+		cfg->p = 0.8;
+		cfg->q = 0.10;
+		cfg->k = 0;
+		cfg->seed = 122;
+		cfg->algo = LOUVAIN;
+		return 1;
+	} else if (strcmp(arg, "-fig5") == 0) {
+		cfg->n = 20;
+		cfg->p = 0.8;
+		cfg->q = 0.10;
+		cfg->k = 3;
+		cfg->seed = 122;
+		cfg->algo = CBLA;
+		return 1;
+	}
+	return 0;
+}
 
+static void run_sdl(run_config_t *cfg)
+{
+	printf("[main] generating graph n=%d p=%.4f q=%.6f\n", cfg->n, cfg->p,
+	       cfg->q);
+
+	graph_t *graph = generate_graph(cfg->n, cfg->p, cfg->q, cfg->seed);
+	if (!graph) {
+		fprintf(stderr, "[main] generate_graph failed\n");
+		return;
+	}
+
+	if (SDL_Init(SDL_INIT_VIDEO) < 0) {
+		fprintf(stderr, "SDL_Init: %s\n", SDL_GetError());
+		free_graph(graph);
+		return;
+	}
+	SDL_Window *window =
+		SDL_CreateWindow("Graph Visualizer", SDL_WINDOWPOS_CENTERED,
+				 SDL_WINDOWPOS_CENTERED, WINDOW_WIDTH,
+				 WINDOW_HEIGHT, SDL_WINDOW_SHOWN);
+
+	SDL_Renderer *renderer = SDL_CreateRenderer(
+		window, -1,
+		SDL_RENDERER_ACCELERATED | SDL_RENDERER_PRESENTVSYNC);
+
+	render_graph(renderer, graph, cfg->algo, cfg->k);
+
+	// Event loop
 	SDL_Event e;
 	int running = 1;
 	while (running) {
-		while (SDL_PollEvent(&e))
+		while (SDL_PollEvent(&e)) {
 			if (e.type == SDL_QUIT)
 				running = 0;
+			if (e.type == SDL_KEYDOWN &&
+			    e.key.keysym.sym == SDLK_ESCAPE)
+				running = 0;
+			// Re-render on window resize
+			if (e.type == SDL_WINDOWEVENT &&
+			    e.window.event == SDL_WINDOWEVENT_RESIZED)
+				render_graph(renderer, graph, cfg->algo,
+					     cfg->k);
+		}
 		SDL_Delay(16);
 	}
 
 	SDL_DestroyRenderer(renderer);
 	SDL_DestroyWindow(window);
 	SDL_Quit();
+	free_graph(graph);
+}
 
-	free_graph(g);
+int main(int argc, char *argv[])
+{
+	if (argc < 2) {
+		print_usage(argv[0]);
+		return 1;
+	}
+
+	if (strcmp(argv[1], "-b") == 0) {
+		run_benchmark();
+		return 0;
+	}
+
+	run_config_t cfg = {
+		.n = 0, .p = 0, .q = 0, .seed = 42, .algo = CLIQUE
+	};
+
+	if (parse_fig(argv[1], &cfg)) {
+		// optionally override algo: ./graphe -fig3 louvain
+		if (argc >= 3 && !parse_algo(argv[2], &cfg.algo)) {
+			fprintf(stderr, "[main] unknown algo '%s'\n", argv[2]);
+			print_usage(argv[0]);
+			return 1;
+		}
+		run_sdl(&cfg);
+		return 0;
+	}
+
+	if (argc < 5) {
+		print_usage(argv[0]);
+		return 1;
+	}
+
+	cfg.n = atoi(argv[1]);
+	cfg.p = atof(argv[2]);
+	cfg.q = atof(argv[3]);
+
+	if (cfg.n <= 0 || cfg.p < 0 || cfg.p > 1 || cfg.q < 0 || cfg.q > 1) {
+		fprintf(stderr,
+			"[main] invalid arguments: n must be > 0, p and q in [0,1]\n");
+		return 1;
+	}
+	if (!parse_algo(argv[4], &cfg.algo)) {
+		fprintf(stderr, "[main] unknown algo '%s'\n", argv[4]);
+		print_usage(argv[0]);
+		return 1;
+	}
+
+	run_sdl(&cfg);
+	return 0;
 
 	return 0;
 }