com.barrybecker4.optimization.viewer.TopControls.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bb4-optimization Show documentation
bb4-optimization java code.
The newest version!
// Copyright by Barry G. Becker, 2000-2018. Licensed under MIT License: http://www.opensource.org/licenses/MIT
package com.barrybecker4.optimization.viewer

import java.awt.BorderLayout
import java.awt.event.{ActionEvent, ActionListener}
import com.barrybecker4.optimization.optimizee.optimizees.OptimizeeProblem
import com.barrybecker4.optimization.strategy.OptimizationStrategyType
import javax.swing.{JComboBox, JPanel}


/**
  * Controls at the top of the OptimizerEvalFrame that allow setting the strategy and optimizee problem.
  *
  * @author Barry Becker
  */
class TopControls(var logFile: String, val testProblems: Seq[OptimizeeProblem],
                  var viewable: OptimizationViewable)
     extends JPanel with ActionListener {

  private var strategyDropDown: JComboBox[OptimizationStrategyType] = _
  private var testProblemDropDown: JComboBox[OptimizeeProblem] = _
  private var testProblem = testProblems.head

  setLayout(new BorderLayout)
  val navBar = new NavigationBar(viewable)
  val comboPanel: JPanel = createStrategyCombo
  if (testProblems.length > 1) {
    testProblemDropDown = new JComboBox[OptimizeeProblem](testProblems.toArray)
    testProblemDropDown.addActionListener(this)
    comboPanel.add(testProblemDropDown)
  }
  add(navBar, BorderLayout.CENTER)
  add(comboPanel, BorderLayout.EAST)
  showOptimization()


  private def createStrategyCombo = {
    val strategyPanel = new JPanel
    strategyDropDown = new JComboBox[OptimizationStrategyType](OptimizationStrategyType.VALUES)
    strategyPanel.add(strategyDropDown)
    strategyDropDown.addActionListener(this)
    strategyPanel
  }

  override def actionPerformed(e: ActionEvent): Unit = {
    if (e.getSource eq strategyDropDown) {
      println("changed strategy to " + strategyDropDown.getSelectedItem)
      showOptimization()
    }
    if (e.getSource eq testProblemDropDown) {
      testProblem = testProblemDropDown.getSelectedItem.asInstanceOf[OptimizeeProblem]
      println("changed test problem to " + testProblem)
      showOptimization()
    }
  }

  def showOptimization(): Unit = {
    val strategy = strategyDropDown.getSelectedItem.asInstanceOf[OptimizationStrategyType]
    viewable.showOptimization(strategy, testProblem, logFile)
  }
}