Перекраска качелей Scala

Так что я очень, очень новичок в scala. Я реализую игру Conways's Game of Life с графическим интерфейсом. Я не могу понять, как заставить мою панель обновляться при изменении 2D-массива. Может ли кто-нибудь указать мне правильное направление? Весь мой код следует:

import swing._
import java.awt.{Color, Graphics2D, Dimension}


// initialize variables
// infinite plane variable
var infCurrent = scala.collection.mutable.ListBuffer[List[Int]]()
var infNext = scala.collection.mutable.ListBuffer[List[Int]]()
var infNext1 = scala.collection.mutable.ListBuffer[List[Int]]()
var infLifeTester = scala.collection.mutable.MutableList[List[Int]]()


// general variables
var lifeCount = 0
var yes = 1


// displayed variables
val xDim = 20
val yDim = 20
var currentState = Array.ofDim[Int](xDim, yDim)
var colorIndexList = scala.collection.mutable.ListBuffer[List[Int]]()
var colorData = Array.ofDim[Color](xDim, yDim)


// initial value
infCurrent = scala.collection.mutable.ListBuffer(List(6,6), List(6,7), List(6,8), List(5,8), List(4,7))


// this function tests if a CURRENTLY ALIVE CELL on the INF PLANE STAYS ALIVE
def infStayAlive(current: scala.collection.mutable.ListBuffer[List[Int]]): scala.collection.mutable.ListBuffer[List[Int]] = {
  for (i <- current) {
    var a = i(0)
    var b = i(1)
    if (current.contains(List(a - 1, b - 1))) lifeCount += 1
    if (current.contains(List(a - 1, b))) lifeCount += 1
    if (current.contains(List(a - 1, b + 1))) lifeCount += 1
    if (current.contains(List(a, b - 1))) lifeCount += 1
    if (current.contains(List(a, b + 1))) lifeCount += 1
    if (current.contains(List(a + 1, b - 1))) lifeCount += 1
    if (current.contains(List(a + 1, b))) lifeCount += 1
    if (current.contains(List(a + 1, b + 1))) lifeCount += 1
    if (lifeCount == 2) infNext = infNext :+ i
    lifeCount = 0
  }
  return infNext
}


// this function gets ALL NEIGHBORS for what's on the INF PLANE
def infGetNeighbors(current: scala.collection.mutable.ListBuffer[List[Int]]): scala.collection.mutable.MutableList[List[Int]] = {
  for (i <- current) {
    var a = i(0)
    var b = i(1)
    infLifeTester = infLifeTester :+ List(a - 1, b - 1)
    infLifeTester = infLifeTester :+ List(a - 1, b)
    infLifeTester = infLifeTester :+ List(a - 1, b + 1)
    infLifeTester = infLifeTester :+ List(a, b - 1)
    infLifeTester = infLifeTester :+ List(a, b + 1)
    infLifeTester = infLifeTester :+ List(a + 1, b - 1)
    infLifeTester = infLifeTester :+ List(a + 1,b)
    infLifeTester = infLifeTester :+ List(a + 1, b + 1)
  }
  infLifeTester = infLifeTester.distinct
  return infLifeTester
}


// this function determines whether cells on the INF PLANE DIE or COME ALIVE
def infComeAlive(infLifeTester: scala.collection.mutable.MutableList[List[Int]]): scala.collection.mutable.ListBuffer[List[Int]] = {
  for(i <- infLifeTester) {
    var a = i(0)
    var b = i(1)
    if (infCurrent.contains(List(a - 1, b - 1))) lifeCount += 1
    if (infCurrent.contains(List(a - 1, b))) lifeCount += 1
    if (infCurrent.contains(List(a - 1, b + 1))) lifeCount += 1
    if (infCurrent.contains(List(a, b - 1))) lifeCount += 1
    if (infCurrent.contains(List(a, b + 1))) lifeCount += 1
    if (infCurrent.contains(List(a + 1, b - 1))) lifeCount += 1
    if (infCurrent.contains(List(a + 1, b))) lifeCount += 1
    if (infCurrent.contains(List(a + 1, b + 1))) lifeCount += 1
    if (lifeCount == 3) infNext1 = infNext1 :+ i
    lifeCount = 0
  }
  infNext1 = infNext1.distinct
  return infNext1
}


def printGrid(infCurrent: scala.collection.mutable.ListBuffer[List[Int]]): Array[Array[Int]] = {
  for(i <- infCurrent) {
    if(i(0) >= 0) {
      if(i(0) < xDim) {
        if(i(1) >= 0) {
          if(i(1) < yDim) {
            currentState(i(0))(i(1)) = 1
            colorIndexList = colorIndexList :+ i
          }
        }
      }
    }
  }
  return currentState
}


def colorGrid(colorIndexList: scala.collection.mutable.ListBuffer[List[Int]]): Array[Array[Color]] = {
  for (i <- colorIndexList) {
    colorData(i(0))(i(1)) = Color.WHITE
  }
  return colorData
}


// define panel class
class DataPanel(data: Array[Array[Color]]) extends Panel {

  override def paintComponent(g: Graphics2D) {
    val dx = g.getClipBounds.width.toFloat  / data.length
    val dy = g.getClipBounds.height.toFloat / data.map(_.length).max
    for {
      x <- 0 until data.length
      y <- 0 until data(x).length
      x1 = (x * dx).toInt
      y1 = (y * dy).toInt
      x2 = ((x + 1) * dx).toInt
      y2 = ((y + 1) * dy).toInt
    } {
      data(x)(y) match {
        case c: Color => g.setColor(c)
        case _ => g.setColor(Color.BLACK)
        repaint
      }
      g.fillRect(x1, y1, x2 - x1, y2 - y1)
      println("hi")
    }
    println("hey")
  }
}


// make swing app
object Draw extends SimpleSwingApplication {

  val data = colorData

  def top = new MainFrame {
    background = Color.RED
    title = "Shombo's Game of Life"
    val button = new Button {
      text = "Stahhhp!!"
    }
    val life = new DataPanel(data) {
      preferredSize = new Dimension(500, 500)
    }
    contents = new BoxPanel(Orientation.Vertical) {
      contents += life
      contents += button
    }
  }
}



Draw.top.visible = true
while(yes == 1) {

  infLifeTester = infGetNeighbors(infCurrent)
  infNext = infStayAlive(infCurrent)
  infNext1 = infComeAlive(infLifeTester)
  infNext = scala.collection.mutable.ListBuffer.concat(infNext, infNext1)
  infCurrent = infNext

  infNext = scala.collection.mutable.ListBuffer[List[Int]]()
  infNext1 = scala.collection.mutable.ListBuffer[List[Int]]()
  infLifeTester = scala.collection.mutable.MutableList[List[Int]]()

  currentState = printGrid(infCurrent)

  println(currentState.deep.mkString("\n"))
  //println("\n")

  colorData = colorGrid(colorIndexList)
  Draw.top.contents.repaint()
  currentState = Array.ofDim[Int](xDim, yDim)
  colorIndexList = scala.collection.mutable.ListBuffer[List[Int]]()
  colorData = Array.ofDim[Color](xDim, yDim)
  yes = 1

}