// compiles Abacus programs directly to Javabyte code

import scala.sys.process._
import lib._
import abacus._
import recs._
import comp2._


def compile_aprog(p: AProg) : String = {

  def compile_inst(i: AInst, l: Int) : List[String] = {
    ("L" + l.toString + ":") :: (i match {
      case Inc(n) => List("iinc " + n.toString + " 1")
      case Dec(n, l) => List("iload " + n.toString, "ifeq L" + l.toString, "iinc " + n.toString + " -1")
      case Goto(l) => List("goto L" + l.toString)
    })
  }
  
  val code = for ((i, l) <- p.zipWithIndex) 
    yield compile_inst(i, l).mkString("", "\n", "\n") 

  code.mkString
}

def init_regs(ns: List[Int]) : String = {
  
  val code = for ((n, i) <- ns.zipWithIndex)
    yield List("bipush " + n.toString, "istore " + i.toString).mkString("", "\n", "\n") 
  
  code.mkString
}

def print_result(l: Int, r: Int, class_name: String) : String = {
  List("L" + l.toString + ":", 
       "iload " + r.toString,
       "invokestatic " + class_name + "/" + class_name + "/write(I)V").mkString("", "\n", "\n")
}
  
// compiler preludes
def beginning(class_name: String) : String = { 
"\n.class public " + class_name + "." + class_name + """
.super java/lang/Object

.method public <init>()V
   aload_0
   invokenonvirtual java/lang/Object/<init>()V
   return
.end method

.method public static write(I)V 
    .limit locals 5 
    .limit stack 5 
    iload 0 
    getstatic java/lang/System/out Ljava/io/PrintStream; 
    swap 
    invokevirtual java/io/PrintStream/println(I)V 
    return 
.end method


.method public static main([Ljava/lang/String;)V
   .limit locals 200
   .limit stack 200

""" }

val ending = """

   return

.end method
"""


def compile(f: Rec, ns: List[Int]) : Unit = {
  val class_name = "Prog" + ns.mkString  // name of the class and program
  val (aprog, res, max) = compile_rec(f)

  val init_code = init_regs(ns.padTo(max, 0))  // initialising registers with input data
  val main_code = compile_aprog(aprog.p) 
  val end_code = print_result(aprog.p.length, res, class_name)
  val code = beginning(class_name) + init_code + main_code + end_code + ending
  
  val fw = new java.io.FileWriter(class_name + ".j") // temporary file
  fw.write(code) 
  fw.close()

  val _ = ("java -jar jvm/jasmin-2.4/jasmin.jar " + class_name + ".j").!!
  val start = System.nanoTime()
  val result = ("java " + class_name + "/" + class_name).!!
  val end = System.nanoTime()
  println("Result: " + result + "  Time: " + (end - start) / 1.0e9)
}

print("Add(69, 30)   "); compile(Add, List(69, 30))
print("Mult(13, 9)   "); compile(recs.Mult, List(13, 9))
print("Power(3, 4)   "); compile(Power, List(3, 4))
print("Strt:  ");  compile(Strt(2), List(2,3))

println("FACTORIAL")

for (i <- 7 to 9) {
  println("Input:  " + i)
  compile(Fact, List(i))
}


println("PRIME TEST")

for (i <- 10 to 20) {
  println("Input:  " + i)
  compile(Prime, List(i))
}