lib/scripts/legacy_lyxpreview2ppm.py

   1 #! /usr/bin/env python
   2
   3 # file legacy_lyxpreview2ppm.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
   9 # Full author contact details are available in file CREDITS
  10
  11 # This script converts a LaTeX file to a bunch of ppm files using the
  12 # deprecated dvi->ps->ppm conversion route.
  13
  14 # If possible, please grab 'dvipng'; it's faster and more robust.
  15 # This legacy support will be removed one day...
  16
  17 import glob, os, re, string, sys
  18 import pipes, shutil, tempfile
  19
  20
  21 # Pre-compiled regular expressions.
  22 latex_file_re = re.compile("\.tex$")
  23
  24
  25 def usage(prog_name):
  26     return "Usage: %s <latex file> <dpi> <fg color> <bg color>\n"\
  27            "\twhere the colors are hexadecimal strings, eg 'faf0e6'"\
  28            % prog_name
  29
  30
  31 def error(message):
  32     sys.stderr.write(message + '\n')
  33     sys.exit(1)
  34
  35
  36 def find_exe(candidates, path):
  37     for prog in candidates:
  38         for directory in path:
  39             full_path = os.path.join(directory, prog)
  40             if os.access(full_path, os.X_OK):
  41                 return full_path
  42
  43     return None
  44
  45
  46 def find_exe_or_terminate(candidates, path):
  47     exe = find_exe(candidates, path)
  48     if exe == None:
  49         error("Unable to find executable from '%s'" % string.join(candidates))
  50
  51     return exe
  52
  53
  54 def run_command(cmd):
  55     handle = os.popen(cmd, 'r')
  56     cmd_stdout = handle.read()
  57     cmd_status = handle.close()
  58
  59     return cmd_status, cmd_stdout
  60
  61
  62 def extract_metrics_info(log_file, metrics_file):
  63     metrics = open(metrics_file, 'w')
  64
  65     log_re = re.compile("Preview: ([ST])")
  66     data_re = re.compile("(-?[0-9]+) (-?[0-9]+) (-?[0-9]+) (-?[0-9]+)")
  67
  68     tp_ascent  = 0.0
  69     tp_descent = 0.0
  70
  71     success = 0
  72     for line in open(log_file, 'r').readlines():
  73         match = log_re.match(line)
  74         if match == None:
  75             continue
  76
  77         snippet = (match.group(1) == 'S')
  78         success = 1
  79         match = data_re.search(line)
  80         if match == None:
  81             error("Unexpected data in %s\n%s" % (log_file, line))
  82
  83         if snippet:
  84             ascent  = string.atof(match.group(2)) + tp_ascent
  85             descent = string.atof(match.group(3)) - tp_descent
  86
  87             frac = 0.5
  88             if abs(ascent + descent) > 0.1:
  89                 frac = ascent / (ascent + descent)
  90
  91                 metrics.write("Snippet %s %f\n" % (match.group(1), frac))
  92
  93         else:
  94             tp_descent = string.atof(match.group(2))
  95             tp_ascent  = string.atof(match.group(4))
  96
  97     return success
  98
  99
 100 def extract_resolution(log_file, dpi):
 101     fontsize_re = re.compile("Preview: Fontsize")
 102     magnification_re = re.compile("Preview: Magnification")
 103     extract_decimal_re = re.compile("([0-9\.]+)")
 104     extract_integer_re = re.compile("([0-9]+)")
 105
 106     found_fontsize = 0
 107     found_magnification = 0
 108
 109     # Default values
 110     magnification = 1000.0
 111     fontsize = 0.0
 112
 113     for line in open(log_file, 'r').readlines():
 114         if found_fontsize and found_magnification:
 115             break
 116
 117         if not found_fontsize:
 118             match = fontsize_re.match(line)
 119             if match != None:
 120                 match = extract_decimal_re.search(line)
 121                 if match == None:
 122                     error("Unable to parse: %s" % line)
 123                 fontsize = string.atof(match.group(1))
 124                 found_fontsize = 1
 125                 continue
 126
 127         if not found_magnification:
 128             match = magnification_re.match(line)
 129             if match != None:
 130                 match = extract_integer_re.search(line)
 131                 if match == None:
 132                     error("Unable to parse: %s" % line)
 133                 magnification = string.atof(match.group(1))
 134                 found_magnification = 1
 135                 continue
 136
 137     return dpi * (10.0 / fontsize) * (1000.0 / magnification)
 138
 139
 140 def get_version_info():
 141     version_re = re.compile("([0-9])\.([0-9])")
 142
 143     match = version_re.match(sys.version)
 144     if match == None:
 145         error("Unable to extract version info from 'sys.version'")
 146
 147     return string.atoi(match.group(1)), string.atoi(match.group(2))
 148
 149
 150 def mkstemp():
 151     major, minor = get_version_info()
 152
 153     if major >= 2 and minor >= 3:
 154         return tempfile.mkstemp()
 155
 156     tmp_name = tempfile.mktemp()
 157     return open(tmp_name, 'w'), tmp_name
 158
 159
 160 def legacy_latex_file(latex_file, fg_color, bg_color):
 161     use_preview_re = re.compile("(\\\\usepackage\[[^]]+)(\]{preview})")
 162
 163     tmp, tmp_name = mkstemp()
 164
 165     success = 0
 166     for line in open(latex_file, 'r').readlines():
 167         match = use_preview_re.match(line)
 168         if match == None:
 169             tmp.write(line)
 170             continue
 171
 172         success = 1
 173         tmp.write("%s,dvips,tightpage%s\n\n" \
 174                   "\\AtBeginDocument{\\AtBeginDvi{%%\n" \
 175                   "\\special{!userdict begin/bop-hook{//bop-hook exec\n" \
 176                   "<%s%s>{255 div}forall setrgbcolor\n" \
 177                   "clippath fill setrgbcolor}bind def end}}}\n" \
 178                   % (match.group(1), match.group(2), fg_color, bg_color))
 179
 180     if success:
 181         tmp.close()
 182         shutil.copy(tmp_name, latex_file)
 183     os.remove(tmp_name)
 184
 185     return success
 186
 187
 188 def crop_files(pnmcrop, basename):
 189     t = pipes.Template()
 190     t.append("%s -left $IN" % pnmcrop, 'f-')
 191     t.append("%s -right > $OUT" % pnmcrop, '-f')
 192
 193     tmp, tmp_name = mkstemp()
 194     tmp.close()
 195     os.remove(tmp_name)
 196
 197     for file in glob.glob("%s*.ppm" % basename):
 198         if t.copy(file, tmp_name):
 199             shutil.copy(tmp_name, file)
 200         os.remove(tmp_name)
 201
 202
 203 def legacy_conversion(argv):
 204     # Parse and manipulate the command line arguments.
 205     if len(argv) != 6:
 206         error(usage(argv[0]))
 207
 208     # Ignore argv[1]
 209
 210     dir, latex_file = os.path.split(argv[2])
 211     if len(dir) != 0:
 212         os.chdir(dir)
 213
 214     dpi = string.atoi(argv[3])
 215     fg_color = argv[4]
 216     bg_color = argv[5]
 217
 218     # External programs used by the script.
 219     path = string.split(os.getenv("PATH"), os.pathsep)
 220     latex   = find_exe_or_terminate(["pplatex", "latex2e", "latex"], path)
 221     dvips   = find_exe_or_terminate(["dvips"], path)
 222     gs      = find_exe_or_terminate(["gs"], path)
 223     pnmcrop = find_exe(["pnmcrop"], path)
 224
 225     # Move color information into the latex file.
 226     if not legacy_latex_file(latex_file, fg_color, bg_color):
 227         error("Unable to move color info into the latex file")
 228
 229     # Compile the latex file.
 230     latex_call = "%s %s" % (latex, latex_file)
 231
 232     latex_status, latex_stdout = run_command(latex_call)
 233     if latex_status != None:
 234         error("%s failed to compile %s" \
 235               % (os.path.basename(latex), latex_file))
 236
 237     # Run the dvi file through dvips.
 238     dvi_file = latex_file_re.sub(".dvi", latex_file)
 239     ps_file  = latex_file_re.sub(".ps",  latex_file)
 240
 241     dvips_call = "%s -o %s %s" % (dvips, ps_file, dvi_file)
 242
 243     dvips_status, dvips_stdout = run_command(dvips_call)
 244     if dvips_status != None:
 245         error("Failed: %s %s" % (os.path.basename(dvips), dvi_file))
 246
 247     # Extract resolution data for gs from the log file.
 248     log_file = latex_file_re.sub(".log", latex_file)
 249     resolution = extract_resolution(log_file, dpi)
 250
 251     # Older versions of gs have problems with a large degree of
 252     # anti-aliasing at high resolutions
 253     alpha = 4
 254     if resolution > 150:
 255         alpha = 2
 256
 257     # Generate the bitmap images
 258     gs_call = "%s -dNOPAUSE -dBATCH -dSAFER -sDEVICE=pnmraw " \
 259               "-sOutputFile=%s%%d.ppm " \
 260               "-dGraphicsAlphaBit=%d -dTextAlphaBits=%d " \
 261               "-r%f %s" \
 262               % (gs, latex_file_re.sub("", latex_file), \
 263                  alpha, alpha, resolution, ps_file)
 264
 265     gs_status, gs_stdout = run_command(gs_call)
 266     if gs_status != None:
 267         error("Failed: %s %s" % (os.path.basename(gs), ps_file))
 268
 269     # Crop the images
 270     if pnmcrop != None:
 271         crop_files(pnmcrop, latex_file_re.sub("", latex_file))
 272
 273     # Extract metrics info from the log file.
 274     metrics_file = latex_file_re.sub(".metrics", latex_file)
 275     if not extract_metrics_info(log_file, metrics_file):
 276         error("Failed to extract metrics info from %s" % log_file)
 277
 278     return 0