player.py

from pyglet.window import key
from pyglet.window import mouse
import math

#Speed constants
WALKING_SPEED = 5
FLYING_SPEED = 10

#Vertical movement constants
GRAV = 20.0
MAX_JUMP = 1.0
JUMP_SPEED = math.sqrt(2 * GRAV * MAX_JUMP)
TERMINAL_VEL = 50

PLAYER_HEIGHT = 2

SIDES = [
    ( 0, 1, 0),
    ( 0,-1, 0),
    (-1, 0, 0),
    ( 1, 0, 0),
    ( 0, 0, 1),
    ( 0, 0,-1),
]

class Player:
    def __init__(self,pos=(0,0,0),rot=(0,0)):
        """Initialize Player object"""
        self.pos = pos
        self.rot = rot
        self.lat = [0,0] #lateral movement
        self.dy = 0 #vertical velocity
        self.flying = False

    def check_hit(self, world, max_dist=8):
        """Return last two blocks hit by sight vector"""
        m = 8
        x,y,z = self.pos
        dx,dy,dz = self.get_sight_vector()
        previous = None
        for _ in range(max_dist * m):
            key = self.round_dis((x,y,z))
            if key != previous and key in world:
                return key, previous
            previous = key
            x,y,z = x + dx / m, y + dy / m, z + dz / m
        return None, None

    def mouse_press(self,x,y,BUTTON,MOD,world):
        """Player function run on mouse press"""
        block, previous = self.check_hit(world)
        if BUTTON == mouse.RIGHT or \
            (BUTTON == mouse.LEFT) and (MOD & key.MOD_CTRL):
            if previous:
                return ("build",previous)
        elif BUTTON == mouse.LEFT and block:
            return ("break",block)

    def mouse_motion(self,dx,dy):
        """Player function run on mouse motion"""
        m = 0.15
        x,y = self.rot
        x,y = x + dx * m, y + dy * m
        y = max(-90, min(90, y))
        self.rot = (x, y)

    def key_press(self,KEY,MOD):
        """Player function run on key press"""
        if KEY == key.W:
            self.lat[0] -= 1
        elif KEY == key.S:
            self.lat[0] += 1
        elif KEY == key.A:
            self.lat[1] -= 1
        elif KEY == key.D:
            self.lat[1] += 1
        elif KEY == key.SPACE:
            if self.dy == 0:
                self.dy = JUMP_SPEED
        elif KEY == key.TAB:
            self.flying = not self.flying

    def key_release(self,KEY,MOD):
        """Player function run on key release"""
        if KEY == key.W:
            self.lat[0] += 1
        elif KEY == key.S:
            self.lat[0] -= 1
        elif KEY == key.A:
            self.lat[1] += 1
        elif KEY == key.D:
            self.lat[1] -= 1

    def get_sight_vector(self):
        """Return player sight vector tuple"""
        x,y = self.rot

        q = math.cos(math.radians(y))
        dx = math.cos(math.radians(x - 90)) * q
        dy = math.sin(math.radians(y))
        dz = math.sin(math.radians(x - 90)) * q

        return (dx,dy,dz)

    def get_motion_vector(self):
        """Return player motion vector tuple"""
        if any(self.lat):
            x,y = self.rot
            lat = math.degrees(math.atan2(*self.lat))
            x_angle = math.radians(x + lat)
            y_angle = math.radians(y)
            if self.flying:
                m = math.cos(y_angle)
                dy = math.sin(y_angle)
                if self.lat[1]:
                    dy = 0.0
                    m = 1
                if self.lat[0] > 0:
                    dy *= -1
                dx = math.cos(x_angle)
                dz = math.sin(x_angle)
            else:
                dy = 0.0
                dx = math.cos(x_angle)
                dz = math.sin(x_angle)
        else:
            dx = 0.0
            dy = 0.0
            dz = 0.0
        return (dx,dy,dz)

    def update(self,dt,world):
        """Player update function set new position"""
        speed = FLYING_SPEED if self.flying else WALKING_SPEED
        d = dt*speed
        dx,dy,dz = self.get_motion_vector()
        dx,dy,dz = dx * d, dy * d, dz * d

        #GRAVITY
        if not self.flying:
            self.dy -= dt * GRAV
            self.dy = max(self.dy, -TERMINAL_VEL)
            dy += self.dy * dt

        x,y,z = self.pos

        # This turns off collision for flying:
        if not self.flying:
            X,Y,Z = self.collide((x + dx,y + dy,z + dz), PLAYER_HEIGHT,world)
        else:
            X,Y,Z = x + dx,y + dy,z + dz
        #X,Y,Z = self.collide((x + dx,y + dy,z + dz), PLAYER_HEIGHT,world)
        self.pos = (X,Y,Z)

    def round_dis(self,pos):
        """Return rounded position tuple"""
        x,y,z = pos
        x,y,z = (int(round(x)),int(round(y)),int(round(z)))
        return (x,y,z)

    def collide(self, pos, height, world):
        """Return position tuple after adjusting for collision"""
        pad = 0.25
        p = list(pos)
        rp = self.round_dis(pos)
        for side in SIDES:
            for i in range(3):
                if not side[i]:
                    continue
                d = (p[i] - rp[i]) * side[i]
                if d < pad:
                    continue
                for dy in range(height):
                    op = list(rp)
                    op[1] -= dy
                    op[i] += side[i]
                    if tuple(op) not in world:
                        continue
                    p[i] -= (d - pad) * side[i]
                    if side == (0,-1,0) or side == (0,1,0):
                        self.dy = 0
                    break
        return tuple(p)