hex

commit e1412842bf102d7c118e60a191016809bfd6bd94
parent aca3eae1effef4c0a09cbb347365168538bcafd3
Author: Mikołaj Lenczewski <mblenczewski@gmail.com>
Date:   Wed, 16 Aug 2023 23:09:33 +0000

hexes: Implemented argument parsing

Diffstat:
M
agents/hexes/include/hexes.h
 | 
1
+
M
agents/hexes/run.sh
 | 
2
+-
M
agents/hexes/src/hexes.c
 | 
62
++++++++++++++++++++++++++++++++++++++++++++++++++++++--------

3 files changed, 56 insertions(+), 9 deletions(-)
diff --git a/agents/hexes/include/hexes.h b/agents/hexes/include/hexes.h
@@ -39,6 +39,7 @@
 
 struct opts {
 	u32 log_level, agent_type;
+	char *host, *port;
 };
 
 extern struct opts opts;
diff --git a/agents/hexes/run.sh b/agents/hexes/run.sh
@@ -1,3 +1,3 @@
 #!/bin/sh
 
-exec $(dirname $0)/hexes $@
+exec $(dirname $0)/hexes -amcts $HEXES_OPTS $@
diff --git a/agents/hexes/src/hexes.c b/agents/hexes/src/hexes.c
@@ -7,9 +7,14 @@
 
 struct opts opts = {
 	.log_level = LOG_INFO,
-	.agent_type = AGENT_MCTS,
+	.agent_type = AGENT_RANDOM,
+	.host = NULL,
+	.port = NULL,
 };
 
+static bool
+argparse(int argc, char **argv, struct opts *opts);
+
 enum game_state {
 	GAME_START,
 	GAME_RECV,
@@ -52,14 +57,13 @@ main(int argc, char **argv)
 {
 	srandom(getpid());
 
-	if (argc < 3) {
-		dbglog(LOG_ERROR, "Usage: %s <host> <port>\n", argv[0]);
-		exit(EXIT_FAILURE);
-	}
+	if (!argparse(argc, argv, &opts)) exit(EXIT_FAILURE);
+
+	dbglog(LOG_DEBUG, "Opts: log_level: %" PRIu32 ", agent_type: %" PRIu32 ", host: %s, port: %s\n",
+			opts.log_level, opts.agent_type, opts.host, opts.port);
 
-	char *host = argv[1], *port = argv[2];
-	if (!network_init(&game.network, host, port)) {
-		dbglog(LOG_ERROR, "Failed to initialise network (connecting to %s:%s)\n", host, port);
+	if (!network_init(&game.network, opts.host, opts.port)) {
+		dbglog(LOG_ERROR, "Failed to initialise network (connecting to %s:%s)\n", opts.host, opts.port);
 		exit(EXIT_FAILURE);
 	}
 
@@ -85,6 +89,48 @@ main(int argc, char **argv)
 	exit(EXIT_SUCCESS);
 }
 
+static bool
+argparse(int argc, char **argv, struct opts *opts)
+{
+	assert(opts);
+
+	char const *optstr = "va:";
+
+	int opt;
+	while ((opt = getopt(argc, argv, optstr)) != -1) {
+		switch (opt) {
+		case 'v':
+			opts->log_level = LOG_DEBUG;
+			break;
+
+		case 'a':
+			if (strcmp(optarg, "random") == 0) {
+				opts->agent_type = AGENT_RANDOM;
+			} else if (strcmp(optarg, "mcts") == 0) {
+				opts->agent_type = AGENT_MCTS;
+			} else {
+				fprintf(stderr, "Unknown agent: %s.\n", optarg);
+				goto error;
+			}
+			break;
+
+		default: goto error; /* ? */
+		}
+	}
+
+	if (optind + 2 > argc) goto error;
+
+	opts->host = argv[optind++];
+	opts->port = argv[optind];
+
+	return true;
+
+error:
+	fprintf(stderr, "Usage: %s [-v] [-a random|mcts] <host> <port>\n", argv[0]);
+
+	return false;
+}
+
 static void
 start_handler(struct game *game)
 {