progs/fun.scala
changeset 628 8067d0a8ba04
parent 626 2d91b2107656
child 644 b4f5714485e1
equal deleted inserted replaced
627:f5214da1976e 628:8067d0a8ba04
   250     case List(t) => t
   250     case List(t) => t
   251     case _ => { println ("Parse Error\n") ; sys.exit(-1) }
   251     case _ => { println ("Parse Error\n") ; sys.exit(-1) }
   252   }
   252   }
   253 }
   253 }
   254 
   254 
       
   255 case class ~[+A, +B](_1: A, _2: B)
       
   256 
   255 class SeqParser[I, T, S](p: => Parser[I, T], 
   257 class SeqParser[I, T, S](p: => Parser[I, T], 
   256                          q: => Parser[I, S])(implicit ev: I => Seq[_]) extends Parser[I, (T, S)] {
   258                          q: => Parser[I, S])(implicit ev: I => Seq[_]) extends Parser[I, ~[T, S]] {
   257   def parse(sb: I) = 
   259   def parse(sb: I) = 
   258     for ((head1, tail1) <- p.parse(sb); 
   260     for ((head1, tail1) <- p.parse(sb); 
   259          (head2, tail2) <- q.parse(tail1)) yield ((head1, head2), tail2)
   261          (head2, tail2) <- q.parse(tail1)) yield (new ~(head1, head2), tail2)
   260 }
   262 }
   261 
   263 
   262 class AltParser[I, T](p: => Parser[I, T], 
   264 class AltParser[I, T](p: => Parser[I, T], 
   263                       q: => Parser[I, T])(implicit ev: I => Seq[_]) extends Parser[I, T] {
   265                       q: => Parser[I, T])(implicit ev: I => Seq[_]) extends Parser[I, T] {
   264   def parse(sb: I) = p.parse(sb) ++ q.parse(sb)   
   266   def parse(sb: I) = p.parse(sb) ++ q.parse(sb)   
   276   def ~[S] (q : => Parser[I, S]) = new SeqParser[I, T, S](p, q)
   278   def ~[S] (q : => Parser[I, S]) = new SeqParser[I, T, S](p, q)
   277 }
   279 }
   278 
   280 
   279 def ListParser[I, T, S](p: => Parser[I, T], 
   281 def ListParser[I, T, S](p: => Parser[I, T], 
   280                         q: => Parser[I, S])(implicit ev: I => Seq[_]): Parser[I, List[T]] = {
   282                         q: => Parser[I, S])(implicit ev: I => Seq[_]): Parser[I, List[T]] = {
   281   (p ~ q ~ ListParser(p, q)) ==> { case ((x, y), z) => x :: z : List[T] } ||
   283   (p ~ q ~ ListParser(p, q)) ==> { case x ~ _ ~ z => x :: z : List[T] } ||
   282   (p ==> ((s) => List(s)))
   284   (p ==> ((s) => List(s)))
   283 }
   285 }
   284 
   286 
   285 case class TokParser(tok: Token) extends Parser[List[Token], Token] {
   287 case class TokParser(tok: Token) extends Parser[List[Token], Token] {
   286   def parse(ts: List[Token]) = ts match {
   288   def parse(ts: List[Token]) = ts match {
   335 // Grammar Rules for Fun
   337 // Grammar Rules for Fun
   336 
   338 
   337 // arithmetic expressions
   339 // arithmetic expressions
   338 lazy val Exp: Parser[List[Token], Exp] = 
   340 lazy val Exp: Parser[List[Token], Exp] = 
   339   (T_KWD("if") ~ BExp ~ T_KWD("then") ~ Exp ~ T_KWD("else") ~ Exp) ==>
   341   (T_KWD("if") ~ BExp ~ T_KWD("then") ~ Exp ~ T_KWD("else") ~ Exp) ==>
   340     { case (((((x, y), z), u), v), w) => If(y, u, w): Exp } ||
   342     { case _ ~ x ~ _ ~ y ~ _ ~ z => If(x, y, z): Exp } ||
   341   (M ~ T_SEMI ~ Exp) ==> { case ((x, y), z) => Sequence(x, z): Exp } || M
   343   (M ~ T_SEMI ~ Exp) ==> { case x ~ _ ~ y => Sequence(x, y): Exp } || M
   342 lazy val M: Parser[List[Token], Exp] =
   344 lazy val M: Parser[List[Token], Exp] =
   343   (T_KWD("write") ~ L) ==> { case (x, y) => Write(y): Exp } || L
   345   (T_KWD("write") ~ L) ==> { case _ ~ y => Write(y): Exp } || L
   344 lazy val L: Parser[List[Token], Exp] = 
   346 lazy val L: Parser[List[Token], Exp] = 
   345   (T ~ T_OP("+") ~ Exp) ==> { case ((x, y), z) => Aop("+", x, z): Exp } ||
   347   (T ~ T_OP("+") ~ Exp) ==> { case x ~ _ ~ z => Aop("+", x, z): Exp } ||
   346   (T ~ T_OP("-") ~ Exp) ==> { case ((x, y), z) => Aop("-", x, z): Exp } || T  
   348   (T ~ T_OP("-") ~ Exp) ==> { case x ~ _ ~ z => Aop("-", x, z): Exp } || T  
   347 lazy val T: Parser[List[Token], Exp] = 
   349 lazy val T: Parser[List[Token], Exp] = 
   348   (F ~ T_OP("*") ~ T) ==> { case ((x, y), z) => Aop("*", x, z): Exp } || 
   350   (F ~ T_OP("*") ~ T) ==> { case x ~ _ ~ z => Aop("*", x, z): Exp } || 
   349   (F ~ T_OP("/") ~ T) ==> { case ((x, y), z) => Aop("/", x, z): Exp } || 
   351   (F ~ T_OP("/") ~ T) ==> { case x ~ _ ~ z => Aop("/", x, z): Exp } || 
   350   (F ~ T_OP("%") ~ T) ==> { case ((x, y), z) => Aop("%", x, z): Exp } || F
   352   (F ~ T_OP("%") ~ T) ==> { case x ~ _ ~ z => Aop("%", x, z): Exp } || F
   351 lazy val F: Parser[List[Token], Exp] = 
   353 lazy val F: Parser[List[Token], Exp] = 
   352   (IdParser ~ T_LPAREN ~ ListParser(Exp, T_COMMA) ~ T_RPAREN) ==> 
   354   (IdParser ~ T_LPAREN ~ ListParser(Exp, T_COMMA) ~ T_RPAREN) ==> 
   353     { case (((x, y), z), w) => Call(x, z): Exp } ||
   355     { case x ~ _ ~ z ~ _ => Call(x, z): Exp } ||
   354   (T_LPAREN ~ Exp ~ T_RPAREN) ==> { case ((x, y), z) => y: Exp } || 
   356   (T_LPAREN ~ Exp ~ T_RPAREN) ==> { case _ ~ y ~ _ => y: Exp } || 
   355   IdParser ==> { case x => Var(x): Exp } || 
   357   IdParser ==> { case x => Var(x): Exp } || 
   356   NumParser ==> { case x => Num(x): Exp }
   358   NumParser ==> { case x => Num(x): Exp }
   357 
   359 
   358 // boolean expressions
   360 // boolean expressions
   359 lazy val BExp: Parser[List[Token], BExp] = 
   361 lazy val BExp: Parser[List[Token], BExp] = 
   360   (Exp ~ T_OP("==") ~ Exp) ==> { case ((x, y), z) => Bop("==", x, z): BExp } || 
   362   (Exp ~ T_OP("==") ~ Exp) ==> { case x ~ _ ~ z => Bop("==", x, z): BExp } || 
   361   (Exp ~ T_OP("!=") ~ Exp) ==> { case ((x, y), z) => Bop("!=", x, z): BExp } || 
   363   (Exp ~ T_OP("!=") ~ Exp) ==> { case x ~ _ ~ z => Bop("!=", x, z): BExp } || 
   362   (Exp ~ T_OP("<") ~ Exp) ==> { case ((x, y), z) => Bop("<", x, z): BExp } || 
   364   (Exp ~ T_OP("<") ~ Exp) ==> { case x ~ _ ~ z => Bop("<", x, z): BExp } || 
   363   (Exp ~ T_OP(">") ~ Exp) ==> { case ((x, y), z) => Bop("<", z, x): BExp } || 
   365   (Exp ~ T_OP(">") ~ Exp) ==> { case x ~ _ ~ z => Bop("<", z, x): BExp } || 
   364   (Exp ~ T_OP("<=") ~ Exp) ==> { case ((x, y), z) => Bop("<=", x, z): BExp } || 
   366   (Exp ~ T_OP("<=") ~ Exp) ==> { case x ~ _ ~ z => Bop("<=", x, z): BExp } || 
   365   (Exp ~ T_OP("=>") ~ Exp) ==> { case ((x, y), z) => Bop("<=", z, x): BExp }  
   367   (Exp ~ T_OP("=>") ~ Exp) ==> { case x ~ _ ~ z => Bop("<=", z, x): BExp }  
   366 
   368 
   367 lazy val Defn: Parser[List[Token], Decl] =
   369 lazy val Defn: Parser[List[Token], Decl] =
   368    (T_KWD("def") ~ IdParser ~ T_LPAREN ~ ListParser(IdParser, T_COMMA) ~ T_RPAREN ~ T_OP("=") ~ Exp) ==>
   370    (T_KWD("def") ~ IdParser ~ T_LPAREN ~ ListParser(IdParser, T_COMMA) ~ T_RPAREN ~ T_OP("=") ~ Exp) ==>
   369      { case ((((((x, y), z), w), u), v), r) => Def(y, w, r): Decl }
   371      { case _ ~ y ~ _ ~ w ~ _ ~ _ ~ r => Def(y, w, r): Decl }
   370 
   372 
   371 lazy val Prog: Parser[List[Token], List[Decl]] =
   373 lazy val Prog: Parser[List[Token], List[Decl]] =
   372   (Defn ~ T_SEMI ~ Prog) ==> { case ((x, y), z) => x :: z : List[Decl] } ||
   374   (Defn ~ T_SEMI ~ Prog) ==> { case x ~ _ ~ z => x :: z : List[Decl] } ||
   373   (Exp ==> ((s) => List(Main(s)) : List[Decl]))
   375   (Exp ==> ((s) => List(Main(s)) : List[Decl]))
   374 
   376 
   375 
   377 
   376 // compiler - built-in functions 
   378 // compiler - built-in functions 
   377 // copied from http://www.ceng.metu.edu.tr/courses/ceng444/link/jvm-cpm.html
   379 // copied from http://www.ceng.metu.edu.tr/courses/ceng444/link/jvm-cpm.html