184 lines
5.9 KiB
Python
184 lines
5.9 KiB
Python
|
#! /usr/bin/env python
|
||
|
|
# encoding: utf-8
|
||
|
|
|
||
|
|
import os,re
|
||
|
|
import Utils,TaskGen,Task,Runner,Build
|
||
|
|
from TaskGen import feature,before
|
||
|
|
from Logs import error,warn,debug
|
||
|
|
re_tex=re.compile(r'\\(?P<type>include|input|import|bringin|lstinputlisting){(?P<file>[^{}]*)}',re.M)
|
||
|
|
def scan(self):
|
||
|
|
node=self.inputs[0]
|
||
|
|
env=self.env
|
||
|
|
nodes=[]
|
||
|
|
names=[]
|
||
|
|
if not node:return(nodes,names)
|
||
|
|
code=Utils.readf(node.abspath(env))
|
||
|
|
curdirnode=self.curdirnode
|
||
|
|
abs=curdirnode.abspath()
|
||
|
|
for match in re_tex.finditer(code):
|
||
|
|
path=match.group('file')
|
||
|
|
if path:
|
||
|
|
for k in['','.tex','.ltx']:
|
||
|
|
debug('tex: trying %s%s'%(path,k))
|
||
|
|
try:
|
||
|
|
os.stat(abs+os.sep+path+k)
|
||
|
|
except OSError:
|
||
|
|
continue
|
||
|
|
found=path+k
|
||
|
|
node=curdirnode.find_resource(found)
|
||
|
|
if node:
|
||
|
|
nodes.append(node)
|
||
|
|
else:
|
||
|
|
debug('tex: could not find %s'%path)
|
||
|
|
names.append(path)
|
||
|
|
debug("tex: found the following : %s and names %s"%(nodes,names))
|
||
|
|
return(nodes,names)
|
||
|
|
latex_fun,_=Task.compile_fun('latex','${LATEX} ${LATEXFLAGS} ${SRCFILE}',shell=False)
|
||
|
|
pdflatex_fun,_=Task.compile_fun('pdflatex','${PDFLATEX} ${PDFLATEXFLAGS} ${SRCFILE}',shell=False)
|
||
|
|
bibtex_fun,_=Task.compile_fun('bibtex','${BIBTEX} ${BIBTEXFLAGS} ${SRCFILE}',shell=False)
|
||
|
|
makeindex_fun,_=Task.compile_fun('bibtex','${MAKEINDEX} ${MAKEINDEXFLAGS} ${SRCFILE}',shell=False)
|
||
|
|
g_bibtex_re=re.compile('bibdata',re.M)
|
||
|
|
def tex_build(task,command='LATEX'):
|
||
|
|
env=task.env
|
||
|
|
bld=task.generator.bld
|
||
|
|
if not env['PROMPT_LATEX']:
|
||
|
|
env.append_value('LATEXFLAGS','-interaction=batchmode')
|
||
|
|
env.append_value('PDFLATEXFLAGS','-interaction=batchmode')
|
||
|
|
fun=latex_fun
|
||
|
|
if command=='PDFLATEX':
|
||
|
|
fun=pdflatex_fun
|
||
|
|
node=task.inputs[0]
|
||
|
|
reldir=node.bld_dir(env)
|
||
|
|
srcfile=node.abspath(env)
|
||
|
|
sr2=node.parent.abspath()+os.pathsep+node.parent.abspath(env)+os.pathsep
|
||
|
|
aux_node=node.change_ext('.aux')
|
||
|
|
idx_node=node.change_ext('.idx')
|
||
|
|
nm=aux_node.name
|
||
|
|
docuname=nm[:len(nm)-4]
|
||
|
|
task.cwd=task.inputs[0].parent.abspath(task.env)
|
||
|
|
warn('first pass on %s'%command)
|
||
|
|
task.env.env={'TEXINPUTS':sr2}
|
||
|
|
task.env.SRCFILE=srcfile
|
||
|
|
ret=fun(task)
|
||
|
|
if ret:
|
||
|
|
return ret
|
||
|
|
try:
|
||
|
|
ct=Utils.readf(aux_node.abspath(env))
|
||
|
|
except(OSError,IOError):
|
||
|
|
error('error bibtex scan')
|
||
|
|
else:
|
||
|
|
fo=g_bibtex_re.findall(ct)
|
||
|
|
if fo:
|
||
|
|
warn('calling bibtex')
|
||
|
|
task.env.env={'BIBINPUTS':sr2,'BSTINPUTS':sr2}
|
||
|
|
task.env.SRCFILE=docuname
|
||
|
|
ret=bibtex_fun(task)
|
||
|
|
if ret:
|
||
|
|
error('error when calling bibtex %s'%docuname)
|
||
|
|
return ret
|
||
|
|
try:
|
||
|
|
idx_path=idx_node.abspath(env)
|
||
|
|
os.stat(idx_path)
|
||
|
|
except OSError:
|
||
|
|
error('error file.idx scan')
|
||
|
|
else:
|
||
|
|
warn('calling makeindex')
|
||
|
|
task.env.SRCFILE=idx_node.name
|
||
|
|
task.env.env={}
|
||
|
|
ret=makeindex_fun(task)
|
||
|
|
if ret:
|
||
|
|
error('error when calling makeindex %s'%idx_path)
|
||
|
|
return ret
|
||
|
|
hash=''
|
||
|
|
i=0
|
||
|
|
while i<10:
|
||
|
|
i+=1
|
||
|
|
prev_hash=hash
|
||
|
|
try:
|
||
|
|
hash=Utils.h_file(aux_node.abspath(env))
|
||
|
|
except KeyError:
|
||
|
|
error('could not read aux.h -> %s'%aux_node.abspath(env))
|
||
|
|
pass
|
||
|
|
if hash and hash==prev_hash:
|
||
|
|
break
|
||
|
|
warn('calling %s'%command)
|
||
|
|
task.env.env={'TEXINPUTS':sr2+os.pathsep}
|
||
|
|
task.env.SRCFILE=srcfile
|
||
|
|
ret=fun(task)
|
||
|
|
if ret:
|
||
|
|
error('error when calling %s %s'%(command,latex_compile_cmd))
|
||
|
|
return ret
|
||
|
|
return None
|
||
|
|
latex_vardeps=['LATEX','LATEXFLAGS']
|
||
|
|
def latex_build(task):
|
||
|
|
return tex_build(task,'LATEX')
|
||
|
|
pdflatex_vardeps=['PDFLATEX','PDFLATEXFLAGS']
|
||
|
|
def pdflatex_build(task):
|
||
|
|
return tex_build(task,'PDFLATEX')
|
||
|
|
class tex_taskgen(TaskGen.task_gen):
|
||
|
|
def __init__(self,*k,**kw):
|
||
|
|
TaskGen.task_gen.__init__(self,*k,**kw)
|
||
|
|
def apply_tex(self):
|
||
|
|
if not getattr(self,'type',None)in['latex','pdflatex']:
|
||
|
|
self.type='pdflatex'
|
||
|
|
tree=self.bld
|
||
|
|
outs=Utils.to_list(getattr(self,'outs',[]))
|
||
|
|
self.env['PROMPT_LATEX']=getattr(self,'prompt',1)
|
||
|
|
deps_lst=[]
|
||
|
|
if getattr(self,'deps',None):
|
||
|
|
deps=self.to_list(self.deps)
|
||
|
|
for filename in deps:
|
||
|
|
n=self.path.find_resource(filename)
|
||
|
|
if not n in deps_lst:deps_lst.append(n)
|
||
|
|
self.source=self.to_list(self.source)
|
||
|
|
for filename in self.source:
|
||
|
|
base,ext=os.path.splitext(filename)
|
||
|
|
node=self.path.find_resource(filename)
|
||
|
|
if not node:raise Utils.WafError('cannot find %s'%filename)
|
||
|
|
if self.type=='latex':
|
||
|
|
task=self.create_task('latex',node,node.change_ext('.dvi'))
|
||
|
|
elif self.type=='pdflatex':
|
||
|
|
task=self.create_task('pdflatex',node,node.change_ext('.pdf'))
|
||
|
|
task.env=self.env
|
||
|
|
task.curdirnode=self.path
|
||
|
|
if deps_lst:
|
||
|
|
variant=node.variant(self.env)
|
||
|
|
try:
|
||
|
|
lst=tree.node_deps[task.unique_id()]
|
||
|
|
for n in deps_lst:
|
||
|
|
if not n in lst:
|
||
|
|
lst.append(n)
|
||
|
|
except KeyError:
|
||
|
|
tree.node_deps[task.unique_id()]=deps_lst
|
||
|
|
if self.type=='latex':
|
||
|
|
if'ps'in outs:
|
||
|
|
tsk=self.create_task('dvips',task.outputs,node.change_ext('.ps'))
|
||
|
|
tsk.env.env={'TEXINPUTS':node.parent.abspath()+os.pathsep+self.path.abspath()+os.pathsep+self.path.abspath(self.env)}
|
||
|
|
if'pdf'in outs:
|
||
|
|
tsk=self.create_task('dvipdf',task.outputs,node.change_ext('.pdf'))
|
||
|
|
tsk.env.env={'TEXINPUTS':node.parent.abspath()+os.pathsep+self.path.abspath()+os.pathsep+self.path.abspath(self.env)}
|
||
|
|
elif self.type=='pdflatex':
|
||
|
|
if'ps'in outs:
|
||
|
|
self.create_task('pdf2ps',task.outputs,node.change_ext('.ps'))
|
||
|
|
self.source=[]
|
||
|
|
def detect(conf):
|
||
|
|
v=conf.env
|
||
|
|
for p in'tex latex pdflatex bibtex dvips dvipdf ps2pdf makeindex pdf2ps'.split():
|
||
|
|
conf.find_program(p,var=p.upper())
|
||
|
|
v[p.upper()+'FLAGS']=''
|
||
|
|
v['DVIPSFLAGS']='-Ppdf'
|
||
|
|
b=Task.simple_task_type
|
||
|
|
b('tex','${TEX} ${TEXFLAGS} ${SRC}',color='BLUE',shell=False)
|
||
|
|
b('bibtex','${BIBTEX} ${BIBTEXFLAGS} ${SRC}',color='BLUE',shell=False)
|
||
|
|
b('dvips','${DVIPS} ${DVIPSFLAGS} ${SRC} -o ${TGT}',color='BLUE',after="latex pdflatex tex bibtex",shell=False)
|
||
|
|
b('dvipdf','${DVIPDF} ${DVIPDFFLAGS} ${SRC} ${TGT}',color='BLUE',after="latex pdflatex tex bibtex",shell=False)
|
||
|
|
b('pdf2ps','${PDF2PS} ${PDF2PSFLAGS} ${SRC} ${TGT}',color='BLUE',after="dvipdf pdflatex",shell=False)
|
||
|
|
b=Task.task_type_from_func
|
||
|
|
cls=b('latex',latex_build,vars=latex_vardeps)
|
||
|
|
cls.scan=scan
|
||
|
|
cls=b('pdflatex',pdflatex_build,vars=pdflatex_vardeps)
|
||
|
|
cls.scan=scan
|
||
|
|
|
||
|
|
feature('tex')(apply_tex)
|
||
|
|
before('apply_core')(apply_tex)
|