From 1dbc7379e5dab5821b9afc9aee2e6dcc3cf815f7 Mon Sep 17 00:00:00 2001
From: Sam Anthony <sam@samanthony.xyz>
Date: Fri, 4 Oct 2024 23:17:42 -0400
Subject: parameterize number of balls

---
 balls.cpp | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/balls.cpp b/balls.cpp
index bdae497..2a0f5b9 100644
--- a/balls.cpp
+++ b/balls.cpp
@@ -50,6 +50,8 @@ static vector<Ball> balls;
 
 int
 main(int argc, char *argv[]) {
+	int nballs;
+
 	glutInit(&argc, argv);
 	glutInitDisplayMode(GLUT_DOUBLE | GLUT_RGBA);
 	glutInitWindowSize(WIDTH, HEIGHT);
@@ -59,7 +61,15 @@ main(int argc, char *argv[]) {
 	glutDisplayFunc(display);
 	glutReshapeFunc(reshape);
 
-	balls = makeBalls(NBALLS_DEFAULT);
+	nballs = NBALLS_DEFAULT;
+	if (argc > 1) {
+		if (sscanf(argv[1], "%d", &nballs) != 1 || nballs < 1) {
+			printf("usage: balls [number of balls]\n");
+			return 1;
+		}
+	}
+
+	balls = makeBalls(nballs);
 
 	glutTimerFunc(FRAME_TIME_MS, animate, 0);
 
-- 
cgit v1.2.3