from agarnet.agarnet import client from agarnet.agarnet import utils import pygame from pygame import gfxdraw from pygame.locals import * import sys import math import time import mechanics from agarnet.agarnet.vec import Vec running = True font_fallback = False try: from pygame import freetype except: font_fallback = True if font_fallback: pygame.font.init() else: pygame.freetype.init() logging = False input = False bot_input = True clock = pygame.time.Clock() marker = [(0,0),(0,0),(0,0)] marker_updated = [True, True, True] screensize=(1280, 800) screen=pygame.display.set_mode(screensize,HWSURFACE|DOUBLEBUF|RESIZABLE) vignette = 1. def enable_vignette(factor=0.15): global vignette if factor: vignette = 1+factor else: vignette = 1 def draw_bar(rect, val, thresh=None, min=0, max=1, color=(0,0,0), barcolor=None, exceedcolor=(255,0,0), threshcolor=None): v = (val-min)/(max-min) t = (thresh-min)/(max-min) if barcolor == None: barcolor_=color else: barcolor_=barcolor if thresh != None and threshcolor==None: threshcolor_ = ((128+color[0])//2, (128+color[1])//2, (128+color[2])//2) else: threshcolor_ = threshcolor for i in range(0, 1 if v= t: draw_box(((rect[0][0]+2 + (rect[1][0]-4)*t , rect[0][1]+2) , ((rect[1][0]-4)*(v-t) , rect[1][1]-4)), exceedcolor, True, False) draw_line((rect[0][0]+2+(rect[1][0]-4)*t, rect[0][1]+1), (rect[0][0]+2+(rect[1][0]-4)*t, rect[0][1]+rect[1][1]-1), threshcolor_, False) def draw_line(p1, p2, color, global_coords=True): if global_coords: p1 = world_to_win_pt(p1, c.player.center) p2 = world_to_win_pt(p2, c.player.center) gfxdraw.line(screen, int(p1[0]), int(p1[1]), int(p2[0]), int(p2[1]), color) def draw_box(rect, color, filled=False, global_coords=True): if global_coords: t = ( world_to_win_pt(rect[0][0], c.player.center), world_to_win_pt(rect[0][1], c.player.center) ) rect = (t, rect[1]) if filled: screen.fill(color, rect) else: gfxdraw.rectangle(screen, rect, color) def draw_circle(pos, r, color, filled=False, global_coords=True): if global_coords: pos = world_to_win_pt(pos, c.player.center) r = world_to_win_length(r) if filled: gfxdraw.filled_circle(screen, pos[0], pos[1], r, color) else: gfxdraw.circle(screen, pos[0], pos[1], r, color) gfxdraw.aacircle(screen, pos[0], pos[1], r, color) def hilight_cell(cell, color_inner, color_outer, r=20): draw_circle(cell.pos, cell.size+r, color_outer, True) draw_circle(cell.pos, cell.size+r/2, color_inner, True) draw_cell(cell) def draw_polygon(polygon, color, filled=False, global_coords=True): if len(polygon) > 2: if global_coords: polygon = list(map(lambda x: world_to_win_pt(x, c.player.center), polygon)) if filled: gfxdraw.filled_polygon(screen, polygon, color) else: gfxdraw.polygon(screen, polygon, color) gfxdraw.aapolygon(screen, polygon, color) def draw_path(path, color, global_coords=True): for i in range(1, len(path)): draw_line(path[i-1], path[i], color, global_coords) def draw_arc(pos, r, bounds, color, global_coords=True): if global_coords: pos = world_to_win_pt(pos, c.player.center) gfxdraw.arc(screen, pos[0], pos[1], r, int(bounds[0]*180/math.pi), int(bounds[1]*180/math.pi), color) def draw_text(pos, text, color, font_size=16, global_coords=True, draw_centered=False): if global_coords: pos = world_to_win_pt(pos, c.player.center) output = None if font_fallback: font = pygame.font.SysFont(pygame.font.get_default_font(), font_size) output = font.render(text, 1, color) else: font = pygame.freetype.SysFont(pygame.freetype.get_default_font(), font_size) output = font.render(text, color)[0] if draw_centered: screen.blit(output, (pos[0] - (output.get_width()/2), pos[1] - (output.get_height()/2))) else: screen.blit(output, pos) return output def update(): pygame.display.update() def update_zoom(): global zoom global calculated_visible_width, calculated_visible_height ratio = 1.7 # reverse engineered value. size = sum(map(lambda cell : cell.size, c.player.own_cells)) # reverse engineered formula diag_server = mechanics.viewport_diag(size) if size > 0 else 10000 # calculate screen diag, if we would have a screen with our width, but correct ratio diag1 = math.sqrt(screensize[0]**2 * (1 + 1/1.7**2)) # calculate screen diag, if we would have a screen with our height, but correct ratio diag2 = math.sqrt(screensize[1]**2 * (1 + 1.7**2)) # what we expect to be visible from server calculated_visible_width = diag_server / math.sqrt(1+1/ratio**2) calculated_visible_height = diag_server / math.sqrt(1+ ratio**2) zoom1 = screensize[0] / calculated_visible_width zoom2 = screensize[1] / calculated_visible_height zoom = min(zoom1,zoom2) / vignette def world_to_win_length(l): return int(l*zoom) def win_to_world_length(l): return int(l/zoom) def world_to_win_pt(pt,center): return (int((pt[0]-center[0])*zoom + screensize[0]/2), int((pt[1]-center[1])*zoom + screensize[1]/2)) def win_to_world_pt(pt,center): return (int((pt[0]-screensize[0]/2)/zoom+center[0]), int((pt[1]-screensize[1]/2)/zoom+center[1])) def generate_virus(spikes, spike_length, radius, global_coords): step = (2*math.pi) / (spikes*2) points = [] for i in range(spikes*2): if i%2 == 0: t = ( int(math.sin(i*step)*radius+global_coords[0]), int(math.cos(i*step)*radius+global_coords[1]) ) else: t = ( int(math.sin(i*step)*(radius-spike_length)+global_coords[0]), int(math.cos(i*step)*(radius-spike_length)+global_coords[1]) ) points.append(t); return points def draw_cell(cell): font_size = 16 virus_sizes = {100:1, 106:2, 113:3, 119:4, 125:5, 131:6, 136:7} cx,cy = world_to_win_pt(cell.pos,c.player.center) try: mov_ang = cell.movement_angle p2 = cell.pos + Vec( math.cos(mov_ang + mechanics.eject_delta*math.pi/180), math.sin(mov_ang + mechanics.eject_delta*math.pi/180) ) * (cell.size+700) p3 = cell.pos + Vec( math.cos(mov_ang - mechanics.eject_delta*math.pi/180), math.sin(mov_ang - mechanics.eject_delta*math.pi/180) ) * (cell.size+700) cx2,cy2 = world_to_win_pt(p2,c.player.center) cx3,cy3 = world_to_win_pt(p3,c.player.center) except (AttributeError, TypeError): cx2,cy2=cx,cy cx3,cy3=cx,cy radius = world_to_win_length(cell.size) if cell.is_virus: color = (0,255,0) color2 = (100,255,0) polygon = generate_virus(int(cell.size*0.3), 10*zoom, radius, (cx, cy)) polygon2 = generate_virus(int(cell.size*0.3), 10*zoom, radius-10, (cx, cy)) gfxdraw.filled_polygon(screen, polygon, color2) gfxdraw.polygon(screen, polygon, (0,0,0)) gfxdraw.aapolygon(screen, polygon, (0,0,0)) gfxdraw.filled_polygon(screen, polygon2, color) gfxdraw.aapolygon(screen, polygon2, color) draw_text((cx, cy), "%s / 7" % virus_sizes.get(cell.size, "?"), (64,0,0), font_size*2, False, True) draw_text((cx, cy + radius + 10), str(cell.cid), (0,0,0), font_size, False, True) else: color=(int(cell.color[0]*255), int(cell.color[1]*255), int(cell.color[2]*255)) if not (cell.is_ejected_mass or cell.is_food): gfxdraw.aapolygon(screen, [(cx,cy),(cx2,cy2),(cx3,cy3),(cx,cy)] ,(255,127,127)) gfxdraw.filled_circle(screen, cx, cy, radius, color) gfxdraw.aacircle(screen, cx, cy, radius, (0,0,0)) gfxdraw.aacircle(screen, cx, cy, int(radius/2), (255,255,255)) gfxdraw.circle(screen, cx, cy, int(radius/2), (255,255,255)) draw_text((cx, cy + radius + 10), cell.name, (0, 0, 0), font_size, False, True) draw_text((cx, cy + radius + 10 + font_size), str(cell.cid), (0,0,0), font_size, False, True) # surface = draw_text(cell.name, (0, 0, 0), font_size) # screen.blit(surface, (cx - (surface.get_width()/2), cy + radius + 5)) draw_text((cx, cy), str(int(cell.mass))+"/"+str(int(cell.size)), (255, 255, 255), font_size, False, True) # surface = draw_text(str(int(cell.mass)), (255, 255, 255), font_size) # screen.blit(surface, (cx - (surface.get_width()/2), cy - (surface.get_height()/2))) else: gfxdraw.aacircle(screen, cx, cy, radius, color) gfxdraw.filled_circle(screen, cx, cy, radius, color) def draw_leaderboard(): def filter(item): return item[1] leaderboard = list(map(filter, c.world.leaderboard_names)) next_y = 5 font_size = 16 for i, s in zip(range(1, len(leaderboard)+1), leaderboard): surface = draw_text((5, next_y), (str(i) + ": " +s), (0, 0, 0), font_size, False) # surface = draw_text((str(i) + ": " +s), (0, 0, 0), font_size) # screen.blit(surface, (5, next_y)) next_y += surface.get_height()+5 def draw_visible_window_borders(): global screen vignette_color=(192,192,192) vignette_width = int(max(0, (screensize[0]-world_to_win_length(calculated_visible_width))/2)) vignette_height = int(max(0, (screensize[1]-world_to_win_length(calculated_visible_height))/2)) screen.fill(vignette_color, rect=((0,0),(vignette_width,screensize[1]))) screen.fill(vignette_color, rect=((screensize[0]-vignette_width,0),(vignette_width,screensize[1]))) screen.fill(vignette_color, rect=((0,0),(screensize[0],vignette_height))) screen.fill(vignette_color, rect=((0,screensize[1]-vignette_height),(screensize[0],vignette_height))) def draw_world_borders(): top = int((c.world.top_left[0] - c.player.center[1])*zoom + screensize[1]/2) left = int((c.world.top_left[1] - c.player.center[0])*zoom + screensize[0]/2) bottom = int((c.world.bottom_right[0] - c.player.center[1])*zoom + screensize[1]/2) right = int((c.world.bottom_right[1] - c.player.center[0])*zoom + screensize[0]/2) if (top >= 0): gfxdraw.hline(screen, 0, screensize[0], top, (0,0,0)) if (bottom <= screensize[1]): gfxdraw.hline(screen, 0, screensize[0], bottom, (0,0,0)) if (left >= 0): gfxdraw.vline(screen, left, 0, screensize[1], (0,0,0)) if (right <= screensize[0]): gfxdraw.vline(screen, right, 0, screensize[1], (0,0,0)) def set_client(cl): global c c=cl def clear_screen(): global screen global c screen.fill((255,255,255)) def draw_marker(pos, color, thick): offset = 40 / math.sqrt(2.) # offset = 15 draw_line((pos[0]+offset, pos[1]+offset),(pos[0]-offset, pos[1]-offset), color) draw_line((pos[0]-offset, pos[1]+offset),(pos[0]+offset, pos[1]-offset), color) for r in [10,20,30,40]: draw_circle(pos, r, color) def draw_markers(): colors=[(255,0,255),(255,255,0),(0,255,255)] for i in [0, 1, 2]: draw_marker(marker[i], colors[i], marker_updated[i]) def draw_debug(): for cell in c.world.cells.values(): parent = None shoot_vec = None settled = None try: parent = cell.parent shoot_vec = cell.shoot_vec settled = cell.calmed_down except AttributeError: pass if parent != None: draw_line(cell.pos, parent.pos,(255,0,0)) draw_circle(parent.pos,3,(255,0,0),True) if shoot_vec != None and settled != True: shoot_vec = 400 * shoot_vec / shoot_vec.len() draw_line(cell.pos, cell.pos+shoot_vec, (0,255,255)) def draw_frame(): global screen, movement, zoom, screensize, input, bot_input, marker, marker_updated, running pygame.event.pump() clock.tick() update_zoom() clear_screen() draw_visible_window_borders() draw_world_borders() food = list(filter(lambda x: x.is_food, c.world.cells.values())) cells = list(filter(lambda x: not x.is_food and not x.is_virus, c.world.cells.values())) cells = sorted(cells, key = lambda x: x.mass) viruses = list(filter(lambda x: x.is_virus, c.world.cells.values())) viruses = sorted(viruses, key = lambda x: x.mass) for cell in food: draw_cell(cell) for cell in cells: draw_cell(cell) for cell in viruses: draw_cell(cell) draw_markers() draw_leaderboard() total_mass = 0 for cell in c.player.own_cells: total_mass += cell.mass pygame.display.set_caption("Agar.io: " + str(c.player.nick) + " - " + str(int(total_mass)) + " Total Mass - " + str(int(clock.get_fps())) + (" FPS - USER INPUT LOCKED" if not input else " FPS") + (" - BOT INPUT LOCKED" if not bot_input else "")) events = pygame.event.get() for event in events: if event.type == VIDEORESIZE: screensize = event.dict['size'] screen=pygame.display.set_mode(screensize, HWSURFACE|DOUBLEBUF|RESIZABLE) update_zoom() pygame.display.update() if event.type == QUIT: pygame.display.quit() if event.type == KEYDOWN: if event.key == K_s: if event.mod & KMOD_SHIFT and (input or bot_input): input = False bot_input = False elif not input and bot_input: input = True bot_input = False else: input = False bot_input = True if event.key == K_ESCAPE: running = False if event.key == K_r: c.send_respawn() if event.type == MOUSEBUTTONDOWN: if event.button in [1,2,3]: marker[event.button-1] = win_to_world_pt(event.pos, c.player.center) marker_updated[event.button-1] = True print("set marker "+str(event.button-1)+" to "+str(event.pos)) if event.type == KEYDOWN: if event.key == K_w: c.send_shoot() if event.key == K_SPACE: c.send_split() if input: if event.type == MOUSEMOTION: c.send_target(*win_to_world_pt(event.pos, c.player.center))