]> git.lyx.org Git - lyx.git/blob - lib/lyx2lyx/lyx_2_0.py
Make sure we delete this line.
[lyx.git] / lib / lyx2lyx / lyx_2_0.py
1 # -*- coding: utf-8 -*-
2 # This file is part of lyx2lyx
3 # -*- coding: utf-8 -*-
4 # Copyright (C) 2010 The LyX team
5 #
6 # This program is free software; you can redistribute it and/or
7 # modify it under the terms of the GNU General Public License
8 # as published by the Free Software Foundation; either version 2
9 # of the License, or (at your option) any later version.
10 #
11 # This program is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU General Public License for more details.
15 #
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
19
20 """ Convert files to the file format generated by lyx 2.0"""
21
22 import re, string
23 import unicodedata
24 import sys, os
25
26 from parser_tools import find_token, find_end_of, find_tokens, \
27   find_token_exact, find_end_of_inset, find_end_of_layout, \
28   find_token_backwards, is_in_inset, get_value, get_quoted_value, \
29   del_token, check_token
30   
31 from lyx2lyx_tools import add_to_preamble, insert_to_preamble, \
32   put_cmd_in_ert, lyx2latex, latex_length, revert_flex_inset, \
33   revert_font_attrs, hex2ratio, str2bool
34
35 ####################################################################
36 # Private helper functions
37
38 def remove_option(lines, m, option):
39     ''' removes option from line m. returns whether we did anything '''
40     l = lines[m].find(option)
41     if l == -1:
42         return False
43     val = lines[m][l:].split('"')[1]
44     lines[m] = lines[m][:l - 1] + lines[m][l+len(option + '="' + val + '"'):]
45     return True
46
47
48 # DO NOT USE THIS ROUTINE ANY MORE. Better yet, replace the uses that
49 # have been made of it with uses of put_cmd_in_ert.
50 def old_put_cmd_in_ert(string):
51     for rep in unicode_reps:
52         string = string.replace(rep[1], rep[0].replace('\\\\', '\\'))
53     string = string.replace('\\', "\\backslash\n")
54     string = "\\begin_inset ERT\nstatus collapsed\n\\begin_layout Plain Layout\n" \
55       + string + "\n\\end_layout\n\\end_inset"
56     return string
57
58
59 ###############################################################################
60 ###
61 ### Conversion and reversion routines
62 ###
63 ###############################################################################
64
65 def revert_swiss(document):
66     " Set language german-ch to ngerman "
67     i = 0
68     if document.language == "german-ch":
69         document.language = "ngerman"
70         i = find_token(document.header, "\\language", 0)
71         if i != -1:
72             document.header[i] = "\\language ngerman"
73     j = 0
74     while True:
75         j = find_token(document.body, "\\lang german-ch", j)
76         if j == -1:
77             return
78         document.body[j] = document.body[j].replace("\\lang german-ch", "\\lang ngerman")
79         j = j + 1
80
81
82 def revert_tabularvalign(document):
83    " Revert the tabular valign option "
84    i = 0
85    while True:
86       i = find_token(document.body, "\\begin_inset Tabular", i)
87       if i == -1:
88           return
89       end = find_end_of_inset(document.body, i)
90       if end == -1:
91           document.warning("Can't find end of inset at line " + str(i))
92           i += 1
93           continue
94       fline = find_token(document.body, "<features", i, end)
95       if fline == -1:
96           document.warning("Can't find features for inset at line " + str(i))
97           i += 1
98           continue
99       p = document.body[fline].find("islongtable")
100       if p != -1:
101           q = document.body[fline].find("tabularvalignment")
102           if q != -1:
103               # FIXME
104               # This seems wrong: It removes everything after 
105               # tabularvalignment, too.
106               document.body[fline] = document.body[fline][:q - 1] + '>'
107           i += 1
108           continue
109
110        # no longtable
111       tabularvalignment = 'c'
112       # which valignment is specified?
113       m = document.body[fline].find('tabularvalignment="top"')
114       if m != -1:
115           tabularvalignment = 't'
116       m = document.body[fline].find('tabularvalignment="bottom"')
117       if m != -1:
118           tabularvalignment = 'b'
119       # delete tabularvalignment
120       q = document.body[fline].find("tabularvalignment")
121       if q != -1:
122           # FIXME
123           # This seems wrong: It removes everything after 
124           # tabularvalignment, too.
125           document.body[fline] = document.body[fline][:q - 1] + '>'
126
127       # don't add a box when centered
128       if tabularvalignment == 'c':
129           i = end
130           continue
131       subst = ['\\end_layout', '\\end_inset']
132       document.body[end:end] = subst # just inserts those lines
133       subst = ['\\begin_inset Box Frameless',
134           'position "' + tabularvalignment +'"',
135           'hor_pos "c"',
136           'has_inner_box 1',
137           'inner_pos "c"',
138           'use_parbox 0',
139           # we don't know the width, assume 50%
140           'width "50col%"',
141           'special "none"',
142           'height "1in"',
143           'height_special "totalheight"',
144           'status open',
145           '',
146           '\\begin_layout Plain Layout']
147       document.body[i:i] = subst # this just inserts the array at i
148       # since there could be a tabular inside a tabular, we cannot
149       # jump to end
150       i += len(subst)
151
152
153 def revert_phantom_types(document, ptype, cmd):
154     " Reverts phantom to ERT "
155     i = 0
156     while True:
157       i = find_token(document.body, "\\begin_inset Phantom " + ptype, i)
158       if i == -1:
159           return
160       end = find_end_of_inset(document.body, i)
161       if end == -1:
162           document.warning("Can't find end of inset at line " + str(i))
163           i += 1
164           continue
165       blay = find_token(document.body, "\\begin_layout Plain Layout", i, end)
166       if blay == -1:
167           document.warning("Can't find layout for inset at line " + str(i))
168           i = end
169           continue
170       bend = find_end_of_layout(document.body, blay)
171       if bend == -1:
172           document.warning("Malformed LyX document: Could not find end of Phantom inset's layout.")
173           i = end
174           continue
175       substi = ["\\begin_inset ERT", "status collapsed", "",
176                 "\\begin_layout Plain Layout", "", "", "\\backslash", 
177                 cmd + "{", "\\end_layout", "", "\\end_inset"]
178       substj = ["\\size default", "", "\\begin_inset ERT", "status collapsed", "",
179                 "\\begin_layout Plain Layout", "", "}", "\\end_layout", "", "\\end_inset"]
180       # do the later one first so as not to mess up the numbering
181       document.body[bend:end + 1] = substj
182       document.body[i:blay + 1] = substi
183       i = end + len(substi) + len(substj) - (end - bend) - (blay - i) - 2
184
185
186 def revert_phantom(document):
187     revert_phantom_types(document, "Phantom", "phantom")
188     
189 def revert_hphantom(document):
190     revert_phantom_types(document, "HPhantom", "hphantom")
191
192 def revert_vphantom(document):
193     revert_phantom_types(document, "VPhantom", "vphantom")
194
195
196 def revert_xetex(document):
197     " Reverts documents that use XeTeX "
198
199     i = find_token(document.header, '\\use_xetex', 0)
200     if i == -1:
201         document.warning("Malformed LyX document: Missing \\use_xetex.")
202         return
203     if not str2bool(get_value(document.header, "\\use_xetex", i)):
204         del document.header[i]
205         return
206     del document.header[i]
207
208     # 1.) set doc encoding to utf8-plain
209     i = find_token(document.header, "\\inputencoding", 0)
210     if i == -1:
211         document.warning("Malformed LyX document: Missing \\inputencoding.")
212     else:
213         document.header[i] = "\\inputencoding utf8-plain"
214
215     # 2.) check font settings
216     # defaults
217     roman = sans = typew = default
218     osf = False
219     sf_scale = tt_scale = 100.0
220     
221     i = find_token(document.header, "\\font_roman", 0)
222     if i == -1:
223         document.warning("Malformed LyX document: Missing \\font_roman.")
224     else:
225         roman = get_value(document.header, "\\font_roman", i)
226         document.header[i] = "\\font_roman default"
227
228     i = find_token(document.header, "\\font_sans", 0)
229     if i == -1:
230         document.warning("Malformed LyX document: Missing \\font_sans.")
231     else:
232         sans = get_value(document.header, "\\font_sans", i)
233         document.header[i] = "\\font_sans default"
234     
235     i = find_token(document.header, "\\font_typewriter", 0)
236     if i == -1:
237         document.warning("Malformed LyX document: Missing \\font_typewriter.")
238     else:
239         typew = get_value(document.header, "\\font_typewriter", i)
240         document.header[i] = "\\font_typewriter default"
241
242     i = find_token(document.header, "\\font_osf", 0)
243     if i == -1:
244         document.warning("Malformed LyX document: Missing \\font_osf.")
245     else:
246         osf = str2bool(get_value(document.header, "\\font_osf", i))
247         document.header[i] = "\\font_osf false"
248
249     i = find_token(document.header, "\\font_sc", 0)
250     if i == -1:
251         document.warning("Malformed LyX document: Missing \\font_sc.")
252     else:
253         # we do not need this value.
254         document.header[i] = "\\font_sc false"
255     
256     i = find_token(document.header, "\\font_sf_scale", 0)
257     if i == -1:
258         document.warning("Malformed LyX document: Missing \\font_sf_scale.")
259     else:
260       val = get_value(document.header, '\\font_sf_scale', i)
261       try:
262         # float() can throw
263         sf_scale = float(val)
264       except:
265         document.warning("Invalid font_sf_scale value: " + val)
266       document.header[i] = "\\font_sf_scale 100"
267
268     i = find_token(document.header, "\\font_tt_scale", 0)
269     if i == -1:
270         document.warning("Malformed LyX document: Missing \\font_tt_scale.")
271     else:
272         val = get_value(document.header, '\\font_tt_scale', i)
273         try:
274           # float() can throw
275           tt_scale = float(val)
276         except:
277           document.warning("Invalid font_tt_scale value: " + val)
278         document.header[i] = "\\font_tt_scale 100"
279
280     # 3.) set preamble stuff
281     pretext = ['%% This document must be processed with xelatex!']
282     pretext.append('\\usepackage{fontspec}')
283     if roman != "default":
284         pretext.append('\\setmainfont[Mapping=tex-text]{' + roman + '}')
285     if sans != "default":
286         sf = '\\setsansfont['
287         if sf_scale != 100.0:
288             sf += 'Scale=' + str(sf_scale / 100.0) + ','
289         sf += 'Mapping=tex-text]{' + sans + '}'
290         pretext.append(sf)
291     if typewriter != "default":
292         tw = '\\setmonofont'
293         if tt_scale != 100.0:
294             tw += '[Scale=' + str(tt_scale / 100.0) + ']'
295         tw += '{' + typewriter + '}'
296         pretext.append(tw)
297     if osf:
298         pretext.append('\\defaultfontfeatures{Numbers=OldStyle}')
299     pretext.append('\usepackage{xunicode}')
300     pretext.append('\usepackage{xltxtra}')
301     insert_to_preamble(document, pretext)
302
303
304 def revert_outputformat(document):
305     " Remove default output format param "
306     
307     if not del_token(document.header, '\\default_output_format', 0):
308         document.warning("Malformed LyX document: Missing \\default_output_format.")
309
310
311 def revert_backgroundcolor(document):
312     " Reverts background color to preamble code "
313     i = find_token(document.header, "\\backgroundcolor", 0)
314     if i == -1:
315         return
316     colorcode = get_value(document.header, '\\backgroundcolor', i)
317     del document.header[i]
318     # don't clutter the preamble if backgroundcolor is not set
319     if colorcode == "#ffffff":
320         return
321     red   = hex2ratio(colorcode[1:3])
322     green = hex2ratio(colorcode[3:5])
323     blue  = hex2ratio(colorcode[5:7])
324     insert_to_preamble(document, \
325         ['% To set the background color',
326         '\\@ifundefined{definecolor}{\\usepackage{color}}{}',
327         '\\definecolor{page_backgroundcolor}{rgb}{' + red + ',' + green + ',' + blue + '}',
328         '\\pagecolor{page_backgroundcolor}'])
329
330
331 def revert_splitindex(document):
332     " Reverts splitindex-aware documents "
333     i = find_token(document.header, '\\use_indices', 0)
334     if i == -1:
335         document.warning("Malformed LyX document: Missing \\use_indices.")
336         return
337     useindices = str2bool(get_value(document.header, "\\use_indices", i))
338     del document.header[i]
339     preamble = []
340     if useindices:
341          preamble.append("\\usepackage{splitidx})")
342     
343     # deal with index declarations in the preamble
344     i = 0
345     while True:
346         i = find_token(document.header, "\\index", i)
347         if i == -1:
348             break
349         k = find_token(document.header, "\\end_index", i)
350         if k == -1:
351             document.warning("Malformed LyX document: Missing \\end_index.")
352             return
353         if useindices:    
354           line = document.header[i]
355           l = re.compile(r'\\index (.*)$')
356           m = l.match(line)
357           iname = m.group(1)
358           ishortcut = get_value(document.header, '\\shortcut', i, k)
359           if ishortcut != "":
360               preamble.append("\\newindex[" + iname + "]{" + ishortcut + "}")
361         del document.header[i:k + 1]
362     if preamble:
363         insert_to_preamble(document, preamble)
364         
365     # deal with index insets
366     # these need to have the argument removed
367     i = 0
368     while True:
369         i = find_token(document.body, "\\begin_inset Index", i)
370         if i == -1:
371             break
372         line = document.body[i]
373         l = re.compile(r'\\begin_inset Index (.*)$')
374         m = l.match(line)
375         itype = m.group(1)
376         if itype == "idx" or indices == "false":
377             document.body[i] = "\\begin_inset Index"
378         else:
379             k = find_end_of_inset(document.body, i)
380             if k == -1:
381                 document.warning("Can't find end of index inset!")
382                 i += 1
383                 continue
384             content = lyx2latex(document, document.body[i:k])
385             # escape quotes
386             content = content.replace('"', r'\"')
387             subst = put_cmd_in_ert("\\sindex[" + itype + "]{" + content + "}")
388             document.body[i:k + 1] = subst
389         i = i + 1
390         
391     # deal with index_print insets
392     i = 0
393     while True:
394         i = find_token(document.body, "\\begin_inset CommandInset index_print", i)
395         if i == -1:
396             return
397         k = find_end_of_inset(document.body, i)
398         ptype = get_quoted_value(document.body, 'type', i, k)
399         if ptype == "idx":
400             j = find_token(document.body, "type", i, k)
401             del document.body[j]
402         elif not useindices:
403             del document.body[i:k + 1]
404         else:
405             subst = put_cmd_in_ert("\\printindex[" + ptype + "]{}")
406             document.body[i:k + 1] = subst
407         i = i + 1
408
409
410 def convert_splitindex(document):
411     " Converts index and printindex insets to splitindex-aware format "
412     i = 0
413     while True:
414         i = find_token(document.body, "\\begin_inset Index", i)
415         if i == -1:
416             break
417         document.body[i] = document.body[i].replace("\\begin_inset Index",
418             "\\begin_inset Index idx")
419         i = i + 1
420     i = 0
421     while True:
422         i = find_token(document.body, "\\begin_inset CommandInset index_print", i)
423         if i == -1:
424             return
425         if document.body[i + 1].find('LatexCommand printindex') == -1:
426             document.warning("Malformed LyX document: Incomplete printindex inset.")
427             return
428         subst = ["LatexCommand printindex", 
429             "type \"idx\""]
430         document.body[i + 1:i + 2] = subst
431         i = i + 1
432
433
434 def revert_subindex(document):
435     " Reverts \\printsubindex CommandInset types "
436     i = find_token(document.header, '\\use_indices', 0)
437     if i == -1:
438         document.warning("Malformed LyX document: Missing \\use_indices.")
439         return
440     useindices = str2bool(get_value(document.header, "\\use_indices", i))
441     i = 0
442     while True:
443         i = find_token(document.body, "\\begin_inset CommandInset index_print", i)
444         if i == -1:
445             return
446         k = find_end_of_inset(document.body, i)
447         ctype = get_value(document.body, 'LatexCommand', i, k)
448         if ctype != "printsubindex":
449             i = k + 1
450             continue
451         ptype = get_quoted_value(document.body, 'type', i, k)
452         if not useindices:
453             del document.body[i:k + 1]
454         else:
455             subst = put_cmd_in_ert("\\printsubindex[" + ptype + "]{}")
456             document.body[i:k + 1] = subst
457         i = i + 1
458
459
460 def revert_printindexall(document):
461     " Reverts \\print[sub]index* CommandInset types "
462     i = find_token(document.header, '\\use_indices', 0)
463     if i == -1:
464         document.warning("Malformed LyX document: Missing \\use_indices.")
465         return
466     useindices = str2bool(get_value(document.header, "\\use_indices", i))
467     i = 0
468     while True:
469         i = find_token(document.body, "\\begin_inset CommandInset index_print", i)
470         if i == -1:
471             return
472         k = find_end_of_inset(document.body, i)
473         ctype = get_value(document.body, 'LatexCommand', i, k)
474         if ctype != "printindex*" and ctype != "printsubindex*":
475             i = k
476             continue
477         if not useindices:
478             del document.body[i:k + 1]
479         else:
480             subst = put_cmd_in_ert("\\" + ctype + "{}")
481             document.body[i:k + 1] = subst
482         i = i + 1
483
484
485 def revert_strikeout(document):
486   " Reverts \\strikeout font attribute "
487   changed = revert_font_attrs(document.body, "\\uuline", "\\uuline")
488   changed = revert_font_attrs(document.body, "\\uwave", "\\uwave") or changed
489   changed = revert_font_attrs(document.body, "\\strikeout", "\\sout")  or changed
490   if changed == True:
491     insert_to_preamble(document, \
492         ['%  for proper underlining',
493         '\\PassOptionsToPackage{normalem}{ulem}',
494         '\\usepackage{ulem}'])
495
496
497 def revert_ulinelatex(document):
498     " Reverts \\uline font attribute "
499     i = find_token(document.body, '\\bar under', 0)
500     if i == -1:
501         return
502     insert_to_preamble(document,\
503             ['%  for proper underlining',
504             '\\PassOptionsToPackage{normalem}{ulem}',
505             '\\usepackage{ulem}',
506             '\\let\\cite@rig\\cite',
507             '\\newcommand{\\b@xcite}[2][\\%]{\\def\\def@pt{\\%}\\def\\pas@pt{#1}',
508             '  \\mbox{\\ifx\\def@pt\\pas@pt\\cite@rig{#2}\\else\\cite@rig[#1]{#2}\\fi}}',
509             '\\renewcommand{\\underbar}[1]{{\\let\\cite\\b@xcite\\uline{#1}}}'])
510
511
512 def revert_custom_processors(document):
513     " Remove bibtex_command and index_command params "
514     
515     if not del_token(document.header, '\\bibtex_command', 0):
516         document.warning("Malformed LyX document: Missing \\bibtex_command.")
517     
518     if not del_token(document.header, '\\index_command', 0):
519         document.warning("Malformed LyX document: Missing \\index_command.")
520
521
522 def convert_nomencl_width(document):
523     " Add set_width param to nomencl_print "
524     i = 0
525     while True:
526       i = find_token(document.body, "\\begin_inset CommandInset nomencl_print", i)
527       if i == -1:
528         break
529       document.body.insert(i + 2, "set_width \"none\"")
530       i = i + 1
531
532
533 def revert_nomencl_width(document):
534     " Remove set_width param from nomencl_print "
535     i = 0
536     while True:
537       i = find_token(document.body, "\\begin_inset CommandInset nomencl_print", i)
538       if i == -1:
539         break
540       j = find_end_of_inset(document.body, i)
541       if not del_token(document.body, "set_width", i, j):
542             document.warning("Can't find set_width option for nomencl_print!")
543       i = j
544
545
546 def revert_nomencl_cwidth(document):
547     " Remove width param from nomencl_print "
548     i = 0
549     while True:
550       i = find_token(document.body, "\\begin_inset CommandInset nomencl_print", i)
551       if i == -1:
552         break
553       j = find_end_of_inset(document.body, i)
554       l = find_token(document.body, "width", i, j)
555       if l == -1:
556         document.warning("Can't find width option for nomencl_print!")
557         i = j
558         continue
559       width = get_quoted_value(document.body, "width", i, j)
560       del document.body[l]
561       insert_to_preamble(document, ["\\setlength{\\nomlabelwidth}{" + width + "}"])
562       i = j - 1
563
564
565 def revert_applemac(document):
566     " Revert applemac encoding to auto "
567     if document.encoding != "applemac":
568       return
569     document.encoding = "auto"
570     i = find_token(document.header, "\\encoding", 0)
571     if i != -1:
572         document.header[i] = "\\encoding auto"
573
574
575 def revert_longtable_align(document):
576     " Remove longtable alignment setting "
577     i = 0
578     while True:
579       i = find_token(document.body, "\\begin_inset Tabular", i)
580       if i == -1:
581           break
582       end = find_end_of_inset(document.body, i)
583       if end == -1:
584           document.warning("Can't find end of inset at line " + str(i))
585           i += 1
586           continue
587       fline = find_token(document.body, "<features", i, end)
588       if fline == -1:
589           document.warning("Can't find features for inset at line " + str(i))
590           i += 1
591           continue
592       j = document.body[fline].find("longtabularalignment")
593       if j == -1:
594           i += 1
595           continue
596       # FIXME Is this correct? It wipes out everything after the 
597       # one we found.
598       document.body[fline] = document.body[fline][:j - 1] + '>'
599       # since there could be a tabular inside this one, we 
600       # cannot jump to end.
601       i += 1
602
603
604 def revert_branch_filename(document):
605     " Remove \\filename_suffix parameter from branches "
606     i = 0
607     while True:
608         i = find_token(document.header, "\\filename_suffix", i)
609         if i == -1:
610             return
611         del document.header[i]
612
613
614 def revert_paragraph_indentation(document):
615     " Revert custom paragraph indentation to preamble code "
616     i = find_token(document.header, "\\paragraph_indentation", 0)
617     if i == -1:
618       return
619     length = get_value(document.header, "\\paragraph_indentation", i)
620     # we need only remove the line if indentation is default
621     if length != "default":
622       # handle percent lengths
623       length = latex_length(length)[1]
624       insert_to_preamble(document, ["\\setlength{\\parindent}{" + length + "}"])
625     del document.header[i]
626
627
628 def revert_percent_skip_lengths(document):
629     " Revert relative lengths for paragraph skip separation to preamble code "
630     i = find_token(document.header, "\\defskip", 0)
631     if i == -1:
632         return
633     length = get_value(document.header, "\\defskip", i)
634     # only revert when a custom length was set and when
635     # it used a percent length
636     if length in ('smallskip', 'medskip', 'bigskip'):
637         return
638     # handle percent lengths
639     percent, length = latex_length(length)
640     if percent:
641         insert_to_preamble(document, ["\\setlength{\\parskip}{" + length + "}"])
642         # set defskip to medskip as default
643         document.header[i] = "\\defskip medskip"
644
645
646 def revert_percent_vspace_lengths(document):
647     " Revert relative VSpace lengths to ERT "
648     i = 0
649     while True:
650       i = find_token(document.body, "\\begin_inset VSpace", i)
651       if i == -1:
652           break
653       # only revert if a custom length was set and if
654       # it used a percent length
655       r = re.compile(r'\\begin_inset VSpace (.*)$')
656       m = r.match(document.body[i])
657       length = m.group(1)
658       if length in ('defskip', 'smallskip', 'medskip', 'bigskip', 'vfill'):
659          i += 1
660          continue
661       # check if the space has a star (protected space)
662       protected = (document.body[i].rfind("*") != -1)
663       if protected:
664           length = length.rstrip('*')
665       # handle percent lengths
666       percent, length = latex_length(length)
667       # revert the VSpace inset to ERT
668       if percent:
669           if protected:
670               subst = put_cmd_in_ert("\\vspace*{" + length + "}")
671           else:
672               subst = put_cmd_in_ert("\\vspace{" + length + "}")
673           document.body[i:i + 2] = subst
674       i += 1
675
676
677 def revert_percent_hspace_lengths(document):
678     " Revert relative HSpace lengths to ERT "
679     i = 0
680     while True:
681       i = find_token_exact(document.body, "\\begin_inset space \\hspace", i)
682       if i == -1:
683           break
684       j = find_end_of_inset(document.body, i)
685       if j == -1:
686           document.warning("Can't find end of inset at line " + str(i))
687           i += 1
688           continue
689       # only revert if a custom length was set...
690       length = get_value(document.body, '\\length', i + 1, j)
691       if length == '':
692           document.warning("Malformed lyx document: Missing '\\length' in Space inset.")
693           i = j
694           continue
695       protected = ""
696       if document.body[i].find("\\hspace*{}") != -1:
697           protected = "*"
698       # ...and if it used a percent length
699       percent, length = latex_length(length)
700       # revert the HSpace inset to ERT
701       if percent:
702           subst = put_cmd_in_ert("\\hspace" + protected + "{" + length + "}")
703           document.body[i:j + 1] = subst
704       # if we did a substitution, this will still be ok
705       i = j
706
707
708 def revert_hspace_glue_lengths(document):
709     " Revert HSpace glue lengths to ERT "
710     i = 0
711     while True:
712       i = find_token_exact(document.body, "\\begin_inset space \\hspace", i)
713       if i == -1:
714           break
715       j = find_end_of_inset(document.body, i)
716       if j == -1:
717           document.warning("Can't find end of inset at line " + str(i))
718           i += 1
719           continue
720       length = get_value(document.body, '\\length', i + 1, j)
721       if length == '':
722           document.warning("Malformed lyx document: Missing '\\length' in Space inset.")
723           i = j
724           continue
725       protected = ""
726       if document.body[i].find("\\hspace*{}") != -1:
727           protected = "*"
728       # only revert if the length contains a plus or minus at pos != 0
729       if length.find('-',1) != -1 or length.find('+',1) != -1:
730           # handle percent lengths
731           length = latex_length(length)[1]
732           # revert the HSpace inset to ERT
733           subst = put_cmd_in_ert("\\hspace" + protected + "{" + length + "}")
734           document.body[i:j+1] = subst
735       i = j
736
737
738 def convert_author_id(document):
739     " Add the author_id to the \\author definition and make sure 0 is not used"
740     i = 0
741     anum = 1
742     re_author = re.compile(r'(\\author) (\".*\")\s*(.*)$')
743     
744     while True:
745         i = find_token(document.header, "\\author", i)
746         if i == -1:
747             break
748         m = re_author.match(document.header[i])
749         if m:
750             name = m.group(2)
751             email = m.group(3)
752             document.header[i] = "\\author %i %s %s" % (anum, name, email)
753         # FIXME Should this really be incremented if we didn't match?
754         anum += 1
755         i += 1
756         
757     i = 0
758     while True:
759         i = find_token(document.body, "\\change_", i)
760         if i == -1:
761             break
762         change = document.body[i].split(' ');
763         if len(change) == 3:
764             type = change[0]
765             author_id = int(change[1])
766             time = change[2]
767             document.body[i] = "%s %i %s" % (type, author_id + 1, time)
768         i += 1
769
770
771 def revert_author_id(document):
772     " Remove the author_id from the \\author definition "
773     i = 0
774     anum = 0
775     rx = re.compile(r'(\\author)\s+(\d+)\s+(\".*\")\s*(.*)$')
776     idmap = dict()
777
778     while True:
779         i = find_token(document.header, "\\author", i)
780         if i == -1:
781             break
782         m = rx.match(document.header[i])
783         if m:
784             author_id = int(m.group(2))
785             idmap[author_id] = anum
786             name = m.group(3)
787             email = m.group(4)
788             document.header[i] = "\\author %s %s" % (name, email)
789         i += 1
790         # FIXME Should this be incremented if we didn't match?
791         anum += 1
792
793     i = 0
794     while True:
795         i = find_token(document.body, "\\change_", i)
796         if i == -1:
797             break
798         change = document.body[i].split(' ');
799         if len(change) == 3:
800             type = change[0]
801             author_id = int(change[1])
802             time = change[2]
803             document.body[i] = "%s %i %s" % (type, idmap[author_id], time)
804         i += 1
805
806
807 def revert_suppress_date(document):
808     " Revert suppressing of default document date to preamble code "
809     i = find_token(document.header, "\\suppress_date", 0)
810     if i == -1:
811         return
812     # remove the preamble line and write to the preamble
813     # when suppress_date was true
814     date = str2bool(get_value(document.header, "\\suppress_date", i))
815     if date:
816         add_to_preamble(document, ["\\date{}"])
817     del document.header[i]
818
819
820 def revert_mhchem(document):
821     "Revert mhchem loading to preamble code"
822
823     mhchem = "off"
824     i = find_token(document.header, "\\use_mhchem", 0)
825     if i == -1:
826         document.warning("Malformed LyX document: Could not find mhchem setting.")
827         mhchem = "auto"
828     else:
829         val = get_value(document.header, "\\use_mhchem", i)
830         if val == "1":
831             mhchem = "auto"
832         elif val == "2":
833             mhchem = "on"
834         del document.header[i]
835
836     if mhchem == "off":
837       # don't load case
838       return 
839
840     if mhchem == "auto":
841         i = 0
842         while True:
843             i = find_token(document.body, "\\begin_inset Formula", i)
844             if i == -1:
845                break
846             line = document.body[i]
847             if line.find("\\ce{") != -1 or line.find("\\cf{") != -1:
848               mhchem = "on"
849               break
850             i += 1
851
852     if mhchem == "on":
853         pre = ["\\PassOptionsToPackage{version=3}{mhchem}", 
854           "\\usepackage{mhchem}"]
855         insert_to_preamble(document, pre) 
856
857
858 def revert_fontenc(document):
859     " Remove fontencoding param "
860     if not del_token(document.header, '\\fontencoding', 0):
861         document.warning("Malformed LyX document: Missing \\fontencoding.")
862
863
864 def merge_gbrief(document):
865     " Merge g-brief-en and g-brief-de to one class "
866
867     if document.textclass != "g-brief-de":
868         if document.textclass == "g-brief-en":
869             document.textclass = "g-brief"
870             document.set_textclass()
871         return
872
873     obsoletedby = { "Brieftext":       "Letter",
874                     "Unterschrift":    "Signature",
875                     "Strasse":         "Street",
876                     "Zusatz":          "Addition",
877                     "Ort":             "Town",
878                     "Land":            "State",
879                     "RetourAdresse":   "ReturnAddress",
880                     "MeinZeichen":     "MyRef",
881                     "IhrZeichen":      "YourRef",
882                     "IhrSchreiben":    "YourMail",
883                     "Telefon":         "Phone",
884                     "BLZ":             "BankCode",
885                     "Konto":           "BankAccount",
886                     "Postvermerk":     "PostalComment",
887                     "Adresse":         "Address",
888                     "Datum":           "Date",
889                     "Betreff":         "Reference",
890                     "Anrede":          "Opening",
891                     "Anlagen":         "Encl.",
892                     "Verteiler":       "cc",
893                     "Gruss":           "Closing"}
894     i = 0
895     while 1:
896         i = find_token(document.body, "\\begin_layout", i)
897         if i == -1:
898             break
899
900         layout = document.body[i][14:]
901         if layout in obsoletedby:
902             document.body[i] = "\\begin_layout " + obsoletedby[layout]
903
904         i += 1
905         
906     document.textclass = "g-brief"
907     document.set_textclass()
908
909
910 def revert_gbrief(document):
911     " Revert g-brief to g-brief-en "
912     if document.textclass == "g-brief":
913         document.textclass = "g-brief-en"
914         document.set_textclass()
915
916
917 def revert_html_options(document):
918     " Remove html options "
919     del_token(document.header, '\\html_use_mathml', 0)
920     del_token(document.header, '\\html_be_strict', 0)
921
922
923 def revert_includeonly(document):
924     i = 0
925     while True:
926         i = find_token(document.header, "\\begin_includeonly", i)
927         if i == -1:
928             return
929         j = find_end_of(document.header, i, "\\begin_includeonly", "\\end_includeonly")
930         if j == -1:
931             document.warning("Unable to find end of includeonly section!!")
932             break
933         document.header[i : j + 1] = []
934
935
936 def revert_includeall(document):
937     " Remove maintain_unincluded_children param "
938     del_token(document.header, '\\maintain_unincluded_children', 0)
939
940
941 def revert_multirow(document):
942     " Revert multirow cells in tables to TeX-code"
943     i = 0
944     multirow = False
945     while True:
946       # cell type 3 is multirow begin cell
947       i = find_token(document.body, '<cell multirow="3"', i)
948       if i == -1:
949           break
950       # a multirow cell was found
951       multirow = True
952       # remove the multirow tag, set the valignment to top
953       # and remove the bottom line
954       # FIXME Are we sure these always have space around them?
955       document.body[i] = document.body[i].replace(' multirow="3" ', ' ')
956       document.body[i] = document.body[i].replace('valignment="middle"', 'valignment="top"')
957       document.body[i] = document.body[i].replace(' bottomline="true" ', ' ')
958       # write ERT to create the multirow cell
959       # use 2 rows and 2cm as default with because the multirow span
960       # and the column width is only hardly accessible
961       cend = find_token(document.body, "</cell>", i)
962       if cend == -1:
963           document.warning("Malformed LyX document: Could not find end of tabular cell.")
964           i += 1
965           continue
966       blay = find_token(document.body, "\\begin_layout", i, cend)
967       if blay == -1:
968           document.warning("Can't find layout for cell!")
969           i = j
970           continue
971       bend = find_end_of_layout(document.body, blay)
972       if blay == -1:
973           document.warning("Can't find end of layout for cell!")
974           i = cend
975           continue
976
977       # do the later one first, so as not to mess up the numbering
978       # we are wrapping the whole cell in this ert
979       # so before the end of the layout...
980       document.body[bend:bend] = put_cmd_in_ert("}")
981       # ...and after the beginning
982       document.body[blay+1:blay+1] = put_cmd_in_ert("\\multirow{2}{2cm}{")
983
984       while True:
985           # cell type 4 is multirow part cell
986           k = find_token(document.body, '<cell multirow="4"', cend)
987           if k == -1:
988               break
989           # remove the multirow tag, set the valignment to top
990           # and remove the top line
991           # FIXME Are we sure these always have space around them?
992           document.body[k] = document.body[k].replace(' multirow="4" ', ' ')
993           document.body[k] = document.body[k].replace('valignment="middle"', 'valignment="top"')
994           document.body[k] = document.body[k].replace(' topline="true" ', ' ')
995           k += 1
996       # this will always be ok
997       i = cend
998
999     if multirow == True:
1000         add_to_preamble(document, ["\\usepackage{multirow}"])
1001
1002
1003 def convert_math_output(document):
1004     " Convert \html_use_mathml to \html_math_output "
1005     i = find_token(document.header, "\\html_use_mathml", 0)
1006     if i == -1:
1007         return
1008     rgx = re.compile(r'\\html_use_mathml\s+(\w+)')
1009     m = rgx.match(document.header[i])
1010     newval = "0" # MathML
1011     if m:
1012       val = str2bool(m.group(1))
1013       if not val:
1014         newval = "2" # Images
1015     else:
1016       document.warning("Can't match " + document.header[i])
1017     document.header[i] = "\\html_math_output " + newval
1018
1019
1020 def revert_math_output(document):
1021     " Revert \html_math_output to \html_use_mathml "
1022     i = find_token(document.header, "\\html_math_output", 0)
1023     if i == -1:
1024         return
1025     rgx = re.compile(r'\\html_math_output\s+(\d)')
1026     m = rgx.match(document.header[i])
1027     newval = "true"
1028     if m:
1029         val = m.group(1)
1030         if val == "1" or val == "2":
1031             newval = "false"
1032     else:
1033         document.warning("Unable to match " + document.header[i])
1034     document.header[i] = "\\html_use_mathml " + newval
1035                 
1036
1037
1038 def revert_inset_preview(document):
1039     " Dissolves the preview inset "
1040     i = 0
1041     while True:
1042       i = find_token(document.body, "\\begin_inset Preview", i)
1043       if i == -1:
1044           return
1045       iend = find_end_of_inset(document.body, i)
1046       if iend == -1:
1047           document.warning("Malformed LyX document: Could not find end of Preview inset.")
1048           i += 1
1049           continue
1050       
1051       # This has several issues.
1052       # We need to do something about the layouts inside InsetPreview.
1053       # If we just leave the first one, then we have something like:
1054       # \begin_layout Standard
1055       # ...
1056       # \begin_layout Standard
1057       # and we get a "no \end_layout" error. So something has to be done.
1058       # Ideally, we would check if it is the same as the layout we are in.
1059       # If so, we just remove it; if not, we end the active one. But it is 
1060       # not easy to know what layout we are in, due to depth changes, etc,
1061       # and it is not clear to me how much work it is worth doing. In most
1062       # cases, the layout will probably be the same.
1063       # 
1064       # For the same reason, we have to remove the \end_layout tag at the
1065       # end of the last layout in the inset. Again, that will sometimes be
1066       # wrong, but it will usually be right. To know what to do, we would
1067       # again have to know what layout the inset is in.
1068       
1069       blay = find_token(document.body, "\\begin_layout", i, iend)
1070       if blay == -1:
1071           document.warning("Can't find layout for preview inset!")
1072           # always do the later one first...
1073           del document.body[iend]
1074           del document.body[i]
1075           # deletions mean we do not need to reset i
1076           continue
1077
1078       # This is where we would check what layout we are in.
1079       # The check for Standard is definitely wrong.
1080       # 
1081       # lay = document.body[blay].split(None, 1)[1]
1082       # if lay != oldlayout:
1083       #     # record a boolean to tell us what to do later....
1084       #     # better to do it later, since (a) it won't mess up
1085       #     # the numbering and (b) we only modify at the end.
1086         
1087       # we want to delete the last \\end_layout in this inset, too.
1088       # note that this may not be the \\end_layout that goes with blay!!
1089       bend = find_end_of_layout(document.body, blay)
1090       while True:
1091           tmp = find_token(document.body, "\\end_layout", bend + 1, iend)
1092           if tmp == -1:
1093               break
1094           bend = tmp
1095       if bend == blay:
1096           document.warning("Unable to find last layout in preview inset!")
1097           del document.body[iend]
1098           del document.body[i]
1099           # deletions mean we do not need to reset i
1100           continue
1101       # always do the later one first...
1102       del document.body[iend]
1103       del document.body[bend]
1104       del document.body[i:blay + 1]
1105       # we do not need to reset i
1106                 
1107
1108 def revert_equalspacing_xymatrix(document):
1109     " Revert a Formula with xymatrix@! to an ERT inset "
1110     i = 0
1111     has_preamble = False
1112     has_equal_spacing = False
1113
1114     while True:
1115       i = find_token(document.body, "\\begin_inset Formula", i)
1116       if i == -1:
1117           break
1118       j = find_end_of_inset(document.body, i)
1119       if j == -1:
1120           document.warning("Malformed LyX document: Could not find end of Formula inset.")
1121           i += 1
1122           continue
1123       
1124       for curline in range(i,j):
1125           found = document.body[curline].find("\\xymatrix@!")
1126           if found != -1:
1127               break
1128  
1129       if found != -1:
1130           has_equal_spacing = True
1131           content = [document.body[i][21:]]
1132           content += document.body[i + 1:j]
1133           subst = put_cmd_in_ert(content)
1134           document.body[i:j + 1] = subst
1135           i += len(subst) - (j - i) + 1
1136       else:
1137           for curline in range(i,j):
1138               l = document.body[curline].find("\\xymatrix")
1139               if l != -1:
1140                   has_preamble = True;
1141                   break;
1142           i = j + 1
1143   
1144     if has_equal_spacing and not has_preamble:
1145         add_to_preamble(document, ['\\usepackage[all]{xy}'])
1146
1147
1148 def revert_notefontcolor(document):
1149     " Reverts greyed-out note font color to preamble code "
1150
1151     i = find_token(document.header, "\\notefontcolor", 0)
1152     if i == -1:
1153         return
1154
1155     colorcode = get_value(document.header, '\\notefontcolor', i)
1156     del document.header[i]
1157
1158     # are there any grey notes?
1159     if find_token(document.body, "\\begin_inset Note Greyedout", 0) == -1:
1160         # no need to do anything else, and \renewcommand will throw 
1161         # an error since lyxgreyedout will not exist.
1162         return
1163
1164     # the color code is in the form #rrggbb where every character denotes a hex number
1165     red = hex2ratio(colorcode[1:3])
1166     green = hex2ratio(colorcode[3:5])
1167     blue = hex2ratio(colorcode[5:7])
1168     # write the preamble
1169     insert_to_preamble(document,
1170       [ '%  for greyed-out notes',
1171         '\\@ifundefined{definecolor}{\\usepackage{color}}{}'
1172         '\\definecolor{note_fontcolor}{rgb}{%s,%s,%s}' % (red, green, blue),
1173         '\\renewenvironment{lyxgreyedout}',
1174         ' {\\textcolor{note_fontcolor}\\bgroup}{\\egroup}'])
1175
1176
1177 def revert_turkmen(document):
1178     "Set language Turkmen to English" 
1179
1180     if document.language == "turkmen": 
1181         document.language = "english" 
1182         i = find_token(document.header, "\\language", 0) 
1183         if i != -1: 
1184             document.header[i] = "\\language english" 
1185
1186     j = 0 
1187     while True: 
1188         j = find_token(document.body, "\\lang turkmen", j) 
1189         if j == -1: 
1190             return 
1191         document.body[j] = document.body[j].replace("\\lang turkmen", "\\lang english") 
1192         j += 1 
1193
1194
1195 def revert_fontcolor(document):
1196     " Reverts font color to preamble code "
1197     i = find_token(document.header, "\\fontcolor", 0)
1198     if i == -1:
1199         return
1200     colorcode = get_value(document.header, '\\fontcolor', i)
1201     del document.header[i]
1202     # don't clutter the preamble if font color is not set
1203     if colorcode == "#000000":
1204         return
1205     # the color code is in the form #rrggbb where every character denotes a hex number
1206     red = hex2ratio(colorcode[1:3])
1207     green = hex2ratio(colorcode[3:5])
1208     blue = hex2ratio(colorcode[5:7])
1209     # write the preamble
1210     insert_to_preamble(document,
1211       ['%  Set the font color',
1212       '\\@ifundefined{definecolor}{\\usepackage{color}}{}',
1213       '\\definecolor{document_fontcolor}{rgb}{%s,%s,%s}' % (red, green, blue),
1214       '\\color{document_fontcolor}'])
1215
1216
1217 def revert_shadedboxcolor(document):
1218     " Reverts shaded box color to preamble code "
1219     i = find_token(document.header, "\\boxbgcolor", 0)
1220     if i == -1:
1221         return
1222     colorcode = get_value(document.header, '\\boxbgcolor', i)
1223     del document.header[i]
1224     # the color code is in the form #rrggbb
1225     red = hex2ratio(colorcode[1:3])
1226     green = hex2ratio(colorcode[3:5])
1227     blue = hex2ratio(colorcode[5:7])
1228     # write the preamble
1229     insert_to_preamble(document,
1230       ['%  Set the color of boxes with shaded background',
1231       '\\@ifundefined{definecolor}{\\usepackage{color}}{}',
1232       "\\definecolor{shadecolor}{rgb}{%s,%s,%s}" % (red, green, blue)])
1233
1234
1235 def revert_lyx_version(document):
1236     " Reverts LyX Version information from Inset Info "
1237     version = "LyX version"
1238     try:
1239         import lyx2lyx_version
1240         version = lyx2lyx_version.version
1241     except:
1242         pass
1243
1244     i = 0
1245     while 1:
1246         i = find_token(document.body, '\\begin_inset Info', i)
1247         if i == -1:
1248             return
1249         j = find_end_of_inset(document.body, i + 1)
1250         if j == -1:
1251             document.warning("Malformed LyX document: Could not find end of Info inset.")
1252             i += 1
1253             continue
1254
1255         # We expect:
1256         # \begin_inset Info
1257         # type  "lyxinfo"
1258         # arg   "version"
1259         # \end_inset
1260         typ = get_quoted_value(document.body, "type", i, j)
1261         arg = get_quoted_value(document.body, "arg", i, j)
1262         if arg != "version" or typ != "lyxinfo":
1263             i = j + 1
1264             continue
1265
1266         # We do not actually know the version of LyX used to produce the document.
1267         # But we can use our version, since we are reverting.
1268         s = [version]
1269         # Now we want to check if the line after "\end_inset" is empty. It normally
1270         # is, so we want to remove it, too.
1271         lastline = j + 1
1272         if document.body[j + 1].strip() == "":
1273             lastline = j + 2
1274         document.body[i: lastline] = s
1275         i = i + 1
1276
1277
1278 def revert_math_scale(document):
1279   " Remove math scaling and LaTeX options "
1280   del_token(document.header, '\\html_math_img_scale', 0)
1281   del_token(document.header, '\\html_latex_start', 0)
1282   del_token(document.header, '\\html_latex_end', 0)
1283
1284
1285 def revert_pagesizes(document):
1286   " Revert page sizes to default "
1287   i = find_token(document.header, '\\papersize', 0)
1288   if i != -1:
1289     size = document.header[i][11:]
1290     if size == "a0paper" or size == "a1paper" or size == "a2paper" \
1291     or size == "a6paper" or size == "b0paper" or size == "b1paper" \
1292     or size == "b2paper" or size == "b6paper" or size == "b0j" \
1293     or size == "b1j" or size == "b2j" or size == "b3j" or size == "b4j" \
1294     or size == "b5j" or size == "b6j":
1295       del document.header[i]
1296
1297
1298 def revert_DIN_C_pagesizes(document):
1299   " Revert DIN C page sizes to default "
1300   i = find_token(document.header, '\\papersize', 0)
1301   if i != -1:
1302     size = document.header[i][11:]
1303     if size == "c0paper" or size == "c1paper" or size == "c2paper" \
1304     or size == "c3paper" or size == "c4paper" or size == "c5paper" \
1305     or size == "c6paper":
1306       del document.header[i]
1307
1308
1309 def convert_html_quotes(document):
1310   " Remove quotes around html_latex_start and html_latex_end "
1311
1312   i = find_token(document.header, '\\html_latex_start', 0)
1313   if i != -1:
1314     line = document.header[i]
1315     l = re.compile(r'\\html_latex_start\s+"(.*)"')
1316     m = l.match(line)
1317     if m:
1318       document.header[i] = "\\html_latex_start " + m.group(1)
1319       
1320   i = find_token(document.header, '\\html_latex_end', 0)
1321   if i != -1:
1322     line = document.header[i]
1323     l = re.compile(r'\\html_latex_end\s+"(.*)"')
1324     m = l.match(line)
1325     if m:
1326       document.header[i] = "\\html_latex_end " + m.group(1)
1327       
1328
1329 def revert_html_quotes(document):
1330   " Remove quotes around html_latex_start and html_latex_end "
1331   
1332   i = find_token(document.header, '\\html_latex_start', 0)
1333   if i != -1:
1334     line = document.header[i]
1335     l = re.compile(r'\\html_latex_start\s+(.*)')
1336     m = l.match(line)
1337     if not m:
1338         document.warning("Weird html_latex_start line: " + line)
1339         del document.header[i]
1340     else:
1341         document.header[i] = "\\html_latex_start \"" + m.group(1) + "\""
1342       
1343   i = find_token(document.header, '\\html_latex_end', 0)
1344   if i != -1:
1345     line = document.header[i]
1346     l = re.compile(r'\\html_latex_end\s+(.*)')
1347     m = l.match(line)
1348     if not m:
1349         document.warning("Weird html_latex_end line: " + line)
1350         del document.header[i]
1351     else:
1352         document.header[i] = "\\html_latex_end \"" + m.group(1) + "\""
1353
1354
1355 def revert_output_sync(document):
1356   " Remove forward search options "
1357   del_token(document.header, '\\output_sync_macro', 0)
1358   del_token(document.header, '\\output_sync', 0)
1359
1360
1361 def revert_align_decimal(document):
1362   i = 0
1363   while True:
1364     i = find_token(document.body, "\\begin_inset Tabular", i)
1365     if i == -1:
1366       return
1367     j = find_end_of_inset(document.body, i)
1368     if j == -1:
1369       document.warning("Unable to find end of Tabular inset at line " + str(i))
1370       i += 1
1371       continue
1372     cell = find_token(document.body, "<cell", i, j)
1373     if cell == -1:
1374       document.warning("Can't find any cells in Tabular inset at line " + str(i))
1375       i = j
1376       continue
1377     k = i + 1
1378     while True:
1379       k = find_token(document.body, "<column", k, cell)
1380       if k == -1:
1381         return
1382       if document.body[k].find('alignment="decimal"') == -1:
1383         k += 1
1384         continue
1385       remove_option(document.body, k, 'decimal_point')
1386       document.body[k] = \
1387         document.body[k].replace('alignment="decimal"', 'alignment="center"')
1388       k += 1
1389
1390
1391 def convert_optarg(document):
1392   " Convert \\begin_inset OptArg to \\begin_inset Argument "
1393   i = 0
1394   while 1:
1395     i = find_token(document.body, '\\begin_inset OptArg', i)
1396     if i == -1:
1397       return
1398     document.body[i] = "\\begin_inset Argument"
1399     i += 1
1400
1401
1402 def revert_argument(document):
1403   " Convert \\begin_inset Argument to \\begin_inset OptArg "
1404   i = 0
1405   while 1:
1406     i = find_token(document.body, '\\begin_inset Argument', i)
1407     if i == -1:
1408       return
1409     document.body[i] = "\\begin_inset OptArg"
1410     i += 1
1411
1412
1413 def revert_makebox(document):
1414   " Convert \\makebox to TeX code "
1415   i = 0
1416   while 1:
1417     i = find_token(document.body, '\\begin_inset Box', i)
1418     if i == -1:
1419       break
1420     z = find_end_of_inset(document.body, i)
1421     if z == -1:
1422       document.warning("Malformed LyX document: Can't find end of box inset.")
1423       i += 1
1424       continue
1425     blay = find_token(document.body, "\\begin_layout", i, z)
1426     if blay == -1:
1427       document.warning("Malformed LyX document: Can't find layout in box.")
1428       i = z
1429       continue
1430     # by looking before the layout we make sure we're actually finding
1431     # an option, not text.
1432     j = find_token(document.body, 'use_makebox', i, blay)
1433     if j == -1:
1434         i = z
1435         continue
1436     
1437     if not check_token(document.body[i], "\\begin_inset Box Frameless") \
1438       or get_value(document.body, 'use_makebox', j) != 1:
1439         del document.body[j]
1440         i = z
1441         continue
1442     bend = find_end_of_layout(document.body, blay)
1443     if bend == -1 or bend > z:
1444         document.warning("Malformed LyX document: Can't find end of layout in box.")
1445         i = z
1446         continue
1447     # determine the alignment
1448     align = get_quoted_value(document.body, 'hor_pos', i, blay, "c")
1449     # determine the width
1450     length = get_quoted_value(document.body, 'width', i, blay, "50col%")
1451     length = latex_length(length)[1]
1452     # remove the \end_layout \end_inset pair
1453     document.body[bend:z + 1] = put_cmd_in_ert("}")
1454     subst = "\\makebox[" + length + "][" \
1455       + align + "]{"
1456     document.body[i:blay + 1] = put_cmd_in_ert(subst)
1457     i += 1
1458
1459
1460 def convert_use_makebox(document):
1461   " Adds use_makebox option for boxes "
1462   i = 0
1463   while 1:
1464     i = find_token(document.body, '\\begin_inset Box', i)
1465     if i == -1:
1466       return
1467     # all of this is to make sure we actually find the use_parbox
1468     # that is an option for this box, not some text elsewhere.
1469     z = find_end_of_inset(document.body, i)
1470     if z == -1:
1471       document.warning("Can't find end of box inset!!")
1472       i += 1
1473       continue
1474     blay = find_token(document.body, "\\begin_layout", i, z)
1475     if blay == -1:
1476       document.warning("Can't find layout in box inset!!")
1477       i = z
1478       continue
1479     # so now we are looking for use_parbox before the box's layout
1480     k = find_token(document.body, 'use_parbox', i, blay)
1481     if k == -1:
1482       document.warning("Malformed LyX document: Can't find use_parbox statement in box.")
1483       i = z
1484       continue
1485     document.body.insert(k + 1, "use_makebox 0")
1486     i = z + 1
1487
1488
1489 def revert_IEEEtran(document):
1490   " Convert IEEEtran layouts and styles to TeX code "
1491   if document.textclass != "IEEEtran":
1492     return
1493   revert_flex_inset(document.body, "IEEE membership", "\\IEEEmembership")
1494   revert_flex_inset(document.body, "Lowercase", "\\MakeLowercase")
1495   layouts = ("Special Paper Notice", "After Title Text", "Publication ID",
1496              "Page headings", "Biography without photo")
1497   latexcmd = {"Special Paper Notice": "\\IEEEspecialpapernotice",
1498               "After Title Text":     "\\IEEEaftertitletext",
1499               "Publication ID":       "\\IEEEpubid"}
1500   obsoletedby = {"Page headings":            "MarkBoth",
1501                  "Biography without photo":  "BiographyNoPhoto"}
1502   for layout in layouts:
1503     i = 0
1504     while True:
1505         i = find_token(document.body, '\\begin_layout ' + layout, i)
1506         if i == -1:
1507           break
1508         j = find_end_of_layout(document.body, i)
1509         if j == -1:
1510           document.warning("Malformed LyX document: Can't find end of " + layout + " layout.")
1511           i += 1
1512           continue
1513         if layout in obsoletedby:
1514           document.body[i] = "\\begin_layout " + obsoletedby[layout]
1515           i = j
1516           continue
1517         content = lyx2latex(document, document.body[i:j + 1])
1518         add_to_preamble(document, [latexcmd[layout] + "{" + content + "}"])
1519         del document.body[i:j + 1]
1520         # no need to reset i
1521
1522
1523 def convert_prettyref(document):
1524         " Converts prettyref references to neutral formatted refs "
1525         re_ref = re.compile("^\s*reference\s+\"(\w+):(\S+)\"")
1526         nm_ref = re.compile("^\s*name\s+\"(\w+):(\S+)\"")
1527
1528         i = 0
1529         while True:
1530                 i = find_token(document.body, "\\begin_inset CommandInset ref", i)
1531                 if i == -1:
1532                         break
1533                 j = find_end_of_inset(document.body, i)
1534                 if j == -1:
1535                         document.warning("Malformed LyX document: No end of InsetRef!")
1536                         i += 1
1537                         continue
1538                 k = find_token(document.body, "LatexCommand prettyref", i, j)
1539                 if k != -1:
1540                         document.body[k] = "LatexCommand formatted"
1541                 i = j + 1
1542         document.header.insert(-1, "\\use_refstyle 0")
1543                 
1544  
1545 def revert_refstyle(document):
1546         " Reverts neutral formatted refs to prettyref "
1547         re_ref = re.compile("^reference\s+\"(\w+):(\S+)\"")
1548         nm_ref = re.compile("^\s*name\s+\"(\w+):(\S+)\"")
1549
1550         i = 0
1551         while True:
1552                 i = find_token(document.body, "\\begin_inset CommandInset ref", i)
1553                 if i == -1:
1554                         break
1555                 j = find_end_of_inset(document.body, i)
1556                 if j == -1:
1557                         document.warning("Malformed LyX document: No end of InsetRef")
1558                         i += 1
1559                         continue
1560                 k = find_token(document.body, "LatexCommand formatted", i, j)
1561                 if k != -1:
1562                         document.body[k] = "LatexCommand prettyref"
1563                 i = j + 1
1564         i = find_token(document.header, "\\use_refstyle", 0)
1565         if i != -1:
1566                 document.header.pop(i)
1567  
1568
1569 def revert_nameref(document):
1570   " Convert namerefs to regular references "
1571   cmds = ["Nameref", "nameref"]
1572   foundone = False
1573   rx = re.compile(r'reference "(.*)"')
1574   for cmd in cmds:
1575     i = 0
1576     oldcmd = "LatexCommand " + cmd
1577     while 1:
1578       # It seems better to look for this, as most of the reference
1579       # insets won't be ones we care about.
1580       i = find_token(document.body, oldcmd, i)
1581       if i == -1:
1582         break
1583       cmdloc = i
1584       i += 1
1585       # Make sure it is actually in an inset!
1586       # A normal line could begin with "LatexCommand nameref"!
1587       val = is_in_inset(document.body, cmdloc, \
1588           "\\begin_inset CommandInset ref")
1589       if not val:
1590           continue
1591       stins, endins = val
1592
1593       # ok, so it is in an InsetRef
1594       refline = find_token(document.body, "reference", stins, endins)
1595       if refline == -1:
1596         document.warning("Can't find reference for inset at line " + stinst + "!!")
1597         continue
1598       m = rx.match(document.body[refline])
1599       if not m:
1600         document.warning("Can't match reference line: " + document.body[ref])
1601         continue
1602       foundone = True
1603       ref = m.group(1)
1604       newcontent = put_cmd_in_ert('\\' + cmd + '{' + ref + '}')
1605       document.body[stins:endins + 1] = newcontent
1606
1607   if foundone:
1608     add_to_preamble(document, ["\usepackage{nameref}"])
1609
1610
1611 def remove_Nameref(document):
1612   " Convert Nameref commands to nameref commands "
1613   i = 0
1614   while 1:
1615     # It seems better to look for this, as most of the reference
1616     # insets won't be ones we care about.
1617     i = find_token(document.body, "LatexCommand Nameref" , i)
1618     if i == -1:
1619       break
1620     cmdloc = i
1621     i += 1
1622     
1623     # Make sure it is actually in an inset!
1624     val = is_in_inset(document.body, cmdloc, \
1625         "\\begin_inset CommandInset ref")
1626     if not val:
1627       continue
1628     document.body[cmdloc] = "LatexCommand nameref"
1629
1630
1631 def revert_mathrsfs(document):
1632     " Load mathrsfs if \mathrsfs us use in the document "
1633     i = 0
1634     for line in document.body:
1635       if line.find("\\mathscr{") != -1:
1636         add_to_preamble(document, ["\\usepackage{mathrsfs}"])
1637         return
1638
1639
1640 def convert_flexnames(document):
1641     "Convert \\begin_inset Flex Custom:Style to \\begin_inset Flex Style and similarly for CharStyle and Element."
1642     
1643     i = 0
1644     rx = re.compile(r'^\\begin_inset Flex (?:Custom|CharStyle|Element):(.+)$')
1645     while True:
1646       i = find_token(document.body, "\\begin_inset Flex", i)
1647       if i == -1:
1648         return
1649       m = rx.match(document.body[i])
1650       if m:
1651         document.body[i] = "\\begin_inset Flex " + m.group(1)
1652       i += 1
1653
1654
1655 flex_insets = {
1656   "Alert" : "CharStyle:Alert",
1657   "Code" : "CharStyle:Code",
1658   "Concepts" : "CharStyle:Concepts",
1659   "E-Mail" : "CharStyle:E-Mail",
1660   "Emph" : "CharStyle:Emph",
1661   "Expression" : "CharStyle:Expression",
1662   "Initial" : "CharStyle:Initial",
1663   "Institute" : "CharStyle:Institute",
1664   "Meaning" : "CharStyle:Meaning",
1665   "Noun" : "CharStyle:Noun",
1666   "Strong" : "CharStyle:Strong",
1667   "Structure" : "CharStyle:Structure",
1668   "ArticleMode" : "Custom:ArticleMode",
1669   "Endnote" : "Custom:Endnote",
1670   "Glosse" : "Custom:Glosse",
1671   "PresentationMode" : "Custom:PresentationMode",
1672   "Tri-Glosse" : "Custom:Tri-Glosse"
1673 }
1674
1675 flex_elements = {
1676   "Abbrev" : "Element:Abbrev",
1677   "CCC-Code" : "Element:CCC-Code",
1678   "Citation-number" : "Element:Citation-number",
1679   "City" : "Element:City",
1680   "Code" : "Element:Code",
1681   "CODEN" : "Element:CODEN",
1682   "Country" : "Element:Country",
1683   "Day" : "Element:Day",
1684   "Directory" : "Element:Directory",
1685   "Dscr" : "Element:Dscr",
1686   "Email" : "Element:Email",
1687   "Emph" : "Element:Emph",
1688   "Filename" : "Element:Filename",
1689   "Firstname" : "Element:Firstname",
1690   "Fname" : "Element:Fname",
1691   "GuiButton" : "Element:GuiButton",
1692   "GuiMenu" : "Element:GuiMenu",
1693   "GuiMenuItem" : "Element:GuiMenuItem",
1694   "ISSN" : "Element:ISSN",
1695   "Issue-day" : "Element:Issue-day",
1696   "Issue-months" : "Element:Issue-months",
1697   "Issue-number" : "Element:Issue-number",
1698   "KeyCap" : "Element:KeyCap",
1699   "KeyCombo" : "Element:KeyCombo",
1700   "Keyword" : "Element:Keyword",
1701   "Literal" : "Element:Literal",
1702   "MenuChoice" : "Element:MenuChoice",
1703   "Month" : "Element:Month",
1704   "Orgdiv" : "Element:Orgdiv",
1705   "Orgname" : "Element:Orgname",
1706   "Postcode" : "Element:Postcode",
1707   "SS-Code" : "Element:SS-Code",
1708   "SS-Title" : "Element:SS-Title",
1709   "State" : "Element:State",
1710   "Street" : "Element:Street",
1711   "Surname" : "Element:Surname",
1712   "Volume" : "Element:Volume",
1713   "Year" : "Element:Year"
1714 }
1715
1716
1717 def revert_flexnames(document):
1718   if document.backend == "latex":
1719     flexlist = flex_insets
1720   else:
1721     flexlist = flex_elements
1722   
1723   rx = re.compile(r'^\\begin_inset Flex\s+(.+)$')
1724   i = 0
1725   while True:
1726     i = find_token(document.body, "\\begin_inset Flex", i)
1727     if i == -1:
1728       return
1729     m = rx.match(document.body[i])
1730     if not m:
1731       document.warning("Illegal flex inset: " + document.body[i])
1732       i += 1
1733       continue
1734     style = m.group(1)
1735     if style in flexlist:
1736       document.body[i] = "\\begin_inset Flex " + flexlist[style]
1737     i += 1
1738
1739
1740 def convert_mathdots(document):
1741     " Load mathdots automatically "
1742     i = find_token(document.header, "\\use_esint" , 0)
1743     if i != -1:
1744       document.header.insert(i + 1, "\\use_mathdots 1")
1745
1746
1747 def revert_mathdots(document):
1748     " Load mathdots if used in the document "
1749
1750     mathdots = find_token(document.header, "\\use_mathdots" , 0)
1751     if mathdots == -1:
1752       document.warning("No \\usemathdots line. Assuming auto.")
1753     else:
1754       val = get_value(document.header, "\\use_mathdots", mathdots)
1755       del document.header[mathdots]
1756       try:
1757         usedots = int(val)
1758       except:
1759         document.warning("Invalid \\use_mathdots value: " + val + ". Assuming auto.")
1760         # probably usedots has not been changed, but be safe.
1761         usedots = 1
1762
1763       if usedots == 0:
1764         # do not load case
1765         return
1766       if usedots == 2:
1767         # force load case
1768         add_to_preamble(["\\usepackage{mathdots}"])
1769         return
1770     
1771     # so we are in the auto case. we want to load mathdots if \iddots is used.
1772     i = 0
1773     while True:
1774       i = find_token(document.body, '\\begin_inset Formula', i)
1775       if i == -1:
1776         return
1777       j = find_end_of_inset(document.body, i)
1778       if j == -1:
1779         document.warning("Malformed LyX document: Can't find end of Formula inset at line " + str(i))
1780         i += 1
1781         continue
1782       code = "\n".join(document.body[i:j])
1783       if code.find("\\iddots") != -1:
1784         add_to_preamble(document, ["\\@ifundefined{iddots}{\\usepackage{mathdots}}"])
1785         return
1786       i = j
1787
1788
1789 def convert_rule(document):
1790     " Convert \\lyxline to CommandInset line. "
1791     i = 0
1792     
1793     inset = ['\\begin_inset CommandInset line',
1794       'LatexCommand rule',
1795       'offset "0.5ex"',
1796       'width "100line%"',
1797       'height "1pt"', '',
1798       '\\end_inset', '', '']
1799
1800     # if paragraphs are indented, we may have to unindent to get the
1801     # line to be full-width.
1802     indent = get_value(document.header, "\\paragraph_separation", 0)
1803     have_indent = (indent == "indent")
1804
1805     while True:
1806       i = find_token(document.body, "\\lyxline" , i)
1807       if i == -1:
1808         return
1809
1810       # we need to find out if this line follows other content
1811       # in its paragraph. find its layout....
1812       lastlay = find_token_backwards(document.body, "\\begin_layout", i)
1813       if lastlay == -1:
1814         document.warning("Can't find layout for line at " + str(i))
1815         # do the best we can.
1816         document.body[i:i+1] = inset
1817         i += len(inset)
1818         continue
1819
1820       # ...and look for other content before it.
1821       lineisfirst = True
1822       for line in document.body[lastlay + 1:i]:
1823         # is it empty or a paragraph option?
1824         if not line or line[0] == '\\':
1825           continue
1826         lineisfirst = False
1827         break
1828
1829       if lineisfirst:
1830         document.body[i:i+1] = inset
1831         if indent:
1832           # we need to unindent, lest the line be too long
1833           document.body.insert(lastlay + 1, "\\noindent")
1834         i += len(inset)
1835       else:
1836         # so our line is in the middle of a paragraph
1837         # we need to add a new line, lest this line follow the
1838         # other content on that line and run off the side of the page
1839         document.body[i:i+1] = inset
1840         document.body[i:i] = ["\\begin_inset Newline newline", "\\end_inset", ""]
1841       i += len(inset)
1842
1843
1844 def revert_rule(document):
1845     " Revert line insets to Tex code "
1846     i = 0
1847     while 1:
1848       i = find_token(document.body, "\\begin_inset CommandInset line" , i)
1849       if i == -1:
1850         return
1851       # find end of inset
1852       j = find_token(document.body, "\\end_inset" , i)
1853       if j == -1:
1854         document.warning("Malformed LyX document: Can't find end of line inset.")
1855         return
1856       # determine the optional offset
1857       offset = get_quoted_value(document.body, 'offset', i, j)
1858       if offset:
1859         offset = '[' + offset + ']'
1860       # determine the width
1861       width = get_quoted_value(document.body, 'width', i, j, "100col%")
1862       width = latex_length(width)[1]
1863       # determine the height
1864       height = get_quoted_value(document.body, 'height', i, j, "1pt")
1865       height = latex_length(height)[1]
1866       # output the \rule command
1867       subst = "\\rule[" + offset + "]{" + width + "}{" + height + "}"
1868       document.body[i:j + 1] = put_cmd_in_ert(subst)
1869       i += len(subst) - (j - i)
1870
1871
1872 def revert_diagram(document):
1873   " Add the feyn package if \\Diagram is used in math "
1874   i = 0
1875   while True:
1876     i = find_token(document.body, '\\begin_inset Formula', i)
1877     if i == -1:
1878       return
1879     j = find_end_of_inset(document.body, i)
1880     if j == -1:
1881         document.warning("Malformed LyX document: Can't find end of Formula inset.")
1882         return 
1883     lines = "\n".join(document.body[i:j])
1884     if lines.find("\\Diagram") == -1:
1885       i = j
1886       continue
1887     add_to_preamble(document, ["\\usepackage{feyn}"])
1888     # only need to do it once!
1889     return
1890
1891
1892 def convert_bibtex_clearpage(document):
1893   " insert a clear(double)page bibliographystyle if bibtotoc option is used "
1894
1895   i = find_token(document.header, '\\papersides', 0)
1896   sides = 0
1897   if i == -1:
1898     document.warning("Malformed LyX document: Can't find papersides definition.")
1899     document.warning("Assuming single sided.")
1900     sides = 1
1901   else:
1902     val = get_value(document.header, "\\papersides", i)
1903     try:
1904       sides = int(val)
1905     except:
1906       pass
1907     if sides != 1 and sides != 2:
1908       document.warning("Invalid papersides value: " + val)
1909       document.warning("Assuming single sided.")
1910       sides = 1
1911
1912   j = 0
1913   while True:
1914     j = find_token(document.body, "\\begin_inset CommandInset bibtex", j)
1915     if j == -1:
1916       return
1917
1918     k = find_end_of_inset(document.body, j)
1919     if k == -1:
1920       document.warning("Can't find end of Bibliography inset at line " + str(j))
1921       j += 1
1922       continue
1923
1924     # only act if there is the option "bibtotoc"
1925     val = get_value(document.body, 'options', j, k)
1926     if not val:
1927       document.warning("Can't find options for bibliography inset at line " + str(j))
1928       j = k
1929       continue
1930     
1931     if val.find("bibtotoc") == -1:
1932       j = k
1933       continue
1934     
1935     # so we want to insert a new page right before the paragraph that
1936     # this bibliography thing is in. 
1937     lay = find_token_backwards(document.body, "\\begin_layout", j)
1938     if lay == -1:
1939       document.warning("Can't find layout containing bibliography inset at line " + str(j))
1940       j = k
1941       continue
1942
1943     if sides == 1:
1944       cmd = "clearpage"
1945     else:
1946       cmd = "cleardoublepage"
1947     subst = ['\\begin_layout Standard',
1948         '\\begin_inset Newpage ' + cmd,
1949         '\\end_inset', '', '',
1950         '\\end_layout', '']
1951     document.body[lay:lay] = subst
1952     j = k + len(subst)
1953
1954
1955 ##
1956 # Conversion hub
1957 #
1958
1959 supported_versions = ["2.0.0","2.0"]
1960 convert = [[346, []],
1961            [347, []],
1962            [348, []],
1963            [349, []],
1964            [350, []],
1965            [351, []],
1966            [352, [convert_splitindex]],
1967            [353, []],
1968            [354, []],
1969            [355, []],
1970            [356, []],
1971            [357, []],
1972            [358, []],
1973            [359, [convert_nomencl_width]],
1974            [360, []],
1975            [361, []],
1976            [362, []],
1977            [363, []],
1978            [364, []],
1979            [365, []],
1980            [366, []],
1981            [367, []],
1982            [368, []],
1983            [369, [convert_author_id]],
1984            [370, []],
1985            [371, []],
1986            [372, []],
1987            [373, [merge_gbrief]],
1988            [374, []],
1989            [375, []],
1990            [376, []],
1991            [377, []],
1992            [378, []],
1993            [379, [convert_math_output]],
1994            [380, []],
1995            [381, []],
1996            [382, []],
1997            [383, []],
1998            [384, []],
1999            [385, []],
2000            [386, []],
2001            [387, []],
2002            [388, []],
2003            [389, [convert_html_quotes]],
2004            [390, []],
2005            [391, []],
2006            [392, []],
2007            [393, [convert_optarg]],
2008            [394, [convert_use_makebox]],
2009            [395, []],
2010            [396, []],
2011            [397, [remove_Nameref]],
2012            [398, []],
2013            [399, [convert_mathdots]],
2014            [400, [convert_rule]],
2015            [401, []],
2016            [402, [convert_bibtex_clearpage]],
2017            [403, [convert_flexnames]],
2018            [404, [convert_prettyref]]
2019 ]
2020
2021 revert =  [[403, [revert_refstyle]],
2022            [402, [revert_flexnames]],
2023            [401, []],
2024            [400, [revert_diagram]],
2025            [399, [revert_rule]],
2026            [398, [revert_mathdots]],
2027            [397, [revert_mathrsfs]],
2028            [396, []],
2029            [395, [revert_nameref]],
2030            [394, [revert_DIN_C_pagesizes]],
2031            [393, [revert_makebox]],
2032            [392, [revert_argument]],
2033            [391, []],
2034            [390, [revert_align_decimal, revert_IEEEtran]],
2035            [389, [revert_output_sync]],
2036            [388, [revert_html_quotes]],
2037            [387, [revert_pagesizes]],
2038            [386, [revert_math_scale]],
2039            [385, [revert_lyx_version]],
2040            [384, [revert_shadedboxcolor]],
2041            [383, [revert_fontcolor]],
2042            [382, [revert_turkmen]],
2043            [381, [revert_notefontcolor]],
2044            [380, [revert_equalspacing_xymatrix]],
2045            [379, [revert_inset_preview]],
2046            [378, [revert_math_output]],
2047            [377, []],
2048            [376, [revert_multirow]],
2049            [375, [revert_includeall]],
2050            [374, [revert_includeonly]],
2051            [373, [revert_html_options]],
2052            [372, [revert_gbrief]],
2053            [371, [revert_fontenc]],
2054            [370, [revert_mhchem]],
2055            [369, [revert_suppress_date]],
2056            [368, [revert_author_id]],
2057            [367, [revert_hspace_glue_lengths]],
2058            [366, [revert_percent_vspace_lengths, revert_percent_hspace_lengths]],
2059            [365, [revert_percent_skip_lengths]],
2060            [364, [revert_paragraph_indentation]],
2061            [363, [revert_branch_filename]],
2062            [362, [revert_longtable_align]],
2063            [361, [revert_applemac]],
2064            [360, []],
2065            [359, [revert_nomencl_cwidth]],
2066            [358, [revert_nomencl_width]],
2067            [357, [revert_custom_processors]],
2068            [356, [revert_ulinelatex]],
2069            [355, []],
2070            [354, [revert_strikeout]],
2071            [353, [revert_printindexall]],
2072            [352, [revert_subindex]],
2073            [351, [revert_splitindex]],
2074            [350, [revert_backgroundcolor]],
2075            [349, [revert_outputformat]],
2076            [348, [revert_xetex]],
2077            [347, [revert_phantom, revert_hphantom, revert_vphantom]],
2078            [346, [revert_tabularvalign]],
2079            [345, [revert_swiss]]
2080           ]
2081
2082
2083 if __name__ == "__main__":
2084     pass