From 0bf7fd38aea996a381717fe4be09817259575531 Mon Sep 17 00:00:00 2001
From: SemperVinco <w.deweijer@hotmail.com>
Date: Wed, 1 Mar 2017 22:24:57 +0100
Subject: Add position argument option -p which places bots in specific
 locations, syntax: ./sim -p b1.rob 1 2 b2.rob -2 4

---
 main.cpp | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

(limited to 'main.cpp')

diff --git a/main.cpp b/main.cpp
index 36f5580..30002a8 100644
--- a/main.cpp
+++ b/main.cpp
@@ -4,6 +4,7 @@
 #include <string>
 #include <unordered_map>
 #include <cstdlib>
+#include <array>
 #include <cctype>
 #include <cassert>
 #include <sys/time.h>
@@ -215,14 +216,32 @@ int main(int argc,char **argv){
 
 	World world;
 	vector<Team> teams;
-	for(int i=1;i<argc;i++){
+
+	bool opt_pos = false;
+	vector<array<int,2>> positions; // TODO: use pair?
+
+	int k = 1; // Increase past options;
+
+	if(string(argv[k]) == "-p") {
+		opt_pos = true;
+		k++;
+	}
+
+	for(int i=k;i<argc;i++){ // Start after options
 		ifstream f(argv[i]);
 		assert(f);
 		teams.push_back(assemble(preprocess(f)));
+		if(opt_pos) {
+			positions.push_back({strtol(argv[i+1],NULL,10), strtol(argv[i+2],NULL,10)}); // TODO: check if i>argc
+			i += 2;
+		}
 	}
 
-	for(const Team &t : teams){
-		Robot &r=world.create(&t,2,t.banks.size(),false);
+	for(int i=0; i<teams.size(); i++) {
+		const Team &t = teams[i];
+		Robot &r = opt_pos ? world.create(&t,2,t.banks.size(),false, positions[i][0], positions[i][1])
+		                   : world.create(&t,2,t.banks.size(),false);
+
 		for(int i=0;i<(int)t.banks.size();i++){
 			r.load(i,t.banks[i]);
 			r.active=1;
-- 
cgit v1.2.3-70-g09d2