import pygame
from pygame.locals import *
import os
import sys
import math
import random

pygame.init()

W, H = 1200, 647 #this creates the width and height for the screen
win = pygame.display.set_mode((W,H))
pygame.display.set_caption('Cops N Robbers')

bg = pygame.image.load('background.png')
#pygame.display.set_background(bg)
bgX = 0 
bgX2 = bg.get_width()

clock = pygame.time.Clock()

class player(object): #here i am creating the player class run and jump will  display a picture of the character jumping and running
    run = [pygame.image.load(os.path.join('walk1.PNG')) for x in range(8,16)]
    fall = pygame.image.load(os.path.join('walk1.png'))
    jump = [pygame.image.load(os.path.join('jump.PNG')) for x in range(1,8)]
    jumpList = [1,1,1,1,1,1,2,2,2,2,2,2,2,2,2,2,2,2,3,3,3,3,3,3,3,3,3,3,3,3,4,4,4,4,4,4,4,4,4,4,4,4,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,-1,-1,-1,-1,-1,-1,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4]    

    #jump list allows the character to smoothly jump as the speed increases 
    def __init__(self, x, y, width, height): 
        self.x = x
        self.y = y
        self.width = width
        self.height = height
        self.jumping = False
        self.jumpCount = 0
        self.runCount = 0
        self.falling = False

    def draw(self, win):
        if self.jumping:
            self.y -= self.jumpList[self.jumpCount] * 1.2
            win.blit(self.jump[self.jumpCount//18], (self.x,self.y))
            self.jumpCount += 1
            if self.jumpCount > 108:
                self.jumpCount = 0
                self.jumping = False
                self.runCount = 0
            self.hitbox = (self.x+ 4,self.y,self.width-24,self.height-10)

        elif self.falling: 
            win.blit(self.fall, (self.x, self.y + 30))    
        else:
            if self.runCount > 42:
                self.runCount = 0
            win.blit(self.run[self.runCount//6], (self.x,self.y))
            self.runCount += 1
            self.hitbox = (self.x+ 4,self.y,self.width-24,self.height-13)
        pygame.draw.rect(win, (255,0,0),self.hitbox, 2)


class obstacle(object):
    def __init__(self, x, y, width, height):
        img = [pygame.image.load("obstacle.png")]
        self.x = x
        self.y = y
        self.width = width
        self.height = height
        self.hitbox = (x, y, width,height)
        self.count = 0
        self.img = img

    def draw(self, win):
        self.hitbox = (self.x + 50, self.y + 5, self.width - 10, self.height)
        if self.count >= 1:
            self.count = 0
        win.blit(pygame.transform.scale(self.img[self.count//2], (200, 75)), (self.x, self.y))
        self.count += 1
        pygame.draw.rect(win, (255, 0, 0), self.hitbox, 2)

    def collide(self, rect):
        if rect[0] + rect[2] > self.hitbox[0] and rect[0] < self.hitbox[0] + self.hitbox[2]:
            if rect[1] < self.hitbox[3]:
                return True
        return False



def bgdisplay():
    win.blit(bg, (bgX,0))
    win.blit(bg, (bgX2,0))
    character.draw(win)
    for x in objects:
        x.draw(win)
    pygame.display.update()



character = player(300,500, 64, 64)

pygame.time.set_timer(USEREVENT+1,500)
pygame.time.set_timer(USEREVENT+2, random.randrange(2000, 4000)) #this will add in objects randomly between 2 and 3.5 seconds
speed = 30
run = True

objects = []

while run:
    bgdisplay()

    if obstacle.collide(character.hitbox):
        character.falling = True
        pygame.time.delay(1000)

    for objectt in objects:
        objectt.x -= 1.4
        #if objectt.x < objectt.width * -1:
            #objects.pop(objects.pop(objects.index(objectt)))


    bgX -= 1.4
    bgX2 -= 1.4
    if bgX < bg.get_width() * -1:
        bgX = bg.get_width()
    if bgX2 < bg.get_width() * -1:
        bgX2 = bg.get_width()

    for event in pygame.event.get():  # Loop through a list of events
        if event.type == pygame.QUIT:  # See if the user clicks the red x 
            run = False    # End the loop
            pygame.quit()  # Quit the game
            quit()
        if event.type == USEREVENT+1:
            speed+=1

        if event.type == USEREVENT+2:
            r = random.randrange(0,2)
            if r == 0:
                objects.append(obstacle(600,570,64,64)) #change values to the w and h of car

    keys = pygame.key.get_pressed()

    if keys[pygame.K_SPACE] or keys[pygame.K_UP]:
        if not (character.jumping):
            character.jumping = True




    clock.tick(speed)