lib/lyx2lyx/parser_tools.py

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