Improve functionality of boards

src/main/scala/model/board/BoardOfColor.scala

@@ -68,6 +68,17 @@ case class BoardOfColor(
     board.get(pos)
   }
 
+  def getPieceTypeAt(pos: Coords): PieceType =
+    getPieceTypeAt(BitPosition.fromCoords(pos))
+  def getPieceTypeAt(pos: BitPosition): PieceType = {
+    if pawns.get(pos) then PieceType.Pawn
+    else if rooks.get(pos) then PieceType.Rook
+    else if knights.get(pos) then PieceType.Knight
+    else if bishops.get(pos) then PieceType.Bishop
+    else if queens.get(pos) then PieceType.Queen
+    else if kings.get(pos) then PieceType.King
+    else PieceType.Empty
+  }
   override def getPositions: Seq[Coords] =
     allPieces.board.positionsOfSetBits().map(_.toCoords())
 

src/main/scala/model/board/ChessBoard.scala

@@ -1,14 +1,49 @@
 package model.board
 
 import model.Color
+import model.board.pieces.Pawn
+import model.board.pieces.Knight
+import model.board.pieces.Bishop
+import model.board.pieces.Rook
+import model.board.pieces.Queen
+import model.board.pieces.King
 
 class ChessBoard(val board: Board, val state: BoardState) {
-  def theoreticalMoves: Seq[Move] = ???
+  // Board that is currently playing a move
+  def playingBoard = state.turn match {
+    case Color.White => board.white
+    case Color.Black => board.black
+  }
+
+  // Board that is waiting for the opponent to move
+  def waitingBoard = state.turn match {
+    case Color.White => board.white
+    case Color.Black => board.black
+  }
+
+  def theoreticalMoves(coords: Coords): BitBoard = {
+    val piece = playingBoard.getPieceTypeAt(coords)
+
+    piece match {
+      case model.PieceType.Pawn   => Pawn.theoreticalMoves(coords, state.turn)
+      case model.PieceType.Knight => Knight.theoreticalMoves(coords, state.turn)
+      case model.PieceType.Bishop => Bishop.theoreticalMoves(coords, state.turn)
+      case model.PieceType.Rook   => Rook.theoreticalMoves(coords, state.turn)
+      case model.PieceType.Queen  => Queen.theoreticalMoves(coords, state.turn)
+      case model.PieceType.King   => King.theoreticalMoves(coords, state.turn)
+      case _                      => BitBoard.empty
+    }
+  }
 }
 
 object ChessBoard {
   def empty = new ChessBoard(
     Board.empty,
     BoardState.empty
   )
+
+  def initial = new ChessBoard(
+    Board.initial,
+    BoardState.empty
+  )
 }