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, format):
 109     if format < 225:
 110         begin_layout = "\\layout"
 111     else:
 112         begin_layout = "\\begin_layout"
 113     while i != -1:
 114         i = find_tokens_backwards(lines, ["\\end_inset", begin_layout], i)
 115         if i == -1: return -1
 116         if check_token(lines[i], begin_layout):
 117             return i
 118         i = find_beginning_of_inset(lines, i)
 119     return -1
 120
 121
 122 # Finds the paragraph after the paragraph that contains line i.
 123 def get_next_paragraph(lines, i, format):
 124     if format < 225:
 125         tokens = ["\\begin_inset", "\\layout", "\\end_float", "\\the_end"]
 126     elif format < 236:
 127         tokens = ["\\begin_inset", "\\begin_layout", "\\end_float", "\\end_document"]
 128     else:
 129         tokens = ["\\begin_inset", "\\begin_layout", "\\end_float", "\\end_body", "\\end_document"]
 130     while i != -1:
 131         i = find_tokens(lines, tokens, i)
 132         if not check_token(lines[i], "\\begin_inset"):
 133             return i
 134         i = find_end_of_inset(lines, i)
 135     return -1
 136
 137
 138 def find_end_of(lines, i, start_token, end_token):
 139     count = 1
 140     n = len(lines)
 141     while i < n:
 142         i = find_tokens(lines, [end_token, start_token], i+1)
 143         if check_token(lines[i], start_token):
 144             count = count+1
 145         else:
 146             count = count-1
 147         if count == 0:
 148             return i
 149     return -1
 150
 151
 152 # Finds the matching \end_inset
 153 def find_beginning_of(lines, i, start_token, end_token):
 154     count = 1
 155     while i > 0:
 156         i = find_tokens_backwards(lines, [start_token, end_token], i-1)
 157         if check_token(lines[i], end_token):
 158             count = count+1
 159         else:
 160             count = count-1
 161         if count == 0:
 162             return i
 163     return -1
 164
 165
 166 # Finds the matching \end_inset
 167 def find_end_of_inset(lines, i):
 168     return find_end_of(lines, i, "\\begin_inset", "\\end_inset")
 169
 170
 171 # Finds the matching \end_inset
 172 def find_beginning_of_inset(lines, i):
 173     return find_beginning_of(lines, i, "\\begin_inset", "\\end_inset")
 174
 175
 176 def find_end_of_tabular(lines, i):
 177     return find_end_of(lines, i, "<lyxtabular", "</lyxtabular")
 178
 179
 180 def get_tabular_lines(lines, i):
 181     result = []
 182     i = i+1
 183     j = find_end_of_tabular(lines, i)
 184     if j == -1:
 185         return []
 186
 187     while i <= j:
 188         if check_token(lines[i], "\\begin_inset"):
 189             i = find_end_of_inset(lines, i)+1
 190         else:
 191             result.append(i)
 192             i = i+1
 193     return result
 194
 195
 196 def is_nonempty_line(line):
 197     return line != " "*len(line)
 198
 199
 200 def find_nonempty_line(lines, start, end = 0):
 201     if end == 0:
 202         end = len(lines)
 203     for i in xrange(start, end):
 204         if is_nonempty_line(lines[i]):
 205             return i
 206     return -1