arcade/main.py

import arcade
from random import randint
import numpy as np

SCREEN_SIZE = (800,600)
TICK_TIME = 0.5

class Wolf(arcade.SpriteCircle):
  ls = arcade.SpriteList()
  def __init__(self,parent=None):
    super().__init__(radius=5, color=arcade.color.RED)
    if parent:
      self.center_x = parent.center_x
      self.center_y = parent.center_y
    else:
      self.center_x = randint(1,SCREEN_SIZE[0])
      self.center_y = randint(1,SCREEN_SIZE[1])
    self.age = 0
    self.age_to_bride = 3
    self.health = 30
    self.speed = 8
    self.move_target = 0
    self.target_timer = 0
    Wolf.ls.append(self)
  @classmethod
  def update(cls):
    if len(cls.ls) == 0:
      Wolf()
    for e in cls.ls:
      e.age += 1
      if e.age >= 30:
        cls.ls.remove(e)
        continue

      if e.health >= 50 and e.age_to_bride < e.age:
        Wolf(e)
        e.health -= 30

      hit_cow = arcade.check_for_collision_with_list(e,Cow.ls)
      for c in hit_cow:
        e.health += c.health
        Cow.ls.remove(c)
      else:
        if e.target_timer <= 0:
          e.move_target = Cow.get_near(e.center_x,e.center_y)
          e.target_timer = 3
        if e.move_target:
          vec = cls.get_vector(e.center_x,e.move_target.center_x,
                             e.center_y,e.move_target.center_y)
          new_x = e.center_x + vec[0]*e.speed
          new_y = e.center_y + vec[1]*e.speed
          if 0 < new_x < SCREEN_SIZE[0] and 0 < new_y < SCREEN_SIZE[1]:
            e.center_x = new_x
            e.center_y = new_y
            e.health -= 1
            if e.health <= 0:
              cls.ls.remove(e)
    
  @classmethod
  def draw(cls):
    cls.ls.draw()
  @staticmethod
  def get_vector(src_x,target_x,src_y,target_y):
    source = np.array([src_x, src_y])
    target = np.array([target_x, target_y])
    direction = target - source
    length = np.linalg.norm(direction)
    return direction / length
    

class Cow(arcade.SpriteCircle):
  ls = arcade.SpriteList()
  def __init__(self,parent=None):
    super().__init__(radius=5, color=arcade.color.BROWN)
    if parent:
      self.center_x = parent.center_x
      self.center_y = parent.center_y
    else:
      self.center_x = randint(1,SCREEN_SIZE[0])
      self.center_y = randint(1,SCREEN_SIZE[1])
    self.age = 0
    self.age_to_bride = 3
    self.health = 30
    self.speed = 5
    self.move_target = 0
    self.target_timer = 0
    Cow.ls.append(self)
  @classmethod
  def get_near(cls,x,y):
    dist_list = list()
    for e in cls.ls:
      dist_list.append((e,((e.center_x - x)**2 + (e.center_y - y)**2)**0.5))
    dist_list.sort(key=lambda x: x[1])
    if len(dist_list) == 0:
      return None
    return dist_list[0][0]
  @classmethod
  def update(cls):
    if len(cls.ls) == 0:
      Cow()
    for c in cls.ls:
      c.age += 1
      if c.age >= 30:
        cls.ls.remove(c)
        continue

      if c.health >= 50 and c.age_to_bride < c.age:
        Cow(c)
        c.health -= 30

      hit_plant = arcade.check_for_collision_with_list(c,Plant.ls)
      for p in hit_plant:
        c.health += p.width
        Plant.ls.remove(p)
      else:
        if c.target_timer <= 0:
          c.move_target = Plant.get_near(c.center_x,c.center_y)
          c.target_timer = 5
        if c.move_target:
          vec = cls.get_vector(c.center_x,c.move_target.center_x,
                             c.center_y,c.move_target.center_y)
          new_x = c.center_x + vec[0]*randint(0,c.speed)
          new_y = c.center_y + vec[1]*randint(0,c.speed)
          if 0 < new_x < SCREEN_SIZE[0] and 0 < new_y < SCREEN_SIZE[1]:
            c.center_x = new_x
            c.center_y = new_y
            c.health -= 1
            if c.health <= 0:
              cls.ls.remove(c)
    
  @classmethod
  def draw(cls):
    cls.ls.draw()
  @staticmethod
  def get_vector(src_x,target_x,src_y,target_y):
    source = np.array([src_x, src_y])
    target = np.array([target_x, target_y])
    direction = target - source
    length = np.linalg.norm(direction)
    return direction / length

class Plant(arcade.SpriteCircle):
  ls = arcade.SpriteList()

  def __init__(self,parent=None):
    super().__init__(radius=1, color=arcade.color.GREEN)
    if parent:
      self.center_x = parent.center_x + randint(-parent.fly_distance,parent.fly_distance)
      self.center_y = parent.center_y + randint(-parent.fly_distance,parent.fly_distance)
    else:
      self.center_x = randint(1,SCREEN_SIZE[0])
      self.center_y = randint(1,SCREEN_SIZE[1])
    self.age = 0
    self.fly_distance = 100
    self.age_to_plant = 10
    if not arcade.check_for_collision_with_list(self,Plant.ls) and 0 < self.center_x < SCREEN_SIZE[0] and 0 < self.center_y < SCREEN_SIZE[1]:
      Plant.ls.append(self)
  @classmethod
  def get_near(cls,x,y):
    dist_list = list()
    for p in cls.ls:
      dist_list.append((p,((p.center_x - x)**2 + (p.center_y - y)**2)**0.5 - p.width ))
    dist_list.sort(key=lambda x: x[1])
    if len(dist_list) == 0:
      return None
    if len(dist_list) < 3:
      return dist_list[0][0]
    return dist_list[randint(0,3)][0]
  @classmethod
  def update(cls):
    for p in cls.ls:
      p.age += 1
      hit_list = arcade.check_for_collision_with_list(p,cls.ls)
      if p.center_x - p.width//2 > 0 and p.center_x + p.width//2 < SCREEN_SIZE[0] and p.center_y - p.height//2 > 0 and p.center_y + p.width//2 < SCREEN_SIZE[1] and not hit_list:
        p.width += 2
        p.height += 2
      else:
        for h in hit_list:
          if p.width < h.width:
            cls.ls.remove(p)
            break
      if p.age >= p.age_to_plant:
        Plant(p)
  @classmethod
  def draw(cls):
    cls.ls.draw()
    for p in cls.ls:
      p.draw_hit_box()

class MainView(arcade.View):
  def __init__(self):
    super().__init__()
    self.background_color = (100,100,200)
    Plant()
    self.time = 0
    self.cur_time = 0
    self.text = None

  def on_update(self, delta_time):
    self.cur_time += delta_time
    if self.cur_time >= TICK_TIME:
      print(f'Total plants: {len(Plant.ls)}, total cows: {len(Cow.ls)}')
      Plant.update()
      Cow.update()
      Wolf.update()
      self.time += 1
      self.cur_time = 0
    
  def on_draw(self):
    self.clear()
    Plant.draw()
    Cow.draw()
    Wolf.draw()
    

if __name__ == '__main__':
  window = arcade.Window(*SCREEN_SIZE,"Игра про жизнь")
  view = MainView()
  window.show_view(view)
  arcade.run()