]> git.donarmstrong.com Git - lilypond.git/blob - scm/ps-to-png.scm
Issue 4451 / 1: Use mkstemp for PNG scale down
[lilypond.git] / scm / ps-to-png.scm
1 ;;;; This file is part of LilyPond, the GNU music typesetter.
2 ;;;;
3 ;;;; Copyright (C) 2005--2015 Jan Nieuwenhuizen <janneke@gnu.org>
4 ;;;;
5 ;;;; LilyPond is free software: you can redistribute it and/or modify
6 ;;;; it under the terms of the GNU General Public License as published by
7 ;;;; the Free Software Foundation, either version 3 of the License, or
8 ;;;; (at your option) any later version.
9 ;;;;
10 ;;;; LilyPond 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 LilyPond.  If not, see <http://www.gnu.org/licenses/>.
17
18 (define-module (scm ps-to-png))
19
20 (use-modules
21  (ice-9 optargs)
22  (ice-9 regex)
23  (ice-9 rw)
24  (srfi srfi-1)
25  (srfi srfi-13)
26  (srfi srfi-14)
27  (lily)
28  )
29
30 (define (re-sub re sub string)
31   (regexp-substitute/global #f re string 'pre sub 'post))
32
33 (define-public (gulp-file file-name . max-size)
34   (ly:gulp-file file-name (if (pair? max-size) (car max-size))))
35
36 (define (search-pngtopam)
37   (search-executable
38    (if (eq? PLATFORM 'windows)
39        '("pngtopam.exe" "pngtopnm.exe")
40        '("pngtopam" "pngtopnm"))))
41
42 (define (search-pamscale)
43   (search-executable
44    (if (eq? PLATFORM 'windows)
45        '("pamscale.exe" "pnmscale.exe")
46        '("pamscale" "pnmscale"))))
47
48 (define (search-pnmtopng)
49   (search-executable
50    (if (eq? PLATFORM 'windows)
51        '("pnmtopng.exe")
52        '("pnmtopng"))))
53
54 (define (scale-down-image factor file)
55   (let* ((port-tmp1 (make-tmpfile))
56          (tmp1-name (port-filename port-tmp1))
57          (port-tmp2 (make-tmpfile))
58          (tmp2-name (port-filename port-tmp2))
59          ;; Netpbm commands (pngtopnm, pnmscale, pnmtopng)
60          ;; outputs only standard output instead of a file.
61          ;; So we need pipe and redirection.
62          ;; However, ly:system can't handle them.
63          ;; Therefore, we use ly:system-with-shell.
64          (cmd
65           (ly:format
66            "~a \"~a\" | ~a -reduce ~a | ~a -compression 9 > \"~a\""
67            (search-pngtopam) tmp1-name
68            (search-pamscale) factor
69            (search-pnmtopng)
70            tmp2-name)))
71
72     (copy-binary-file file tmp1-name)
73     (ly:system-with-shell cmd)
74     (copy-binary-file tmp2-name file)
75     (delete-file tmp1-name)
76     (delete-file tmp2-name)))
77
78 (define-public (ps-page-count ps-name)
79   (let* ((byte-count 10240)
80          (header (gulp-file ps-name byte-count))
81          (first-null (string-index header #\nul))
82          (match (string-match "%%Pages: ([0-9]+)"
83                               (if (number? first-null)
84                                   (substring header 0 first-null)
85                                   header))))
86     (if match (string->number (match:substring match 1)) 0)))
87
88 (define-public (make-ps-images base-name tmp-name is-eps . rest)
89   (let-keywords*
90    rest #f
91    ((resolution 90)
92     (page-width  100)
93     (page-height 100)
94     (rename-page-1 #f)
95     (be-verbose (ly:get-option 'verbose))
96     (pixmap-format 'png16m)
97     (anti-alias-factor 1))
98
99    (let* ((format-str (format #f "~a" pixmap-format))
100           (extension (cond
101                       ((string-contains format-str "png") "png")
102                       ((string-contains format-str "jpg") "jpeg")
103                       ((string-contains format-str "jpeg") "jpeg")
104                       (else
105                        (ly:error "Unknown pixmap format ~a" pixmap-format))))
106           (png1 (format #f "~a.~a" base-name extension))
107           (pngn (format #f "~a-page%d.~a" base-name extension))
108           (page-count (ps-page-count tmp-name))
109           (multi-page? (> page-count 1))
110
111           ;; Escape `%' (except `page%d') for ghostscript
112           (base-name-gs (string-join
113                          (string-split base-name #\%)
114                          "%%"))
115           (png1-gs (format #f "~a.~a" base-name-gs extension))
116           (pngn-gs (format #f "~a-page%d.~a" base-name-gs extension))
117           (output-file (if multi-page? pngn-gs png1-gs))
118
119           (*unspecified* (if #f #f))
120           (cmd
121            (remove (lambda (x) (eq? x *unspecified*))
122                    (list
123                     (search-gs)
124                     (if (ly:get-option 'verbose) *unspecified* "-q")
125                     (if (or (ly:get-option 'gs-load-fonts)
126                             (ly:get-option 'gs-load-lily-fonts)
127                             (eq? PLATFORM 'windows))
128                         "-dNOSAFER"
129                         "-dSAFER")
130
131                     (if is-eps
132                         "-dEPSCrop"
133                         (ly:format "-dDEVICEWIDTHPOINTS=~$" page-width))
134                     (if is-eps
135                         *unspecified*
136                         (ly:format "-dDEVICEHEIGHTPOINTS=~$" page-height))
137                     "-dGraphicsAlphaBits=4"
138                     "-dTextAlphaBits=4"
139                     "-dNOPAUSE"
140                     "-dBATCH"
141                     (ly:format "-sDEVICE=~a" pixmap-format)
142                     (string-append "-sOutputFile=" output-file)
143                     (ly:format "-r~a" (* anti-alias-factor resolution))
144                     (string-append "-f" tmp-name))))
145           (files '()))
146
147      (ly:system cmd)
148
149      (set! files
150            (if multi-page?
151                (map
152                 (lambda (n)
153                   (format #f "~a-page~a.png" base-name (1+ n)))
154                 (iota page-count))
155                (list (format #f "~a.png" base-name))))
156
157      (if (and rename-page-1 multi-page?)
158          (begin
159            (rename-file (re-sub "%d" "1" pngn) png1)
160            (set! files
161                  (cons png1
162                        (cdr files)))
163            ))
164
165      (if (not (= 1 anti-alias-factor))
166          (for-each
167           (lambda (f) (scale-down-image anti-alias-factor f)) files))
168      files)))