class TestCamera(unittest.TestCase):
def setUp(self):
self.camera = Camera(100, 100, 20, 20)
def test_Focus(self):
self.assertEqual(self.camera.GetFocusPos(), None)
focus = focusMock()
self.camera.SetFocusPos(focus)
self.assertEqual(self.camera.GetFocusPos(), (12, 34))
def test_CenterScreenAtFocus(self):
focus = focusMock()
self.assertEqual(self.camera.cameraFrame.x, 0)
self.assertEqual(self.camera.cameraFrame.y, 0)
self.camera.CenterScreenAtFocus()
self.assertEqual(self.camera.cameraFrame.x, 0)
self.assertEqual(self.camera.cameraFrame.y, 0)
self.camera.SetFocusPos(focus)
self.assertEqual(self.camera.cameraFrame.x, 0)
self.assertEqual(self.camera.cameraFrame.y, 0)
self.camera.CenterScreenAtFocus()
self.assertEqual(self.camera.cameraFrame.x, -38)
self.assertEqual(self.camera.cameraFrame.y, -16)
def test_CameraCornerWorldFrame(self):
x, y = self.camera.CameraCornerWorldFrame((100, 100))
self.assertEqual(x, 50)
self.assertEqual(y, 50)
def test_CameraToWorld(self):
x, y = self.camera.CameraToWorld((100, 100))
self.assertEqual(x, 100)
self.assertEqual(y, 100)
focus = focusMock()
self.camera.SetFocusPos(focus)
self.camera.CenterScreenAtFocus()
x, y = self.camera.CameraToWorld((100, 100))
self.assertEqual(x, 62)
self.assertEqual(y, 84)
def test_CameraToBlockgrid(self):
pos = (27.1, 200.7)
self.assertEqual(self.camera.CameraToBlockgrid(pos), (1, 10))
focus = focusMock()
self.camera.SetFocusPos(focus)
self.camera.CenterScreenAtFocus()
self.assertEqual(self.camera.CameraToBlockgrid(pos), (-1, 9))
def test_WorldToBlockgrid(self):
pos = (27.1, 200.7)
self.assertEqual(self.camera.WorldToBlockgrid(pos), (1, 10))
class App:
def __init__(self, seed):
self.seed = seed
self._running = True
self._close = False
self.size = self.weight, self.height = Screen.WIDTH, Screen.HEIGHT
self.fps = Physics.FPS
self.playtime = 0.0
self.lastTime = time.time()
self.blockQueryTime = time.time()
self.currFps = 0
self.clock = pygame.time.Clock()
if not self.seed:
self.seed = int(random.random()*100000)
print("Starting Survive with seed: {}".format(self.seed))
pygame.mixer.init()
pygame.init()
pygame.display.set_caption("Survive")
def on_init(self):
self.camera = Camera(Screen.WIDTH, Screen.HEIGHT, Physics.BLOCKWIDTH, Physics.BLOCKHEIGHT)
enviroment = Enviroment(Physics.MAPWIDTH, Physics.MAPDEPTH, self.seed)
self.world = Engine(pygame.display.set_mode(self.size, pygame.DOUBLEBUF | pygame.HWSURFACE), self.camera, enviroment)
self.player = Player(0, -35)
self.entities = [self.player]
self._running = True
self.visionLines = []
self.font = pygame.font.SysFont('mono', 16)
self.n = 0
self.camera.SetFocusPos(self.entities[self.n])
def on_event(self, event):
if event.type == pygame.QUIT:
self._close = True
elif event.type == pygame.KEYDOWN:
if event.key == pygame.K_d:
self.player.Move(Direction.RIGHT, True)
elif event.key == pygame.K_a:
self.player.Move(Direction.LEFT, True)
elif event.key == pygame.K_s:
self.player.Move(Direction.DOWN, True)
elif event.key == pygame.K_SPACE:
self.player.Move(Direction.UP, True)
elif event.key == pygame.K_ESCAPE:
self._close = True
elif event.key == pygame.K_r:
self.on_init()
elif event.key == pygame.K_p:
self._running = not self._running
elif event.key == pygame.K_n:
self.n += 1
if self.n >= len(self.entities):
self.n = 0
self.camera.SetFocusPos(self.entities[self.n])
elif event.key == pygame.K_1:
self.player.currentItem = Item.PICKAXE
elif event.key == pygame.K_2:
self.player.currentItem = BlockType.STONE
elif event.key == pygame.K_3:
self.player.currentItem = BlockType.GRASS
elif event.key == pygame.K_4:
self.player.currentItem = BlockType.DIRT
elif event.key == pygame.K_5:
self.player.currentItem = BlockType.BEACH
elif event.key == pygame.K_6:
self.player.currentItem = BlockType.LIGHT
elif event.type == pygame.KEYUP:
if event.key == pygame.K_d:
self.player.Move(Direction.RIGHT, False)
elif event.key == pygame.K_a:
self.player.Move(Direction.LEFT, False)
elif event.key == pygame.K_s:
self.player.Move(Direction.DOWN, False)
elif event.key == pygame.K_SPACE:
self.player.Move(Direction.UP, False)
elif event.type == pygame.MOUSEBUTTONDOWN and event.button == 1:
self.player.Action(self.world, self.camera.CameraToWorld(pygame.mouse.get_pos()))
def on_loop(self):
self.world.SpawnCreatures(self.player, self.entities)
for entity in self.entities:
entity.Step()
self.world.CollisionCheck(self.entities)
self.camera.PlaceInScene(self.entities)
self.player.SelectBlock(self.camera.CameraToWorld(pygame.mouse.get_pos()), self.world)
self.world.renderedBlocks.clear()
for block in self.world.enviroment.BlocksAt(self.player.GetPosition(), 500):
self.world.renderedBlocks[block.id] = block
self.world.LightSource(self.player.GetPosition())
def on_render(self):
self.draw_background()
self.world.RenderBlocks()
self.draw_hud()
self.draw_entities()
pygame.display.update()
def draw_background(self):
self.world._display_surf.fill((3, 3, 3))
def draw_entities(self):
for entity in self.entities:
pygame.draw.rect(self.world._display_surf, entity.color, entity.hitbox)
def draw_hud(self):
self.draw_playerInteraction()
self.draw_text("({}, {}) @ World: {} | {}".format(*self.player.GetPosition(), self.seed, self.player.onGround), (0,0))
self.draw_text("FPS: {}".format(self.currFps), (0, 10))
self.draw_text("Press R to restart", (0, 20))
self.draw_text("Press ESC to exit", (0, 30))
self.draw_text("Press p to pause", (0, 40))
blocks = []
currTime = time.time()
self.world.enviroment.quadtree.query_radius(self.camera.CameraToWorld(pygame.mouse.get_pos()), Physics.BLOCKWIDTH, blocks)
self.draw_text("({}, {}) blocks: {} in :{}s".format(*self.camera.CameraToWorld(pygame.mouse.get_pos()), len(blocks), currTime - self.blockQueryTime), (0, 70))
self.blockQueryTime = currTime
n = 0
for point in blocks:
block = point.payload
if block:
self.draw_text("({}, {}) Color: {}".format(*self.camera.WorldToBlockgrid((block.hitboxWorldFrame.x, block.hitboxWorldFrame.y)), block.color), (0, 80+n*10))
n += 1
def draw_playerInteraction(self):
center = self.camera.WorlToCamera(self.player.GetPositionCentered())
mouse = self.camera.CameraToWorld(pygame.mouse.get_pos())
player = self.player.GetPositionCentered()
distPos = self.world.GetPointAlongLineAtDistance((player, mouse), self.player.reach)
pygame.draw.line(self.world._display_surf, self.player.currentItem, center, (center[0] + distPos[0], center[1] + distPos[1]), 5)
def draw_text(self, text, position):
surface = self.font.render(text, True, (255, 255, 255))
self.world._display_surf.blit(surface, position)
def on_cleanup(self):
pygame.quit()
def on_execute(self):
if self.on_init() == False:
self._running = False
while( not self._close ):
currTime = time.time()
self.currFps = int(1 /(currTime - self.lastTime))
self.lastTime = currTime
for event in pygame.event.get():
self.on_event(event)
if self._running:
self.on_loop()
self.on_render()
else:
pass
self.playtime += self.clock.tick(self.fps) / 1000.0
pygame.display.flip()
self.on_cleanup()
