LaTeXCompiler.py 20.8 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
#!/usr/bin/python

#######################################################################
#
# SYMBOLS SECTION - Can be edited. Changes will be preserved.
#
#######################################################################


import os
import sys
12
13
from functools import partial

14
15
16
17
try:
    import regex as re
except ImportError:
    import re
Eckhart Arnold's avatar
Eckhart Arnold committed
18
19
20
from DHParser import logging, is_filename, Grammar, Compiler, Lookbehind, Alternative, Pop, \
    Required, Token, Synonym, \
    Optional, NegativeLookbehind, OneOrMore, RegExp, Series, RE, Capture, \
21
    ZeroOrMore, Forward, NegativeLookahead, mixin_comment, compile_source, \
Eckhart Arnold's avatar
Eckhart Arnold committed
22
23
24
25
26
    PreprocessorFunc, \
    Node, TransformationFunc, \
    traverse, join, \
    reduce_single_child, replace_by_single_child, remove_expendables, remove_empty, flatten, \
    collapse, replace_content, remove_brackets
27
28
29
30


#######################################################################
#
Eckhart Arnold's avatar
Eckhart Arnold committed
31
# PREPROCESSOR SECTION - Can be edited. Changes will be preserved.
32
33
34
#
#######################################################################

Eckhart Arnold's avatar
Eckhart Arnold committed
35
def LaTeXPreprocessor(text):
36
37
    return text

38

Eckhart Arnold's avatar
Eckhart Arnold committed
39
40
def get_preprocessor() -> PreprocessorFunc:
    return LaTeXPreprocessor
41
42
43
44
45
46
47
48
49
50
51


#######################################################################
#
# PARSER SECTION - Don't edit! CHANGES WILL BE OVERWRITTEN!
#
#######################################################################

class LaTeXGrammar(Grammar):
    r"""Parser for a LaTeX source file, with this grammar:
    
52
    # LaTeX-Grammar for DHParser
53
54
55
56
57
58
    
    @ testing    = True
    @ whitespace = /[ \t]*(?:\n(?![ \t]*\n)[ \t]*)?/    # optional whitespace, including at most one linefeed
    @ comment    = /%.*(?:\n|$)/
    
    
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
    latexdoc       = preamble document
    preamble       = { command }+
    
    document       = [PARSEP] "\begin{document}" [PARSEP]
                     frontpages [PARSEP]
                     (Chapters | Sections) [PARSEP]
                     [Bibliography] [Index] [PARSEP]
                     "\end{document}" [PARSEP] §EOF
    frontpages     = sequence
    
    
    #######################################################################
    #
    # document structure
    #
    #######################################################################
    
    Chapters       = { Chapter [PARSEP] }+
    Chapter        = "\Chapter" block [PARSEP] { sequence | Sections }
    
    Sections       = { Section [PARSEP] }+
    Section        = "\Section" block [PARSEP] { sequence | SubSections }
    
    SubSections    = { SubSection [PARSEP] }+
    SubSection     = "\SubSection" block [PARSEP] { sequence | SubSubSections }
    
    SubSubSections = { SubSubSection [PARSEP] }+
    SubSubSection  = "\SubSubSection" block [PARSEP] { sequence | Paragraphs }
    
    Paragraphs     = { Paragraph [PARSEP] }+
    Paragraph      = "\paragraph" block [PARSEP] { sequence | SubParagraphs }
    
    SubParagraphs  = { SubParagraph [PARSEP] }+
    SubParagraph   = "\subparagpaph" block [PARSEP] { sequence }
    
    Bibliography   = "\bibliography" block [PARSEP]
    Index          = "\printindex" [PARSEP]
    
    
    #######################################################################
    #
    # document content
    #
    #######################################################################
    
    
    #### block environments ####
106
    
Eckhart Arnold's avatar
Eckhart Arnold committed
107
    block_environment   = known_environment | generic_block
108
    known_environment   = itemize | enumerate | figure | table | quotation
109
                        | verbatim
Eckhart Arnold's avatar
Eckhart Arnold committed
110
111
112
    generic_block       = begin_generic_block sequence §end_generic_block
    begin_generic_block = -&SUCC_LB begin_environment -&SUCC_LB
    end_generic_block   = -&SUCC_LB  end_environment  -&SUCC_LB
113
    
114
    itemize             = "\begin{itemize}" [PARSEP] { item } §"\end{itemize}"
115
    enumerate           = "\begin{enumerate}" [PARSEP] {item } §"\end{enumerate}"
116
    item                = "\item" [PARSEP] sequence
117
    
118
119
120
121
122
123
    figure              = "\begin{figure}" sequence "\end{figure}"
    quotation           = ("\begin{quotation}" sequence "\end{quotation}")
                        | ("\begin{quote}" sequence "\end{quote}")
    verbatim            = "\begin{verbatim}" sequence "\end{verbatim}"
    table               = "\begin{tabular}" table_config sequence "\end{tabular}"
    table_config        = "{" /[lcr|]+/~ "}"
124
125
    
    
126
127
128
    #### paragraphs and sequences of paragraphs ####
    
    block_of_paragraphs = /{/ sequence §/}/
129
    sequence            = { (paragraph | block_environment ) [PARSEP] }+
130
131
    
    paragraph           = { !blockcmd text_elements //~ }+
132
    text_elements       = command | text | block | inline_environment
133
134
135
136
    
    
    #### inline enivronments ####
    
137
    inline_environment  = known_inline_env | generic_inline_env
138
    known_inline_env    = inline_math
Eckhart Arnold's avatar
Eckhart Arnold committed
139
140
141
    generic_inline_env  = begin_inline_env { text_elements }+ §end_inline_env
    begin_inline_env    = (-!SUCC_LB begin_environment) | (begin_environment -!SUCC_LB)
    end_inline_env      = (-!SUCC_LB end_environment)   | (end_environment   -!SUCC_LB)
142
143
    begin_environment   = "\begin{" §NAME §"}"
    end_environment     = "\end{" §::NAME §"}"
144
145
146
147
148
149
150
    
    inline_math         = "$" MATH "$"
    
    
    #### commands ####
    
    command             = known_command | generic_command
Eckhart Arnold's avatar
Eckhart Arnold committed
151
    known_command       = footnote | includegraphics | caption
152
153
154
    generic_command     = CMDNAME [[ //~ config ] //~ block ]
    
    footnote            = "\footnote" block_of_paragraphs
Eckhart Arnold's avatar
Eckhart Arnold committed
155
156
    includegraphics     = "\includegraphics" config block
    caption             = "\caption" block
157
158
159
160
161
162
163
    
    #######################################################################
    #
    # low-level text and character sequences
    #
    #######################################################################
    
164
    config     = "[" cfgtext §"]"
165
    block      = /{/ { text_elements } §/}/
166
167
168
169
170
    
    text       = { cfgtext | (BRACKETS //~) }+
    cfgtext    = { word_sequence | (ESCAPED //~) }+
    word_sequence = { TEXTCHUNK //~ }+
    
171
172
173
174
175
176
177
    blockcmd   = /[\\]/ ( ( "begin{" | "end{" )
                          ( "enumerate" | "itemize" | "figure" | "quote"
                          | "quotation" | "tabular") "}"
                        | structural)
    
    structural = "subsection" | "section" | "chapter" | "subsubsection"
               | "paragraph" | "subparagraph" | "item"
178
179
180
181
182
183
184
    
    
    #######################################################################
    #
    # Primitives
    #
    #######################################################################
185
186
187
    
    CMDNAME    = /\\(?:(?!_)\w)+/~
    NAME       = /\w+/~
188
    MATH       = /[\w_^{}[\]]*/~
189
190
191
192
193
194
    
    ESCAPED    = /\\[%$&_\/]/
    BRACKETS   = /[\[\]]/                       # left or right square bracket: [ ]
    TEXTCHUNK  = /[^\\%$&\{\}\[\]\s\n]+/        # some piece of text excluding whitespace,
                                                # linefeed and special characters
    WSPC       = /[ \t]+/                       # (horizontal) whitespace
Eckhart Arnold's avatar
Eckhart Arnold committed
195
    LF         = !PARSEP /[ \t]*\n[ \t]*/       # linefeed but not an empty line
196
197
    PARSEP     = /[ \t]*(?:\n[ \t]*)+\n[ \t]*/  # at least one empty line, i.e.
                                                # [whitespace] linefeed [whitespace] linefeed
Eckhart Arnold's avatar
Eckhart Arnold committed
198
199
200
201
    EOF        = /(?!.)/
    
    SUCC_LB    = /(?!.)|(?:.*\n)+\s*$/          # linebreak succeeding an arbitrary chunk of text
    # PRED_LB    = /\s*(?!.)|\s*?\n/              # linebreak preeceding any text
202
    """
203
    block_environment = Forward()
204
205
    block_of_paragraphs = Forward()
    text_elements = Forward()
Eckhart Arnold's avatar
Eckhart Arnold committed
206
    source_hash__ = "eb91cd592f8a8c60a796ba705a121b72"
207
208
209
210
211
    parser_initialization__ = "upon instantiation"
    COMMENT__ = r'%.*(?:\n|$)'
    WSP__ = mixin_comment(whitespace=r'[ \t]*(?:\n(?![ \t]*\n)[ \t]*)?', comment=r'%.*(?:\n|$)')
    wspL__ = ''
    wspR__ = WSP__
Eckhart Arnold's avatar
Eckhart Arnold committed
212
213
214
215
216
217
218
219
    SUCC_LB = RegExp('(?!.)|(?:.*\\n)+\\s*$')
    EOF = RegExp('(?!.)')
    PARSEP = RegExp('[ \\t]*(?:\\n[ \\t]*)+\\n[ \\t]*')
    LF = Series(NegativeLookahead(PARSEP), RegExp('[ \\t]*\\n[ \\t]*'))
    WSPC = RegExp('[ \\t]+')
    TEXTCHUNK = RegExp('[^\\\\%$&\\{\\}\\[\\]\\s\\n]+')
    BRACKETS = RegExp('[\\[\\]]')
    ESCAPED = RegExp('\\\\[%$&_/]')
220
    MATH = RE('[\\w_^{}[\\]]*')
221
222
    NAME = Capture(RE('\\w+'))
    CMDNAME = RE('\\\\(?:(?!_)\\w)+')
Eckhart Arnold's avatar
Eckhart Arnold committed
223
224
225
226
227
228
229
    structural = Alternative(Token("subsection"), Token("section"), Token("chapter"),
                             Token("subsubsection"), Token("paragraph"), Token("subparagraph"),
                             Token("item"))
    blockcmd = Series(RegExp('[\\\\]'), Alternative(
        Series(Alternative(Token("begin{"), Token("end{")),
               Alternative(Token("enumerate"), Token("itemize"), Token("figure"), Token("quote"),
                           Token("quotation"), Token("tabular")), Token("}")), structural))
230
231
232
    word_sequence = OneOrMore(Series(TEXTCHUNK, RE('')))
    cfgtext = OneOrMore(Alternative(word_sequence, Series(ESCAPED, RE(''))))
    text = OneOrMore(Alternative(cfgtext, Series(BRACKETS, RE(''))))
Eckhart Arnold's avatar
Eckhart Arnold committed
233
    block = Series(RegExp('{'), ZeroOrMore(text_elements), Required(RegExp('}')))
234
    config = Series(Token("["), cfgtext, Required(Token("]")))
Eckhart Arnold's avatar
Eckhart Arnold committed
235
236
    caption = Series(Token("\\caption"), block)
    includegraphics = Series(Token("\\includegraphics"), config, block)
237
238
    footnote = Series(Token("\\footnote"), block_of_paragraphs)
    generic_command = Series(CMDNAME, Optional(Series(Optional(Series(RE(''), config)), RE(''), block)))
Eckhart Arnold's avatar
Eckhart Arnold committed
239
    known_command = Alternative(footnote, includegraphics, caption)
240
241
    command = Alternative(known_command, generic_command)
    inline_math = Series(Token("$"), MATH, Token("$"))
242
243
    end_environment = Series(Token("\\end{"), Required(Pop(NAME)), Required(Token("}")))
    begin_environment = Series(Token("\\begin{"), Required(NAME), Required(Token("}")))
Eckhart Arnold's avatar
Eckhart Arnold committed
244
245
246
247
248
249
    end_inline_env = Alternative(Series(NegativeLookbehind(SUCC_LB), end_environment),
                                 Series(end_environment, NegativeLookbehind(SUCC_LB)))
    begin_inline_env = Alternative(Series(NegativeLookbehind(SUCC_LB), begin_environment),
                                   Series(begin_environment, NegativeLookbehind(SUCC_LB)))
    generic_inline_env = Series(begin_inline_env, OneOrMore(text_elements),
                                Required(end_inline_env))
250
    known_inline_env = Synonym(inline_math)
251
252
    inline_environment = Alternative(known_inline_env, generic_inline_env)
    text_elements.set(Alternative(command, text, block, inline_environment))
253
    paragraph = OneOrMore(Series(NegativeLookahead(blockcmd), text_elements, RE('')))
254
    sequence = OneOrMore(Series(Alternative(paragraph, block_environment), Optional(PARSEP)))
Eckhart Arnold's avatar
Eckhart Arnold committed
255
    block_of_paragraphs.set(Series(RegExp('{'), sequence, Required(RegExp('}'))))
256
257
258
    table_config = Series(Token("{"), RE('[lcr|]+'), Token("}"))
    table = Series(Token("\\begin{tabular}"), table_config, sequence, Token("\\end{tabular}"))
    verbatim = Series(Token("\\begin{verbatim}"), sequence, Token("\\end{verbatim}"))
Eckhart Arnold's avatar
Eckhart Arnold committed
259
    quotation = Alternative(Series(Token("\\begin{quotation}"), sequence, Token("\\end{quotation}")), Series(Token("\\begin{quote}"), sequence, Token("\\end{quote}")))
260
261
    figure = Series(Token("\\begin{figure}"), sequence, Token("\\end{figure}"))
    item = Series(Token("\\item"), Optional(PARSEP), sequence)
262
263
    enumerate = Series(Token("\\begin{enumerate}"), Optional(PARSEP), ZeroOrMore(item),
                       Required(Token("\\end{enumerate}")))
264
    itemize = Series(Token("\\begin{itemize}"), Optional(PARSEP), ZeroOrMore(item), Required(Token("\\end{itemize}")))
Eckhart Arnold's avatar
Eckhart Arnold committed
265
266
267
    end_generic_block = Series(Lookbehind(SUCC_LB), end_environment, Lookbehind(SUCC_LB))
    begin_generic_block = Series(Lookbehind(SUCC_LB), begin_environment, Lookbehind(SUCC_LB))
    generic_block = Series(begin_generic_block, sequence, Required(end_generic_block))
268
    known_environment = Alternative(itemize, enumerate, figure, table, quotation, verbatim)
Eckhart Arnold's avatar
Eckhart Arnold committed
269
    block_environment.set(Alternative(known_environment, generic_block))
270
271
272
273
274
275
    Index = Series(Token("\\printindex"), Optional(PARSEP))
    Bibliography = Series(Token("\\bibliography"), block, Optional(PARSEP))
    SubParagraph = Series(Token("\\subparagpaph"), block, Optional(PARSEP), ZeroOrMore(sequence))
    SubParagraphs = OneOrMore(Series(SubParagraph, Optional(PARSEP)))
    Paragraph = Series(Token("\\paragraph"), block, Optional(PARSEP), ZeroOrMore(Alternative(sequence, SubParagraphs)))
    Paragraphs = OneOrMore(Series(Paragraph, Optional(PARSEP)))
Eckhart Arnold's avatar
Eckhart Arnold committed
276
    SubSubSection = Series(Token("\\SubSubSection"), block, Optional(PARSEP), ZeroOrMore(Alternative(sequence, Paragraphs)))
277
    SubSubSections = OneOrMore(Series(SubSubSection, Optional(PARSEP)))
Eckhart Arnold's avatar
Eckhart Arnold committed
278
    SubSection = Series(Token("\\SubSection"), block, Optional(PARSEP), ZeroOrMore(Alternative(sequence, SubSubSections)))
279
280
281
282
283
284
    SubSections = OneOrMore(Series(SubSection, Optional(PARSEP)))
    Section = Series(Token("\\Section"), block, Optional(PARSEP), ZeroOrMore(Alternative(sequence, SubSections)))
    Sections = OneOrMore(Series(Section, Optional(PARSEP)))
    Chapter = Series(Token("\\Chapter"), block, Optional(PARSEP), ZeroOrMore(Alternative(sequence, Sections)))
    Chapters = OneOrMore(Series(Chapter, Optional(PARSEP)))
    frontpages = Synonym(sequence)
Eckhart Arnold's avatar
Eckhart Arnold committed
285
    document = Series(Optional(PARSEP), Token("\\begin{document}"), Optional(PARSEP), frontpages, Optional(PARSEP), Alternative(Chapters, Sections), Optional(PARSEP), Optional(Bibliography), Optional(Index), Optional(PARSEP), Token("\\end{document}"), Optional(PARSEP), Required(EOF))
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
    preamble = OneOrMore(command)
    latexdoc = Series(preamble, document)
    root__ = latexdoc
    
def get_grammar() -> LaTeXGrammar:
    global thread_local_LaTeX_grammar_singleton
    try:
        grammar = thread_local_LaTeX_grammar_singleton
        return grammar
    except NameError:
        thread_local_LaTeX_grammar_singleton = LaTeXGrammar()
        return thread_local_LaTeX_grammar_singleton


#######################################################################
#
# AST SECTION - Can be edited. Changes will be preserved.
#
#######################################################################


def streamline_whitespace(node):
    assert node.tag_name in ['WSPC', ':Whitespace']
    s = str(node)
    c = s.find('%')
    n = s.find('\n')
    if c >= 0:
        node.result = ('  ' if (n >= c) or (n < 0) else '\n')+ s[c:].rstrip(' \t')
    elif s.find('\n') >= 0:
        node.result = '\n'
    else:
        node.result = ' '


LaTeX_AST_transformation_table = {
    # AST Transformations for the LaTeX-grammar
    "+":
        remove_empty,
    "latexdoc": [],
    "preamble": [],
    "document": [],
    "blockenv": [],
    "parblock": [],
    "sequence":
        flatten,
    "paragraph":
        [flatten(lambda node: not node.parser.name or node.parser.name == "text"),
         join('text', ':Whitespace')],
    "inlineenv": [],
    "beginenv": [],
    "endenv": [],
    "command": [],
    "config": [],
    "block": [remove_brackets, reduce_single_child],
    "text":
        [reduce_single_child, join('text', 'word_sequence', ':Whitespace')],
    "cfgtext": [flatten, reduce_single_child],
    "word_sequence":
        [collapse],
    "blockcmd": [],
    "CMDNAME":
        [remove_expendables, reduce_single_child],
    "NAME": [],
    "ESCAPED": [reduce_single_child],
    "BRACKETS": [],
    "TEXTCHUNK": [],
    "WSPC, :Whitespace":
        streamline_whitespace,
    "LF":
        replace_content(lambda node: '\n'),
    "PARSEP":
        replace_content(lambda node: '\n\n'),
    "EOF": [],
    "*":
        replace_by_single_child,
}

LaTeXTransform = partial(traverse, processing_table=LaTeX_AST_transformation_table)
# LaTeXTransform = lambda tree : 1

def get_transformer() -> TransformationFunc:
    return LaTeXTransform


#######################################################################
#
# COMPILER SECTION - Can be edited. Changes will be preserved.
#
#######################################################################

class LaTeXCompiler(Compiler):
    """Compiler for the abstract-syntax-tree of a LaTeX source file.
    """

    def __init__(self, grammar_name="LaTeX", grammar_source=""):
        super(LaTeXCompiler, self).__init__(grammar_name, grammar_source)
        assert re.match('\w+\Z', grammar_name)

    def on_latexdoc(self, node):
Eckhart Arnold's avatar
Eckhart Arnold committed
385
        return node
386
387
388
389
390
391
392

    def on_preamble(self, node):
        pass

    def on_document(self, node):
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
    def on_frontpages(self, node):
        pass

    def on_Chapters(self, node):
        pass

    def on_Chapter(self, node):
        pass

    def on_Sections(self, node):
        pass

    def on_Section(self, node):
        pass

    def on_SubSections(self, node):
        pass

    def on_SubSection(self, node):
        pass

    def on_SubSubSections(self, node):
        pass

    def on_SubSubSection(self, node):
        pass

    def on_Paragraphs(self, node):
        pass

    def on_Paragraph(self, node):
        pass

    def on_SubParagraphs(self, node):
        pass

    def on_SubParagraph(self, node):
        pass

    def on_Bibliography(self, node):
        pass

    def on_Index(self, node):
        pass

    def on_block_environment(self, node):
        pass

    def on_known_environment(self, node):
        pass

    def on_generic_block(self, node):
        pass

    def on_begin_generic_block(self, node):
        pass

    def on_end_generic_block(self, node):
        pass

    def on_itemize(self, node):
        pass

    def on_enumerate(self, node):
        pass

    def on_item(self, node):
        pass

    def on_figure(self, node):
        pass

    def on_quotation(self, node):
        pass

    def on_verbatim(self, node):
        pass

    def on_table(self, node):
472
473
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
474
475
476
477
    def on_table_config(self, node):
        pass

    def on_block_of_paragraphs(self, node):
478
479
480
481
482
483
484
485
        pass

    def on_sequence(self, node):
        pass

    def on_paragraph(self, node):
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
486
487
488
489
490
491
492
    def on_text_elements(self, node):
        pass

    def on_inline_environment(self, node):
        pass

    def on_known_inline_env(self, node):
493
494
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
495
    def on_generic_inline_env(self, node):
496
497
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
498
499
500
501
502
503
504
505
506
507
508
509
510
    def on_begin_inline_env(self, node):
        pass

    def on_end_inline_env(self, node):
        pass

    def on_begin_environment(self, node):
        pass

    def on_end_environment(self, node):
        pass

    def on_inline_math(self, node):
511
512
513
514
515
        pass

    def on_command(self, node):
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
    def on_known_command(self, node):
        pass

    def on_generic_command(self, node):
        pass

    def on_footnote(self, node):
        pass

    def on_includegraphics(self, node):
        pass

    def on_caption(self, node):
        pass

531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
    def on_config(self, node):
        pass

    def on_block(self, node):
        pass

    def on_text(self, node):
        pass

    def on_cfgtext(self, node):
        pass

    def on_word_sequence(self, node):
        pass

    def on_blockcmd(self, node):
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
549
550
551
    def on_structural(self, node):
        pass

552
553
554
555
556
557
    def on_CMDNAME(self, node):
        pass

    def on_NAME(self, node):
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
558
559
560
    def on_MATH(self, node):
        pass

561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
    def on_ESCAPED(self, node):
        pass

    def on_BRACKETS(self, node):
        pass

    def on_TEXTCHUNK(self, node):
        pass

    def on_WSPC(self, node):
        pass

    def on_LF(self, node):
        pass

    def on_PARSEP(self, node):
        pass

    def on_EOF(self, node):
        pass

Eckhart Arnold's avatar
Eckhart Arnold committed
582
583
584
585
586
587
    def on_SUCC_LB(self, node):
        pass

    def on_PRED_LB(self, node):
        pass

588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614

def get_compiler(grammar_name="LaTeX", grammar_source="") -> LaTeXCompiler:
    global thread_local_LaTeX_compiler_singleton
    try:
        compiler = thread_local_LaTeX_compiler_singleton
        compiler.set_grammar_name(grammar_name, grammar_source)
        return compiler
    except NameError:
        thread_local_LaTeX_compiler_singleton = \
            LaTeXCompiler(grammar_name, grammar_source)
        return thread_local_LaTeX_compiler_singleton 


#######################################################################
#
# END OF DHPARSER-SECTIONS
#
#######################################################################


def compile_src(source):
    """Compiles ``source`` and returns (result, errors, ast).
    """
    with logging("LOGS"):
        compiler = get_compiler()
        cname = compiler.__class__.__name__
        log_file_name = os.path.basename(os.path.splitext(source)[0]) \
Eckhart Arnold's avatar
Eckhart Arnold committed
615
616
            if is_filename(source) < 0 else cname[:cname.find('.')] + '_out'
        result = compile_source(source, get_preprocessor(),
617
618
619
620
621
622
623
624
625
626
627
628
629
                                get_grammar(),
                                get_transformer(), compiler)
    return result


if __name__ == "__main__":
    if len(sys.argv) > 1:
        result, errors, ast = compile_src(sys.argv[1])
        if errors:
            for error in errors:
                print(error)
            sys.exit(1)
        else:
Eckhart Arnold's avatar
Eckhart Arnold committed
630
            print(result.as_xml() if isinstance(result, Node) else result)
631
632
    else:
        print("Usage: LaTeXCompiler.py [FILENAME]")