1
0
Fork 0
mirror of https://github.com/kastdeur/lilydrum.git synced 2024-12-23 01:33:32 +01:00
lilydrum/lybook/makebook
2015-12-09 19:30:20 +01:00

162 lines
5.1 KiB
Python

#!/usr/bin/python3.4
# To be Run: python3.4 makebook -v -o ./tex/main_pipes.tex pipes
# Uses a copy of the filestructure of pipeband-drumming to generate the body of a tex file
import os
from argparse import ArgumentParser
class MakeBook:
OUT = './tex/main.tex'
MASTER_DIR = os.path.dirname(os.path.abspath(__file__))
RUN_DIR = os.path.abspath(os.curdir)
SCORES_DIR = './scores/'
LY_DIR = '../'
INSTRUMENTS = ['full','drums', 'pipes', 'bass', 'tenor', 'side','snare']
def __init__(self):
usage = __file__
parser = ArgumentParser(usage)
parser.add_argument('-v','--verbose',default=False,action='store_true',dest='verbose', help='verbose')
parser.add_argument('instrument',default=False,nargs='*',help='Instruments to be included in the book')
parser.add_argument('-o','--output',default=self.OUT,help='Output file')
self.args = parser.parse_args()
if self.args.verbose:
self.vprint('Verbose output')
standards = ['standards']
marches = ['2-4_marches', '3-4_marches', '4-4_marches', '5-4_marches', '6-8_marches']
watch_folders = [ 'hornpipes', 'jigs', 'strathspeys', 'reels','other']
watch_folders = marches + watch_folders
# Run Path_walker over dirs
tune_dirs = []
for d in watch_folders:
if d in marches:
continue
tune_dirs.append(d)
# Ready f.out
try:
self.fout = open(self.args.output,'w+')
self.fwrite(u'\\addcontentsline{toc}{section}{Standards}')
for d in standards:
self.vprint(os.path.join(self.SCORES_DIR,d))
self.path_walker(os.path.join(self.SCORES_DIR,d), level=1, content_line_level=1)
self.fwrite(u'\\addcontentsline{toc}{section}{Marches}')
for d in marches:
self.vprint(os.path.join(self.SCORES_DIR,d))
self.path_walker(os.path.join(self.SCORES_DIR,d), level=1, content_line_level=2)
for d in tune_dirs:
self.vprint(os.path.join(self.SCORES_DIR,d))
self.path_walker(os.path.join(self.SCORES_DIR,d))
except IOError:
print('Cannot open {}'.format(self.OUT))
except:
self.fout.close()
raise
def vprint(self,line):
if self.args.verbose:
print(line)
def fwrite(self,line):
self.fout.write(line+'\n')
def include_pdf (self,file, level):
title = False
self.vprint(file)
lilydir = file[:file.rfind('/')].replace(self.SCORES_DIR,self.LY_DIR)
if os.path.isdir(lilydir):
path = file[:file.rfind('/')]
#Try to load title from config file
if os.path.isfile(os.path.join(lilydir,'config.ily')):
self.vprint('Loading config file')
with open(os.path.join(lilydir,'config.ily'),'r') as f:
for line in iter(f):
if line.startswith('title'):
title = line[line.find('=')+1:]#Strip 'title='
title = title[1:-1] #Strip Quotes
continue
lilyfile = os.path.join(lilydir, file[file.rfind('-')+1:].replace('.pdf','.ly'))
if not title and os.path.isfile(lilyfile):
self.vprint('Look through lilyfile')
with open(lilyfile, 'r') as f:
for line in iter(f):
if line.strip().startswith('title'):
title = line[line.find('=')+1:]#Strip 'title='
title = title[1:-1] #Strip Quotes
continue
if title:
ref = title
forbid = ' /!@#$%^&*()<>?\|;:\'"'
for s in forbid:
ref = ref.replace(s,'_')
if not title:
self.vprint('No title yet Found!')
last_slash = file.rfind('/')
last_dot = file.rfind('.')
if file[file.rfind('.',0,last_dot)+1:file.rfind('.')] in self.INSTRUMENTS:
ref = file[file.rfind('/')+1:file.rfind('.')]
else :
ref = file[file.rfind('/',0,last_slash-1)+1:file.rfind('.')].replace('/','-')
# Make title from filename
title = ref
#Remove references to Instruments
for inst in self.INSTRUMENTS:
title = title.replace('-'+inst, '')
title = title.replace('.'+inst, '')
#Remove chars
title = title.replace('-','').replace('_',' ').title()
title = title[1:-1].replace('\\n',' ')
if not self.args.instrument:
for inst in self.INSTRUMENTS:
if inst in file:
title = title + ' ('+inst+')'
self.vprint('= '+title)
ref = 'p'+ref.strip('_').lower()
#string = u'\\includepdf[pages=-, addtotoc={1,'+('sub'*level)+'section,'+str(level+1)+','+title+','+ref+'}, pagecommand={}]{'+file+'}'
string = u'\\includepdf[pages=-, addtotoc={1,'+('sub'*level)+'section,'+str(level+1)+',{'+title+'},'+ref+'}, pagecommand={}]{'+file+'}'
self.fwrite(string)
return string
def path_walker(self, a, level=None, content_line_level=None):
if not a:
return
if level is None:
level = 0
if content_line_level is None:
content_line_level = 1
# Do stuff
if level < content_line_level:
self.fwrite(u'\\mysection{'+('sub'*level)+'section}{'+a[a.rfind('/')+1:].replace('_',' ').replace('-','/').title()+'}')
newline = False
for root, dirs, files in os.walk(a,False):
for f in files:
if not f.endswith('.pdf'):
continue
if self.args.instrument:
instr_in_file = False
for instr in self.args.instrument:
if instr in f:
self.vprint(instr)
instr_in_file = True
break
if not instr_in_file:
continue
# Include file
if newline:
self.fwrite(u'\\newpage')
newline = True
self.include_pdf(os.path.join(root,f), level+1)
# Run above
MakeBook();