aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/firrtl/Parser.scala
blob: 41426357352a3cdcc34cb8e07557f59f5aa2fb24 (plain)
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
package firrtl

import org.antlr.v4.runtime._;
import org.antlr.v4.runtime.atn._;
import org.antlr.v4.runtime.tree._;
import com.typesafe.scalalogging.LazyLogging
import java.io.FileInputStream
import scala.collection.JavaConverters._
import scala.io.Source
import Utils._
import antlr._

class ParserException(message: String) extends Exception(message)
case class ParameterNotSpecifiedException(message: String) extends ParserException(message)
case class ParameterRedefinedException(message: String) extends ParserException(message)

object Parser extends LazyLogging
{
  /** Takes Iterator over lines of FIRRTL, returns AST (root node is Circuit)
    *
    * Parser performs conversion to machine firrtl
    */
  def parse(filename: String, lines: Iterator[String]): Circuit = {
    val fixedInput = Translator.addBrackets(lines)
    //logger.debug("Preprocessed Input:\n" + fixedInput.result) 
    val antlrStream = new ANTLRInputStream(fixedInput.result)
    val lexer = new FIRRTLLexer(antlrStream)
    val tokens = new CommonTokenStream(lexer)
    val parser = new FIRRTLParser(tokens)

    // FIXME Dangerous (TODO)
    parser.getInterpreter.setPredictionMode(PredictionMode.SLL)

    // Concrete Syntax Tree
    val cst = parser.circuit

    val numSyntaxErrors = parser.getNumberOfSyntaxErrors
    if (numSyntaxErrors > 0) throw new ParserException(s"${numSyntaxErrors} syntax error(s) detected")

    val visitor = new Visitor(filename) 
    //val ast = visitor.visitCircuit(cst) match {
    val ast = visitor.visit(cst) match {
      case c: Circuit => c
      case x => throw new ClassCastException("Error! AST not rooted with Circuit node!")
    }

    ast
  }

  def parse(lines: Seq[String]): Circuit = parse("<None>", lines.iterator)

}