Scala的计算机代数系统(CAS)

时间:2011-05-18 09:12:59

标签: scala implementation cas computer-algebra-systems

我正在为scala寻找一个简单的CAS系统。

它应该具有以下功能:

  • 允许访问抽象语法树(最好通过案例类以便于匹配)
  • String解析为AST
  • 简化表达

如果不存在,我必须自己写一些基本的东西,那么最好的代表是什么?

我在想这样的事情:

abstract trait Term
{
  def simplify:Term
  def evaluate(assignment:Var => Double):Double
  def derivative:Term
}

case class Const(c:Int) extends Term
case class Var(x:String) extends Term

case class Negate(x:Term) extends Term
case class Subtract(x:Term, y:Term) extends Term
case class Divide(x:Term, y:Term) extends Term


object Add { def apply(x:Term*):Add = Add(x.toList) }
case class Add(xs : List[Term]) extends Term

object Multiply { def apply(x:Term*):Multiply = Multiply(x.toList) }
case class Multiply(xs:List[Term]) extends Term

case class Power(x:Term, y:Term) extends Term
case class Exp(x:Term) extends Term

我实施了simplification algorithm described here,这似乎很乏味。 (但是,在简化代数表达式方面,也许是不可避免的?)

对此具体实施的一些批评是:

  • 我会在案例类的参数上以递归方式调用simplify(似乎它可能以某种方式集中化)
  • 处理ListAdd的varargs / Mutliply参数似乎可能会变得混乱

1 个答案:

答案 0 :(得分:4)

我不知道Scala现有的CAS。

在进行语言处理时,我通常发现在密封的层次结构中使用模式匹配比使用OO样式多态更令人愉快。由于添加新术语类型很少(这意味着更改语言)并且添加新操作,因此表达式问题的这一方面看起来更合适。

sealed trait Term
case class Const(c : Double) extends Term
case class Var(x : String) extends Term
case class Negate(x : Term) extends Term
case class Multiply(xs : List[Term]) extends Term
// etc

object CAS {

  // I assume that the assignment map may be incomplete, thus
  // evaluation is really a partial substitution and then simplification
  def evaluate(t : Term, assignment : Var => Option[Double]) : Term = t match {
    case _ : Const => t
    case v : Var => assignment(v) map Const getOrElse v
    case Negate(x) => evaluate(Multiply(Const(-1) :: evaluate(x, assignment) :: Nil), assignment)
    case Multiply(ts) => {
      val evalTs = ts map { t => evaluate(t, assignment) }
      val flattened = evalTs flatMap {
         case Multiply(subs) => subs
         case t => List(t)
      }
      val constTotal = Const((flattened collect { case Const(c) => c }).product)
      val otherTerms = flattened filter { case t : Const => false; case _ => true }
      (constTotal, otherTerms) match {
         case (Const(0), _) => Const(0)
         case (Const(1), Nil) => Const(1)
         case (Const(1), _) => Multiply(otherTerms)
         case _ => Multiply(constTotal +: otherTerms)
      }
    }
    // etc

  }

  private val emptyAssignment : (Var => Option[Double]) = { x : Var => None }

  // simplfication is just evaluation with an empty assignment
  def simplify(t : Term) : Term = evaluate(t, emptyAssignment)
}

我一直想要学习的一点技术,但没有属性语法。他们应该从这种AST处理中获取大部分单调乏味的东西。有关Scala实现,请参阅kiama http://code.google.com/p/kiama/

顺便说一下,虽然我在这里使用双打为你的域名,你可能最好使用“大理性” - 一对BigIntegers。它们很慢但很精确。