1 ################################################################
2 # lilylib.py -- options and stuff
4 # source file of the GNU LilyPond music typesetter
6 # (c) 1998--2006 Han-Wen Nienhuys <hanwen@cs.uu.nl>
7 # Jan Nieuwenhuizen <janneke@gnu.org>
9 ### subst:\(^\|[^._a-z]\)\(abspath\|identify\|warranty\|progress\|warning\|error\|exit\|getopt_args\|option_help_str\|options_help_str\|help\|setup_temp\|read_pipe\|system\|cleanup_temp\|strip_extension\|cp_to_dir\|mkdir_p\|init\) *(
12 ### subst: \(help_summary\|keep_temp_dir_p\|option_definitions\|original_dir\|program_name\|pseudo_filter_p\|temp_dir\|verbose_p\)
24 ################################################################
25 # Users of python modules should include this snippet
26 # and customize variables below.
28 # We'll suffer this path init stuff as long as we don't install our
29 # python packages in <prefix>/lib/pythonx.y (and don't kludge around
30 # it as we do with teTeX on Red Hat Linux: set some environment var
31 # (PYTHONPATH) in profile)
33 # If set, LILYPONDPREFIX must take prevalence
34 # if datadir is not set, we're doing a build and LILYPONDPREFIX
36 datadir = '@local_lilypond_datadir@'
37 if not os.path.isdir (datadir):
38 datadir = '@lilypond_datadir@'
39 if os.environ.has_key ('LILYPONDPREFIX') :
40 datadir = os.environ['LILYPONDPREFIX']
41 while datadir[-1] == os.sep:
44 sys.path.insert (0, os.path.join (datadir, 'python'))
49 localedir = '@localedir@'
52 gettext.bindtextdomain ('lilypond', localedir)
53 gettext.textdomain ('lilypond')
64 sys.stderr.write (__main__.program_name + ": " + _ ("warning: %s") % s + '\n')
67 sys.stderr.write (__main__.program_name + ": " + _ ("error: %s") % s + '\n')
70 be_verbose = get_global_option('verbose_p', 'verbose')
72 raise _ ('Exiting (%d)...') % i
76 def lilypond_version (binary):
77 p = read_pipe ('%s --version ' % binary)
82 m = re.search ('GNU LilyPond ([0-9a-z.]+)', p)
88 def lilypond_version_check (binary, req):
90 v = lilypond_version (binary)
92 error (_("Binary %s has version %s, looking for version %s") % \
99 ''' Create a temporary directory, and return its name. '''
101 if not __main__.keep_temp_dir_p:
102 __main__.temp_dir = tempfile.mktemp (__main__.program_name)
104 os.mkdir (__main__.temp_dir, 0700)
108 return __main__.temp_dir
110 def command_name (cmd):
111 # Strip all stuf after command,
112 # deal with "((latex ) >& 1 ) .." too
113 cmd = re.match ('([\(\)]*)([^\\\ ]*)', cmd).group (2)
114 return os.path.basename (cmd)
116 def error_log (name):
117 name = re.sub('[^a-z]','x', name)
118 return tempfile.mktemp ('%s.errorlog' % name)
120 def read_pipe (cmd, mode = 'r'):
126 progress (_ ("Opening pipe `%s\'") % cmd)
128 error_log_file = error_log (command_name (cmd))
129 redirect = ' 2>%s' % error_log_file
131 pipe = os.popen (cmd + redirect, mode)
132 output = pipe.read ()
133 status = pipe.close ()
134 # successful pipe close returns 'None'
137 signal = 0x0f & status
138 exit_status = status >> 8
141 error (_ ("`%s\' failed (%d)") % (cmd, exit_status))
144 contents = open (error_log_file).read ()
146 error (_ ("The error log is as follows:"))
147 sys.stderr.write (contents)
151 os.unlink (error_log_file)
159 os.unlink (error_log_file)
163 def get_global_option (old, new):
165 return __main__.__dict__[old]
167 return __main__.global_options.__dict__[new]
169 def system (cmd, ignore_error = 0, progress_p = 0):
171 '''System CMD. If IGNORE_ERROR, do not complain when CMD
172 returns non zero. If PROGRESS_P, always show progress.
176 Exit status of CMD '''
178 name = command_name (cmd)
182 be_verbose = get_global_option('verbose_p', 'verbose')
183 pseudo_filter = get_global_option ('pseudo_filter_p', 'pseudo_filter')
187 progress (_ ("Invoking `%s\'") % cmd)
189 progress ( _("Running %s...") % name)
193 error_log_file = error_log (name)
194 redirect = ' 1>/dev/null 2>' + error_log_file
196 redirect = ' 1>/dev/null'
198 status = os.system (cmd + redirect)
199 signal = 0x0f & status
200 exit_status = status >> 8
204 exit_type = 'status %d' % exit_status
206 exit_type = 'signal %d' % signal
208 msg = _ ("`%s\' failed (%s)") % (name, exit_type)
211 warning (msg + ' ' + _ ("(ignored)"))
214 if not progress_p and error_log_file:
215 error (_ ("The error log is as follows:"))
216 sys.stderr.write (open (error_log_file).read ())
218 os.unlink (error_log_file)
222 os.unlink (error_log_file)
227 be_verbose = get_global_option('verbose_p', 'verbose')
228 if not __main__.keep_temp_dir_p:
230 progress (_ ("Cleaning %s...") % __main__.temp_dir)
231 shutil.rmtree (__main__.temp_dir)
236 def strip_extension (f, ext):
237 (p, e) = os.path.splitext (f)
243 def cp_to_dir (pattern, dir):
244 "Copy files matching re PATTERN from cwd to DIR"
246 # Duh. Python style portable: cp *.EXT OUTDIR
247 # system ('cp *.%s %s' % (ext, outdir), 1)
249 files = filter (lambda x, p=pattern: re.match (p, x), os.listdir ('.'))
250 map (lambda x, d=dir: shutil.copy2 (x, os.path.join (d, x)), files)
253 def search_exe_path (name):
254 p = os.environ['PATH']
255 exe_paths = string.split (p, ':')
257 full = os.path.join (e, name)
258 if os.path.exists (full):
263 def mkdir_p (dir, mode=0777):
264 if not os.path.isdir (dir):
267 def print_environment ():
268 for (k,v) in os.environ.items ():
269 sys.stderr.write ("%s=\"%s\"\n" % (k, v))
272 def ps_page_count (ps_name):
273 header = open (ps_name).read (1024)
274 m = re.search ('\n%%Pages: ([0-9]+)', header)
276 return string.atoi (m.group (1))
279 def make_ps_images (ps_name, resolution = 90, papersize = "a4",
281 base = os.path.basename (re.sub (r'\.e?ps', '', ps_name))
282 header = open (ps_name).read (1024)
285 pngn = base + '-page%d.png'
287 multi_page = re.search ('\n%%Pages: ', header)
289 # png16m is because Lily produces color nowadays.
292 # GS can produce empty 2nd page if pngn is used.
296 -dGraphicsAlphaBits=4\
300 -sOutputFile='%(output_file)s'\
301 -sPAPERSIZE=%(papersize)s\
306 -c quit ''' % vars ()
310 -dGraphicsAlphaBits=4\
314 -sOutputFile='%(output_file)s'\
315 -sPAPERSIZE=%(papersize)s\
321 remove = glob.glob (png1) + glob.glob (base + '-page*.png')
322 map (os.unlink, remove)
324 status = system (cmd)
325 signal = 0xf & status
326 exit_status = status >> 8
329 remove = glob.glob (png1) + glob.glob (base + '-page*.png')
330 map (os.unlink, remove)
331 error (_ ("%s exited with status: %d") % ('GS', status))
334 if rename_page1_p and multi_page:
335 os.rename (pngn % 1, png1)
336 files = glob.glob (png1) + glob.glob (re.sub ('%d', '*', pngn))
339 class NonDentedHeadingFormatter (optparse.IndentedHelpFormatter):
340 def format_heading(self, heading):
342 return heading[0].upper() + heading[1:] + ':\n'
344 def format_option_strings(self, option):
346 if option._short_opts and option._long_opts:
350 if option.takes_value():
351 metavar = '=%s' % option.metavar or option.dest.upper()
353 return "%3s%s %s%s" % (" ".join (option._short_opts),
355 " ".join (option._long_opts),
358 def format_usage(self, usage):
359 return _("Usage: %s\n") % usage
361 def format_description(self, description):
364 def get_option_parser (*args, **kwargs):
365 p = optparse.OptionParser (*args, **kwargs)
366 p.formatter = NonDentedHeadingFormatter ()