scripts/lilypond-book.py

   1 #!@PYTHON@
   2
   3 '''
   4 Example usage:
   5
   6 test:
   7      lilypond-book --filter="tr '[a-z]' '[A-Z]'" BOOK
   8
   9 convert-ly on book:
  10      lilypond-book --filter="convert-ly --no-version --from=1.6.11 -" BOOK
  11
  12 classic lilypond-book:
  13      lilypond-book --process="lilypond" BOOK.tely
  14
  15 TODO:
  16
  17     *  this script is too complex. Modularize.
  18
  19     *  ly-options: intertext?
  20     *  --line-width?
  21     *  eps in latex / eps by lilypond -b ps?
  22     *  check latex parameters, twocolumn, multicolumn?
  23     *  use --png --ps --pdf for making images?
  24
  25     *  Converting from lilypond-book source, substitute:
  26        @mbinclude foo.itely -> @include foo.itely
  27        \mbinput -> \input
  28
  29 '''
  30
  31 import __main__
  32 import glob
  33 import stat
  34 import string
  35 import tempfile
  36 import commands
  37
  38 # Users of python modules should include this snippet
  39 # and customize variables below.
  40
  41 # We'll suffer this path initialization stuff as long as we don't install
  42 # our python packages in <prefix>/lib/pythonX.Y
  43
  44 # If set, LILYPONDPREFIX must take prevalence.
  45 # if datadir is not set, we're doing a build and LILYPONDPREFIX.
  46 import getopt, os, sys
  47 datadir = '@local_lilypond_datadir@'
  48 if not os.path.isdir (datadir):
  49         datadir = '@lilypond_datadir@'
  50 if os.environ.has_key ('LILYPONDPREFIX'):
  51         datadir = os.environ['LILYPONDPREFIX']
  52         while datadir[-1] == os.sep:
  53                 datadir= datadir[:-1]
  54
  55         datadir = os.path.join (datadir, "share/lilypond/current/")
  56
  57 sys.path.insert (0, os.path.join (datadir, 'python'))
  58
  59 # Customize these.
  60 #if __name__ == '__main__':
  61
  62 import lilylib as ly
  63 import fontextract
  64 global _;_=ly._
  65 global re;re = ly.re
  66
  67 # Lilylib globals.
  68 program_version = '@TOPLEVEL_VERSION@'
  69 program_name = os.path.basename (sys.argv[0])
  70 verbose_p = 0
  71 pseudo_filter_p = 0
  72 original_dir = os.getcwd ()
  73 backend = 'ps'
  74
  75 help_summary = _ (
  76 '''Process LilyPond snippets in hybrid HTML, LaTeX, or texinfo document.
  77 Example usage:
  78
  79    lilypond-book --filter="tr '[a-z]' '[A-Z]'" BOOK
  80    lilypond-book --filter="convert-ly --no-version --from=2.0.0 -" BOOK
  81    lilypond-book --process='lilypond -I include' BOOK
  82 ''')
  83
  84 copyright = ('Jan Nieuwenhuizen <janneke@gnu.org>',
  85              'Han-Wen Nienhuys <hanwen@cs.uu.nl>')
  86
  87 option_definitions = [
  88         (_ ("FMT"), 'f', 'format',
  89           _ ('''use output format FMT (texi [default], texi-html,
  90                 latex, html)''')),
  91         (_ ("FILTER"), 'F', 'filter',
  92           _ ("pipe snippets through FILTER [convert-ly -n -]")),
  93         ('', 'h', 'help',
  94           _ ("print this help")),
  95         (_ ("DIR"), 'I', 'include',
  96           _ ("add DIR to include path")),
  97         (_ ("DIR"), 'o', 'output',
  98           _ ("write output to DIR")),
  99         (_ ("COMMAND"), 'P', 'process',
 100           _ ("process ly_files using COMMAND FILE...")),
 101         ('', '', 'psfonts',
 102           _ ('''extract all PostScript fonts into INPUT.psfonts for LaTeX
 103           must use this with dvips -h INPUT.psfonts''')),
 104         ('', '', 'keep-line-breaks',
 105           _ ('''do not alter the number of newline characters in output
 106           .tex files. Useful when the LaTeX package srcltx is used''')),
 107         ('', 'V', 'verbose',
 108           _ ("be verbose")),
 109         ('', 'v', 'version',
 110           _ ("print version information")),
 111         ('', 'w', 'warranty',
 112           _ ("show warranty and copyright")),
 113 ]
 114
 115 include_path = [os.path.abspath (os.getcwd ())]
 116 lilypond_binary = os.path.join ('@bindir@', 'lilypond')
 117
 118 # Only use installed binary when we are installed too.
 119 if '@bindir@' == ('@' + 'bindir@') or not os.path.exists (lilypond_binary):
 120         lilypond_binary = 'lilypond'
 121
 122 psfonts_p = 0
 123 srcltx_p = 0
 124 use_hash_p = 1
 125 format = 0
 126 output_name = ''
 127 filter_cmd = 0
 128 process_cmd = ''
 129 default_ly_options = { 'alt': "[image of music]" }
 130
 131 #
 132 # Is this pythonic?  Personally, I find this rather #define-nesque. --hwn
 133 #
 134 AFTER = 'after'
 135 BEFORE = 'before'
 136 EXAMPLEINDENT = 'exampleindent'
 137 FILTER = 'filter'
 138 FRAGMENT = 'fragment'
 139 HTML = 'html'
 140 INDENT = 'indent'
 141 LATEX = 'latex'
 142 LAYOUT = 'layout'
 143 LINE_WIDTH = 'line-width'
 144 NOFRAGMENT = 'nofragment'
 145 NOINDENT = 'noindent'
 146 NOQUOTE = 'noquote'
 147 NOTES = 'body'
 148 NOTIME = 'notime'
 149 OUTPUT = 'output'
 150 OUTPUTIMAGE = 'outputimage'
 151 PACKED = 'packed'
 152 PAPER = 'paper'
 153 PREAMBLE = 'preamble'
 154 PRINTFILENAME = 'printfilename'
 155 QUOTE = 'quote'
 156 RAGGED_RIGHT = 'ragged-right'
 157 RELATIVE = 'relative'
 158 STAFFSIZE = 'staffsize'
 159 TEXIDOC = 'texidoc'
 160 TEXINFO = 'texinfo'
 161 VERBATIM = 'verbatim'
 162 FONTLOAD = 'fontload'
 163 FILENAME = 'filename'
 164 ALT = 'alt'
 165
 166
 167 # NOTIME has no opposite so it isn't part of this dictionary.
 168 # NOQUOTE is used internally only.
 169 no_options = {
 170         NOFRAGMENT: FRAGMENT,
 171         NOINDENT: INDENT,
 172 }
 173
 174
 175 # Recognize special sequences in the input.
 176 #
 177 #   (?P<name>regex) -- Assign result of REGEX to NAME.
 178 #   *? -- Match non-greedily.
 179 #   (?m) -- Multiline regex: Make ^ and $ match at each line.
 180 #   (?s) -- Make the dot match all characters including newline.
 181 #   (?x) -- Ignore whitespace in patterns.
 182 no_match = 'a\ba'
 183 snippet_res = {
 184         ##
 185         HTML: {
 186                 'include':
 187                   no_match,
 188
 189                 'lilypond':
 190                   r'''(?mx)
 191                     (?P<match>
 192                     <lilypond
 193                       (\s*(?P<options>.*?)\s*:)?\s*
 194                       (?P<code>.*?)
 195                     />)''',
 196
 197                 'lilypond_block':
 198                   r'''(?msx)
 199                     (?P<match>
 200                     <lilypond
 201                       \s*(?P<options>.*?)\s*
 202                     >
 203                     (?P<code>.*?)
 204                     </lilypond>)''',
 205
 206                 'lilypond_file':
 207                   r'''(?mx)
 208                     (?P<match>
 209                     <lilypondfile
 210                       \s*(?P<options>.*?)\s*
 211                     >
 212                     \s*(?P<filename>.*?)\s*
 213                     </lilypondfile>)''',
 214
 215                 'multiline_comment':
 216                   r'''(?smx)
 217                     (?P<match>
 218                     \s*(?!@c\s+)
 219                     (?P<code><!--\s.*?!-->)
 220                     \s)''',
 221
 222                 'singleline_comment':
 223                   no_match,
 224
 225                 'verb':
 226                   r'''(?x)
 227                     (?P<match>
 228                       (?P<code><pre>.*?</pre>))''',
 229
 230                 'verbatim':
 231                   r'''(?x)
 232                     (?s)
 233                     (?P<match>
 234                       (?P<code><pre>\s.*?</pre>\s))''',
 235         },
 236
 237         ##
 238         LATEX: {
 239                 'include':
 240                   r'''(?smx)
 241                     ^[^%\n]*?
 242                     (?P<match>
 243                     \\input\s*{
 244                       (?P<filename>\S+?)
 245                     })''',
 246
 247                 'lilypond':
 248                   r'''(?smx)
 249                     ^[^%\n]*?
 250                     (?P<match>
 251                     \\lilypond\s*(
 252                     \[
 253                       \s*(?P<options>.*?)\s*
 254                     \])?\s*{
 255                       (?P<code>.*?)
 256                     })''',
 257
 258                 'lilypond_block':
 259                   r'''(?smx)
 260                     ^[^%\n]*?
 261                     (?P<match>
 262                     \\begin\s*(
 263                     \[
 264                       \s*(?P<options>.*?)\s*
 265                     \])?\s*{lilypond}
 266                       (?P<code>.*?)
 267                     ^[^%\n]*?
 268                     \\end\s*{lilypond})''',
 269
 270                 'lilypond_file':
 271                   r'''(?smx)
 272                     ^[^%\n]*?
 273                     (?P<match>
 274                     \\lilypondfile\s*(
 275                     \[
 276                       \s*(?P<options>.*?)\s*
 277                     \])?\s*\{
 278                       (?P<filename>\S+?)
 279                     })''',
 280
 281                 'multiline_comment':
 282                   no_match,
 283
 284                 'singleline_comment':
 285                   r'''(?mx)
 286                     ^.*?
 287                     (?P<match>
 288                       (?P<code>
 289                       %.*$\n+))''',
 290
 291                 'verb':
 292                   r'''(?mx)
 293                     ^[^%\n]*?
 294                     (?P<match>
 295                       (?P<code>
 296                       \\verb(?P<del>.)
 297                         .*?
 298                       (?P=del)))''',
 299
 300                 'verbatim':
 301                   r'''(?msx)
 302                     ^[^%\n]*?
 303                     (?P<match>
 304                       (?P<code>
 305                       \\begin\s*{verbatim}
 306                         .*?
 307                       \\end\s*{verbatim}))''',
 308         },
 309
 310         ##
 311         TEXINFO: {
 312                 'include':
 313                   r'''(?mx)
 314                     ^(?P<match>
 315                     @include\s+
 316                       (?P<filename>\S+))''',
 317
 318                 'lilypond':
 319                   r'''(?smx)
 320                     ^[^\n]*?(?!@c\s+)[^\n]*?
 321                     (?P<match>
 322                     @lilypond\s*(
 323                     \[
 324                       \s*(?P<options>.*?)\s*
 325                     \])?\s*{
 326                       (?P<code>.*?)
 327                     })''',
 328
 329                 'lilypond_block':
 330                   r'''(?msx)
 331                     ^(?P<match>
 332                     @lilypond\s*(
 333                     \[
 334                       \s*(?P<options>.*?)\s*
 335                     \])?\s+?
 336                     ^(?P<code>.*?)
 337                     ^@end\s+lilypond)\s''',
 338
 339                 'lilypond_file':
 340                   r'''(?mx)
 341                     ^(?P<match>
 342                     @lilypondfile\s*(
 343                     \[
 344                       \s*(?P<options>.*?)\s*
 345                     \])?\s*{
 346                       (?P<filename>\S+)
 347                     })''',
 348
 349                 'multiline_comment':
 350                   r'''(?smx)
 351                     ^(?P<match>
 352                       (?P<code>
 353                       @ignore\s
 354                         .*?
 355                       @end\s+ignore))\s''',
 356
 357                 'singleline_comment':
 358                   r'''(?mx)
 359                     ^.*
 360                     (?P<match>
 361                       (?P<code>
 362                       @c([ \t][^\n]*|)\n))''',
 363
 364         # Don't do this: It interferes with @code{@{}.
 365         #       'verb': r'''(?P<code>@code{.*?})''',
 366
 367                 'verbatim':
 368                   r'''(?sx)
 369                     (?P<match>
 370                       (?P<code>
 371                       @example
 372                         \s.*?
 373                       @end\s+example\s))''',
 374         },
 375 }
 376
 377 format_res = {
 378         HTML: {
 379                 'intertext': r',?\s*intertext=\".*?\"',
 380                 'option_sep': '\s*',
 381         },
 382
 383         LATEX: {
 384                 'intertext': r',?\s*intertext=\".*?\"',
 385                 'option_sep': '\s*,\s*',
 386         },
 387
 388         TEXINFO: {
 389                 'intertext': r',?\s*intertext=\".*?\"',
 390                 'option_sep': '\s*,\s*',
 391         },
 392 }
 393
 394 # Options without a pattern in ly_options.
 395 simple_options = [
 396         EXAMPLEINDENT,
 397         FRAGMENT,
 398         NOFRAGMENT,
 399         NOINDENT,
 400         PRINTFILENAME,
 401         TEXIDOC,
 402         VERBATIM,
 403         FONTLOAD,
 404         FILENAME,
 405         ALT
 406 ]
 407
 408 ly_options = {
 409         ##
 410         NOTES: {
 411                 RELATIVE: r'''\relative c%(relative_quotes)s''',
 412         },
 413
 414         ##
 415         PAPER: {
 416                 INDENT: r'''indent = %(indent)s''',
 417
 418                 LINE_WIDTH: r'''line-width = %(line-width)s''',
 419
 420                 QUOTE: r'''line-width = %(line-width)s - 2.0 * %(exampleindent)s''',
 421
 422                 RAGGED_RIGHT: r'''ragged-right = ##t''',
 423
 424                 PACKED: r'''packed = ##t''',
 425         },
 426
 427         ##
 428         LAYOUT: {
 429                 NOTIME: r'''
 430   \context {
 431     \Score
 432     timing = ##f
 433   }
 434   \context {
 435     \Staff
 436     \remove Time_signature_engraver
 437   }''',
 438         },
 439
 440         ##
 441         PREAMBLE: {
 442                 STAFFSIZE: r'''#(set-global-staff-size %(staffsize)s)''',
 443         },
 444 }
 445
 446 output = {
 447         ##
 448         HTML: {
 449                 FILTER: r'''<lilypond %(options)s>
 450 %(code)s
 451 </lilypond>
 452 ''',
 453
 454                 AFTER: r'''
 455   </a>
 456 </p>''',
 457
 458                 BEFORE: r'''<p>
 459   <a href="%(base)s.ly">''',
 460
 461                 OUTPUT: r'''
 462     <img align="center" valign="center"
 463          border="0" src="%(image)s" alt="%(alt)s">''',
 464
 465                 PRINTFILENAME: '<p><tt><a href="%(base)s.ly">%(filename)s</a></tt></p>',
 466
 467                 QUOTE: r'''<blockquote>
 468 %(str)s
 469 </blockquote>
 470 ''',
 471
 472                 VERBATIM: r'''<pre>
 473 %(verb)s</pre>''',
 474         },
 475
 476         ##
 477         LATEX: {
 478                 OUTPUT: r'''{
 479 \parindent 0pt
 480 \catcode`\@=12
 481 \ifx\preLilyPondExample \undefined
 482   \relax
 483 \else
 484   \preLilyPondExample
 485 \fi
 486 \def\lilypondbook{}
 487 \input %(base)s-systems.tex
 488 \ifx\postLilyPondExample \undefined
 489   \relax
 490 \else
 491   \postLilyPondExample
 492 \fi
 493 }''',
 494
 495                 PRINTFILENAME: '''\\texttt{%(filename)s}
 496         ''',
 497
 498                 QUOTE: r'''\begin{quotation}%(str)s
 499 \end{quotation}''',
 500
 501                 VERBATIM: r'''\noindent
 502 \begin{verbatim}%(verb)s\end{verbatim}''',
 503
 504                 FILTER: r'''\begin{lilypond}[%(options)s]
 505 %(code)s
 506 \end{lilypond}''',
 507         },
 508
 509         ##
 510         TEXINFO: {
 511                 FILTER: r'''@lilypond[%(options)s]
 512 %(code)s
 513 @lilypond''',
 514
 515                 OUTPUT: r'''
 516 @iftex
 517 @include %(base)s-systems.texi
 518 @end iftex
 519 ''',
 520
 521                 OUTPUTIMAGE: r'''@noindent
 522 @ifinfo
 523 @image{%(base)s,,,%(alt)s,%(ext)s}
 524 @end ifinfo
 525 @html
 526 <p>
 527   <a href="%(base)s.ly">
 528     <img align="center" valign="center"
 529          border="0" src="%(image)s" alt="%(alt)s">
 530   </a>
 531 </p>
 532 @end html
 533 ''',
 534
 535                 PRINTFILENAME: '''@file{%(filename)s}
 536         ''',
 537
 538                 QUOTE: r'''@quotation
 539 %(str)s@end quotation
 540 ''',
 541
 542                 NOQUOTE: r'''@format
 543 %(str)s@end format
 544 ''',
 545
 546                 VERBATIM: r'''@exampleindent 0
 547 @example
 548 %(verb)s@end example
 549 ''',
 550         },
 551 }
 552
 553 PREAMBLE_LY = '''%%%% Generated by %(program_name)s
 554 %%%% Options: [%(option_string)s]
 555
 556 #(set! toplevel-score-handler print-score-with-defaults)
 557 #(set! toplevel-music-handler
 558   (lambda (p m)
 559    (if (not (eq? (ly:music-property m \'void) #t))
 560         (print-score-with-defaults
 561          p (scorify-music m p)))))
 562
 563 #(ly:set-option (quote no-point-and-click))
 564 #(define inside-lilypond-book #t)
 565 #(define version-seen? #t)
 566 %(preamble_string)s
 567
 568
 569
 570
 571
 572
 573 %% ****************************************************************
 574 %% Start cut-&-pastable-section
 575 %% ****************************************************************
 576
 577 \paper {
 578   #(define dump-extents #t)
 579   %(font_dump_setting)s
 580   %(paper_string)s
 581 }
 582
 583 \layout {
 584   %(layout_string)s
 585 }
 586 '''
 587
 588 FRAGMENT_LY = r'''
 589 %(notes_string)s
 590 {
 591
 592
 593 %% ****************************************************************
 594 %% ly snippet contents follows:
 595 %% ****************************************************************
 596 %(code)s
 597
 598
 599 %% ****************************************************************
 600 %% end ly snippet
 601 %% ****************************************************************
 602 }
 603 '''
 604
 605 FULL_LY = '''
 606
 607
 608 %% ****************************************************************
 609 %% ly snippet:
 610 %% ****************************************************************
 611 %(code)s
 612
 613
 614 %% ****************************************************************
 615 %% end ly snippet
 616 %% ****************************************************************
 617 '''
 618
 619 texinfo_line_widths = {
 620         '@afourpaper': '160\\mm',
 621         '@afourwide': '6.5\\in',
 622         '@afourlatex': '150\\mm',
 623         '@smallbook': '5\\in',
 624         '@letterpaper': '6\\in',
 625 }
 626
 627 def classic_lilypond_book_compatibility (key, value):
 628         if key == 'singleline' and value == None:
 629                 return (RAGGED_RIGHT, None)
 630
 631         m = re.search ('relative\s*([-0-9])', key)
 632         if m:
 633                 return ('relative', m.group (1))
 634
 635         m = re.match ('([0-9]+)pt', key)
 636         if m:
 637                 return ('staffsize', m.group (1))
 638
 639         if key == 'indent' or key == 'line-width':
 640                 m = re.match ('([-.0-9]+)(cm|in|mm|pt|staffspace)', value)
 641                 if m:
 642                         f = float (m.group (1))
 643                         return (key, '%f\\%s' % (f, m.group (2)))
 644
 645         return (None, None)
 646
 647 def find_file (name):
 648         for i in include_path:
 649                 full = os.path.join (i, name)
 650                 if os.path.exists (full):
 651                         return full
 652         ly.error (_ ("file not found: %s") % name + '\n')
 653         ly.exit (1)
 654         return ''
 655
 656 def verbatim_html (s):
 657         return re.sub ('>', '&gt;',
 658                        re.sub ('<', '&lt;',
 659                                re.sub ('&', '&amp;', s)))
 660
 661 def verbatim_texinfo (s):
 662         return re.sub ('{', '@{',
 663                        re.sub ('}', '@}',
 664                                re.sub ('@', '@@', s)))
 665
 666 def split_options (option_string):
 667         if option_string:
 668                 if format == HTML:
 669                         options = re.findall('[\w\.-:]+(?:\s*=\s*(?:"[^"]*"|\'[^\']*\'|\S+))?',option_string)
 670                         for i in range(len(options)):
 671                                 options[i] = re.sub('^([^=]+=\s*)(?P<q>["\'])(.*)(?P=q)','\g<1>\g<3>',options[i])
 672                         return options
 673                 else:
 674                         return re.split (format_res[format]['option_sep'],
 675                                          option_string)
 676         return []
 677
 678 def invokes_lilypond ():
 679         return re.search ('^[\'\"0-9A-Za-z/.]*lilypond', process_cmd)
 680
 681 def set_default_options (source):
 682         global default_ly_options
 683         if not default_ly_options.has_key (LINE_WIDTH):
 684                 if format == LATEX:
 685                         textwidth = get_latex_textwidth (source)
 686                         default_ly_options[LINE_WIDTH] = \
 687                           '''%.0f\\pt''' % textwidth
 688                 elif format == TEXINFO:
 689                         for (k, v) in texinfo_line_widths.items ():
 690                                 # FIXME: @layout is usually not in
 691                                 # chunk #0:
 692                                 #
 693                                 #  \input texinfo @c -*-texinfo-*-
 694                                 #
 695                                 # Bluntly search first K items of
 696                                 # source.
 697                                 # s = chunks[0].replacement_text ()
 698                                 if re.search (k, source[:1024]):
 699                                         default_ly_options[LINE_WIDTH] = v
 700                                         break
 701
 702 class Chunk:
 703         def replacement_text (self):
 704                 return ''
 705
 706         def filter_text (self):
 707                 return self.replacement_text ()
 708
 709         def ly_is_outdated (self):
 710                 return 0
 711
 712         def png_is_outdated (self):
 713                 return 0
 714
 715         def is_plain (self):
 716                 return False
 717
 718 class Substring (Chunk):
 719         def __init__ (self, source, start, end, line_number):
 720                 self.source = source
 721                 self.start = start
 722                 self.end = end
 723                 self.line_number = line_number
 724                 self.override_text = None
 725
 726         def is_plain (self):
 727                 return True
 728
 729         def replacement_text (self):
 730                 if self.override_text:
 731                         return self.override_text
 732                 else:
 733                         return self.source[self.start:self.end]
 734
 735 class Snippet (Chunk):
 736         def __init__ (self, type, match, format, line_number):
 737                 self.type = type
 738                 self.match = match
 739                 self.hash = 0
 740                 self.option_dict = {}
 741                 self.format = format
 742                 self.line_number = line_number
 743
 744         def replacement_text (self):
 745                 return self.match.group ('match')
 746
 747         def substring (self, s):
 748                 return self.match.group (s)
 749
 750         def __repr__ (self):
 751                 return `self.__class__` + ' type = ' + self.type
 752
 753 class Include_snippet (Snippet):
 754         def processed_filename (self):
 755                 f = self.substring ('filename')
 756                 return os.path.splitext (f)[0] + format2ext[format]
 757
 758         def replacement_text (self):
 759                 s = self.match.group ('match')
 760                 f = self.substring ('filename')
 761
 762                 return re.sub (f, self.processed_filename (), s)
 763
 764 class Lilypond_snippet (Snippet):
 765         def __init__ (self, type, match, format, line_number):
 766                 Snippet.__init__ (self, type, match, format, line_number)
 767                 os = match.group ('options')
 768                 self.do_options (os, self.type)
 769
 770         def ly (self):
 771                 return self.substring ('code')
 772
 773         def full_ly (self):
 774                 s = self.ly ()
 775                 if s:
 776                         return self.compose_ly (s)
 777                 return ''
 778
 779         def do_options (self, option_string, type):
 780                 self.option_dict = {}
 781
 782                 options = split_options (option_string)
 783
 784                 for i in options:
 785                         if string.find (i, '=') > 0:
 786                                 (key, value) = re.split ('\s*=\s*', i)
 787                                 self.option_dict[key] = value
 788                         else:
 789                                 if i in no_options.keys ():
 790                                         if no_options[i] in self.option_dict.keys ():
 791                                                 del self.option_dict[no_options[i]]
 792                                 else:
 793                                         self.option_dict[i] = None
 794
 795                 has_line_width = self.option_dict.has_key (LINE_WIDTH)
 796                 no_line_width_value = 0
 797
 798                 # If LINE_WIDTH is used without parameter, set it to default.
 799                 if has_line_width and self.option_dict[LINE_WIDTH] == None:
 800                         no_line_width_value = 1
 801                         del self.option_dict[LINE_WIDTH]
 802
 803                 for i in default_ly_options.keys ():
 804                         if i not in self.option_dict.keys ():
 805                                 self.option_dict[i] = default_ly_options[i]
 806
 807                 if not has_line_width:
 808                         if type == 'lilypond' or FRAGMENT in self.option_dict.keys ():
 809                                 self.option_dict[RAGGED_RIGHT] = None
 810
 811                         if type == 'lilypond':
 812                                 if LINE_WIDTH in self.option_dict.keys ():
 813                                         del self.option_dict[LINE_WIDTH]
 814                         else:
 815                                 if RAGGED_RIGHT in self.option_dict.keys ():
 816                                         if LINE_WIDTH in self.option_dict.keys ():
 817                                                 del self.option_dict[LINE_WIDTH]
 818
 819                         if QUOTE in self.option_dict.keys () or type == 'lilypond':
 820                                 if LINE_WIDTH in self.option_dict.keys ():
 821                                         del self.option_dict[LINE_WIDTH]
 822
 823                 if not INDENT in self.option_dict.keys ():
 824                         self.option_dict[INDENT] = '0\\mm'
 825
 826                 # The QUOTE pattern from ly_options only emits the `line-width'
 827                 # keyword.
 828                 if has_line_width and QUOTE in self.option_dict.keys ():
 829                         if no_line_width_value:
 830                                 del self.option_dict[LINE_WIDTH]
 831                         else:
 832                                 del self.option_dict[QUOTE]
 833
 834         def compose_ly (self, code):
 835                 if FRAGMENT in self.option_dict.keys ():
 836                         body = FRAGMENT_LY
 837                 else:
 838                         body = FULL_LY
 839
 840                 # Defaults.
 841                 relative = 1
 842                 override = {}
 843                 # The original concept of the `exampleindent' option is broken.
 844                 # It is not possible to get a sane value for @exampleindent at all
 845                 # without processing the document itself.  Saying
 846                 #
 847                 #   @exampleindent 0
 848                 #   @example
 849                 #   ...
 850                 #   @end example
 851                 #   @exampleindent 5
 852                 #
 853                 # causes ugly results with the DVI backend of texinfo since the
 854                 # default value for @exampleindent isn't 5em but 0.4in (or a smaller
 855                 # value).  Executing the above code changes the environment
 856                 # indentation to an unknown value because we don't know the amount
 857                 # of 1em in advance since it is font-dependent.  Modifying
 858                 # @exampleindent in the middle of a document is simply not
 859                 # supported within texinfo.
 860                 #
 861                 # As a consequence, the only function of @exampleindent is now to
 862                 # specify the amount of indentation for the `quote' option.
 863                 #
 864                 # To set @exampleindent locally to zero, we use the @format
 865                 # environment for non-quoted snippets.
 866                 override[EXAMPLEINDENT] = r'0.4\in'
 867                 override[LINE_WIDTH] = texinfo_line_widths['@smallbook']
 868                 override.update (default_ly_options)
 869
 870                 option_list = []
 871                 for (key, value) in self.option_dict.items ():
 872                         if value == None:
 873                                 option_list.append (key)
 874                         else:
 875                                 option_list.append (key + '=' + value)
 876                 option_string = string.join (option_list, ',')
 877
 878                 compose_dict = {}
 879                 compose_types = [NOTES, PREAMBLE, LAYOUT, PAPER]
 880                 for a in compose_types:
 881                         compose_dict[a] = []
 882
 883                 for (key, value) in self.option_dict.items ():
 884                         (c_key, c_value) = \
 885                           classic_lilypond_book_compatibility (key, value)
 886                         if c_key:
 887                                 if c_value:
 888                                         ly.warning \
 889                                           (_ ("deprecated ly-option used: %s=%s" \
 890                                             % (key, value)))
 891                                         ly.warning \
 892                                           (_ ("compatibility mode translation: %s=%s" \
 893                                             % (c_key, c_value)))
 894                                 else:
 895                                         ly.warning \
 896                                           (_ ("deprecated ly-option used: %s" \
 897                                             % key))
 898                                         ly.warning \
 899                                           (_ ("compatibility mode translation: %s" \
 900                                             % c_key))
 901
 902                                 (key, value) = (c_key, c_value)
 903
 904                         if value:
 905                                 override[key] = value
 906                         else:
 907                                 if not override.has_key (key):
 908                                         override[key] = None
 909
 910                         found = 0
 911                         for type in compose_types:
 912                                 if ly_options[type].has_key (key):
 913                                         compose_dict[type].append (ly_options[type][key])
 914                                         found = 1
 915                                         break
 916
 917                         if not found and key not in simple_options:
 918                                 ly.warning (_ ("ignoring unknown ly option: %s") % key)
 919
 920                 # URGS
 921                 if RELATIVE in override.keys () and override[RELATIVE]:
 922                         relative = string.atoi (override[RELATIVE])
 923
 924                 relative_quotes = ''
 925
 926                 # 1 = central C
 927                 if relative < 0:
 928                         relative_quotes += ',' * (- relative)
 929                 elif relative > 0:
 930                         relative_quotes += "'" * relative
 931
 932                 program_name = __main__.program_name
 933
 934                 paper_string = string.join (compose_dict[PAPER],
 935                                             '\n  ') % override
 936                 layout_string = string.join (compose_dict[LAYOUT],
 937                                              '\n  ') % override
 938                 notes_string = string.join (compose_dict[NOTES],
 939                                             '\n  ') % vars ()
 940                 preamble_string = string.join (compose_dict[PREAMBLE],
 941                                                '\n  ') % override
 942
 943                 font_dump_setting = ''
 944                 if FONTLOAD in self.option_dict:
 945                         font_dump_setting = '#(define-public force-eps-font-include #t)\n'
 946
 947                 return (PREAMBLE_LY + body) % vars ()
 948
 949         # TODO: Use md5?
 950         def get_hash (self):
 951                 if not self.hash:
 952                         self.hash = abs (hash (self.full_ly ()))
 953                 return self.hash
 954
 955         def basename (self):
 956                 if FILENAME in self.option_dict:
 957                         return self.option_dict[FILENAME]
 958                 if use_hash_p:
 959                         return 'lily-%d' % self.get_hash ()
 960                 raise 'to be done'
 961
 962         def write_ly (self):
 963                 outf = open (self.basename () + '.ly', 'w')
 964                 outf.write (self.full_ly ())
 965
 966                 open (self.basename () + '.txt', 'w').write ('image of music')
 967
 968         def ly_is_outdated (self):
 969                 base = self.basename ()
 970
 971                 tex_file = '%s.tex' % base
 972                 eps_file = '%s.eps' % base
 973                 system_file = '%s-systems.tex' % base
 974                 ly_file = '%s.ly' % base
 975                 ok = os.path.exists (ly_file) \
 976                      and os.path.exists (system_file)\
 977                      and os.stat (system_file)[stat.ST_SIZE] \
 978                      and re.match ('% eof', open (system_file).readlines ()[-1])
 979                 if ok and (not use_hash_p or FILENAME in self.option_dict):
 980                         ok = (self.full_ly () == open (ly_file).read ())
 981                 if ok:
 982                         # TODO: Do something smart with target formats
 983                         #       (ps, png) and m/ctimes.
 984                         return None
 985                 return self
 986
 987         def png_is_outdated (self):
 988                 base = self.basename ()
 989                 ok = self.ly_is_outdated ()
 990                 if format == HTML or format == TEXINFO:
 991                         ok = ok and os.path.exists (base + '.eps')
 992
 993                         page_count = 0
 994                         if ok:
 995                                 page_count = ly.ps_page_count (base + '.eps')
 996
 997                         if page_count == 1:
 998                                 ok = ok and os.path.exists (base + '.png')
 999                         elif page_count > 1:
1000                                 for a in range (1, page_count + 1):
1001                                                 ok = ok and os.path.exists (base + '-page%d.png' % a)
1002
1003                 return not ok
1004
1005         def texstr_is_outdated (self):
1006                 if backend == 'ps':
1007                         return 0
1008
1009                 base = self.basename ()
1010                 ok = self.ly_is_outdated ()
1011                 ok = ok and (os.path.exists (base + '.texstr'))
1012                 return not ok
1013
1014         def filter_text (self):
1015                 code = self.substring ('code')
1016                 s = run_filter (code)
1017                 d = {
1018                         'code': s,
1019                         'options': self.match.group ('options')
1020                 }
1021                 # TODO
1022                 return output[self.format][FILTER] % d
1023
1024         def replacement_text (self):
1025                 func = Lilypond_snippet.__dict__['output_' + self.format]
1026                 return func (self)
1027
1028         def get_images (self):
1029                 base = self.basename ()
1030                 # URGUGHUGHUGUGH
1031                 single = '%(base)s.png' % vars ()
1032                 multiple = '%(base)s-page1.png' % vars ()
1033                 images = (single,)
1034                 if os.path.exists (multiple) \
1035                    and (not os.path.exists (single) \
1036                         or (os.stat (multiple)[stat.ST_MTIME] \
1037                             > os.stat (single)[stat.ST_MTIME])):
1038                         count = ly.ps_page_count ('%(base)s.eps' % vars ())
1039                         images = ['%s-page%d.png' % (base, a) for a in range (1, count+1)]
1040                         images = tuple (images)
1041                 return images
1042
1043         def output_html (self):
1044                 str = ''
1045                 base = self.basename ()
1046                 if format == HTML:
1047                         str += self.output_print_filename (HTML)
1048                         if VERBATIM in self.option_dict:
1049                                 verb = verbatim_html (self.substring ('code'))
1050                                 str += write (output[HTML][VERBATIM] % vars ())
1051                         if QUOTE in self.option_dict:
1052                                 str = output[HTML][QUOTE] % vars ()
1053
1054                 str += output[HTML][BEFORE] % vars ()
1055                 for image in self.get_images ():
1056                         (base, ext) = os.path.splitext (image)
1057                         alt = self.option_dict[ALT]
1058                         str += output[HTML][OUTPUT] % vars ()
1059                 str += output[HTML][AFTER] % vars ()
1060                 return str
1061
1062         def output_info (self):
1063                 str = ''
1064                 for image in self.get_images ():
1065                         (base, ext) = os.path.splitext (image)
1066
1067                         # URG, makeinfo implicitly prepends dot to extension.
1068                         # Specifying no extension is most robust.
1069                         ext = ''
1070                         alt = self.option_dict[ALT]
1071                         str += output[TEXINFO][OUTPUTIMAGE] % vars ()
1072
1073                 base = self.basename ()
1074                 str += output[format][OUTPUT] % vars ()
1075                 return str
1076
1077         def output_latex (self):
1078                 str = ''
1079                 base = self.basename ()
1080                 if format == LATEX:
1081                         str += self.output_print_filename (LATEX)
1082                         if VERBATIM in self.option_dict:
1083                                 verb = self.substring ('code')
1084                                 str += (output[LATEX][VERBATIM] % vars ())
1085                         elif srcltx_p:
1086                                 breaks = self.ly ().count ("\n")
1087                                 str += "".ljust (breaks, "\n").replace ("\n","%\n")
1088                 str += (output[LATEX][OUTPUT] % vars ())
1089                 if QUOTE in self.option_dict:
1090                         str = output[LATEX][QUOTE] % vars ()
1091                 return str
1092
1093         def output_print_filename (self, format):
1094                 str = ''
1095                 if PRINTFILENAME in self.option_dict:
1096                         base = self.basename ()
1097                         filename = self.substring ('filename')
1098                         str = output[format][PRINTFILENAME] % vars ()
1099
1100                 return str
1101
1102         def output_texinfo (self):
1103                 str = ''
1104                 if self.output_print_filename (TEXINFO):
1105                         str += ('@html\n'
1106                                 + self.output_print_filename (HTML)
1107                                 + '\n@end html\n')
1108                         str += ('@tex\n'
1109                                 + self.output_print_filename (LATEX)
1110                                 + '\n@end tex\n')
1111                 base = self.basename ()
1112                 if TEXIDOC in self.option_dict:
1113                         texidoc = base + '.texidoc'
1114                         if os.path.exists (texidoc):
1115                                 str += '@include %(texidoc)s\n\n' % vars ()
1116
1117                 if VERBATIM in self.option_dict:
1118                         verb = verbatim_texinfo (self.substring ('code'))
1119                         str += (output[TEXINFO][VERBATIM] % vars ())
1120                         if not QUOTE in self.option_dict:
1121                                 str = output[TEXINFO][NOQUOTE] % vars ()
1122
1123                 str += self.output_info ()
1124
1125 #               str += ('@ifinfo\n' + self.output_info () + '\n@end ifinfo\n')
1126 #               str += ('@tex\n' + self.output_latex () + '\n@end tex\n')
1127 #               str += ('@html\n' + self.output_html () + '\n@end html\n')
1128
1129                 if QUOTE in self.option_dict:
1130                         str = output[TEXINFO][QUOTE] % vars ()
1131
1132                 # need par after image
1133                 str += '\n'
1134
1135                 return str
1136
1137 class Lilypond_file_snippet (Lilypond_snippet):
1138         def ly (self):
1139                 name = self.substring ('filename')
1140                 return '\\sourcefilename \"%s\"\n%s' \
1141                          % (name, open (find_file (name)).read ())
1142
1143 snippet_type_to_class = {
1144         'lilypond_file': Lilypond_file_snippet,
1145         'lilypond_block': Lilypond_snippet,
1146         'lilypond': Lilypond_snippet,
1147         'include': Include_snippet,
1148 }
1149
1150 def find_linestarts (s):
1151         nls = [0]
1152         start = 0
1153         end = len (s)
1154         while 1:
1155                 i = s.find ('\n', start)
1156                 if i < 0:
1157                         break
1158
1159                 i = i + 1
1160                 nls.append (i)
1161                 start = i
1162
1163         nls.append (len (s))
1164         return nls
1165
1166 def find_toplevel_snippets (s, types):
1167         res = {}
1168         for i in types:
1169                 res[i] = ly.re.compile (snippet_res[format][i])
1170
1171         snippets = []
1172         index = 0
1173         ## found = dict (map (lambda x: (x, None),
1174         ##                    types))
1175         ## urg python2.1
1176         found = {}
1177         map (lambda x, f = found: f.setdefault (x, None),
1178              types)
1179
1180         line_starts = find_linestarts (s)
1181         line_start_idx = 0
1182         # We want to search for multiple regexes, without searching
1183         # the string multiple times for one regex.
1184         # Hence, we use earlier results to limit the string portion
1185         # where we search.
1186         # Since every part of the string is traversed at most once for
1187         # every type of snippet, this is linear.
1188
1189         while 1:
1190                 first = None
1191                 endex = 1 << 30
1192                 for type in types:
1193                         if not found[type] or found[type][0] < index:
1194                                 found[type] = None
1195                                 m = res[type].search (s[index:endex])
1196                                 if not m:
1197                                         continue
1198
1199                                 cl = Snippet
1200                                 if snippet_type_to_class.has_key (type):
1201                                         cl = snippet_type_to_class[type]
1202
1203
1204                                 start = index + m.start ('match')
1205                                 line_number = line_start_idx
1206                                 while (line_starts[line_number] < start):
1207                                         line_number += 1
1208
1209                                 line_number += 1
1210                                 snip = cl (type, m, format, line_number)
1211                                 found[type] = (start, snip)
1212
1213                         if found[type] \
1214                            and (not first \
1215                                 or found[type][0] < found[first][0]):
1216                                 first = type
1217
1218                                 # FIXME.
1219
1220                                 # Limiting the search space is a cute
1221                                 # idea, but this *requires* to search
1222                                 # for possible containing blocks
1223                                 # first, at least as long as we do not
1224                                 # search for the start of blocks, but
1225                                 # always/directly for the entire
1226                                 # @block ... @end block.
1227
1228                                 endex = found[first][0]
1229
1230                 if not first:
1231                         snippets.append (Substring (s, index, len (s), line_start_idx))
1232                         break
1233
1234                 while (start > line_starts[line_start_idx+1]):
1235                         line_start_idx += 1
1236
1237                 (start, snip) = found[first]
1238                 snippets.append (Substring (s, index, start, line_start_idx + 1))
1239                 snippets.append (snip)
1240                 found[first] = None
1241                 index = start + len (snip.match.group ('match'))
1242
1243         return snippets
1244
1245 def filter_pipe (input, cmd):
1246         if verbose_p:
1247                 ly.progress (_ ("Opening filter `%s'") % cmd)
1248
1249         (stdin, stdout, stderr) = os.popen3 (cmd)
1250         stdin.write (input)
1251         status = stdin.close ()
1252
1253         if not status:
1254                 status = 0
1255                 output = stdout.read ()
1256                 status = stdout.close ()
1257                 error = stderr.read ()
1258
1259         if not status:
1260                 status = 0
1261         signal = 0x0f & status
1262         if status or (not output and error):
1263                 exit_status = status >> 8
1264                 ly.error (_ ("`%s' failed (%d)") % (cmd, exit_status))
1265                 ly.error (_ ("The error log is as follows:"))
1266                 sys.stderr.write (error)
1267                 sys.stderr.write (stderr.read ())
1268                 ly.exit (status)
1269
1270         if verbose_p:
1271                 ly.progress ('\n')
1272
1273         return output
1274
1275 def run_filter (s):
1276         return filter_pipe (s, filter_cmd)
1277
1278 def is_derived_class (cl, baseclass):
1279         if cl == baseclass:
1280                 return 1
1281         for b in cl.__bases__:
1282                 if is_derived_class (b, baseclass):
1283                         return 1
1284         return 0
1285
1286 def process_snippets (cmd, ly_snippets, texstr_snippets, png_snippets):
1287         ly_names = filter (lambda x: x,
1288                            map (Lilypond_snippet.basename, ly_snippets))
1289         texstr_names = filter (lambda x: x,
1290                            map (Lilypond_snippet.basename, texstr_snippets))
1291         png_names = filter (lambda x: x,
1292                             map (Lilypond_snippet.basename, png_snippets))
1293
1294         status = 0
1295         def my_system (cmd):
1296                 status = ly.system (cmd,
1297                                     ignore_error = 1, progress_p = 1)
1298
1299                 if status:
1300                         ly.error ('Process %s exited unsuccessfully.' % cmd)
1301                         raise Compile_error
1302
1303         # UGH
1304         # the --process=CMD switch is a bad idea
1305         # it is too generic for lilypond-book.
1306         if texstr_names and invokes_lilypond ():
1307                 my_system (string.join ([cmd, '--backend texstr',
1308                                          'snippet-map.ly'] + texstr_names))
1309                 for l in texstr_names:
1310                         my_system ('latex %s.texstr' % l)
1311
1312         if ly_names:
1313                 my_system (string.join ([cmd, 'snippet-map.ly'] + ly_names))
1314
1315 LATEX_INSPECTION_DOCUMENT = r'''
1316 \nonstopmode
1317 %(preamble)s
1318 \begin{document}
1319 \typeout{textwidth=\the\textwidth}
1320 \typeout{columnsep=\the\columnsep}
1321 \makeatletter\if@twocolumn\typeout{columns=2}\fi\makeatother
1322 \end{document}
1323 '''
1324
1325 # Do we need anything else besides `textwidth'?
1326 def get_latex_textwidth (source):
1327         m = re.search (r'''(?P<preamble>\\begin\s*{document})''', source)
1328         preamble = source[:m.start (0)]
1329         latex_document = LATEX_INSPECTION_DOCUMENT % vars ()
1330         # Workaround problems with unusable $TMP on Cygwin:
1331         tempfile.tempdir = ''
1332         tmpfile = tempfile.mktemp('.tex')
1333         logfile = os.path.splitext (tmpfile)[0] + '.log'
1334         open (tmpfile,'w').write (latex_document)
1335         ly.system ('latex %s' % tmpfile)
1336         parameter_string = open (logfile).read()
1337         os.unlink (tmpfile)
1338         os.unlink (logfile)
1339
1340         columns = 0
1341         m = re.search ('columns=([0-9.]*)', parameter_string)
1342         if m:
1343                 columns = string.atoi (m.group (1))
1344
1345         columnsep = 0
1346         m = re.search ('columnsep=([0-9.]*)pt', parameter_string)
1347         if m:
1348                 columnsep = string.atof (m.group (1))
1349
1350         textwidth = 0
1351         m = re.search ('textwidth=([0-9.]*)pt', parameter_string)
1352         if m:
1353                 textwidth = string.atof (m.group (1))
1354                 if columns:
1355                         textwidth = (textwidth - columnsep) / columns
1356
1357         return textwidth
1358
1359 def modify_preamble (chunk):
1360         str = chunk.replacement_text ()
1361         if (re.search (r"\\begin *{document}", str)
1362             and not re.search ("{graphic[sx]", str)):
1363                 str = re.sub (r"\\begin{document}",
1364                               r"\\usepackage{graphics}" + '\n'
1365                               + r"\\begin{document}",
1366                               str)
1367                 chunk.override_text = str
1368
1369
1370
1371 ext2format = {
1372         '.html': HTML,
1373         '.itely': TEXINFO,
1374         '.latex': LATEX,
1375         '.lytex': LATEX,
1376         '.tely': TEXINFO,
1377         '.tex': LATEX,
1378         '.texi': TEXINFO,
1379         '.texinfo': TEXINFO,
1380         '.xml': HTML,
1381 }
1382
1383 format2ext = {
1384         HTML: '.html',
1385         # TEXINFO: '.texinfo',
1386         TEXINFO: '.texi',
1387         LATEX: '.tex',
1388 }
1389
1390 class Compile_error:
1391         pass
1392
1393 def write_file_map (lys, name):
1394         snippet_map = open ('snippet-map.ly', 'w')
1395         snippet_map.write ("""
1396 #(define version-seen? #t)
1397 #(ly:add-file-name-alist '(
1398 """)
1399         for ly in lys:
1400                 snippet_map.write ('("%s.ly" . "%s:%d (%s.ly)")\n'
1401                                    % (ly.basename (),
1402                                       name,
1403                                       ly.line_number,
1404                                       ly.basename ()))
1405
1406         snippet_map.write ('))\n')
1407
1408 def do_process_cmd (chunks, input_name):
1409         all_lys = filter (lambda x: is_derived_class (x.__class__,
1410                                                       Lilypond_snippet),
1411                           chunks)
1412
1413         write_file_map (all_lys, input_name)
1414         ly_outdated = \
1415           filter (lambda x: is_derived_class (x.__class__,
1416                                               Lilypond_snippet)
1417                             and x.ly_is_outdated (),
1418                   chunks)
1419         texstr_outdated = \
1420           filter (lambda x: is_derived_class (x.__class__,
1421                                               Lilypond_snippet)
1422                             and x.texstr_is_outdated (),
1423                   chunks)
1424         png_outdated = \
1425           filter (lambda x: is_derived_class (x.__class__,
1426                                               Lilypond_snippet)
1427                             and x.png_is_outdated (),
1428                   chunks)
1429
1430         ly.progress (_ ("Writing snippets..."))
1431         map (Lilypond_snippet.write_ly, ly_outdated)
1432         ly.progress ('\n')
1433
1434         if ly_outdated:
1435                 ly.progress (_ ("Processing..."))
1436                 ly.progress ('\n')
1437                 process_snippets (process_cmd, ly_outdated, texstr_outdated, png_outdated)
1438         else:
1439                 ly.progress (_ ("All snippets are up to date..."))
1440         ly.progress ('\n')
1441
1442 def guess_format (input_filename):
1443         format = None
1444         e = os.path.splitext (input_filename)[1]
1445         if e in ext2format.keys ():
1446                 # FIXME
1447                 format = ext2format[e]
1448         else:
1449                 ly.error (_ ("can't determine format for: %s" \
1450                              % input_filename))
1451                 ly.exit (1)
1452         return format
1453
1454 def write_if_updated (file_name, lines):
1455         try:
1456                 f = open (file_name)
1457                 oldstr = f.read ()
1458                 new_str = string.join (lines, '')
1459                 if oldstr == new_str:
1460                         ly.progress (_ ("%s is up to date.") % file_name)
1461                         ly.progress ('\n')
1462                         return
1463         except:
1464                 pass
1465
1466         ly.progress (_ ("Writing `%s'...") % file_name)
1467         open (file_name, 'w').writelines (lines)
1468         ly.progress ('\n')
1469
1470 def do_file (input_filename):
1471         # Ugh.
1472         if not input_filename or input_filename == '-':
1473                 in_handle = sys.stdin
1474                 input_fullname = '<stdin>'
1475         else:
1476                 if os.path.exists (input_filename):
1477                         input_fullname = input_filename
1478                 elif format == LATEX and ly.search_exe_path ('kpsewhich'):
1479                         # urg python interface to libkpathsea?
1480                         input_fullname = ly.read_pipe ('kpsewhich '
1481                                                        + input_filename)[:-1]
1482                 else:
1483                         input_fullname = find_file (input_filename)
1484                 in_handle = open (input_fullname)
1485
1486         if input_filename == '-':
1487                 input_base = 'stdin'
1488         else:
1489                 input_base = os.path.basename \
1490                              (os.path.splitext (input_filename)[0])
1491
1492         # Only default to stdout when filtering.
1493         if output_name == '-' or (not output_name and filter_cmd):
1494                 output_filename = '-'
1495                 output_file = sys.stdout
1496         else:
1497                 # don't complain when output_name is existing
1498                 output_filename = input_base + format2ext[format]
1499                 if output_name:
1500                         if not os.path.isdir (output_name):
1501                                 os.mkdir (output_name, 0777)
1502                         os.chdir (output_name)
1503                 else:
1504                         if os.path.exists (input_filename) \
1505                            and os.path.exists (output_filename) \
1506                            and os.path.samefile (output_filename, input_fullname):
1507                            ly.error (
1508                            _ ("Output would overwrite input file; use --output."))
1509                            ly.exit (2)
1510
1511         try:
1512                 ly.progress (_ ("Reading %s...") % input_fullname)
1513                 source = in_handle.read ()
1514                 ly.progress ('\n')
1515
1516                 set_default_options (source)
1517
1518
1519                 # FIXME: Containing blocks must be first, see
1520                 #        find_toplevel_snippets.
1521                 snippet_types = (
1522                         'multiline_comment',
1523                         'verbatim',
1524                         'lilypond_block',
1525         #               'verb',
1526                         'singleline_comment',
1527                         'lilypond_file',
1528                         'include',
1529                         'lilypond',
1530                 )
1531                 ly.progress (_ ("Dissecting..."))
1532                 chunks = find_toplevel_snippets (source, snippet_types)
1533
1534                 if format == LATEX:
1535                         for c in chunks:
1536                                 if (c.is_plain () and
1537                                     re.search (r"\\begin *{document}", c.replacement_text())):
1538                                         modify_preamble (c)
1539                                         break
1540                 ly.progress ('\n')
1541
1542                 if filter_cmd:
1543                         write_if_updated (output_filename,
1544                                           [c.filter_text () for c in chunks])
1545                 elif process_cmd:
1546                         do_process_cmd (chunks, input_fullname)
1547                         ly.progress (_ ("Compiling %s...") % output_filename)
1548                         ly.progress ('\n')
1549                         write_if_updated (output_filename,
1550                                           [s.replacement_text ()
1551                                            for s in chunks])
1552
1553                 def process_include (snippet):
1554                         os.chdir (original_dir)
1555                         name = snippet.substring ('filename')
1556                         ly.progress (_ ("Processing include: %s") % name)
1557                         ly.progress ('\n')
1558                         return do_file (name)
1559
1560                 include_chunks = map (process_include,
1561                                       filter (lambda x: is_derived_class (x.__class__,
1562                                                                           Include_snippet),
1563                                               chunks))
1564
1565
1566                 return chunks + reduce (lambda x,y: x + y, include_chunks, [])
1567
1568         except Compile_error:
1569                 os.chdir (original_dir)
1570                 ly.progress (_ ("Removing `%s'") % output_filename)
1571                 ly.progress ('\n')
1572                 raise Compile_error
1573
1574 def do_options ():
1575         global format, output_name, psfonts_p, srcltx_p
1576         global filter_cmd, process_cmd, verbose_p
1577
1578         (sh, long) = ly.getopt_args (option_definitions)
1579         try:
1580                 (options, files) = getopt.getopt (sys.argv[1:], sh, long)
1581         except getopt.error, s:
1582                 sys.stderr.write ('\n')
1583                 ly.error (_ ("getopt says: `%s'" % s))
1584                 sys.stderr.write ('\n')
1585                 ly.help ()
1586                 ly.exit (2)
1587
1588         for opt in options:
1589                 o = opt[0]
1590                 a = opt[1]
1591
1592                 if 0:
1593                         pass
1594                 elif o == '--filter' or o == '-F':
1595                         filter_cmd = a
1596                         process_cmd = 0
1597                 elif o == '--format' or o == '-f':
1598                         format = a
1599                         if a == 'texi-html' or a == 'texi':
1600                                 format = TEXINFO
1601                 elif o == '--tex-backend ':
1602                         backend = 'tex'
1603                 elif o == '--help' or o == '-h':
1604                         ly.help ()
1605                         sys.exit (0)
1606                 elif o == '--include' or o == '-I':
1607                         include_path.append (os.path.join (original_dir,
1608                                                            os.path.abspath (a)))
1609                 elif o == '--output' or o == '-o':
1610                         output_name = a
1611                 elif o == '--process' or o == '-P':
1612                         process_cmd = a
1613                         filter_cmd = 0
1614                 elif o == '--version' or o == '-v':
1615                         ly.identify (sys.stdout)
1616                         sys.exit (0)
1617                 elif o == '--verbose' or o == '-V':
1618                         verbose_p = 1
1619                 elif o == '--psfonts':
1620                         psfonts_p = 1
1621                 elif o == '--keep-line-breaks':
1622                         srcltx_p = 1
1623                         for s in (OUTPUT, QUOTE, VERBATIM):
1624                                 output[LATEX][s] = output[LATEX][s].replace("\n"," ")
1625                 elif o == '--warranty' or o == '-w':
1626                         if 1 or status:
1627                                 ly.warranty ()
1628                         sys.exit (0)
1629         return files
1630
1631 def main ():
1632         files = do_options ()
1633         if not files or len (files) > 1:
1634                 ly.help ()
1635                 ly.exit (2)
1636
1637         file = files[0]
1638
1639         basename = os.path.splitext (file)[0]
1640         basename = os.path.split (basename)[1]
1641
1642         global process_cmd, format
1643         if not format:
1644                 format = guess_format (files[0])
1645
1646         formats = 'ps'
1647         if format == TEXINFO or format == HTML:
1648                 formats += ',png'
1649         if process_cmd == '':
1650                 process_cmd = lilypond_binary \
1651                               + ' --formats=%s --backend eps ' % formats
1652
1653         if process_cmd:
1654                 process_cmd += string.join ([(' -I %s' % commands.mkarg (p))
1655                                              for p in include_path])
1656
1657         ly.identify (sys.stderr)
1658
1659         try:
1660                 chunks = do_file (file)
1661                 if psfonts_p and invokes_lilypond ():
1662                         fontextract.verbose = verbose_p
1663                         snippet_chunks = filter (lambda x: is_derived_class (x.__class__,
1664                                                                               Lilypond_snippet),
1665                                                  chunks)
1666
1667                         psfonts_file = basename + '.psfonts'
1668                         if not verbose_p:
1669                                 ly.progress (_ ("Writing fonts to %s...") % psfonts_file)
1670                         fontextract.extract_fonts (psfonts_file,
1671                                                    [x.basename() + '.eps'
1672                                                     for x in snippet_chunks])
1673                         if not verbose_p:
1674                                 ly.progress ('\n')
1675
1676         except Compile_error:
1677                 ly.exit (1)
1678
1679         if format == TEXINFO or format == LATEX:
1680                 if not psfonts_p:
1681                         ly.warning (_ ("option --psfonts not used"))
1682                         ly.warning (_ ("processing with dvips will have no fonts"))
1683
1684                 psfonts_file = os.path.join (output_name, basename + '.psfonts')
1685                 output = os.path.join (output_name, basename +  '.dvi' )
1686
1687                 ly.progress ('\n')
1688                 ly.progress (_ ("DVIPS usage:"))
1689                 ly.progress ('\n')
1690                 ly.progress ("    dvips -h %(psfonts_file)s %(output)s" % vars ())
1691                 ly.progress ('\n')
1692
1693 if __name__ == '__main__':
1694         main ()