main.lua

tilesize = 40
require "piece"
require "card"
require "chessmode"

function love.load(arg)
  math.randomseed(os.time())
  lg = love.graphics
  lk = love.keyboard
  lw = love.window
  lm = love.mouse
  lw.setTitle("ChessTCG")
  lw.setMode(tilesize * 16, tilesize * 16, {borderless = true})
  chessfont = lg.newFont("Cases.ttf", tilesize)
  font = lg.newFont("Roboto-Regular.ttf", tilesize / 2)
  board = {}
  for r = 1, 8 do
    board[r] = {}
    for c = 1, 8 do
      board[r][c] = {}
    end
  end
  colors = {
    white = {0.9, 0.9, 0.9},
    gray = {0.5, 0.5, 0.5},
    black = {0.1, 0.1, 0.1},
    whitetransparent = {0.9, 0.9, 0.9, 0.5},
    blacktransparent = {0.1, 0.1, 0.1, 0.5},
    lightbrown = {196 / 255, 145 / 255, 79 / 255},
    darkbrown = {136 / 255, 75 / 255, 38 / 255},
    accent = {74 / 255, 134 / 255, 236 / 255},
    accent2 = {219 / 255, 177 / 255, 58 / 255},
  }
  game = {
    stage = "game",
    turn = "white",
  }

  -- create players
  white = {
    deck = {},
    hand = {},
    mana = 1,
    maxmana = 1,
  }
  black = {
    deck = {},
    hand = {},
    mana = 1,
    maxmana = 1,
  }
  -- create pieces
  -- chessplace()
  pieces = {}
  for c = 1, 8 do
    table.insert(pieces, Piece:Create{r = 1, c = c, team = "black", type = "generic", spr = "."})
    table.insert(pieces, Piece:Create{r = 2, c = c, team = "black", type = "generic", spr = "."})
    table.insert(pieces, Piece:Create{r = 7, c = c, team = "white", type = "generic", spr = "."})
    table.insert(pieces, Piece:Create{r = 8, c = c, team = "white", type = "generic", spr = "."})
  end
  -- create cards
  for i = 1, 8 do
    table.insert(white.deck, Card:Create{team = "white", type = "pawn", spr = "o", cost = 1})
  end
  table.insert(white.deck, Card:Create{team = "white", type = "rook", spr = "t", cost = 5})
  table.insert(white.deck, Card:Create{team = "white", type = "knight", spr = "m", cost = 2})
  table.insert(white.deck, Card:Create{team = "white", type = "bishop", spr = "v", cost = 3})
  table.insert(white.deck, Card:Create{team = "white", type = "queen", spr = "w", cost = 8})
  table.insert(white.deck, Card:Create{team = "white", type = "king", spr = "l", cost = 2})
  table.insert(white.deck, Card:Create{team = "white", type = "bishop", spr = "v", cost = 3})
  table.insert(white.deck, Card:Create{team = "white", type = "knight", spr = "m", cost = 2})
  table.insert(white.deck, Card:Create{team = "white", type = "rook", spr = "t", cost = 5})
  for i = 1, 8 do
    table.insert(black.deck, Card:Create{team = "black", type = "pawn", spr = "o", cost = 1})
  end
  table.insert(black.deck, Card:Create{team = "black", type = "rook", spr = "t", cost = 5})
  table.insert(black.deck, Card:Create{team = "black", type = "knight", spr = "m", cost = 2})
  table.insert(black.deck, Card:Create{team = "black", type = "bishop", spr = "v", cost = 3})
  table.insert(black.deck, Card:Create{team = "black", type = "queen", spr = "w", cost = 8})
  table.insert(black.deck, Card:Create{team = "black", type = "king", spr = "l", cost = 2})
  table.insert(black.deck, Card:Create{team = "black", type = "bishop", spr = "v", cost = 3})
  table.insert(black.deck, Card:Create{team = "black", type = "knight", spr = "m", cost = 2})
  table.insert(black.deck, Card:Create{team = "black", type = "rook", spr = "t", cost = 5})

  --shuffle decks
  shuffle(white.deck)
  shuffle(black.deck)

  --draw starting hand
  for i = 1, 4 do
    white.deck[1]:Move("hand")
    black.deck[1]:Move("hand")
  end
end

function love.update()
  if lk.isDown("escape") or ((lk.isDown("lctrl") or lk.isDown("rctrl")) and lk.isDown("w")) then love.event.quit() end
end

function love.draw()
  lg.setColor(StringToColor(game.turn))
  lg.rectangle("fill", 0, 0, tilesize * 16, tilesize * 16)
  -- draw board
  for r = 1, 8 do
    for c = 1, 8 do
      if (r % 2 == 0 and c % 2 == 0) or (r % 2 == 1 and c % 2 == 1) then lg.setColor(colors.darkbrown) else lg.setColor(colors.lightbrown) end
      lg.rectangle("fill", c * tilesize + tilesize * 3, r * tilesize + tilesize * 3, tilesize, tilesize)
    end
  end
  -- draw mana
  lg.setColor(colors.accent)
  lg.rectangle("fill", tilesize * 12, tilesize * (12 - white.mana), tilesize * 0.5, tilesize * white.mana)
  lg.rectangle("fill", tilesize * 3.5, tilesize * 4, tilesize * 0.5, tilesize * black.mana)
  if selection then
    if board[selection.r][selection.c].team == "white" then
      lg.setColor(colors.accent2)
      lg.rectangle("fill", tilesize * 12, tilesize * (12 - board[selection.r][selection.c].cost), tilesize * 0.5, tilesize * board[selection.r][selection.c].cost)
      if board[selection.r][selection.c].cost > white.mana then
        lg.setColor(colors.accent)
        lg.rectangle("fill", tilesize * 12, tilesize * (12 - white.mana), tilesize * 0.5, tilesize * white.mana)
      end
    else
      lg.setColor(colors.accent2)
      lg.rectangle("fill", tilesize * 3.5, tilesize * 4, tilesize * 0.5, tilesize * board[selection.r][selection.c].cost)
      if board[selection.r][selection.c].cost > black.mana then
        lg.setColor(colors.accent)
        lg.rectangle("fill", tilesize * 3.5, tilesize * 4, tilesize * 0.5, tilesize * black.mana)
      end
    end
  end
  -- draw mana cost overlay
  if cardselection then
    if cardselection.team == "white" then
      lg.setColor(colors.accent2)
      lg.rectangle("fill", tilesize * 12, tilesize * (12 - white.hand[cardselection.val].cost), tilesize * 0.5, tilesize * white.hand[cardselection.val].cost)
      if white.hand[cardselection.val].cost > white.mana then
        lg.setColor(colors.accent)
        lg.rectangle("fill", tilesize * 12, tilesize * (12 - white.mana), tilesize * 0.5, tilesize * white.mana)
      end
    else
      lg.setColor(colors.accent2)
      lg.rectangle("fill", tilesize * 3.5, tilesize * 4, tilesize * 0.5, tilesize * black.hand[cardselection.val].cost)
      if black.hand[cardselection.val].cost > black.mana then
        lg.setColor(colors.accent)
        lg.rectangle("fill", tilesize * 3.5, tilesize * 4, tilesize * 0.5, tilesize * black.mana)
      end
    end
  end

  for i, p in ipairs(pieces) do
    p:Draw()
  end
  if #white.deck > 0 then
    white.deck[1]:Draw(tilesize * 1, tilesize * 12.5)
  end
  if #black.deck > 0 then
    black.deck[1]:Draw(tilesize * 13, tilesize * 0.5)
  end

  -- draw end turn buttons
  lg.setColor(colors.black)
  lg.rectangle("fill", tilesize * 13, tilesize * 12.5, tilesize * 2, tilesize * 3, 10)
  lg.setColor(colors.white)
  lg.rectangle("fill", tilesize * 1, tilesize * 0.5, tilesize * 2, tilesize * 3, 10)
  lg.setFont(font)
  lg.setColor(colors.white)
  lg.printf("End\nTurn", tilesize * 13, tilesize * 13.25, tilesize * 2, "center")
  lg.setColor(colors.black)
  lg.printf("End\nTurn", tilesize * 1, tilesize * 1.25, tilesize * 2, "center")

  for i, c in ipairs(white.hand) do
    if cardselection and i == cardselection.val and cardselection.team == "white" then
      c:Draw(lm.getX() - tilesize * 1, lm.getY() - tilesize * 1.5, true)
    else
      c:Draw(tilesize * 2 * (i + 1), tilesize * 12.5)
    end
  end
  for i, c in ipairs(black.hand) do
    if cardselection and i == cardselection.val and cardselection.team == "black" then
      c:Draw(lm.getX() - tilesize * 1, lm.getY() - tilesize * 1.5, true)
    else
      c:Draw(tilesize * 2 * (i + 1), tilesize * 0.5)
    end
  end
  if selection then
    -- draw valid moves
    for r = 1, 8 do
      for c = 1, 8 do
        if board[selection.r][selection.c]:Check(r, c) then
          lg.setColor(colors.gray)
          lg.circle("fill", (c + 3.5) * tilesize, (r + 3.5) * tilesize, tilesize * 0.2)
        end
      end
    end
    board[selection.r][selection.c]:Draw() -- draw the piece being moved on top of other pieces
  end
end

function love.mousepressed(x, y, button, istouch, presses)
  row = math.floor(y / tilesize) - 3
  column = math.floor(x / tilesize) - 3
  if 0 < column and column <= 8 then
    if 0 < row and row <= 8 then
      if board[row][column].team and (selection == nil or board[row][column].team == board[selection.r][selection.c].team) then
        selection = {r = row, c = column}
      end
    elseif 8 < row and #white.hand >= math.ceil(column * 0.5) then
      cardselection = {val = math.ceil(column * 0.5), team = "white"}
    elseif row <= 0 and #black.hand >= math.ceil(column * 0.5) then
      cardselection = {val = math.ceil(column * 0.5), team = "black"}
    end
  end
end

function love.mousereleased(x, y, button, isTouch)
  row = math.floor(y / tilesize) - 3
  column = math.floor(x / tilesize) - 3
  if 0 < row and row <= 8 and 0 < column and column <= 8 then
    if selection and board[selection.r][selection.c]:Check(row, column) then
      print(string.char(selection.c + 96) .. selection.r .. " " .. string.char(column + 96) .. row)
      if board[selection.r][selection.c].team == "white" and white.mana >= board[selection.r][selection.c].cost then
        white.mana = white.mana - board[selection.r][selection.c].cost
        board[selection.r][selection.c]:Move(row, column)
      elseif board[selection.r][selection.c].team == "black" and black.mana >= board[selection.r][selection.c].cost then
        black.mana = black.mana - board[selection.r][selection.c].cost
        board[selection.r][selection.c]:Move(row, column)
      end
    end
    if cardselection and game.turn == cardselection.team and board[row][column].team == cardselection.team then
      if cardselection.team == "white" and white.mana >= white.hand[cardselection.val].cost then
        white.mana = white.mana - white.hand[cardselection.val].cost
        white.hand[cardselection.val]:Play(row, column)
      elseif cardselection.team == "black" and black.mana >= black.hand[cardselection.val].cost then
        black.mana = black.mana - black.hand[cardselection.val].cost
        black.hand[cardselection.val]:Play(row, column)
      end
    end
  end
  -- end turn
  if tilesize * 13 < x and x < tilesize * 15 and tilesize * 12.5 < y and y < tilesize * 15.5 and game.turn == "white" then
    game.turn = "black"
    DrawCard("black")
    if black.maxmana < 8 then
      black.maxmana = black.maxmana + 1
    end
    black.mana = black.maxmana
    for i, p in ipairs(pieces) do
      p.moved = false
    end
  end
  if tilesize * 1 < x and x < tilesize * 3 and tilesize * 0.5 < y and y < tilesize * 3.5 and game.turn == "black" then
    game.turn = "white"
    DrawCard("white")
    if white.maxmana < 8 then
      white.maxmana = white.maxmana + 1
    end
    white.mana = white.maxmana
    for i, p in ipairs(pieces) do
      p.moved = false
    end
  end
  selection = nil
  cardselection = nil
end

function StringToColor(string)
  if string == "white" then
    return colors.white
  elseif string == "black" then
    return colors.black
  else
    return colors.gray
  end
end

function shuffle(table)
  for i = #table, 2, - 1 do
    local j = math.random(i)
    table[i], table[j] = table[j], table[i]
  end
  return table
end