lib/scripts/fig_copy.py

   1 # -*- coding: utf-8 -*-
   2
   3 # file fig_copy.py
   4 # This file is part of LyX, the document processor.
   5 # Licence details can be found in the file COPYING.
   6 #
   7 # \author Angus Leeming
   8 # \author Bo Peng
   9 #
  10 # Full author contact details are available in file CREDITS
  11
  12 # Usage:
  13 # fig_copy.py <from file> <to file>
  14
  15 # This script will copy an XFIG .fig file "$1" to "$2". In the process,
  16 # it will modify the contents of the .fig file so that the names of any
  17 # picture files that are stored as relative paths are replaced
  18 # with the absolute path.
  19
  20 from __future__ import print_function
  21 import os, sys
  22
  23 if len(sys.argv) != 3:
  24     print ("Usage: fig_copy.py <from file> <to file>", file=sys.stderr)
  25     sys.exit(1)
  26
  27 if not os.path.isfile(sys.argv[1]):
  28     print ("Unable to read", sys.argv[1], file=sys.stderr)
  29     sys.exit(1)
  30
  31 from_dir = os.path.split(os.path.realpath(sys.argv[1]))[0]
  32 to_dir = os.path.split(os.path.realpath(sys.argv[2]))[0]
  33
  34 # The work is trivial if "to" and "from" are in the same directory.
  35 if from_dir == to_dir:
  36     import shutil
  37     try:
  38         shutil.copy(sys.argv[1], sys.argv[2])
  39     except:
  40         sys.exit(1)
  41     sys.exit(0)
  42
  43 # Ok, they're in different directories. The .fig file must be modified.
  44 import re
  45
  46 # We're looking for a line of text that defines an entry of
  47 # type '2' (a polyline), subtype '5' (an external picture file).
  48 # The line has 14 other data fields.
  49 patternline = re.compile(br'^\s*2\s+5(\s+[0-9.+-]+){14}\s*$')
  50 emptyline   = re.compile(br'^\s*$')
  51 commentline = re.compile(br'^\s*#.*$')
  52 # we allow space in path name
  53 figureline  = re.compile(br'^(\s*[01]\s*)(\S[\S ]*)(\s*)$')
  54
  55 input = open(sys.argv[1], 'rb')
  56 output = open(sys.argv[2], 'wb')
  57
  58 # path in the fig is relative to this path
  59 os.chdir(from_dir)
  60
  61 found = False
  62 for line in input:
  63     if found and not emptyline.match(line) and not commentline.match(line):
  64         # The contents of the final line containing the file name
  65         # are ' X <file name>', where X = 0 or 1.
  66         # We extract filename and replace it with the absolute filename.
  67         (pre, path, post) = figureline.match(line).groups()
  68         line = pre + os.path.realpath(path) + post
  69         found = False
  70     elif patternline.match(line):
  71         found = True
  72     output.write(line)
  73
  74 input.close()
  75 output.close()