update

Author: vangj

python/pygame/space-battle/game.py

@@ -0,0 +1,48 @@
+import os
+import sys
+
+import pygame
+from pygame.constants import KEYUP, QUIT
+
+from objects import Ship
+
+
+def start():
+    pygame.init()
+
+    FPS = 30
+    width = 500
+    height = 400
+    DISPLAYSURF = pygame.display.set_mode((width, height))
+    DISPLAYSURF.fill((255, 255, 255))
+    pygame.display.set_caption('Space Battle')
+    fps_clock = pygame.time.Clock()
+
+    bullets = pygame.sprite.Group()
+    rocks = pygame.sprite.Group()
+    ship = Ship((width / 2, height - 50), 0.5)
+
+    while True:
+        kwargs = {
+            'width': width,
+            'height': height
+        }
+
+        for event in pygame.event.get():
+            if event.type == QUIT:
+                pygame.quit()
+                sys.exit()
+            elif event.type == KEYUP:
+                kwargs['event'] = event
+
+        DISPLAYSURF.fill((255, 255, 255, 0))
+
+        ship.draw(DISPLAYSURF, **kwargs)
+
+        pygame.display.update()
+        fps_clock.tick(FPS)
+
+
+if __name__ == '__main__':
+    os.environ['SDL_VIDEO_CENTERED'] = '1'
+    start()

python/pygame/space-battle/objects.py

@@ -0,0 +1,78 @@
+import pygame
+from pygame.constants import K_LEFT, K_a, K_RIGHT, K_d
+
+
+class BaseObject(pygame.sprite.Sprite):
+    def __init__(self, ipath, position, scale=0.5):
+        pygame.sprite.Sprite.__init__(self)
+
+        img = pygame.image.load(ipath)
+        rect = img.get_rect(center=position)
+        w, h = rect.size[0], rect.size[1]
+        w, h = int(w * scale), int(h * scale)
+        self.image = pygame.transform.scale(img, (w, h))
+        self.rect = self.image.get_rect(center=position)
+        self.should_kill = False
+
+    def get_center(self):
+        return self.rect.center
+
+    def draw(self, surface, **kwargs):
+        raise NotImplementedError('Not yet implemented')
+
+
+class Bullet(BaseObject):
+    def __init__(self, position, scale=1.0):
+        BaseObject.__init__(self, './images/bullet.png', position, scale)
+
+    def draw(self, surface, **kwargs):
+        if self.should_kill:
+            return
+
+        x, y = self.rect.center
+        self.rect.center = x, y - 1
+
+        surface.blit(self.image, self.rect.center)
+
+        if y <= 0:
+            self.should_kill = True
+
+
+class Rock(BaseObject):
+    def __init__(self, position, scale=1.0):
+        BaseObject.__init__(self, './images/rock.png', position, scale)
+
+    def draw(self, surface, **kwargs):
+        if self.should_kill:
+            return
+
+        x, y = self.rect.center
+        self.rect.center = x, y + 1
+
+        surface.blit(self.image, self.rect.center)
+
+        if y >= kwargs['height']:
+            self.should_kill = True
+
+
+class Ship(BaseObject):
+    def __init__(self, position, scale=1.0):
+        BaseObject.__init__(self, './images/ship.png', position, scale)
+
+    def draw(self, surface, **kwargs):
+        width, height = kwargs['width'], kwargs['height']
+        x, y = self.rect.center
+        w, h = self.rect.size
+
+        if 'event' in kwargs:
+            event = kwargs['event']
+            if event.key in (K_LEFT, K_a):
+                if x - w / 2.0 >= 0:
+                    x -= 10
+            elif event.key in (K_RIGHT, K_d):
+                if x + w / 2.0 <= width:
+                    x += 10
+
+        self.rect.center = x, y
+
+        surface.blit(self.image, self.rect.center)