diff --git a/flower_game_env.py b/flower_game_env.py index 38fa0d7..100c069 100644 --- a/flower_game_env.py +++ b/flower_game_env.py @@ -357,18 +357,21 @@ class FlowerGameEnv(gym.Env): self.prev_dist_to_flower_nd = None # Eat-Event (Wenn die Blume woanders neu spawnt) - if self._cooldown == 0 and tf and ff: - if self.prev_flower_x is None: + if self._cooldown == 0 and tf: + if not ff: + pyautogui.press("r") + else: + if self.prev_flower_x is None: + self.prev_flower_x = fx + if self.prev_flower_y is None: + self.prev_flower_y = fy + if abs(self.prev_flower_x - fx) >= float(w/20 if w > 0 else 1) and abs(self.prev_flower_y - fy) >= float(h/15 if h > 0 else 1): + print("Blume gegessen!") + reward += self.eat_reward + self._cooldown = self.contact_cooldown_frames + self.flowers_eaten += 1 self.prev_flower_x = fx - if self.prev_flower_y is None: self.prev_flower_y = fy - if abs(self.prev_flower_x - fx) >= float(w/20 if w > 0 else 1) and abs(self.prev_flower_y - fy) >= float(h/15 if h > 0 else 1): - print("Blume gegessen!") - reward += self.eat_reward - self._cooldown = self.contact_cooldown_frames - self.flowers_eaten += 1 - self.prev_flower_x = fx - self.prev_flower_y = fy # Überprüfung auf Kollision mit Bombe / Game Over Screen Farben rgb_data = self.raw[:, :, :3]