def setup(self):
# Set up game here. Call this fn to restart game
self.setup_ground()
if self.state == State.MAIN_MENU:
self.start_screen_sprite_list = arcade.SpriteList()
self.bee_sprite = Bee.setup_bee()
self.start_screen_sprite_list.append(self.bee_sprite)
# initiate & place starting screen
self.start_screen = arcade.Sprite("sprites/start_screen.png", 1)
self.start_screen.center_x = SCREEN_WIDTH / 2
self.start_screen.center_y = SCREEN_HEIGHT / 2
self.start_screen_sprite_list.append(self.start_screen)
elif self.state == State.PLAYING:
#initiate sprite lists
self.bottom_obstacles_list = arcade.SpriteList()
self.top_obstacles_list = arcade.SpriteList()
self.flowers_list = arcade.SpriteList()
self.player_sprite_list = arcade.SpriteList()
#set up bee
self.bee_sprite = Bee.setup_bee()
self.player_sprite_list.append(self.bee_sprite)
#set up bottom and top obstacles
self.bottom_obstacles_list, self.top_obstacles_list = Obstacle.setup_obstacles(
self.bottom_obstacles_list, self.top_obstacles_list)
#place flowers for points
self.flowers_list = Flower.setup(self.flowers_list,
self.bottom_obstacles_list,
self.top_obstacles_list)
#set up score
self.score = 0
self.physics_engine = arcade.PhysicsEnginePlatformer(
self.bee_sprite, self.ground_list, gravity_constant=GRAVITY)
elif self.state == State.GAME_OVER:
self.end_screen_sprite_list = arcade.SpriteList()
self.end_screen = arcade.Sprite("sprites/game_over_screen.png", 1)
self.end_screen.center_x = SCREEN_WIDTH / 2
self.end_screen.center_y = SCREEN_HEIGHT / 2
self.end_screen_sprite_list.append(self.end_screen)
