polish example

Author: AuthentiCuber

example.py

@@ -1,4 +1,3 @@
-import time
 import pygame
 from pygame.locals import *
 import timestep
@@ -12,54 +11,64 @@
 pygame.display.set_caption("Timestep Test")
 SCREEN_WIDTH, SCREEN_HEIGHT = screen.get_size()
 
-font = pygame.font.SysFont("Calibri", 40)
-start = False
-start_time = 0
-
 
 class Player(timestep.Character):
     def __init__(self, x: int, y: int) -> None:
         self.image = pygame.Surface((100, 100))
         self.image.fill("red")
         super().__init__(x, y, self.image)
-        self.gravity = pgvec2(0, 0.1)
+        self.gravity = pgvec2(0, 1)
+        self.friction = 0.8
+        self.jumped = False
 
     def update(self) -> None:
         super().update()
 
         self.vel.y += self.gravity.y
+        self.vel.x *= self.friction
+
+        keys = pygame.key.get_pressed()
+        if keys[K_RIGHT]:
+            self.vel.x = 10
+        if keys[K_LEFT]:
+            self.vel.x = -10
+        if keys[K_UP] and not self.jumped:
+            self.vel.y = -15
+            self.jumped = True
+
+        self.rect.x += round(self.vel.x)
         self.rect.y += round(self.vel.y)
 
+        if self.rect.bottom > SCREEN_HEIGHT:
+            self.rect.bottom = SCREEN_HEIGHT
+            self.vel.y = 0
+            self.jumped = False
+        if self.rect.right > SCREEN_WIDTH:
+            self.rect.right = SCREEN_WIDTH
+            self.vel.x = 0
+        elif self.rect.left < 0:
+            self.rect.left = 0
+            self.vel.x = 0
+
 
 player = Player(500, 0)
 
 
 class game_loop(timestep.Timestep):
     def update(self):
-        global game_running, start_time, start
+        global game_running
         for event in pygame.event.get():
             if event.type == QUIT or (event.type == KEYDOWN and event.key == K_ESCAPE):
                 game_running = False
-            if event.type == KEYDOWN:
-                if event.key == K_SPACE and not start:
-                    start = True
-                    start_time = time.time()
 
-        if start:
-            player.update()
+        player.update()
 
     def render(self, alpha):
-        global game_running
 
         screen.fill((0, 0, 0))
 
         player.draw(screen, alpha)
 
-        if player.rect.y >= SCREEN_HEIGHT:
-            finish = time.time() - start_time
-            print(finish)
-            game_running = False
-
         pygame.display.flip()
 
 

timestep.py

@@ -3,7 +3,6 @@
 import pygame
 import time
 
-
 class Character:
     """Base class for any sprites or objects"""
 
@@ -12,18 +11,20 @@ def __init__(self, x: int, y: int, image: pygame.Surface) -> None:
         self.rect = self.image.get_rect()
         self.rect.topleft = x, y
         self.vel = pygame.math.Vector2(0, 0)
-        self.prev_pos = pygame.math.Vector2(self.rect.topleft)
+        self.pos = pygame.math.Vector2(self.rect.topleft)
+        self.prev_pos = self.pos
 
     def update(self) -> None:
-        """Update """
+        """Override this method with movement, input, collisions etc."""
         self.prev_pos = self.__get_rect_pos()
 
     def draw(self, surface: pygame.Surface, alpha: float) -> None:
-        pos = self.prev_pos.lerp(self.__get_rect_pos(), alpha)
-        surface.blit(self.image, pos)
+        """Draw the Character to the screen."""
+        self.pos = self.prev_pos.lerp(self.__get_rect_pos(), alpha)
+        surface.blit(self.image, (round(self.pos.x), round(self.pos.y)))
 
     def __get_rect_pos(self) -> pygame.math.Vector2:
-        """Return the position of the Charater's rect as a Vec2"""
+        """Return the position of the Charater's rect as a Vec2."""
         return pygame.math.Vector2(self.rect.topleft)
 
 
@@ -37,9 +38,9 @@ def __init__(self, step: float) -> None:
 
     def __calc_dt(self) -> float:
         now = time.time()
-        dt = now - self.__last_time
+        self.dt = now - self.__last_time
         self.__last_time = now
-        return dt
+        return self.dt
 
     def update(self) -> None:
         """Override this function with event loop and movement"""