miksilo.modularLanguages.deltas.statement.IfThenDelta.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of modularlanguages_sjs0.6_2.13 Show documentation
modularLanguages
The newest version!
package miksilo.modularLanguages.deltas.statement

import miksilo.modularLanguages.core.deltas._
import miksilo.modularLanguages.core.deltas.grammars.LanguageGrammars
import miksilo.modularLanguages.core.deltas.path.NodePath
import miksilo.modularLanguages.core.node._
import miksilo.languageServer.core.language.{Compilation, Language}
import miksilo.languageServer.core.smarts.ConstraintBuilder
import miksilo.languageServer.core.smarts.scopes.objects.Scope
import miksilo.modularLanguages.deltas.ConstraintSkeleton
import miksilo.modularLanguages.deltas.expression.ExpressionDelta
import miksilo.modularLanguages.deltas.javac.types.BooleanTypeDelta

object IfThenDelta extends DeltaWithGrammar with StatementInstance {

  override def description: String = "Enables using the if-then (no else) construct."

  def neww(condition: Node, thenBody: Node): Node = Shape.create(Condition -> condition, Then -> thenBody)

  object Shape extends NodeShape

  object Condition extends NodeField

  object Then extends NodeField

  override def dependencies: Set[Contract] = super.dependencies ++ Set(BlockDelta)

  implicit class IfThen[T <: NodeLike](val node: T) extends NodeWrapper[T] {
    def condition: T = node(Condition).asInstanceOf[T]
    def thenStatement: T = node(Then).asInstanceOf[T]
  }

  override def transformGrammars(grammars: LanguageGrammars, state: Language): Unit = {
    import grammars._
    val statementGrammar = find(StatementDelta.Grammar)
    val expressionGrammar = find(ExpressionDelta.FirstPrecedenceGrammar)
    val bodyGrammar = find(BlockDelta.BlockOrStatementGrammar)
    val ifThenGrammar = create(Shape, ("if" ~> ("(" ~> expressionGrammar.as(Condition) ~< ")") % bodyGrammar.as(Then)).
      asNode(Shape))
    statementGrammar.addAlternative(ifThenGrammar)
  }

  override def collectConstraints(compilation: Compilation, builder: ConstraintBuilder, statement: NodePath, parentScope: Scope): Unit = {
    val ifThen: IfThen[NodePath] = statement
    ConstraintSkeleton.constraints(compilation, builder, ifThen.thenStatement, parentScope)
    ExpressionDelta.addConstraints(compilation, builder, ifThen.condition, BooleanTypeDelta.constraintType, parentScope)
  }

  override def shape: NodeShape = Shape

  override def getControlFlowGraph(language: Language, statement: NodePath, labels: Map[Any, NodePath]): ControlFlowGraph = {
    val thenStatement = statement.thenStatement
    val thenGraph = ControlFlowGraph.getControlFlowGraph(language, thenStatement, labels)
    val conditionGraph = ControlFlowGraph.singleton(statement)
    val trueGraph = conditionGraph.sequence(thenGraph)
    val elseGraph = conditionGraph
    trueGraph.parallel(elseGraph)
  }
}