GameActivity.kt

val state : State<Array<IntArray>> = gameViewModel.state.collectAsState()
Canvas(
    modifier = Modifier
        .fillMaxHeight(0.8f)
        .aspectRatio(0.5f)
        .border(1.dp, Color.Black)
) {
    val squareDim = size.height / state.value.size
    state.value.forEachIndexed { y, it ->
        it.forEachIndexed { x, value ->
            if (value > 0) {
                drawGameElement(
                    squareColor = when (value) {
                        1 -> Color.Red
                        2 -> Color.Blue
                        3 -> Color.Yellow
                        4 -> Color.Green
                        else -> Color.Red
                    },
                    borderColor = Color.White,
                    offset = Offset(
                        x = (x) * squareDim,
                        y = (y) * squareDim
                    ),
                    size = Size(
                        width = squareDim,
                        height = squareDim
                    )
                )
            }
        }
    }
}