lib/lyx2lyx/parser_tools.py

   1 # This file is part of lyx2lyx
   2 # Copyright (C) 2002 Dekel Tsur <dekel@lyx.org>, José Matos <jamatos@lyx.org>
   3 #
   4 # This program is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU General Public License
   6 # as published by the Free Software Foundation; either version 2
   7 # of the License, or (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program; if not, write to the Free Software
  16 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  17
  18 import string
  19
  20 def check_token(line, token):
  21     if line[:len(token)] == token:
  22         return 1
  23     return 0
  24
  25 # We need to check that the char after the token is space, but I think
  26 # we can ignore this
  27 def find_token(lines, token, start, end = 0):
  28     if end == 0:
  29         end = len(lines)
  30     m = len(token)
  31     for i in xrange(start, end):
  32         if lines[i][:m] == token:
  33             return i
  34     return -1
  35
  36 def find_token2(lines, token, start, end = 0):
  37     if end == 0:
  38         end = len(lines)
  39     for i in xrange(start, end):
  40         x = string.split(lines[i])
  41         if len(x) > 0 and x[0] == token:
  42             return i
  43     return -1
  44
  45 def find_tokens(lines, tokens, start, end = 0):
  46     if end == 0:
  47         end = len(lines)
  48     for i in xrange(start, end):
  49         line = lines[i]
  50         for token in tokens:
  51             if line[:len(token)] == token:
  52                 return i
  53     return -1
  54
  55 def find_re(lines, rexp, start, end = 0):
  56     if end == 0:
  57         end = len(lines)
  58     for i in xrange(start, end):
  59         if rexp.match(lines[i]):
  60                 return i
  61     return -1
  62
  63 def find_token_backwards(lines, token, start):
  64     m = len(token)
  65     for i in xrange(start, -1, -1):
  66         line = lines[i]
  67         if line[:m] == token:
  68             return i
  69     return -1
  70
  71 def find_tokens_backwards(lines, tokens, start):
  72     for i in xrange(start, -1, -1):
  73         line = lines[i]
  74         for token in tokens:
  75             if line[:len(token)] == token:
  76                 return i
  77     return -1
  78
  79 def get_value(lines, token, start, end = 0):
  80     i = find_token2(lines, token, start, end)
  81     if i == -1:
  82         return ""
  83     if len(string.split(lines[i])) > 1:
  84         return string.split(lines[i])[1]
  85     else:
  86         return ""
  87
  88 def del_token(lines, token, i, j):
  89     k = find_token2(lines, token, i, j)
  90     if k == -1:
  91         return j
  92     else:
  93         del lines[k]
  94         return j-1
  95
  96 # Finds the paragraph that contains line i.
  97 def get_paragraph(lines, i):
  98     while i != -1:
  99         i = find_tokens_backwards(lines, ["\\end_inset", "\\layout"], i)
 100         if i == -1: return -1
 101         if check_token(lines[i], "\\layout"):
 102             return i
 103         i = find_beginning_of_inset(lines, i)
 104
 105 # Finds the paragraph after the paragraph that contains line i.
 106 def get_next_paragraph(lines, i):
 107     while i != -1:
 108         i = find_tokens(lines, ["\\begin_inset", "\\layout", "\\end_float", "\\the_end"], i)
 109         if not check_token(lines[i], "\\begin_inset"):
 110             return i
 111         i = find_end_of_inset(lines, i)
 112
 113 def find_end_of(lines, i, start_token, end_token):
 114     count = 1
 115     n = len(lines)
 116     while i < n:
 117         i = find_tokens(lines, [end_token, start_token], i+1)
 118         if check_token(lines[i], start_token):
 119             count = count+1
 120         else:
 121             count = count-1
 122         if count == 0:
 123             return i
 124     return -1
 125
 126 # Finds the matching \end_inset
 127 def find_beginning_of(lines, i, start_token, end_token):
 128     count = 1
 129     while i > 0:
 130         i = find_tokens_backwards(lines, [start_token, end_token], i-1)
 131         if check_token(lines[i], end_token):
 132             count = count+1
 133         else:
 134             count = count-1
 135         if count == 0:
 136             return i
 137     return -1
 138
 139 # Finds the matching \end_inset
 140 def find_end_of_inset(lines, i):
 141     return find_end_of(lines, i, "\\begin_inset", "\\end_inset")
 142
 143 # Finds the matching \end_inset
 144 def find_beginning_of_inset(lines, i):
 145     return find_beginning_of(lines, i, "\\begin_inset", "\\end_inset")
 146
 147 def find_end_of_tabular(lines, i):
 148     return find_end_of(lines, i, "<lyxtabular", "</lyxtabular")
 149
 150 def get_tabular_lines(lines, i):
 151     result = []
 152     i = i+1
 153     j = find_end_of_tabular(lines, i)
 154     if j == -1:
 155         return []
 156
 157     while i <= j:
 158         if check_token(lines[i], "\\begin_inset"):
 159             i = find_end_of_inset(lines, i)+1
 160         else:
 161             result.append(i)
 162             i = i+1
 163     return result
 164
 165 def is_nonempty_line(line):
 166     return line != " "*len(line)
 167
 168 def find_nonempty_line(lines, start, end = 0):
 169     if end == 0:
 170         end = len(lines)
 171     for i in xrange(start, end):
 172         if is_nonempty_line(lines[i]):
 173             return i
 174     return -1
 175
 176 def set_comment(lines, version):
 177     lines[0] = "#LyX %s created this file. For more info see http://www.lyx.org/" % version
 178     if lines[1][0] == '#':
 179         del lines[1]
 180
 181 def set_format(lines, number):
 182     if int(number) <= 217:
 183         number = float(number)/100
 184     i = find_token(lines, "\\lyxformat", 0)
 185     lines[i] = "\\lyxformat %s" % number