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()