1 files changed, 39 insertions, 0 deletions
diff --git a/game/server/tf/bot/map_entities/tf_bot_roster.h b/game/server/tf/bot/map_entities/tf_bot_roster.h
new file mode 100644
index 0000000..e3931ba
--- /dev/null
+++ b/game/server/tf/bot/map_entities/tf_bot_roster.h
@@ -0,0 +1,39 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+// tf_bot_roster.h
+// entity that dictates what classes a bot can choose when spawning
+// Tom Bui, April 2010
+
+#ifndef TF_BOT_ROSTER_H
+#define TF_BOT_ROSTER_H
+
+class CTFBotRoster : public CPointEntity
+{
+	DECLARE_CLASS( CTFBotRoster, CPointEntity );
+public:
+	DECLARE_DATADESC();
+
+	CTFBotRoster( void );
+	virtual ~CTFBotRoster() {}
+
+	// input
+	void InputSetAllowScout( inputdata_t &inputdata );
+	void InputSetAllowSniper( inputdata_t &inputdata );
+	void InputSetAllowSoldier( inputdata_t &inputdata );
+	void InputSetAllowDemoman( inputdata_t &inputdata );
+	void InputSetAllowMedic( inputdata_t &inputdata );
+	void InputSetAllowHeavy( inputdata_t &inputdata );
+	void InputSetAllowPyro( inputdata_t &inputdata );
+	void InputSetAllowSpy( inputdata_t &inputdata );
+	void InputSetAllowEngineer( inputdata_t &inputdata );
+
+	// misc.
+	bool IsClassAllowed( int iBotClass ) const;
+	bool IsClassChangeAllowed() const;
+
+public:
+	string_t m_teamName;
+	bool m_bAllowClassChanges;
+	bool m_bAllowedClasses[TF_LAST_NORMAL_CLASS];
+};
+
+#endif // TF_BOT_ROSTER_H