MineMapGenerator.kt

package domain

import util.RandomLocationGenerator

object MineMapGenerator {

    fun createMineMap(height: Int, width: Int, numOfMine: Int): MineMap {
        require(numOfMine <= height * width) { "지뢰의 수는 지도의 크기(가로*세로)보다 많을 수 없습니다." }

        // mine 위치 정보 받아오기
        val mineLocations = mutableSetOf<Location>()
        while (mineLocations.size < numOfMine) {
            val locationGenerator = RandomLocationGenerator(height, width)
            val randomLocation = locationGenerator.generate()
            mineLocations.add(randomLocation)
        }

        // mine 위치 정보를 바탕으로 지도 제작
        val elementLocations = mutableListOf<List<MapElement>>()
        for (i in 0 until height) {
            elementLocations += createRow(i, width, mineLocations)
        }

        val mineMap = MineMap(elementLocations.toList())

        // EmptyElement에 대해 주변 지뢰의 수 기입
        mineMap.elements.flatMap { innerList -> innerList.filterIsInstance<EmptyElement>() }
            .forEach { emptyElement -> emptyElement.countMine(mineMap) }

        return mineMap
    }

    private fun createRow(
        rowNum: Int,
        rowLen: Int,
        mineLocations: MutableSet<Location>
    ): List<MapElement> {

        val row = mutableListOf<MapElement>()

        // 해당 row에서 mine의 index만 가지는 set
        val minesOfRow = mineLocations.filter { it.row == rowNum }
            .map { it.column }

        for (y in 0 until rowLen) {
            row += getElement(rowNum, y, minesOfRow)
        }

        return row.toList()
    }

    private fun getElement(x: Int, y: Int, minesOfRow: List<Int>): MapElement {
        return if (minesOfRow.contains(y)) Mine.create(x, y) else EmptyElement.create(x, y)
    }
}