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
|
##########################################################################
## # The Coq Proof Assistant / The Coq Development Team ##
## v # Copyright INRIA, CNRS and contributors ##
## <O___,, # (see version control and CREDITS file for authors & dates) ##
## \VV/ ###############################################################
## // # This file is distributed under the terms of the ##
## # GNU Lesser General Public License Version 2.1 ##
## # (see LICENSE file for the text of the license) ##
##########################################################################
"""A visitor for ANTLR notation ASTs, producing plain text with ellipses.
Somewhat-closely approximates the rendering of the original manual.
"""
from io import StringIO
from .parsing import parse
from .TacticNotationsParser import TacticNotationsParser
from .TacticNotationsVisitor import TacticNotationsVisitor
class TacticNotationsToDotsVisitor(TacticNotationsVisitor):
def __init__(self):
self.buffer = StringIO()
def visitAlternative(self, ctx:TacticNotationsParser.AlternativeContext):
self.buffer.write("[")
self.visitChildren(ctx)
self.buffer.write("]")
def visitAltsep(self, ctx:TacticNotationsParser.AltsepContext):
self.buffer.write("|")
def visitRepeat(self, ctx:TacticNotationsParser.RepeatContext):
separator = ctx.ATOM() or ctx.PIPE()
self.visitChildren(ctx)
if ctx.LGROUP().getText()[1] == "+":
spacer = (separator.getText() + " " if separator else "")
self.buffer.write(spacer + "…" + spacer)
self.visitChildren(ctx)
def visitCurlies(self, ctx:TacticNotationsParser.CurliesContext):
self.buffer.write("{")
self.visitChildren(ctx)
self.buffer.write("}")
def visitAtomic(self, ctx:TacticNotationsParser.AtomicContext):
self.buffer.write(ctx.ATOM().getText())
def visitPipe(self, ctx:TacticNotationsParser.PipeContext):
self.buffer.write("|")
def visitHole(self, ctx:TacticNotationsParser.HoleContext):
self.buffer.write("‘{}’".format(ctx.ID().getText()[1:]))
def visitEscaped(self, ctx:TacticNotationsParser.EscapedContext):
self.buffer.write(ctx.ESCAPED().getText().replace("%", ""))
def visitWhitespace(self, ctx:TacticNotationsParser.WhitespaceContext):
self.buffer.write(" ")
def stringify_with_ellipses(notation):
vs = TacticNotationsToDotsVisitor()
vs.visit(parse(notation))
return vs.buffer.getvalue()
|
