it.unibo.scafi.simulation.frontend.SimulationImpl.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scafi-simulator-gui_2.12 Show documentation
scafi-simulator-gui
The newest version!
/*
 * Copyright (C) 2016-2019, Roberto Casadei, Mirko Viroli, and contributors.
 * See the LICENSE file distributed with this work for additional information regarding copyright ownership.
*/

package it.unibo.scafi.simulation.frontend

import it.unibo.scafi.simulation.frontend.BasicSpatialIncarnation._
import it.unibo.scafi.simulation.frontend.controller.GeneralController
import it.unibo.scafi.simulation.frontend.model.implementation.SensorEnum
import it.unibo.scafi.simulation.frontend.model.{EuclideanDistanceNbr, Node, SimulationManager}
import it.unibo.scafi.space.Point3D

class SimulationImpl(val configurationSeed: Long = System.nanoTime(),
                     val simulationSeed: Long = System.nanoTime(),
                     simulatorManager: SimulationManager) extends Simulation {
  //private Thread runProgram;  //should implements runnable
  private var controller: GeneralController = null
  private var net: SpaceAwareSimulator = null
  var network: model.Network = null
  var runProgram: Function0[(Int,Export)] = null
  var deltaRound: Double = .0
  var strategy: Any = null

  this.deltaRound = 0.00
  this.strategy = null

  def setRunProgram(program: Any): Unit = {

    val devsToPos: Map[Int, Point3D] = network.nodes.mapValues(n => new Point3D(n.position.x, n.position.y, n.position.z)).toMap // Map id->position
    net = new SpaceAwareSimulator(
      space = new Basic3DSpace(devsToPos,
        proximityThreshold = this.network.neighbourhoodPolicy match {
          case EuclideanDistanceNbr(radius) => radius
        }),
        devs = devsToPos.map { case (d, p) => d -> new DevInfo(d, p,
          Map.empty,
          nsns => nbr => null)
        },
      simulationSeed = simulationSeed,
      randomSensorSeed = configurationSeed
    )

    network.setNeighbours(net.getAllNeighbours)

    SensorEnum.sensors.foreach(se => {
      // TODO: println(se);
      net.addSensor(se.name, se.value) }
    )

    val ap = Class.forName(program.toString).newInstance().asInstanceOf[CONTEXT=>EXPORT]
    this.runProgram = () => net.exec(ap)
  }

  def setDeltaRound(deltaRound: Double): Unit = {
    this.deltaRound = deltaRound
    simulatorManager.setPauseFire(deltaRound)
  }

  def getDeltaRound(): Double = this.deltaRound

  def setStrategy(strategy: Any): Unit = {
    this.strategy = strategy
  }

  def getRunProgram: ()=>(Int,Export) = this.runProgram

  private var sensors = Map[String,Any]()

  override def setSensor(sensor: String, value: Any, nodes: Set[Node] = Set()): Unit = {
    val idSet: Set[Int] = nodes.map(_.id)
    if(nodes.size==0 && !controller.selectionAttempted) {
      net.addSensor(sensor, value)
      sensors += sensor -> value
    } else {
      net.chgSensorValue(sensor, idSet, value)
    }
  }

  def getSensorValue(sensorName: String): Option[Any] = {
    net.getSensor(sensorName)
  }

  override def setPosition(n: Node): Unit = {
    net.setPosition(n.id, new Point3D(n.position.x, n.position.y, n.position.z))
    network.setNodeNeighbours(n.id, net.neighbourhood(n.id))
  }

  override def setController(controller: GeneralController): Unit =
    this.controller = controller
}

object SimulationImpl {
  def apply(simulatorManager: SimulationManager): SimulationImpl =
    new SimulationImpl(System.nanoTime(), System.nanoTime(), simulatorManager)
}