From a11119abe39d75af29ad381d3ec264973e3365b2 Mon Sep 17 00:00:00 2001
From: ineedbots <ineedbots@outlook.com>
Date: Mon, 14 Jun 2021 13:20:39 -0600
Subject: [PATCH] abstract class choice

---
 main_shared/maps/mp/bots/_bot_script.gsc  |  4 ++--
 main_shared/maps/mp/bots/_bot_utility.gsc | 16 ++++++++++++++++
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/main_shared/maps/mp/bots/_bot_script.gsc b/main_shared/maps/mp/bots/_bot_script.gsc
index e291c3d..c934ac4 100644
--- a/main_shared/maps/mp/bots/_bot_script.gsc
+++ b/main_shared/maps/mp/bots/_bot_script.gsc
@@ -248,7 +248,7 @@ classWatch()
 
 	for(;;)
 	{
-		while(!isdefined(self.pers["team"]) || level.oldschool)
+		while(!isdefined(self.pers["team"]) || !allowClassChoice())
 			wait .05;
 			
 		wait 0.5;
@@ -271,7 +271,7 @@ teamWatch()
 
 	for(;;)
 	{
-		while(!isdefined(self.pers["team"]))
+		while(!isdefined(self.pers["team"])  || !allowTeamChoice())
 			wait .05;
 			
 		wait 0.05;
diff --git a/main_shared/maps/mp/bots/_bot_utility.gsc b/main_shared/maps/mp/bots/_bot_utility.gsc
index 9ba79f6..c97e1b6 100644
--- a/main_shared/maps/mp/bots/_bot_utility.gsc
+++ b/main_shared/maps/mp/bots/_bot_utility.gsc
@@ -517,6 +517,22 @@ waittill_either_return(str1, str2)
 	return str2;
 }
 
+/*
+	iw5
+*/
+allowClassChoice()
+{
+	return true;
+}
+
+/*
+	iw5
+*/
+allowTeamChoice()
+{
+	return true;
+}
+
 /*
 	Taken from iw4 script
 */