main.lua

-- main.lua
require "ball"
require "player"

-- State machines

State = 0

GameStates = {
    Menu = 0,
    Game = 1,
    Credits = 2
}

MenuState = 0

MenuStates = {
    Play = 0,
    Credits = 1
}

-- Game Over
GameOver = false
Goals = 5
VictoriousPlayer = "No one"


function love.load()
    -- Sets the random seed
    math.randomseed( os.time() )
end

function love.draw()
    if (State == GameStates.Menu) then
        drawMenu()
    elseif (State == GameStates.Game) then
        drawGame()
    elseif (State == GameStates.Credits) then
        drawCredits()
    end
end

function love.update(dt)
    if (State == GameStates.Menu) then
        menuUpdate()
    elseif (State == GameStates.Game) then
        gameUpdate( dt )
    elseif (State == GameStates.Credits) then
        -- creditsUpdate(dt) -- Unnecessary
    end
end

function checkBallWallCollision( )
	local x, y = ball:getPosition()
	local speed_x, speed_y = ball:getSpeed()
	if ball.y < 0 then -- Upper wall collision
        y = 0
        speed_y = -speed_y
    elseif ball.y + ball.size > love.graphics.getHeight() then -- Lower wall collision
        y = love.graphics.getHeight() - ball.size
        speed_y = -speed_y
    end
    

    if ball.x < 0 then -- Left wall collision (Player 1 field)
    	x = 0
        speed_x = -speed_x
        player2.score = player2.score + 1
        ball = nil
        next_player_1 = true
    elseif ball.x + ball.size > love.graphics.getWidth() then -- Right wall collision (Player 2 field)
    	x = love.graphics.getWidth() - ball.size
    	speed_x = -speed_x
    	player1.score = player1.score + 1
        ball = nil
        next_player_1 = false
    end

    if (ball) then
        ball:setPosition(x, y)
        ball:setSpeed(speed_x, speed_y)
    end
end

function drawScores( )
    love.graphics.setColor(Colors.White)
	love.graphics.print( player1.score, 310, 0, 0, 5 )
    love.graphics.print( player2.score, 450, 0, 0, 5 )
end

function drawField()
    local rect_sizex = 10
    local rect_sizey = 40
    local space = 10
    local repeats = love.graphics.getHeight()/rect_sizey
    
    love.graphics.setColor(Colors.White)
    for i = 0, repeats do
        love.graphics.rectangle( 'fill', love.graphics.getWidth()/2 - rect_sizex/2, 
        	i*(space + rect_sizey) + 5, rect_sizex, rect_sizey)
    end
end

function setBallInGame( )
    ball = Ball:new()
    ball:setPosition( love.graphics.getWidth()/2 - ball.size/2, love.graphics.getHeight()/2 - ball.size/2 )

    if next_player_1 then
        local _, y = player1:getPosition()
        local _, height = player1:getDimensions()
        y = y + height/2
        ball:setPosition( love.graphics.getWidth()/2 - ball.size/2, y - ball.size/2)
        ball:setSpeed( ball.speed_starting*math.random(0.9, 1.4), ball.speed_starting*math.random(0.9, 1.4) )
    else
        local _, y = player2:getPosition()
        local _, height = player2:getDimensions()
        y = y + height/2
        ball:setPosition( love.graphics.getWidth()/2 + ball.size/2, y - ball.size/2)
        ball:setSpeed( -ball.speed_starting*math.random(0.9, 1.4), ball.speed_starting*math.random(0.9, 1.4) )
    end
end

function menuUpdate( )
    if(MenuState == MenuStates.Play) then
        if love.keyboard.isDown("down") then
            MenuState = MenuStates.Credits
        elseif love.keyboard.isDown("return") then
            unloadMenu()
            loadGame()
        end
    elseif(MenuState == MenuStates.Credits) then
        if love.keyboard.isDown("up") then
            MenuState = MenuStates.Play
        elseif love.keyboard.isDown("return") then
            unloadMenu()
            loadCredits()
        end
    end
end

function drawMenu( )

    local starting_y_pos = 80

    love.graphics.setColor(Colors.White)
    love.graphics.printf({Colors.Green, "R", Colors.Yellow, "E", Colors.Red, "P", Colors.Gray, "A",
        Colors.Purple, "I", Colors.Orange, "R", Colors.White, " Pong"}, 110, starting_y_pos, 200, "center", 0, 3)

    local color = {}

    if(MenuState == MenuStates.Play) then
        color = Colors.Orange
    else
        color = Colors.SmokyGray
    end
    love.graphics.printf({color, "Play"}, 200, starting_y_pos + 80, 200, "center", 0, 2)

    if(MenuState == MenuStates.Credits) then
        color = Colors.Orange
    else
        color = Colors.SmokyGray
    end

    love.graphics.printf({color, "Credits"}, 200, starting_y_pos + 120, 200, "center", 0, 2)

    love.graphics.printf("How to play:", 300, 300, 200, "center")
    love.graphics.print("Player 1\n\nMove up - W\nMove down - S\nRepair paddle - X", 40, 330, 0, 1)
    love.graphics.print("Player 2\n\nMove up - I\nMove down - K\nRepair paddle - M", 650, 330, 0, 1)
    love.graphics.print("New ball - Spacebar", 40, 430, 0, 1)
end

function drawGame( )
    drawScores()
    drawField()
    player1:draw()
    player2:draw()

    if (ball) then
       ball:draw()
    end

    if (GameOver) then
        drawVictorious()
    end
end

function drawVictorious()
    local color = {}
    if VictoriousPlayer == "Player 1" then
        color = player1:getColor()
    else
        color = player2:getColor()
    end

    love.graphics.setColor( Colors.Purple )
    local rect_w, rect_h = 240, 60
    love.graphics.rectangle( "fill", love.graphics.getWidth()/2 - rect_w/2,
        love.graphics.getHeight()/2 - rect_h/2, rect_w, rect_h )

    love.graphics.setColor( Colors.White )
    love.graphics.printf( {color, VictoriousPlayer, Colors.White, " is victorious!"},
        love.graphics.getWidth()/2 - rect_w/2 + 15, love.graphics.getHeight()/2 - rect_h/2 + 20, 200, "center" )
end

function gameUpdate( dt )
    local gameover = checkVictoryCondition( )

    if not gameover then
        player1:update( dt )
        player2:update( dt )

        if (ball) then
            ball:update( dt )
            ball:checkPlayerCollision( player1 )
            ball:checkPlayerCollision( player2 )
            checkBallWallCollision( )
        else
            if love.keyboard.isDown("space") then
                setBallInGame()
            end
        end
    else
        -- Just don't get any input
    end
end

function checkVictoryCondition( )
    local player1_dead = (player1:getHealth() <= 0) or (player2:getScore() == Goals)
    local player2_dead = (player2:getHealth() <= 0) or (player1:getScore() == Goals)
    GameOver = player1_dead or player2_dead

    if GameOver then
        if player1_dead then
            VictoriousPlayer = "Player 2"
        else
            VictoriousPlayer = "Player 1"
        end
    end

    return GameOver
end

function unloadMenu( )
    MenuState = MenuStates.Play
end

function loadGame( )
    State = GameStates.Game
    GameOver = false

    player1 = Player:new()
    player2 = Player:new()
    ball = nil
    next_player_1 = true
    
    local width
    local height
    width, height = player1:getDimensions()
    y = love.graphics.getHeight()/2 - height/2

    player1:setPosition( 20, y )
    player2:setPosition( love.graphics.getWidth() - width - 20 , y )
    player2:setUpKey( "i" )
    player2:setDownKey( "k" )
    player2:setRepairKey( "m" )
end

function unloadGame( )
    State = GameStates.Menu
end

function love.keyreleased(key)
    if key == "escape" then
        if (State == GameStates.Menu) then
            love.event.quit()
        elseif (State == GameStates.Game) then
            unloadGame()
        elseif (State == GameStates.Credits) then
            unloadCredits()
        end
   end
end

function loadCredits( )
    State = GameStates.Credits
end

function unloadCredits( )
    State = GameStates.Menu
end

function drawCredits( )
    love.graphics.setColor(Colors.White)
    love.graphics.print("Repair Pong", 0, 0, 0, 1)
    love.graphics.print("Global Game Jam 2020 - PUC-Rio - Rio de Janeiro, Brazil", 0, 30, 0, 1)
    love.graphics.print({Colors.White, "Programmer and game designer:\n", Colors.Green, "Bruno Baère", Colors.White,
        " (", Colors.Blue, "https://killerasus.github.io", Colors.White, ")"}, 0, 60, 0, 1)
    love.graphics.print("Thanks:\nTo all participants of the GGJ 2020 PUC-Rio jam site. It was an awesome experience!", 0, 100, 0, 1)
    love.graphics.print({Colors.White, "Repair Pong icon is a derivative work of ", Colors.Green, "Freepik", Colors.White, " (",
        Colors.Blue, "https://www.flaticon.com/authors/freepik", Colors.White, ") from ", Colors.Green,
        "Flaticon", Colors.White, " (", Colors.Blue, "https://www.flaticon.com/", Colors.White, ")"}, 0, 180, 0, 1)
end