Thrust-II-reloaded/main.lua

--[========================================================================[--

Main file for Thrust II Reloaded.

Copyright © 2015-2017 Pedro Gimeno Fortea

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.

--]========================================================================]--

-- For performance and licensing reasons, strict.lua is only run by the
-- developer, not packaged. Include it in the strict/ directory if you want
-- to develop using it.
pcall(function() require 'strict.strict' end)

require('lovespaces')

local volscale = 40
local volgrabbed = false

main = {}
local active, canvas

function main.activate(new)
  if active and active.deactivate then active.deactivate() end
  active = new
  if active.activate then active.activate() end
end

-- screens
screens = {
  splash     = require 'splash';
  menu       = require 'menu';
  redef      = require 'redef';
  getready   = require 'getready';
  game       = require 'game';
  gameover   = require 'gameover';
  gamewon    = require 'gamewon';
  orbexplode = require 'orbexplode';
}

-- modules that are not screens
main.map,
  main.enemytypes,
  main.enemies,
  main.orbs,
  main.decoys,
  main.targets,
  main.respawn,
  main.agents = require 'layout' ()


function main.setVolume(vol, save)
  vol = (vol < 0) and 0 or vol
  vol = (vol > 1) and 1 or vol
  la.setVolume(vol^2)
  main.vol = vol

  if save then
    local tmp = main.vol .. (main.music and "y" or "n")
    lfs.write("vol.txt", tmp, #tmp)
  end
end

local wparamtable = {fullscreen=true,vsync=false,resizable=true,fullscreentype="desktop"}

function main.saveScreen()
  local tmp = string.format("%.17g\n%s\n%d\n%d\n",
      main.pixelsize,
      main.fullscreen and "y" or "n",
      main.wwNFS, main.whNFS)
  lfs.write("screen.txt", tmp, #tmp)
end

function main.setMode()
  wparamtable.fullscreen = main.fullscreen
  lw.setMode(main.wwNFS, main.whNFS, wparamtable)
  lw.setTitle("Thrust II reloaded")
  main.saveScreen()
end

-- deepcopy from http://lua-users.org/wiki/CopyTable
-- minus the parts that we don't need
function main.deepcopy(orig)
  local orig_type = type(orig)
  local copy
  if orig_type == 'table' then
    copy = {}
    for orig_key, orig_value in next, orig, nil do
--      copy[deepcopy(orig_key)] = deepcopy(orig_value)
      copy[orig_key] = main.deepcopy(orig_value)
    end
    -- setmetatable(copy, deepcopy(getmetatable(orig)))
  else -- number, string, boolean, etc
    copy = orig
  end
  return copy
end

function main.drawaligned(alignment, drawable, x, y, angle, xzoom, yzoom)
  -- alignment's value works as a numeric keypad ("normal" is 7):
  --
  --    7 8 9
  --    4 5 6
  --    1 2 3
  if type(drawable) == "userdata" and drawable.typeOf and drawable:typeOf("Drawable") then
    -- it's a drawable
    local iw, ih = drawable:getDimensions()
    local ox, oy = 0, 0
    if alignment <= 3 then
      oy = ih
    elseif alignment <= 6 then
      oy = ih / 2
    end
    if alignment % 3 == 0 then
      ox = iw
    elseif alignment % 3 == 2 then
      ox = iw / 2
    end

    return lg.draw(drawable, math.floor(x), math.floor(y), angle, xzoom, yzoom, ox, oy)
  end
end


function love.load(args)
  lfs.setIdentity("ThrustIIreloaded")

  main.argv = args

  main.pixelsize = 1
  main.fullscreen = false
  main.wwNFS = 640 -- non-fullscreen width
  main.whNFS = 480 -- non-fullscreen height
  if lfs.isFile("screen.txt") then
    local lines = lfs.read("screen.txt")
    local pix, fs, ww, wh = lines:match("^(%d+%.?%d*[Ee]?%d*)\n([yn]?)\n(%d+)\n(%d+)\n")
    main.pixelsize = tonumber(pix)
    main.fullscreen = fs == "y"
    main.wwNFS = tonumber(ww)
    main.whNFS = tonumber(wh)
  else
    main.saveScreen()
  end
  main.setMode()

  lg.setDefaultFilter("nearest", "nearest")
  main.ww = math.ceil(lg.getWidth()/main.pixelsize)
  main.wh = math.ceil(lg.getHeight()/main.pixelsize)
  main.wcx, main.wcy = main.ww/2, main.wh/2
  canvas = lg.newCanvas(main.ww, main.wh)

  main.font = lg.newImageFont("img/Font16x16.png",
      " !\"#$%&'()*+,-./0123456789:;<=>?\127ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuvwxyz{|}~@",
      1) -- required for 0.10 compatibility, ignored by 0.9
  lg.setFont(main.font)

  local space = love._version_major == 0 and love._version_minor < 10 and " " or "space"
  main.keys =
    { left = "q", right = "w", thrust = "p", pickup = "l", fire = space }
  if lfs.isFile("keys.txt") then
    local line = 0
    for s in lfs.lines("keys.txt") do
      if s == " " or s == "space" then s = space end -- 0.10 compat
      line = line + 1
      if line == 1 then
        main.keys.left = s
      elseif line == 2 then
        main.keys.right = s
      elseif line == 3 then
        main.keys.thrust = s
      elseif line == 4 then
        main.keys.pickup = s
      elseif line == 5 then
        main.keys.fire = s
      else
        break
      end
    end
  else
    local s = main.keys.left   .. "\n"
           .. main.keys.right  .. "\n"
           .. main.keys.thrust .. "\n"
           .. main.keys.pickup .. "\n"
           .. main.keys.fire
    lfs.write("keys.txt", s, #s)
  end

  main.vol = 0.5
  main.music = true
  if lfs.isFile("vol.txt") then
    local line = lfs.read("vol.txt")
    local vol, music = line:match("^(%d+%.?%d*[Ee]?%d*)([yn]?)$")
    main.setVolume(tonumber(vol))
    main.music = music == "y"
  else
    main.setVolume(main.vol, true)
  end

  for k, mod in next, screens do
    if mod and mod.load then mod.load(arg) end
  end

  -- Bootstrap activation
  main.activate(screens.splash)

  -- Allow volume change with auto-repeat
  lk.setKeyRepeat(true)

  -- Used to end quit status on key release. Otherwise the key up event
  -- that led to releasing the quit status is registered.
  main.dontquit = false
  main.unpause = false
  main.singlestep = false
  main.savedelay = false
end

-- HACK: rather than altering love.run, we alter love.timer.sleep.
-- This guarantees that it's run between lg.present and fetching
-- events, giving maximum responsivity.
do
  local ltisleep = love.timer.sleep
  function love.timer.sleep(t)
    ltisleep(0.02)
  end
end

function love.update(dt)
  if volgrabbed then
    main.setVolume((lmo.getX() / main.pixelsize - (main.wcx + 108)) / volscale)
  end
  if main.savedelay then
    main.savedelay = main.savedelay - dt
    if main.savedelay <= 0 then
      main.savedelay = false
    end
  end
  if main.paused and not main.singlestep or main.quitrequest then return end
  if active.update then active.update(main.singlestep and 0.03125 or dt) end
end

local function roundrect(x1, y1, w, h, r)
  local nsegments = 8
  local x2, y2 = x1+w, y1+h
  local npoints = nsegments + 1 -- points per corner
  local poly = {}
  poly[1] = x2
  poly[2] = y2 - r
  poly[npoints*2-1] = x2 - r
  poly[npoints*2] = y2
  poly[npoints*2+1] = x1 + r
  poly[npoints*2+2] = y2
  poly[npoints*4-1] = x1
  poly[npoints*4] = y2 - r
  poly[npoints*4+1] = x1
  poly[npoints*4+2] = y1 + r
  poly[npoints*6-1] = x1 + r
  poly[npoints*6] = y1
  poly[npoints*6+1] = x2 - r
  poly[npoints*6+2] = y1
  poly[npoints*8-1] = x2
  poly[npoints*8] = y1 + r
  local angle, x, y

  for i = 2, (nsegments-1)*2, 2 do
    angle = math.pi * 0.25 * (i / nsegments)
    x = math.cos(angle) * r
    y = math.sin(angle) * r
    poly[i+1] = x2 - r + x
    poly[i+2] = y2 - r + y
    poly[npoints*4-i-1] = x1 + r - x
    poly[npoints*4-i] = y2 - r + y
    poly[npoints*4+i+1] = x1 + r - x
    poly[npoints*4+i+2] = y1 + r - y
    poly[npoints*8-i-1] = x2 - r + x
    poly[npoints*8-i] = y1 + r - y
  end
  lg.polygon("fill", poly)
end


function main.centertext(text, x, y)
  lg.print(text, math.floor(x-main.font:getWidth(text)/2), math.floor(y-main.font:getHeight()/2))
end

local function drawActive()
  -- HACK: Allow scroll-out in get ready screen, by not clearing the canvas
  if active ~= screens.getready then
    lg.clear()
  end
  active.draw()
end

local function volumeKnobStencil()
  lg.setColor(255,255,255,255)
  lg.rectangle("fill", main.wcx + 103 + main.vol*volscale, main.wh-32, 10, 24)
end

function love.draw()
  if main.quitrequest then
    lg.push()
    lg.scale(main.pixelsize)
    lg.setColor(85, 85, 85, 255)
    lg.rectangle("fill", 0, 0, main.ww, main.wh)
    lg.setColor(255, 255, 255, 120)
    lg.draw(canvas)
    lg.setColor(0, 0, 0, 200)

    roundrect(main.wcx-130, main.wcy-50, 260, 100, 7)
    lg.setColor(255, 255, 255, 255)
    main.centertext("REALLY QUIT?", main.wcx, main.wcy - 20)

    lg.setColor(160,160,160)
    roundrect(main.wcx-70, main.wcy+4, 60, 24, 4)
    roundrect(main.wcx+10, main.wcy+4, 60, 24, 4)
    lg.setColor(0, 0, 0, 255)
    main.centertext("YES", main.wcx-40, main.wcy+16)
    main.centertext("NO", main.wcx+40, main.wcy+16)
    lg.setColor(255, 255, 255, 255)
    lg.pop()
  else
    lg.setColor(255, 255, 255, 255)
    if main.paused and not main.singlestep then
      lg.push()
      lg.scale(main.pixelsize)
      lg.draw(canvas)
      lg.setColor(64, 64, 64, 150)
      roundrect(main.wcx-60, main.wh-44, 120, 40, 6)
      lg.setColor(255, 255, 255, 255)
      main.centertext("PAUSED", main.wcx, main.wh - 24)
      lg.pop()
    else
      main.canvas = canvas
      if active.draw then
        canvas:renderTo(drawActive)
      end
      lg.push()
      lg.scale(main.pixelsize)
      lg.draw(canvas)
      lg.pop()
    end
    if main.savedelay then
      lg.push()
      lg.scale(main.pixelsize)
      lg.setColor(64, 64, 64, 200)
      roundrect(main.wcx-60, main.wh-44, 120, 40, 6)
      lg.setColor(255, 255, 255, 255)
      main.centertext("SAVED", main.wcx, main.wh - 24)
      lg.pop()
    end
  end

  -- volume control

  lg.push()
  lg.scale(main.pixelsize)
  local half = main.wcx
  local volx = main.vol*volscale

  if lg.stencil then
    -- 0.10 update
    lg.stencil(volumeKnobStencil)
    lg.setStencilTest("equal", 0) -- Inverted stencil
  else
    lg.setInvertedStencil(volumeKnobStencil)
  end
  lg.setColor(0,0,0, 40)
  lg.polygon("fill", half + 92, main.wh - 11,
                     half + 155, main.wh - 11,
                     half + 155, main.wh - 29)
  lg.setColor(255, 255, 255, 60)
  lg.polygon("fill", half + 100, main.wh - 12,
                     half + 154, main.wh - 12,
                     half + 154, main.wh - 28)
  if lg.setStencilTest then
    lg.setStencilTest()
  else
    lg.setStencil()
  end
  lg.setColor(0, 0, 0, 40)
  lg.rectangle("fill", half + 103 + volx, main.wh-32, 10, 24)
  lg.setColor(255, 255, 255, 70)
  lg.rectangle("fill", half + 104 + volx, main.wh-31, 8, 22)
  lg.setColor(255, 255, 255)
  if main.singlestep then main.singlestep = false end
  lg.pop()
end

function love.quit()
  if not main.quitaccepted then
    main.quitrequest = true
    main.dontquit = false
    if not main.paused and active.pause then active.pause(true) end
    return true -- don't quit
  end
end

function love.mousepressed(x, y, b)
  x = x / main.pixelsize
  y = y / main.pixelsize
  if main.quitrequest then
    if x >= main.wcx-70 and x <= main.wcx-10
       and y >= main.wcy+4 and y <= main.wcy+28
    then
      main.quitaccepted = true
      le.quit()
    elseif x >= main.wcx+10 and x <= main.wcx+70
       and y >= main.wcy+4 and y <= main.wcy+28
    then
      main.quitrequest = false
      if not main.paused and active.pause then active.pause(false) end
    end
  end

  if y >= main.wh-32 and y <= main.wh-8
    and  x >= main.wcx + 104 + main.vol*volscale
    and x <= main.wcx + 112 + main.vol*volscale
  then
    lmo.setGrabbed(true)
    volgrabbed = true
    main.setVolume((x - (main.wcx + 108)) / volscale)
  end
end

--[[ (not in 0.9.1 - handled in love.update())
function love.mousemoved(x, y, dx, dy)
  if volgrabbed then
    main.setVolume((x - (main.wcx + 108)) / volscale)
  end
end
]]

function love.mousereleased(x, y, b)
  if volgrabbed then
    lmo.setGrabbed(false)
    volgrabbed = false
    main.setVolume((x / main.pixelsize - (main.wcx + 108)) / volscale, true)
  end
end

function love.keypressed(k, r, rr)
  -- 0.10 compat
  if rr ~= nil then r = rr end

  if k == "kp-" or k == "kp+" then
    main.setVolume(main.vol + (k == "kp+" and 0.0625 or -0.0625), true)
  elseif r then
    -- nothing
  elseif main.quitrequest then
    if k == "escape" then
      main.dontquit = true
    end
  elseif k == "escape" then
    le.quit()
  elseif k == "pause" and not main.quitrequest and
      -- ctrl + pause = break, that breaks the game and must be passed down
      not (lk.isDown("lctrl") or lk.isDown("rctrl") or lk.isDown("ctrl"))
  then
    main.paused = not main.paused
    if active.pause then active.pause(main.paused) end
  elseif active.keypressed and not main.paused then
    active.keypressed(k, r) 
  end
  --[[ debug ]]
  --if k == "1" and main.paused then main.singlestep = true end
  --]]
end

function love.textinput(c)
  if main.quitrequest then
    if c:lower() == "y" then
      main.quitaccepted = true
      le.quit()
    elseif c:lower() == "n" then
      main.dontquit = true
    end
    return
  end
  if not main.paused then
    if active.textinput then active.textinput(c) end
  end
end

function love.keyreleased(k)
  if main.quitrequest then
    if main.dontquit then
      main.dontquit = false
      main.quitrequest = false
      if not main.paused and active.pause then active.pause(false) end
    end
  elseif k ~= "pause" and not main.paused then
    if active.keyreleased then active.keyreleased(k) end
  end
end

function love.resize(neww, newh)
  main.ww = math.ceil(neww/main.pixelsize)
  main.wh = math.ceil(newh/main.pixelsize)
  main.wcx = main.ww/2
  main.wcy = main.wh/2
  local newcanvas = lg.newCanvas(main.ww, main.wh)
  local saveCanvas = lg.getCanvas()
  lg.setCanvas(newcanvas)
  main.drawaligned(5, canvas, main.wcx, main.wcy)
  lg.setCanvas(saveCanvas)
  canvas = newcanvas
  if not main.fullscreen then
    main.wwNFS = neww
    main.whNFS = newh
    main.saveScreen()
  end
  for k, mod in next, screens do
    if mod.resize then
      mod.resize(neww, newh)
    end
  end
end