blob: 40956ab70c396ea5657aaf03c08180e53754151f (
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
package firrtl
import org.antlr.v4.runtime._;
import org.antlr.v4.runtime.atn._;
import org.antlr.v4.runtime.tree._;
import java.io.FileInputStream
import scala.collection.JavaConverters._
import scala.io.Source
import Utils._
import antlr._
object Parser
{
def parseModule(string: String): Module = {
val fixedInput = Translator.addBrackets(Iterator(string))
val antlrStream = new ANTLRInputStream(fixedInput.result)
val lexer = new FIRRTLLexer(antlrStream)
val tokens = new CommonTokenStream(lexer)
val parser = new FIRRTLParser(tokens)
// FIXME Dangerous
parser.getInterpreter.setPredictionMode(PredictionMode.SLL)
// Concrete Syntax Tree
val cst = parser.module
val visitor = new Visitor("none")
//val ast = visitor.visitCircuit(cst) match {
val ast = visitor.visit(cst) match {
case m: Module => m
case x => throw new ClassCastException("Error! AST not rooted with Module node!")
}
ast
}
/** Takes a firrtl filename, returns AST (root node is Circuit)
*
* Currently must be standard FIRRTL file
* Parser performs conversion to machine firrtl
*/
def parse(filename: String): Circuit = {
//val antlrStream = new ANTLRInputStream(input.reader)
val fixedInput = Translator.addBrackets(Source.fromFile(filename).getLines)
val antlrStream = new ANTLRInputStream(fixedInput.result)
val lexer = new FIRRTLLexer(antlrStream)
val tokens = new CommonTokenStream(lexer)
val parser = new FIRRTLParser(tokens)
// FIXME Dangerous
parser.getInterpreter.setPredictionMode(PredictionMode.SLL)
// Concrete Syntax Tree
val cst = parser.circuit
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
}
}
|
