]> git.lyx.org Git - lyx.git/blobdiff - lib/scripts/legacy_lyxpreview2ppm.py
A little more work on prefs2prefs.
[lyx.git] / lib / scripts / legacy_lyxpreview2ppm.py
index c08c9e63477ad21a250f72342b4f769c3b345d88..be242cc19247d08ad5a3c0172efafe78876b633d 100644 (file)
@@ -1,64 +1,73 @@
 #! /usr/bin/env python
+# -*- coding: utf-8 -*-
 
 # file legacy_lyxpreview2ppm.py
 # This file is part of LyX, the document processor.
 # Licence details can be found in the file COPYING.
 
 # author Angus Leeming
-
 # Full author contact details are available in file CREDITS
 
-# This script converts a LaTeX file to a bunch of ppm files using the
-# deprecated dvi->ps->ppm conversion route.
-
+# with much advice from members of the preview-latex project:
+#   David Kastrup, dak@gnu.org and
+#   Jan-Åke Larsson, jalar@mai.liu.se.
+# and with much help testing the code under Windows from
+#   Paul A. Rubin, rubin@msu.edu.
+
+# This script takes a LaTeX file and generates a collection of
+# ppm image files, one per previewed snippet.
+# Example usage:
+# legacy_lyxpreview2bitmap.py 0lyxpreview.tex 128 ppm 000000 faf0e6
+
+# This script takes five arguments:
+# TEXFILE:       the name of the .tex file to be converted.
+# SCALEFACTOR:   a scale factor, used to ascertain the resolution of the
+#                generated image which is then passed to gs.
+# OUTPUTFORMAT:  the format of the output bitmap image files.
+#                This particular script can produce only "ppm" format output.
+# FG_COLOR:      the foreground color as a hexadecimal string, eg '000000'.
+# BG_COLOR:      the background color as a hexadecimal string, eg 'faf0e6'.
+
+# Decomposing TEXFILE's name as DIR/BASE.tex, this script will,
+# if executed successfully, leave in DIR:
+# * a (possibly large) number of image files with names
+#   like BASE[0-9]+.ppm
+# * a file BASE.metrics, containing info needed by LyX to position
+#   the images correctly on the screen.
+
+# The script uses several external programs and files:
+# * A latex executable;
+# * preview.sty;
+# * dvips;
+# * gs;
+# * pnmcrop (optional).
+
+# preview.sty is part of the preview-latex project
+#   http://preview-latex.sourceforge.net/
+# Alternatively, it can be obtained from
+#   CTAN/support/preview-latex/
+
+# The script uses the deprecated dvi->ps->ppm conversion route.
 # If possible, please grab 'dvipng'; it's faster and more robust.
-# This legacy support will be removed one day...
+# If you have it then this script will not be invoked by
+# lyxpreview2bitmap.py.
+# Warning: this legacy support will be removed one day...
 
-import glob, os, re, string, sys
-import pipes, shutil, tempfile
+import glob, os, pipes, re, string, sys
 
+from lyxpreview_tools import copyfileobj, error, find_exe, \
+     find_exe_or_terminate, make_texcolor, mkstemp, run_command, warning
 
-# Pre-compiled regular expressions.
+# Pre-compiled regular expression.
 latex_file_re = re.compile("\.tex$")
 
 
 def usage(prog_name):
-    return "Usage: %s <latex file> <dpi> <fg color> <bg color>\n"\
+    return "Usage: %s <latex file> <dpi> ppm <fg color> <bg color>\n"\
            "\twhere the colors are hexadecimal strings, eg 'faf0e6'"\
            % prog_name
 
 
-def error(message):
-    sys.stderr.write(message + '\n')
-    sys.exit(1)
-
-
-def find_exe(candidates, path):
-    for prog in candidates:
-        for directory in path:
-            full_path = os.path.join(directory, prog)
-            if os.access(full_path, os.X_OK):
-                return full_path
-
-    return None
-
-
-def find_exe_or_terminate(candidates, path):
-    exe = find_exe(candidates, path)
-    if exe == None:
-        error("Unable to find executable from '%s'" % string.join(candidates))
-
-    return exe
-
-
-def run_command(cmd):
-    handle = os.popen(cmd, 'r')
-    cmd_stdout = handle.read()
-    cmd_status = handle.close()
-
-    return cmd_status, cmd_stdout
-
-
 def extract_metrics_info(log_file, metrics_file):
     metrics = open(metrics_file, 'w')
 
@@ -69,30 +78,45 @@ def extract_metrics_info(log_file, metrics_file):
     tp_descent = 0.0
 
     success = 0
-    for line in open(log_file, 'r').readlines():
-        match = log_re.match(line)
-        if match == None:
-            continue
+    try:
+        for line in open(log_file, 'r').readlines():
+            match = log_re.match(line)
+            if match == None:
+                continue
+
+            snippet = (match.group(1) == 'S')
+            success = 1
+            match = data_re.search(line)
+            if match == None:
+                error("Unexpected data in %s\n%s" % (log_file, line))
+
+            if snippet:
+                ascent  = string.atoi(match.group(2))
+                descent = string.atoi(match.group(3))
 
-        snippet = (match.group(1) == 'S')
-        success = 1
-        match = data_re.search(line)
-        if match == None:
-            error("Unexpected data in %s\n%s" % (log_file, line))
+                frac = 0.5
+                if ascent >= 0 and descent >= 0:
+                    ascent = float(ascent) + tp_ascent
+                    descent = float(descent) - tp_descent
 
-        if snippet:
-            ascent  = string.atof(match.group(2)) + tp_ascent
-            descent = string.atof(match.group(3)) - tp_descent
+                    if abs(ascent + descent) > 0.1:
+                        frac = ascent / (ascent + descent)
 
-            frac = 0.5
-            if abs(ascent + descent) > 0.1:
-                frac = ascent / (ascent + descent)
+                    # Sanity check
+                    if frac < 0 or frac > 1:
+                            frac = 0.5
 
                 metrics.write("Snippet %s %f\n" % (match.group(1), frac))
 
-        else:
-            tp_descent = string.atof(match.group(2))
-            tp_ascent  = string.atof(match.group(4))
+            else:
+                tp_descent = string.atof(match.group(2))
+                tp_ascent  = string.atof(match.group(4))
+
+    except:
+        # Unable to open the file, but do nothing here because
+        # the calling function will act on the value of 'success'.
+        warning('Warning in extract_metrics_info! Unable to open "%s"' % log_file)
+        warning(`sys.exc_type` + ',' + `sys.exc_value`)
 
     return success
 
@@ -108,141 +132,156 @@ def extract_resolution(log_file, dpi):
 
     # Default values
     magnification = 1000.0
-    fontsize = 0.0
-
-    for line in open(log_file, 'r').readlines():
-        if found_fontsize and found_magnification:
-            break
-
-        if not found_fontsize:
-            match = fontsize_re.match(line)
-            if match != None:
-                match = extract_decimal_re.search(line)
-                if match == None:
-                    error("Unable to parse: %s" % line)
-                fontsize = string.atof(match.group(1))
-                found_fontsize = 1
-                continue
-
-        if not found_magnification:
-            match = magnification_re.match(line)
-            if match != None:
-                match = extract_integer_re.search(line)
-                if match == None:
-                    error("Unable to parse: %s" % line)
-                magnification = string.atof(match.group(1))
-                found_magnification = 1
-                continue
-
+    fontsize = 10.0
+
+    try:
+        for line in open(log_file, 'r').readlines():
+            if found_fontsize and found_magnification:
+                break
+
+            if not found_fontsize:
+                match = fontsize_re.match(line)
+                if match != None:
+                    match = extract_decimal_re.search(line)
+                    if match == None:
+                        error("Unable to parse: %s" % line)
+                    fontsize = string.atof(match.group(1))
+                    found_fontsize = 1
+                    continue
+
+            if not found_magnification:
+                match = magnification_re.match(line)
+                if match != None:
+                    match = extract_integer_re.search(line)
+                    if match == None:
+                        error("Unable to parse: %s" % line)
+                    magnification = string.atof(match.group(1))
+                    found_magnification = 1
+                    continue
+
+    except:
+        warning('Warning in extract_resolution! Unable to open "%s"' % log_file)
+        warning(`sys.exc_type` + ',' + `sys.exc_value`)
+
+    # This is safe because both fontsize and magnification have
+    # non-zero default values.
     return dpi * (10.0 / fontsize) * (1000.0 / magnification)
 
-    
-def get_version_info():
-    version_re = re.compile("([0-9])\.([0-9])")
 
-    match = version_re.match(sys.version)
-    if match == None:
-        error("Unable to extract version info from 'sys.version'")
+def legacy_latex_file(latex_file, fg_color, bg_color, bg_color_gr):
+    use_preview_dvi_re = re.compile("(\s*\\\\usepackage\[[^]]+)(dvips\]{preview})")
+    use_preview_pdf_re = re.compile("(\s*\\\\usepackage\[[^]]+)(pdftex\]{preview})")
 
-    return string.atoi(match.group(1)), string.atoi(match.group(2))
+    tmp = mkstemp()
 
+    success = 0
+    try:
+        for line in open(latex_file, 'r').readlines():
+            match = use_preview_dvi_re.match(line)
+            if match == None:
+                match = use_preview_pdf_re.match(line)
+                if match == None:
+                    tmp.write(line)
+                    continue
+                success = 1
+                tmp.write("  \\usepackage{color}\n" \
+                      "  \\pagecolor[rgb]{%s}\n" \
+                      "%s\n" \
+                      % (bg_color_gr, match.group()))
+                continue
 
-def mkstemp():
-    major, minor = get_version_info()
-
-    if major >= 2 and minor >= 3:
-        return tempfile.mkstemp()
-
-    tmp_name = tempfile.mktemp()
-    return open(tmp_name, 'w'), tmp_name
-
-
-def legacy_latex_file(latex_file, fg_color, bg_color):
-    use_preview_re = re.compile("(\\\\usepackage\[[^]]+)(\]{preview})")
-
-    tmp, tmp_name = mkstemp()
+            success = 1
+            tmp.write("%stightpage,%s\n" \
+                      "  \\AtBeginDocument{\\AtBeginDvi{%%\n" \
+                      "  \\special{!userdict begin/bop-hook{//bop-hook exec\n" \
+                      "  <%s%s>{255 div}forall setrgbcolor\n" \
+                      "  clippath fill setrgbcolor}bind def end}}}\n" \
+                      % (match.group(1), match.group(2), fg_color, bg_color))
 
-    success = 0
-    for line in open(latex_file, 'r').readlines():
-        match = use_preview_re.match(line)
-        if match == None:
-            tmp.write(line)
-            continue
-
-        success = 1
-        tmp.write("%s,dvips,tightpage%s\n\n" \
-                  "\\AtBeginDocument{\\AtBeginDvi{%%\n" \
-                  "\\special{!userdict begin/bop-hook{//bop-hook exec\n" \
-                  "<%s%s>{255 div}forall setrgbcolor\n" \
-                  "clippath fill setrgbcolor}bind def end}}}\n" \
-                  % (match.group(1), match.group(2), fg_color, bg_color))
+    except:
+        # Unable to open the file, but do nothing here because
+        # the calling function will act on the value of 'success'.
+        warning('Warning in legacy_latex_file! Unable to open "%s"' % latex_file)
+        warning(`sys.exc_type` + ',' + `sys.exc_value`)
 
     if success:
-        tmp.close()
-        shutil.copy(tmp_name, latex_file)
-    os.remove(tmp_name)
+        copyfileobj(tmp, open(latex_file,"wb"), 1)
 
     return success
 
 
 def crop_files(pnmcrop, basename):
     t = pipes.Template()
-    t.append("%s -left $IN" % pnmcrop, 'f-')
-    t.append("%s -right > $OUT" % pnmcrop, '-f')
-
-    tmp, tmp_name = mkstemp()
-    tmp.close()
-    os.remove(tmp_name)
+    t.append('%s -left' % pnmcrop, '--')
+    t.append('%s -right' % pnmcrop, '--')
 
     for file in glob.glob("%s*.ppm" % basename):
-        if t.copy(file, tmp_name):
-            shutil.copy(tmp_name, file)
-        os.remove(tmp_name)
+        tmp = mkstemp()
+        new = t.open(file, "r")
+        copyfileobj(new, tmp)
+        if not new.close():
+            copyfileobj(tmp, open(file,"wb"), 1)
 
 
 def legacy_conversion(argv):
+    latex_commands = ["latex", "pplatex", "platex", "latex2e"]
     # Parse and manipulate the command line arguments.
-    if len(argv) != 6:
+    if len(argv) == 7:
+        latex_commands = [argv[6]]
+    elif len(argv) != 6:
         error(usage(argv[0]))
 
-    # Ignore argv[1]
-
-    dir, latex_file = os.path.split(argv[2])
+    dir, latex_file = os.path.split(argv[1])
     if len(dir) != 0:
         os.chdir(dir)
 
-    dpi = string.atoi(argv[3])
+    dpi = string.atoi(argv[2])
+
+    output_format = argv[3]
+
     fg_color = argv[4]
     bg_color = argv[5]
+    bg_color_gr = make_texcolor(argv[5], True)
 
     # External programs used by the script.
-    path = string.split(os.getenv("PATH"), os.pathsep)
-    latex   = find_exe_or_terminate(["pplatex", "latex2e", "latex"], path)
-    dvips   = find_exe_or_terminate(["dvips"], path)
-    gs      = find_exe_or_terminate(["gs"], path)
-    pnmcrop = find_exe(["pnmcrop"], path)
+    path  = string.split(os.environ["PATH"], os.pathsep)
+    latex = find_exe_or_terminate(latex_commands, path)
 
     # Move color information into the latex file.
-    if not legacy_latex_file(latex_file, fg_color, bg_color):
+    if not legacy_latex_file(latex_file, fg_color, bg_color, bg_color_gr):
         error("Unable to move color info into the latex file")
 
     # Compile the latex file.
-    latex_call = "%s %s" % (latex, latex_file)
+    latex_call = '%s "%s"' % (latex, latex_file)
 
     latex_status, latex_stdout = run_command(latex_call)
     if latex_status != None:
         error("%s failed to compile %s" \
               % (os.path.basename(latex), latex_file))
 
+    return legacy_conversion_step2(latex_file, dpi, output_format)
+
+
+def legacy_conversion_step2(latex_file, dpi, output_format):
+    # External programs used by the script.
+    path    = string.split(os.environ["PATH"], os.pathsep)
+    dvips   = find_exe_or_terminate(["dvips"], path)
+    gs      = find_exe_or_terminate(["gswin32c", "gs"], path)
+    pnmcrop = find_exe(["pnmcrop"], path)
+
     # Run the dvi file through dvips.
     dvi_file = latex_file_re.sub(".dvi", latex_file)
     ps_file  = latex_file_re.sub(".ps",  latex_file)
+    pdf_file  = latex_file_re.sub(".pdf", latex_file)
+
+    dvips_call = '%s -o "%s" "%s"' % (dvips, ps_file, dvi_file)
+    dvips_failed = False
 
-    dvips_call = "%s -o %s %s" % (dvips, ps_file, dvi_file)
-    
     dvips_status, dvips_stdout = run_command(dvips_call)
     if dvips_status != None:
-        error("Failed: %s %s" % (os.path.basename(dvips), dvi_file))
+        warning('Failed: %s %s ... looking for PDF' \
+            % (os.path.basename(dvips), dvi_file))
+        dvips_failed = True
 
     # Extract resolution data for gs from the log file.
     log_file = latex_file_re.sub(".log", latex_file)
@@ -254,13 +293,27 @@ def legacy_conversion(argv):
     if resolution > 150:
         alpha = 2
 
+    gs_device = "png16m"
+    gs_ext = "png"
+    if output_format == "ppm":
+        gs_device = "pnmraw"
+        gs_ext = "ppm"
+
     # Generate the bitmap images
-    gs_call = "%s -dNOPAUSE -dBATCH -dSAFER -sDEVICE=pnmraw " \
-              "-sOutputFile=%s%%d.ppm " \
-              "-dGraphicsAlphaBit=%d -dTextAlphaBits=%d " \
-              "-r%f %s" \
-              % (gs, latex_file_re.sub("", latex_file), \
-                 alpha, alpha, resolution, ps_file)
+    gs_call = '%s -dNOPAUSE -dBATCH -dSAFER -sDEVICE=%s ' \
+              '-sOutputFile="%s%%d.%s" ' \
+              '-dGraphicsAlphaBit=%d -dTextAlphaBits=%d ' \
+              '-r%f "%s"' \
+              % (gs, gs_device, latex_file_re.sub("", latex_file), \
+                 gs_ext, alpha, alpha, resolution, ps_file)
+
+    if dvips_failed:
+        gs_call = '%s -dNOPAUSE -dBATCH -dSAFER -sDEVICE=%s ' \
+                  '-sOutputFile="%s%%d.%s" ' \
+                  '-dGraphicsAlphaBit=%d -dTextAlphaBits=%d ' \
+                  '-r%f "%s"' \
+                  % (gs, gs_device, latex_file_re.sub("", latex_file), \
+                     gs_ext, alpha, alpha, resolution, pdf_file)
 
     gs_status, gs_stdout = run_command(gs_call)
     if gs_status != None:
@@ -276,3 +329,7 @@ def legacy_conversion(argv):
         error("Failed to extract metrics info from %s" % log_file)
 
     return 0
+
+
+if __name__ == "__main__":
+    legacy_conversion(sys.argv)