X-Git-Url: http://nitlanguage.org diff --git a/src/test_parser.nit b/src/test_parser.nit index 193a734..6ad8038 100644 --- a/src/test_parser.nit +++ b/src/test_parser.nit @@ -19,10 +19,12 @@ module test_parser import parser import parser_util +import astutil +# A basic visitor that prints AST trees to the screen class PrintTreeVisitor super Visitor - private var rank: Int + private var rank: Int = 0 redef fun visit(n) do if n isa Token then @@ -34,11 +36,6 @@ class PrintTreeVisitor n.visit_all(self) rank = rank - 1 end - - init - do - rank = 0 - end end var no_print = false @@ -46,6 +43,7 @@ var only_lexer = false var need_help = false var no_file = false var interactive = false +var xml = false while not args.is_empty and args.first.chars.first == '-' do if args.first == "-n" then @@ -54,6 +52,8 @@ while not args.is_empty and args.first.chars.first == '-' do only_lexer = true else if args.first == "-p" then only_lexer = false + else if args.first == "-x" then + xml = true else if args.first == "-e" then no_file = true else if args.first == "-i" then @@ -76,15 +76,16 @@ if (args.is_empty and not interactive) or need_help then print(" -n do not print anything") print(" -l only lexer") print(" -p lexer and parser (default)") + print(" -x instead of a ascii tree, output a XML document") print(" -e instead on files, each argument is a content to parse") print(" -i tree to parse are read interactively") print(" -h print this help") else if interactive then if only_lexer then - print "Error: -l and -i are incompatibles" + print "Error: -l and -i are incompatible" exit 1 else if no_file then - print "Error: -e and -i are incompatibles" + print "Error: -e and -i are incompatible" exit 1 else if not args.is_empty then print "Error: -i works without arguments" @@ -120,7 +121,7 @@ else if no_file then source = new SourceFile.from_string("", a) else - var f = new IFStream.open(a) + var f = new FileReader.open(a) source = new SourceFile(a, f) f.close end @@ -143,7 +144,10 @@ else return end - if not no_print then + if xml then + tree.parentize_tokens + tree.to_xml.write_to(stdout) + else if not no_print then (new PrintTreeVisitor).enter_visit(tree) end end