import sys

from antlr4 import *
from HelloLexer import HelloLexer
from HelloParser import HelloParser
from HelloListener import HelloListener

from antlr4.tree.Trees import Trees


def main(argv):
    if len(argv) == 1:
        InputStream = StdinStream()
    else:
        inpname = argv[1]
        InputStream = FileStream(inpname)
    lexer = HelloLexer(InputStream)
    tokstream = CommonTokenStream(lexer)
    parser = HelloParser(tokstream)

    tree = parser.start()

    print(Trees.toStringTree(tree, None, parser))
if __name__ == '__main__':
    main(sys.argv)