def get_sam_vs_fed(cls, sam_brain_foo, fed_brain_foo):
scr = Screen(700, 700, (-10, -10), (60, 60))
sc = Scene.get_standart(50, 50, 3, 3)
poss = [sc.pos_1, sc.pos_2]
np.random.shuffle(poss)
p1, p2 = poss
u1 = Unit.get_some(p1,
'Samoilov',
color=(36, 235, 130),
brain_foo=sam_brain_foo)
u2 = Unit.get_some(p2,
'Fedulov',
color=(233, 44, 44),
brain_foo=fed_brain_foo)
return cls(sc, scr, u2, u1)
def get_standart(cls, you_brain_foo, enemy_brain_foo=None):
scr = Screen(700, 700, (-10, -10), (60, 60))
sc = Scene.get_standart(50, 50, 3, 3)
poss = [sc.pos_1, sc.pos_2]
np.random.shuffle(poss)
p1, p2 = poss
u1 = Unit.get_some(p1,
'you',
color=(36, 235, 130),
brain_foo=you_brain_foo)
u2 = Unit.get_some(p2,
'enemy',
color=(233, 44, 44),
brain_foo=enemy_brain_foo)
u2.draw_stuff = False
return cls(sc, scr, u2, u1)
def get_standart2(cls, you_brain_foo, enemy_brain_foo=None):
scr = Screen(700, 700, (-10, -10), (60, 60))
sc = Scene.get_standart(50, 50, 3, 3)
p1, p2 = sc.pos_1, sc.pos_1 + (10, 10)
u1 = Unit.get_some(p1,
'you',
color=(36, 235, 130),
brain_foo=you_brain_foo)
u2 = Unit.get_some(p2,
'enemy',
color=(233, 44, 44),
brain_foo=enemy_brain_foo)
u1.set_alpha(30)
u2.set_alpha(170)
u2.draw_stuff = False
res = cls(sc, scr, u2, u1)
# res.units = []
return res
def __del__(self):
"""Destructor
"""
self.close()
def _initpygame(self):
if self.pygame:
return
import pygame
self.pygame = pygame
self.pygame.init()
self.pygame.font.init()
self.screen_main = pygame.display.set_mode(
(self.display_width, self.display_height))
self.clock = pygame.time.Clock()
self.myfont = pygame.font.SysFont('Arial', 30)
if __name__ == "__main__":
from scene import Scene
scr = Screen(700, 700, (-10, -10), (60, 60))
sc = Scene.get_standart(50, 50, 3, 3)
u = Unit.get_some((10, 10))
scr.fill()
for pl in sc.obstacles:
scr.draw(pl)
scr.draw(u)
scr.update()
input()
