All files / web/src/arcade-games/rithmomachia/utils parseBoardCSV.ts

0% Statements 0/211
0% Branches 0/1
0% Functions 0/1
0% Lines 0/211

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212                                                                                                                                                                                                                                                                                                                                                                                                                                       
/**
 * Parse the Rithmomachia board setup CSV and generate piece setup.
 *
 * CSV Format:
 * - Portrait orientation: black at top, white at bottom
 * - 8 columns (CSV horizontal) = 8 rows in game (1-8)
 * - 16 ranks (CSV vertical, triplets) = 16 columns in game (A-P)
 * - Each rank is 3 CSV rows: [color, shape, number]
 *
 * Game Rotation:
 * - Board is rotated 90° counterclockwise from CSV
 * - CSV column 0 → game row 1 (bottom)
 * - CSV column 7 → game row 8 (top)
 * - CSV rank 0 → game column A (leftmost, black side)
 * - CSV rank 15 → game column P (rightmost, white side)
 */

import type { Color, Piece, PieceType } from '../types'

interface CSVPiece {
  color: Color
  type: PieceType
  value?: number
  pyramidFaces?: number[]
  square: string
}

/**
 * Parse CSV content into structured board layout.
 */
export function parseCSV(csvContent: string): CSVPiece[] {
  const lines = csvContent.trim().split('\n')
  const pieces: CSVPiece[] = []

  // Process in triplets (color, shape, number)
  for (let rankIndex = 0; rankIndex < 16; rankIndex++) {
    const colorRowIndex = rankIndex * 3
    const shapeRowIndex = rankIndex * 3 + 1
    const numberRowIndex = rankIndex * 3 + 2

    if (numberRowIndex >= lines.length) break

    const colorRow = lines[colorRowIndex].split(',')
    const shapeRow = lines[shapeRowIndex].split(',')
    const numberRow = lines[numberRowIndex].split(',')

    // Process each column (8 total)
    for (let colIndex = 0; colIndex < 8; colIndex++) {
      const color = colorRow[colIndex]?.trim()
      const shape = shapeRow[colIndex]?.trim()
      const numberStr = numberRow[colIndex]?.trim()

      // Skip empty cells
      if (!color || !shape || !numberStr) continue

      // Map CSV position to game square
      // CSV column → game row (1-8)
      // CSV rank → game column (A-P)
      const gameRow = colIndex + 1 // CSV col 0 → row 1, col 7 → row 8
      const gameCol = String.fromCharCode(65 + rankIndex) // rank 0 → A, rank 15 → P
      const square = `${gameCol}${gameRow}`

      // Parse color
      const pieceColor: Color = color.toLowerCase() === 'black' ? 'B' : 'W'

      // Parse type
      let pieceType: PieceType
      const shapeLower = shape.toLowerCase()
      if (shapeLower === 'circle') pieceType = 'C'
      else if (shapeLower === 'triangle' || shapeLower === 'traingle')
        pieceType = 'T' // Handle typo
      else if (shapeLower === 'square') pieceType = 'S'
      else if (shapeLower === 'pyramid') pieceType = 'P'
      else {
        console.warn(`Unknown shape "${shape}" at ${square}`)
        continue
      }

      // Parse value/pyramid faces
      if (pieceType === 'P') {
        // Pyramid - for now use default faces, we'll need to determine these
        pieces.push({
          color: pieceColor,
          type: pieceType,
          pyramidFaces: pieceColor === 'B' ? [36, 25, 16, 4] : [64, 49, 36, 25],
          square,
        })
      } else {
        const value = parseInt(numberStr, 10)
        if (isNaN(value)) {
          console.warn(`Invalid number "${numberStr}" at ${square}`)
          continue
        }

        pieces.push({
          color: pieceColor,
          type: pieceType,
          value,
          square,
        })
      }
    }
  }

  return pieces
}

/**
 * Convert CSV pieces to full Piece objects with IDs.
 */
export function createBoardFromCSV(csvPieces: CSVPiece[]): Record<string, Piece> {
  const pieces: Record<string, Piece> = {}

  // Count pieces by color and type for ID generation
  const counts = {
    B: { C: 0, T: 0, S: 0, P: 0 },
    W: { C: 0, T: 0, S: 0, P: 0 },
  }

  for (const csvPiece of csvPieces) {
    const color = csvPiece.color
    const type = csvPiece.type

    // Generate piece ID
    const count = ++counts[color][type]
    const id = `${color}_${type}_${String(count).padStart(2, '0')}`

    // Create full piece
    const piece: Piece = {
      id,
      color,
      type,
      square: csvPiece.square,
      captured: false,
    }

    if (type === 'P') {
      piece.pyramidFaces = csvPiece.pyramidFaces
      piece.activePyramidFace = null
    } else {
      piece.value = csvPiece.value
    }

    pieces[id] = piece
  }

  return pieces
}

/**
 * Main entry point: read CSV and generate board.
 */
export async function loadBoardFromCSV(csvPath: string): Promise<Record<string, Piece>> {
  const fs = await import('fs')
  const csvContent = fs.readFileSync(csvPath, 'utf-8')
  const csvPieces = parseCSV(csvContent)
  return createBoardFromCSV(csvPieces)
}

/**
 * Generate board layout summary for verification.
 */
export function generateBoardSummary(pieces: Record<string, Piece>): string {
  const lines: string[] = []

  // Generate grid view (A-P columns, 1-8 rows)
  lines.push('\n=== Board Layout (Game Orientation) ===\n')
  lines.push('   A    B    C    D    E    F    G    H    I    J    K    L    M    N    O    P')

  for (let row = 8; row >= 1; row--) {
    let line = `${row} `
    for (let colCode = 65; colCode <= 80; colCode++) {
      const col = String.fromCharCode(colCode)
      const square = `${col}${row}`
      const piece = Object.values(pieces).find((p) => p.square === square)

      if (piece) {
        const colorChar = piece.color
        const typeChar = piece.type
        const value = piece.type === 'P' ? 'P' : piece.value?.toString().padStart(3, ' ')
        line += ` ${colorChar}${typeChar}${value}`
      } else {
        line += '  ---'
      }
    }
    lines.push(line)
  }

  // Piece counts
  lines.push('\n=== Piece Counts ===')
  const blackPieces = Object.values(pieces).filter((p) => p.color === 'B')
  const whitePieces = Object.values(pieces).filter((p) => p.color === 'W')

  const countByType = (pieces: Piece[]) => {
    const counts = { C: 0, T: 0, S: 0, P: 0 }
    for (const p of pieces) counts[p.type]++
    return counts
  }

  const blackCounts = countByType(blackPieces)
  const whiteCounts = countByType(whitePieces)

  lines.push(
    `Black: ${blackPieces.length} total (C:${blackCounts.C}, T:${blackCounts.T}, S:${blackCounts.S}, P:${blackCounts.P})`
  )
  lines.push(
    `White: ${whitePieces.length} total (C:${whiteCounts.C}, T:${whiteCounts.T}, S:${whiteCounts.S}, P:${whiteCounts.P})`
  )

  return lines.join('\n')
}