• Home
• com.github.danielkorzekwa
• bayes-scala_2.10
• 0.5
• source code
• Gate.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

dk.bayes.model.factorgraph.Gate.scala Maven / Gradle / Ivy

package dk.bayes.model.factorgraph

import dk.bayes.model.factor.api.Factor
import dk.bayes.model.factor.GaussianFactor
import dk.bayes.model.factor.api.SingleFactor

/**
 * This class represents an outgoing gate from a factor/variable node in a factor graph.
 *
 * @author Daniel Korzekwa
 *
 * @param message The initial outgoing message sent through the gate
 */
sealed abstract class Gate(initialMsg: SingleFactor) {

  type END_GATE <: Gate

  private var endGate: Option[END_GATE] = None

  private var message: SingleFactor = initialMsg
  private var oldMessage: SingleFactor = initialMsg

  /**Allows for comparing the age between different messages and finding the message that was updated least recently.*/
  private var msgIndex: Long = -1

  def setEndGate(gate: END_GATE) { endGate = Some(gate) }
  def getEndGate(): END_GATE = endGate.get

  def setMessage(newMessage: SingleFactor, msgIndex: Long) {
    oldMessage = message
    message = newMessage

    this.msgIndex = msgIndex
  }

  def getMsgIndex(): Long = msgIndex
  def getMessage(): SingleFactor = message
  def getOldMessage(): SingleFactor = oldMessage
}

case class FactorGate(initialMsg: SingleFactor) extends Gate(initialMsg) {

  type END_GATE = VarGate

  var _factorNode: Option[FactorNode] = None

  def setFactorNode(factorNode: FactorNode) = _factorNode = Some(factorNode)
  def getFactorNode() = _factorNode.get

}
case class VarGate(initialMsg: SingleFactor, varNode: VarNode) extends Gate(initialMsg) {
  type END_GATE = FactorGate
}