scripts/lilypond-book.py

   1 #!@PYTHON@
   2
   3 '''
   4 TODO:
   5       ly-options: intertext, quote ?
   6       --linewidth?
   7       eps in latex?
   8       check latex parameters, twocolumn
   9       multicolumn?
  10       papersizes?
  11       ly2dvi/notexidoc?
  12
  13 Example usage:
  14
  15 test:
  16      lilypond-book --filter="tr '[a-z]' '[A-Z]'" BOOK
  17
  18 convert-ly on book:
  19      lilypond-book --filter="convert-ly --no-version --from=1.6.11 -" BOOK
  20
  21 classic lilypond-book:
  22      lilypond-book --process="lilypond-bin" BOOK.tely
  23
  24    must substitute:
  25      @mbinclude foo.itely -> @include foo.itely
  26      \mbinput -> \input
  27
  28 '''
  29
  30 import string
  31 import __main__
  32
  33 ################################################################
  34 # Users of python modules should include this snippet
  35 # and customize variables below.
  36
  37 # We'll suffer this path init stuff as long as we don't install our
  38 # python packages in <prefix>/lib/pythonx.y (and don't kludge around
  39 # it as we do with teTeX on Red Hat Linux: set some environment var
  40 # (PYTHONPATH) in profile)
  41
  42 # If set, LILYPONDPREFIX must take prevalence
  43 # if datadir is not set, we're doing a build and LILYPONDPREFIX
  44 import getopt, os, sys
  45 datadir = '@local_lilypond_datadir@'
  46 if not os.path.isdir (datadir):
  47         datadir = '@lilypond_datadir@'
  48 if os.environ.has_key ('LILYPONDPREFIX') :
  49         datadir = os.environ['LILYPONDPREFIX']
  50         while datadir[-1] == os.sep:
  51                 datadir= datadir[:-1]
  52
  53 sys.path.insert (0, os.path.join (datadir, 'python'))
  54
  55 # Customize these
  56 #if __name__ == '__main__':
  57
  58 import lilylib as ly
  59 global _;_=ly._
  60 global re;re = ly.re
  61
  62
  63 # lilylib globals
  64 program_version = '@TOPLEVEL_VERSION@'
  65 program_name = 'lilypond-book'
  66 verbose_p = 0
  67 pseudo_filter_p = 0
  68 original_dir = os.getcwd ()
  69
  70
  71 help_summary = _ ("""Process LilyPond snippets in hybrid HTML, LaTeX or texinfo document.  Example usage:
  72
  73    lilypond-book --filter="tr '[a-z]' '[A-Z]'" BOOK
  74    lilypond-book --filter="convert-ly --no-version --from=2.0.0 -" BOOK
  75    lilypond-book --process='lilypond-bin -I include' BOOK
  76
  77 """)
  78
  79 copyright = ('Jan Nieuwenhuizen <janneke@gnu.org>>',
  80              'Han-Wen Nienhuys <hanwen@cs.uu.nl>')
  81
  82 option_definitions = [
  83         (_ ("EXT"), 'f', 'format', _ ("use output format EXT (texi [default], texi-html, latex, html)")),
  84         (_ ("FILTER"), 'F', 'filter', _ ("pipe snippets through FILTER [convert-ly -n -]")),
  85         ('', 'h', 'help', _ ("print this help")),
  86         (_ ("DIR"), 'I', 'include', _ ("add DIR to include path")),
  87         (_ ("COMMAND"), 'P', 'process', _ ("process ly_files using COMMAND FILE...")),
  88         (_ ("DIR"), 'o', 'output', _ ("write output to DIR")),
  89         ('', 'V', 'verbose', _ ("be verbose")),
  90         ('', 'v', 'version', _ ("print version information")),
  91         ('', 'w', 'warranty', _ ("show warranty and copyright")),
  92         ]
  93
  94 include_path = [ly.abspath (os.getcwd ())]
  95 lilypond_binary = os.path.join ('@bindir@', 'lilypond-bin')
  96
  97 # only use installed binary  when we're installed too.
  98 if '@bindir@' == ('@' + 'bindir@') or not os.path.exists (lilypond_binary):
  99         lilypond_binary = 'lilypond-bin'
 100
 101
 102 use_hash_p = 1
 103 format = 0
 104 output_name = 0
 105 latex_filter_cmd = 'latex "\\nonstopmode \input /dev/stdin"'
 106 filter_cmd = 0
 107 process_cmd = lilypond_binary
 108 default_ly_options = { }
 109
 110 AFTER = 'after'
 111 BEFORE = 'before'
 112 HTML = 'html'
 113 LATEX = 'latex'
 114 LINEWIDTH = 'linewidth'
 115 NOTES = 'body'
 116 OUTPUT = 'output'
 117 PAPER = 'paper'
 118 PREAMBLE = 'preamble'
 119 TEXINFO = 'texinfo'
 120 VERBATIM = 'verbatim'
 121
 122
 123 # Recognize special sequences in the input
 124 #
 125 # (?P<name>regex) -- assign result of REGEX to NAME
 126 # *? -- match non-greedily.
 127 # (?m) -- multiline regex: make ^ and $ match at each line
 128 # (?s) -- make the dot match all characters including newline
 129 no_match = 'a\ba'
 130 snippet_res = {
 131         HTML: {
 132         'include':  no_match,
 133         'lilypond' : '(?m)(?P<match><lilypond((?P<options>[^:]*):)(?P<code>.*?)/>)',
 134         'lilypond_block': r'''(?ms)(?P<match><lilypond(?P<options>[^>]+)?>(?P<code>.*?)</lilypond>)''',
 135         'lilypond_file': r'(?m)(?P<match><lilypondfile(?P<options>[^>]+)?>\s*(?P<filename>[^<]+)\s*</lilypondfile>)',
 136         'multiline_comment': r"(?sm)\s*(?!@c\s+)(?P<code><!--\s.*?!-->)\s",
 137         'singleline_comment': no_match,
 138         'verb': r'''(?P<code><pre>.*?</pre>)''',
 139         'verbatim': r'''(?s)(?P<code><pre>\s.*?</pre>\s)''',
 140         },
 141
 142         LATEX: {
 143         'include': r'(?m)^[^%\n]*?(?P<match>\\input{(?P<filename>[^}]+)})',
 144         'lilypond' : r'(?m)^[^%\n]*?(?P<match>\\lilypond\s*(\[(?P<options>.*?)\])?\s*{(?P<code>.*?)})',
 145         'lilypond_block': r"(?sm)^[^%\n]*?(?P<match>\\begin\s*(\[(?P<options>.*?)\])?\s*{lilypond}(?P<code>.*?)\\end{lilypond})",
 146         'lilypond_file': r'(?m)^[^%\n]*?(?P<match>\\lilypondfile\s*(\[(?P<options>.*?)\])?\s*\{(?P<filename>.+)})',
 147         'multiline_comment': no_match,
 148         'singleline_comment': r"(?m)^.*?(?P<match>(?P<code>^%.*$\n+))",
 149         'verb': r"(?P<code>\\verb(?P<del>.).*?(?P=del))",
 150         'verbatim': r"(?s)(?P<code>\\begin\s*{verbatim}.*?\\end{verbatim})",
 151         },
 152
 153         TEXINFO: {
 154         'include':  '(?m)^[^%\n]*?(?P<match>@include\s+(?P<filename>\S*))',
 155         'lilypond' : '(?m)^(?P<match>@lilypond(\[(?P<options>[^]]*)\])?{(?P<code>.*?)})',
 156         'lilypond_block': r'''(?ms)^(?P<match>@lilypond(\[(?P<options>[^]]*)\])?\s(?P<code>.*?)@end lilypond)\s''',
 157         'lilypond_file': '(?m)^(?P<match>@lilypondfile(\[(?P<options>[^]]*)\])?{(?P<filename>[^}]+)})',
 158         'multiline_comment': r"(?sm)^\s*(?!@c\s+)(?P<code>@ignore\s.*?@end ignore)\s",
 159         'singleline_comment': r"(?m)^.*?(?P<match>(?P<code>@c([ \t][^\n]*|)\n))",
 160         'verb': r'''(?P<code>@code{.*?})''',
 161         'verbatim': r'''(?s)(?P<code>@example\s.*?@end example\s)''',
 162         },
 163         }
 164
 165 format_res = {
 166         HTML: {
 167         'option-sep' : '\s*',
 168         'intertext': r',?\s*intertext=\".*?\"',
 169         },
 170         LATEX: {
 171         'intertext': r',?\s*intertext=\".*?\"',
 172         'option-sep' : ',\s*',
 173         },
 174         TEXINFO: {
 175         'intertext': r',?\s*intertext=\".*?\"',
 176         'option-sep' : ',\s*',
 177         },
 178         }
 179
 180 ly_options = {
 181         NOTES: {
 182         'relative': r'''\relative c%(relative_quotes)s''',
 183         },
 184         PAPER: {
 185         'indent' : r'''
 186     indent = %(indent)s''',
 187         'linewidth' : r'''
 188     linewidth = %(linewidth)s''',
 189         'noindent' : r'''
 190     indent = 0.0\mm''',
 191         'notime' : r'''
 192     \translator {
 193         \StaffContext
 194         \remove Time_signature_engraver
 195     }''',
 196         'raggedright' : r'''
 197     indent = 0.0\mm
 198     raggedright = ##t''',
 199         },
 200         PREAMBLE: {
 201         'staffsize': r'''
 202 #(set-global-staff-size %(staffsize)s)''',
 203         },
 204         }
 205
 206 output = {
 207         HTML : {
 208         AFTER: '',
 209         BEFORE: '',
 210         OUTPUT: r'''<img align="center" valign="center"
 211 border="0" src="%(base)s.png" alt="[picture of music]">''',
 212         VERBATIM: r'''<pre>
 213 %(verb)s</pre>''',
 214         },
 215
 216         LATEX : {
 217         AFTER: '',
 218         BEFORE: '',
 219         OUTPUT: r'''{\parindent 0pt
 220 \catcode`\@=12
 221 \ifx\preLilyPondExample\preLilyPondExample\fi
 222 \def\lilypondbook{}
 223 \input %(base)s.tex
 224 \ifx\preLilyPondExample\postLilyPondExample\fi
 225 \catcode`\@=0}''',
 226         VERBATIM: r'''\begin{verbatim}
 227 %(verb)s\end{verbatim}
 228 ''',
 229         },
 230
 231         TEXINFO :       {
 232         BEFORE: '',
 233         AFTER: '',
 234         VERBATIM: r'''@example
 235 %(verb)s@end example''',
 236
 237         },
 238
 239         }
 240
 241 PREAMBLE_LY = r'''%% Generated by %(program_name)s
 242 %% Options: [%(option_string)s]
 243 %(preamble_string)s
 244 \paper {%(paper_string)s
 245 }
 246 '''
 247
 248 FRAGMENT_LY = r'''\score{
 249     \notes%(notes_string)s{
 250         %(code)s    }
 251 }'''
 252 FULL_LY = '%(code)s'
 253
 254 def classic_lilypond_book_compatibility (o):
 255         if o == 'singleline':
 256                 return 'raggedright'
 257         m = re.search ('relative\s*([-0-9])', o)
 258         if m:
 259                 return 'relative=%s' % m.group (1)
 260         m = re.match ('([0-9]+)pt', o)
 261         if m:
 262                 return 'staffsize=%s' % m.group (1)
 263         m = re.match ('indent=([-.0-9]+)(cm|in|mm|pt)', o)
 264         if m:
 265                 f = float (m.group (1))
 266                 return 'indent=%f\\%s' % (f, m.group (2))
 267         m = re.match ('linewidth=([-.0-9]+)(cm|in|mm|pt)', o)
 268         if m:
 269                 f = float (m.group (1))
 270                 return 'linewidth=%f\\%s' % (f, m.group (2))
 271         return None
 272
 273 def compose_ly (code, option_string):
 274         options = []
 275         # urg
 276         for i in default_ly_options.keys ():
 277                 options.append (i)
 278                 vars ()[i] = default_ly_options[i]
 279
 280         if option_string:
 281                 options = options + split_options (option_string)
 282
 283         m = re.search (r'''\\score''', code)
 284         if not m and (not options \
 285                       or not 'nofragment' in options \
 286                       or 'fragment' in options):
 287                 options.append ('raggedright')
 288                 body = FRAGMENT_LY
 289         else:
 290                 body = FULL_LY
 291
 292         # defaults
 293         relative = 0
 294         staffsize = '16'
 295
 296         notes_options = []
 297         paper_options = []
 298         preamble_options = []
 299         for i in options:
 300                 c = classic_lilypond_book_compatibility (i)
 301                 if c:
 302                         ly.warning (_ ("deprecated ly-option used: %s" % i))
 303                         ly.warning (_ ("compatibility mode translation: %s" \
 304                                        % c))
 305                         i = c
 306
 307                 if string.find (i, '=') > 0:
 308                         key, value = string.split (i, '=')
 309                         # hmm
 310                         vars ()[key] = value
 311                 else:
 312                         key = i
 313
 314                 if key in ly_options[NOTES].keys ():
 315                         notes_options.append (ly_options[NOTES][key])
 316                 elif key in ly_options[PREAMBLE].keys ():
 317                         preamble_options.append (ly_options[PREAMBLE][key])
 318                 elif key in ly_options[PAPER].keys ():
 319                         paper_options.append (ly_options[PAPER][key])
 320                 elif key not in ('fragment', 'nofragment',
 321                                  'relative', 'verbatim'):
 322                         ly.warning (_("ignoring unknown ly option: %s") % i)
 323
 324         relative_quotes = (",,,", ",,", ",", "", "'", "''", "'''")[relative-3]
 325         program_name = __main__.program_name
 326         notes_string = string.join (notes_options, '\n    ') % vars ()
 327         paper_string = string.join (paper_options, '\n    ') % vars ()
 328         preamble_string = string.join (preamble_options, '\n    ') % vars ()
 329         return (PREAMBLE_LY + body) % vars ()
 330
 331
 332 # BARF
 333 # use lilypond-bin for latex (.lytex) books,
 334 # and lilypond --preview for html, texinfo books?
 335 def to_eps (file):
 336         cmd = r'latex "\nonstopmode \input %s"' % file
 337         # Ugh.  (La)TeX writes progress and error messages on stdout
 338         # Redirect to stderr
 339         cmd = '(( %s  >&2 ) >&- )' % cmd
 340         ly.system (cmd)
 341         ly.system ('dvips -Ppdf -u+lilypond.map -E -o %s.eps %s' \
 342                    % (file, file))
 343
 344 def find_file (name):
 345         for i in include_path:
 346                 full = os.path.join (i, name)
 347                 if os.path.exists (full):
 348                         return full
 349         ly.error (_ ('file not found: %s\n' % name))
 350         ly.exit (1)
 351         return ''
 352
 353 def verbatim_html (s):
 354         return re.sub ('>', '&gt;',
 355                        re.sub ('<', '&lt;',
 356                                re.sub ('&', '&amp;', s)))
 357
 358 def verbatim_texinfo (s):
 359         return re.sub ('{', '@{',
 360                        re.sub ('}', '@}',
 361                                re.sub ('@', '@@', s)))
 362
 363 def split_options (option_string):
 364         return re.split (format_res[format]['option-sep'], option_string)
 365
 366
 367 ## make index static of Snippet?
 368 index = 0
 369
 370 class Snippet:
 371
 372         ## huh? index is redundant? --hwn
 373         def __init__ (self, type, source, index, match):
 374                 self.type = type
 375                 self.source = source
 376                 self.index = index
 377                 self.match = match
 378                 self.hash = 0
 379
 380         def start (self, s):
 381                 return self.index + self.match.start (s)
 382
 383         def end (self, s):
 384                 return self.index + self.match.end (s)
 385
 386         def substring (self, s):
 387                 return self.source[self.start (s):self.end (s)]
 388
 389         def ly (self):
 390                 s = ''
 391                 if self.type == 'lilypond_block' or self.type == 'lilypond':
 392                         s = self.substring ('code')
 393                 elif self.type == 'lilypond_file':
 394                         name = self.substring ('filename')
 395                         s = open (find_file (name)).read ()
 396                 return s
 397
 398         def full_ly (self):
 399                 s = self.ly ()
 400                 if s:
 401                         return compose_ly (s, self.match.group ('options'))
 402                 return ''
 403
 404         def get_hash (self):
 405                 if not self.hash:
 406                         self.hash = abs (hash (self.ly ()))
 407                 return self.hash
 408
 409         def basename (self):
 410                 if use_hash_p:
 411                         return 'lily-%d' % self.get_hash ()
 412                 raise 'to be done'
 413
 414         def write_ly (self):
 415                 if self.type == 'lilypond_block' or self.type == 'lilypond'\
 416                        or self.type == 'lilypond_file':
 417                         h = open (self.basename () + '.ly', 'w')
 418                         h.write (self.full_ly ())
 419
 420         def output_html (self):
 421                 base = self.basename ()
 422                 option_string = self.match.group ('options')
 423                 if option_string and VERBATIM in split_options (option_string)\
 424                    and format == HTML:
 425                         verb = verbatim_html (self.substring ('code'))
 426                         h.write (output[HTML][VERBATIM] % vars ())
 427                 h.write (output[HTML][BEFORE])
 428                 h.write (output[HTML][OUTPUT] % vars ())
 429                 h.write (output[HTML][AFTER])
 430
 431         def output_latex (self):
 432                 option_string = self.match.group ('options')
 433                 if option_string and VERBATIM in split_options (option_string)\
 434                    and format == LATEX:
 435                         verb = self.substring ('code')
 436                         h.write (output[LATEX][VERBATIM] % vars ())
 437                 h.write (output[LATEX][BEFORE])
 438                 base = self.basename ()
 439                 h.write (output[LATEX][OUTPUT] % vars ())
 440                 h.write (output[LATEX][AFTER])
 441
 442         def output_texinfo (self):
 443                 option_string = self.match.group ('options')
 444                 if option_string and VERBATIM in split_options (option_string):
 445                         verb = verbatim_texinfo (self.substring ('code'))
 446                         h.write (output[TEXINFO][VERBATIM] % vars ())
 447                 h.write ('\n@tex\n')
 448                 self.output_latex ()
 449                 h.write ('\n@end tex\n')
 450
 451                 h.write ('\n@html\n')
 452                 self.output_html ()
 453                 h.write ('\n@end html\n')
 454
 455         def outdated_p (self):
 456                 if self.type != 'lilypond_block' and self.type != 'lilypond'\
 457                        and self.type != 'lilypond_file':
 458                         return None
 459                 base = self.basename ()
 460                 if os.path.exists (base + '.ly') \
 461                    and os.path.exists (base + '.tex') \
 462                    and (use_hash_p \
 463                         or self.ly () == open (base + '.ly').read ()):
 464                         # TODO: something smart with target formats
 465                         # (ps, png) and m/ctimes
 466                         return None
 467                 return self
 468
 469         def filter_code (self):
 470                 global index
 471                 # Hmm, why is verbatim's group called 'code'; rename to 'verb'?
 472                 #if snippet.match.group ('code'):
 473                 # urg
 474                 if self.type == 'lilypond' or self.type == 'lilypond_block':
 475                         h.write (self.source[index:self.start ('code')])
 476                         h.write (run_filter (self.substring ('code')))
 477                         h.write (self.source[self.end ('code'):self.end (0)])
 478                 else:
 479                         h.write (self.source[index:self.end (0)])
 480                 index = self.end (0)
 481
 482         def compile_output (self):
 483                 global index
 484                 # Hmm, why is verbatim's group called 'code'; rename to 'verb'?
 485                 # if snippet.match.group ('code'):
 486                 # urg
 487                 if self.type == 'lilypond' \
 488                        or self.type == 'lilypond_block'\
 489                        or self.type == 'lilypond_file':
 490                         h.write (self.source[index:self.start (0)])
 491                         snippet_output = eval ("Snippet.output_" + format)
 492                         snippet_output (self)
 493                 elif self.type == 'include':
 494                         h.write (self.source[index:self.start ('filename')])
 495                         base = os.path.splitext (self.substring ('filename'))[0]
 496                         h.write (base + format2ext[format])
 497                         h.write (self.source[self.end ('filename'):self.end (0)])
 498                 else:
 499                         h.write (self.source[index:self.end (0)])
 500                 index = self.end (0)
 501
 502
 503 # this isn't working - <params> are doubly used.
 504 # a solution is to strip P<> from the regexes,
 505 # and redo the match with  the original in notice_snippet
 506 def other_toplevel_snippets (str, types):
 507         res = ['(?P<regex%s>%s)' % (t, snippet_res[format][t])
 508                for t in types]
 509
 510         big_re = re.compile (string.join (res, '|'))
 511         snippets = []
 512
 513         def notice_snippet (match, snips = snippets, types = types):
 514                 snip = None
 515                 for t in types:
 516                         try:
 517                                 key = 'regex' + t
 518                                 gr = match.group (key)
 519                                 snip =Snippet (t, str, match.start (key), match)
 520                                 break
 521                         except IndexError:
 522                                 pass
 523                 if snip:
 524                         snips.append (snip)
 525                 else:
 526                         raise "Huh?"
 527
 528                 return match.group (0)
 529
 530         str = big_re.sub (notice_snippet, str)
 531
 532         return snippets
 533
 534 def simple_toplevel_snippets (str, types):
 535         snippets  = []
 536         for t in types:
 537                 regex = re.compile (snippet_res[format][t])
 538
 539                 # ugh, this can be done nicer in python 2.x
 540                 def notice_snippet (match, snippets = snippets,
 541                                     t = t, str = str):
 542                         s = Snippet (t, str, 0, match)
 543                         snippets.append (s)
 544                         return ''
 545
 546                 regex.sub (notice_snippet, str)
 547         return snippets
 548
 549 def find_toplevel_snippets (s, types):
 550         res = {}
 551         for i in types:
 552                 res[i] = ly.re.compile (snippet_res[format][i])
 553
 554         snippets = []
 555         index = 0
 556         found = {}.fromkeys (types)
 557         while 1:
 558                 first = 0
 559                 endex = 1 << 30
 560                 for i in types:
 561                         if not found[i] or found[i].start (0) < index:
 562                                 found[i] = 0
 563                                 m = res[i].search (s[index:endex])
 564                                 if m:
 565                                         found[i] = Snippet (i, s, index, m)
 566                         if found[i] \
 567                                and (not first \
 568                                     or found[i].start (0) < found[first].start (0)):
 569                                 first = i
 570                                 endex = found[first].start (0)
 571                 if not first:
 572                         break
 573                 snippets.append (found[first])
 574                 index = found[first].end (0)
 575
 576         return snippets
 577
 578 def filter_pipe (input, cmd):
 579         if verbose_p:
 580                 ly.progress (_ ("Opening filter `%s\'") % cmd)
 581
 582         stdin, stdout, stderr = os.popen3 (cmd)
 583         stdin.write (input)
 584         status = stdin.close ()
 585
 586         if not status:
 587                 status = 0
 588                 output = stdout.read ()
 589                 status = stdout.close ()
 590                 error = stderr.read ()
 591
 592         if not status:
 593                 status = 0
 594         signal = 0x0f & status
 595         if status or (not output and error):
 596                 exit_status = status >> 8
 597                 ly.error (_ ("`%s\' failed (%d)") % (cmd, exit_status))
 598                 ly.error (_ ("The error log is as follows:"))
 599                 sys.stderr.write (error)
 600                 sys.stderr.write (stderr.read ())
 601                 ly.exit (status)
 602
 603         if verbose_p:
 604                 ly.progress ('\n')
 605
 606         return output
 607
 608 def run_filter (s):
 609         return filter_pipe (s, filter_cmd)
 610
 611 def process_snippets (cmd, snippets):
 612         names = filter (lambda x:x, map (Snippet.basename, snippets))
 613         if names:
 614                 ly.system (string.join ([cmd] + names))
 615
 616         if format == HTML or format == TEXINFO:
 617                 for i in names:
 618                         if os.path.exists (i + '.tex'):
 619                                 to_eps (i)
 620                                 ly.make_ps_images (i + '.eps', resolution=110)
 621
 622 LATEX_DOCUMENT = r'''
 623 %(preamble)s
 624 \begin{document}
 625 \typeout{textwidth=\the\textwidth}
 626 \typeout{columnsep=\the\columnsep}
 627 \makeatletter\if@twocolumn\typeout{columns=2}\fi\makeatother
 628 \end{document}
 629 '''
 630 #need anything else besides textwidth?
 631 def get_latex_textwidth (source):
 632         m = re.search (r'''(?P<preabmle>\\begin\s*{document})''', source)
 633         preamble = source[:m.start (0)]
 634         latex_document = LATEX_DOCUMENT % vars ()
 635         parameter_string = filter_pipe (latex_document, latex_filter_cmd)
 636
 637         columns = 0
 638         m = re.search ('columns=([0-9.]*)', parameter_string)
 639         if m:
 640                 columns = string.atoi (m.group (1))
 641
 642         columnsep = 0
 643         m = re.search ('columnsep=([0-9.]*)pt', parameter_string)
 644         if m:
 645                 columnsep = string.atof (m.group (1))
 646
 647         textwidth = 0
 648         m = re.search('textwidth=([0-9.]*)pt', parameter_string)
 649         if m:
 650                 textwidth = string.atof (m.group (1))
 651                 if columns:
 652                         textwidth = (textwidth - columnsep) / columns
 653
 654         return textwidth
 655
 656
 657 ext2format = {
 658         '.html' : HTML,
 659         '.itely' : TEXINFO,
 660         '.lytex' : LATEX,
 661         '.tely' : TEXINFO,
 662         '.tex': LATEX,
 663         '.texi' : TEXINFO,
 664         '.texinfo' : TEXINFO,
 665         '.xml' : HTML,
 666         }
 667
 668 format2ext = {
 669         HTML: '.html',
 670         #TEXINFO: '.texinfo',
 671         TEXINFO: '.texi',
 672         LATEX: '.tex',
 673         }
 674
 675 def do_file (input_filename):
 676         #ugh
 677         global format
 678         if not format:
 679                 e = os.path.splitext (input_filename)[1]
 680                 if e in ext2format.keys ():
 681                         #FIXME
 682                         format = ext2format[e]
 683                 else:
 684                         ly.error (_ ("cannot determine format for: %s" \
 685                                      % input_filename))
 686
 687         ly.progress (_ ("Reading %s...") % input_filename)
 688         if not input_filename or input_filename == '-':
 689                 ih = sys.stdin
 690         else:
 691                 ih = open (input_filename)
 692         source = ih.read ()
 693         ly.progress ('\n')
 694
 695         ly.progress (_ ("Dissecting..."))
 696         #snippets = find_toplevel_snippets (source, snippet_res[format].keys ())
 697         snippet_types = (
 698                 'lilypond_block',
 699                 'verb',
 700                 'verbatim',
 701                 'singleline_comment',
 702                 'multiline_comment',
 703                 'lilypond_file',
 704                 'include',
 705                 'lilypond', )
 706
 707         snippets = simple_toplevel_snippets (source, snippet_types)
 708         ly.progress ('\n')
 709
 710         global h
 711         if output_name == '-' or not output_name:
 712                 h = sys.stdout
 713                 output_filename = '-'
 714         else:
 715                 if not os.path.isdir (output_name):
 716                         os.mkdir (output_name, 0777)
 717                 if input_filename == '-':
 718                         input_base = 'stdin'
 719                 else:
 720                         input_base = os.path.splitext (input_filename)[0]
 721                 output_filename = output_name + '/' + input_base \
 722                                   + format2ext[format]
 723                 h = open (output_filename, 'w')
 724                 os.chdir (output_name)
 725
 726         global default_ly_options
 727         textwidth = 0
 728         if format == LATEX and LINEWIDTH not in default_ly_options.keys ():
 729                 textwidth = get_latex_textwidth (source)
 730                 default_ly_options[LINEWIDTH] = '''%.0f\pt''' % textwidth
 731
 732         global index
 733         if filter_cmd:
 734                 index = 0
 735                 map (Snippet.filter_code, snippets)
 736                 h.write (source[index:])
 737         elif process_cmd:
 738                 outdated = filter (lambda x:x,
 739                                    map (Snippet.outdated_p, snippets))
 740                 ly.progress (_ ("Writing snippets..."))
 741                 map (Snippet.write_ly, snippets)
 742                 ly.progress ('\n')
 743
 744                 if outdated:
 745                         ly.progress (_ ("Processing..."))
 746                         process_snippets (process_cmd, outdated)
 747                 else:
 748                         ly.progress (_ ("All snippets are up to date..."))
 749                 ly.progress ('\n')
 750
 751                 ly.progress (_ ("Compiling %s...") % output_filename)
 752                 index = 0
 753                 map (Snippet.compile_output, snippets)
 754                 h.write (source[index:])
 755                 ly.progress ('\n')
 756
 757         if h != sys.stdout:
 758                 h.close ()
 759
 760         def process_include (snippet):
 761                 os.chdir (original_dir)
 762                 name = snippet.substring ('filename')
 763                 ly.progress (_ ('Processing include: %s') % name)
 764                 ly.progress ('\n')
 765                 do_file (name)
 766
 767         map (process_include, filter (lambda x: x.type == 'include', snippets))
 768
 769 def do_options ():
 770         global format, output_name
 771         global filter_cmd, process_cmd, verbose_p
 772
 773         (sh, long) = ly.getopt_args (option_definitions)
 774         try:
 775                 (options, files) = getopt.getopt (sys.argv[1:], sh, long)
 776         except getopt.error, s:
 777                 sys.stderr.write ('\n')
 778                 ly.error (_ ("getopt says: `%s\'" % s))
 779                 sys.stderr.write ('\n')
 780                 ly.help ()
 781                 ly.exit (2)
 782
 783         for opt in options:
 784                 o = opt[0]
 785                 a = opt[1]
 786
 787                 if 0:
 788                         pass
 789                 elif o == '--filter' or o == '-F':
 790                         filter_cmd = a
 791                         process_cmd = 0
 792                 elif o == '--format' or o == '-f':
 793                         format = a
 794                         if a == 'texi-html' or a == 'texi':
 795                                 format = TEXINFO
 796                 elif o == '--help' or o == '-h':
 797                         ly.help ()
 798                         sys.exit (0)
 799                 elif o == '--include' or o == '-I':
 800                         include_path.append (os.path.join (original_dir,
 801                                                            ly.abspath (a)))
 802                 elif o == '--output' or o == '-o':
 803                         output_name = a
 804                 elif o == '--outdir':
 805                         output_name = a
 806                 elif o == '--process' or o == '-P':
 807                         process_cmd = a
 808                         filter_cmd = 0
 809                 elif o == '--version' or o == '-v':
 810                         ly.identify (sys.stdout)
 811                         sys.exit (0)
 812                 elif o == '--verbose' or o == '-V':
 813                         verbose_p = 1
 814                 elif o == '--warranty' or o == '-w':
 815                         if 1 or status:
 816                                 ly.warranty ()
 817                         sys.exit (0)
 818         return files
 819
 820 def main ():
 821         files = do_options ()
 822         ly.identify (sys.stderr)
 823         ly.setup_environment ()
 824         if files:
 825                 do_file (files[0])
 826
 827 if __name__ == '__main__':
 828         main ()