aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/scalam/m/ast/expressions.scala
blob: db99ae6aa6b367afb7c9f26bfd2081dcc7a68389 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
package scalam.m.ast

trait Expression extends Mable with Root{
  def m: String = this match {
    case IntLiteral(x) => x.toString
    case DoubleLiteral(x) => x.toString
    case StringLiteral(x) => "'" + x + "'"
    case SliceLiteral => ":"
    case ArrayLiteral(elements @ _*) => elements.map(_.m).mkString("[", ",", "]")
    case MatrixLiteral(rows @ _*) => rows.map(_.m).mkString("[", ";", "]")

    case Variable(id) => id.m
    case IndexMatrix(id, indices @ _*) => id.m + indices.map(_.m).mkString("(", ",", ")")
    case IndexStructure(id, indices @ _*) => id.m + indices.map(_.m).mkString("{", ",", "}")
    case Function(id, params @ _*) => id.m + params.map(_.m).mkString("(", ",", ")")

    case _ => throw new IllegalArgumentException("unkown expression: " + this)
  }
}

case class IntLiteral(x: Int) extends Expression
case class DoubleLiteral(x: Double) extends Expression
case class StringLiteral(x: String) extends Expression
case object SliceLiteral extends Expression
case class ArrayLiteral(elements: Expression*) extends Expression
case class MatrixLiteral(rows: Expression*) extends Expression
case class Variable(id: Identifier) extends Expression
case class IndexMatrix(id: Identifier, indices: Expression*) extends Expression
case class IndexStructure(id: Identifier, indices: Expression*) extends Expression
case class Function(function: Identifier, params: Expression*) extends Expression