def createTeam(self, requested_team_name, play_offense): """ Given a team name, returns the team object. """ teams_dir = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'teams') if requested_team_name == 'helios': print('Creating team Helios') team_name = 'HELIOS_' + ('left' if play_offense else 'right') team_dir = os.path.join(teams_dir, 'helios', 'helios-13Eindhoven') lib_dir = os.path.join(teams_dir, 'helios', 'local', 'lib') return Teams.Helios(team_name, team_dir, lib_dir, binaryName='helios_player', host='localhost', port=self._serverPort) elif requested_team_name == 'base': print('Creating team Agent2d (base)') team_name = 'base_' + ('left' if play_offense else 'right') team_dir = os.path.join(teams_dir, 'base') lib_dir = os.path.join(teams_dir, 'base', 'lib') return Teams.Agent2d(team_name, team_dir, lib_dir, binaryName='sample_player', logDir=self._logDir, record=self._record, host='localhost', port=self._serverPort) elif requested_team_name == 'cyrus': print('Creating team Cyrus') team_name = 'CYRUS_' + ('left' if play_offense else 'right') team_dir = os.path.join(teams_dir, 'cyrus') lib_dir = os.path.join(teams_dir, 'cyrus', 'lib') return Teams.Cyrus(team_name, team_dir, lib_dir, binaryName='sample_player', host='localhost', port=self._serverPort) elif requested_team_name == 'gliders': print('Creating team Gliders2d') team_name = 'Gliders_' + ('left' if play_offense else 'right') team_dir = os.path.join(teams_dir, 'gliders') lib_dir = None return Teams.Gliders(team_name, team_dir, lib_dir, binaryName='sample_player', host='localhost', port=self._serverPort) else: print('Unknown team requested: ' + requested_team_name) sys.exit(1)