227 lines
9.6 KiB
Python
227 lines
9.6 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import xml.etree.ElementTree as etree
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import re
|
||
|
|
from subprocess import call
|
||
|
|
|
||
|
|
FG_ROOT="../"
|
||
|
|
|
||
|
|
def tex_escape(text):
|
||
|
|
"""
|
||
|
|
:param text: a plain text message
|
||
|
|
:return: the message escaped to appear correctly in LaTeX
|
||
|
|
"""
|
||
|
|
conv = {
|
||
|
|
'&': r'\&',
|
||
|
|
'%': r'\%',
|
||
|
|
'$': r'\$',
|
||
|
|
'#': r'\#',
|
||
|
|
'_': r'\_',
|
||
|
|
'{': r'\{',
|
||
|
|
'}': r'\}',
|
||
|
|
'~': r'\textasciitilde{}',
|
||
|
|
'^': r'\^{}',
|
||
|
|
'\\': r'\textbackslash{}',
|
||
|
|
'<': r'\textless',
|
||
|
|
'>': r'\textgreater',
|
||
|
|
}
|
||
|
|
regex = re.compile('|'.join(re.escape(unicode(key)) for key in sorted(conv.keys(), key = lambda item: - len(item))))
|
||
|
|
return regex.sub(lambda match: conv[match.group()], text)
|
||
|
|
|
||
|
|
def generate_airplane_latex(source):
|
||
|
|
"""
|
||
|
|
Generates the LaTeX files and directory structure for the procedures
|
||
|
|
"""
|
||
|
|
root = os.path.join("procedures", source['name'])
|
||
|
|
checklists_path = os.path.join(root, "checklists")
|
||
|
|
if(not os.path.isdir(root)):
|
||
|
|
os.mkdir(root)
|
||
|
|
if(not os.path.isdir(checklists_path)):
|
||
|
|
os.mkdir(checklists_path)
|
||
|
|
|
||
|
|
# Write some warning notes
|
||
|
|
preamble = open(os.path.join(root, "preamble.tex"), "w")
|
||
|
|
|
||
|
|
preamble.write("\\section{Preamble}")
|
||
|
|
preamble.write("This procedure list should be considered incomplete and it is not intended for real world use. It is automatically generated documentation intended for FlightGear flight simulator.")
|
||
|
|
|
||
|
|
documentation_tex = open(os.path.join(root, source['dir'] + "_documentation.tex"), "w")
|
||
|
|
documentation_tex.write("\\documentclass{article}\n")
|
||
|
|
documentation_tex.write("\\usepackage{hyperref}\n\usepackage{graphicx}\n"
|
||
|
|
"\\usepackage{fancyhdr}\\pagestyle{fancy}\n\\lfoot{Intended for FlightGear}\\rfoot{Not for real world use!}")
|
||
|
|
documentation_tex.write("\\title{"+source['name']+" documentation}\n")
|
||
|
|
documentation_tex.write("\\author{FlightGear team}\n")
|
||
|
|
documentation_tex.write("\\renewcommand{\\familydefault}{\\sfdefault}")
|
||
|
|
documentation_tex.write("\\begin{document}\n")
|
||
|
|
documentation_tex.write("\\maketitle\n")
|
||
|
|
|
||
|
|
checklist_tex = open(os.path.join(root, source['dir'] + "_checklist.tex"), "w")
|
||
|
|
checklist_tex.write("\\documentclass{article}\n")
|
||
|
|
checklist_tex.write("\\usepackage{fancyhdr}\\pagestyle{fancy}\n\\lfoot{Intended for FlightGear}\\rfoot{Not for real world use!}")
|
||
|
|
checklist_tex.write("\\begin{document}\n")
|
||
|
|
|
||
|
|
# If there's a thumbnail available, copy it into the documentation and include it
|
||
|
|
# just before the beginning of the document, just after the title.
|
||
|
|
if(source['thumbnail']):
|
||
|
|
shutil.copyfile(FG_ROOT + "Aircraft/" + source['dir'] + "/thumbnail.jpg",
|
||
|
|
os.path.join(root, "thumbnail.jpg"))
|
||
|
|
documentation_tex.write("\\begin{figure}[h!]\\centering")
|
||
|
|
documentation_tex.write("\\includegraphics[width=5cm,height=5cm,keepaspectratio]{thumbnail.jpg}")
|
||
|
|
documentation_tex.write("\\end{figure}")
|
||
|
|
documentation_tex.write("\\tableofcontents\n")
|
||
|
|
documentation_tex.write("\\input{preamble.tex}\n")
|
||
|
|
|
||
|
|
# If any additional documentation, copy it to the procedures directory and
|
||
|
|
# add it to the procedures.tex file
|
||
|
|
if source['extra_documentation']:
|
||
|
|
extra_docs_dir = os.path.join(FG_ROOT + "Aircraft/" + source['dir'], "Docs")
|
||
|
|
extra_docs_dest_dir = os.path.join("procedures/" + source['name'], "Docs")
|
||
|
|
if os.path.isdir(extra_docs_dest_dir):
|
||
|
|
shutil.rmtree(extra_docs_dest_dir)
|
||
|
|
shutil.copytree(extra_docs_dir, extra_docs_dest_dir)
|
||
|
|
documentation_tex.write("\\input{Docs/" + source['dir'] + "_documentation}\n")
|
||
|
|
|
||
|
|
# If there are any checklist files parsed, write a title
|
||
|
|
if(source['sources'] > 0):
|
||
|
|
documentation_tex.write("\\input{checklists.tex}\n")
|
||
|
|
checklists_tex = open(os.path.join(root, "checklists.tex"), "w")
|
||
|
|
checklists_tex.write("\\section{Checklists}\n")
|
||
|
|
|
||
|
|
# and the checklists themselves.
|
||
|
|
for xmlfile in source['sources']:
|
||
|
|
tree = etree.parse(xmlfile)
|
||
|
|
chkl_root = tree.getroot()
|
||
|
|
for checklist in chkl_root:
|
||
|
|
if checklist.tag != "checklist":
|
||
|
|
print "Unrecognised tag in", xmlfile
|
||
|
|
continue
|
||
|
|
title = checklist.find("title")
|
||
|
|
if title is None:
|
||
|
|
title = "Untitled"
|
||
|
|
else:
|
||
|
|
title = tex_escape(title.text)
|
||
|
|
items = []
|
||
|
|
|
||
|
|
for item in checklist.findall("item"):
|
||
|
|
name = item.find("name")
|
||
|
|
value = item.find("value")
|
||
|
|
if name is None or value is None:
|
||
|
|
continue
|
||
|
|
if name.text is None or value.text is None:
|
||
|
|
continue
|
||
|
|
items.append({
|
||
|
|
'name': tex_escape(name.text),
|
||
|
|
'value': tex_escape(value.text)
|
||
|
|
})
|
||
|
|
filename = title + ".tex"
|
||
|
|
filename = filename.replace("/", "_")
|
||
|
|
filename = filename.replace(" ", "_")
|
||
|
|
checklists_tex.write("\\subsection{" + title + "}\n")
|
||
|
|
checklist_tex.write("\\section*{" + title + "}\n")
|
||
|
|
checklists_tex.write("\\input{checklists/"+filename+"}\n")
|
||
|
|
checklist_tex.write("\\input{checklists/"+filename+"}\n")
|
||
|
|
f = open(os.path.join(checklists_path,filename), "w")
|
||
|
|
if len(items) > 0:
|
||
|
|
f.write("\\begin{description}\n")
|
||
|
|
for item in items:
|
||
|
|
f.write("\\item["+item['name']+"] \dotfill " + item['value']+"\n")
|
||
|
|
f.write("\\end{description}\n")
|
||
|
|
checklist_tex.write("\\clearpage\n")
|
||
|
|
|
||
|
|
documentation_tex.write("\\end{document}\n")
|
||
|
|
checklist_tex.write("\\end{document}\n")
|
||
|
|
|
||
|
|
def gather_aircraft_metadata(directory = FG_ROOT + "Aircraft"):
|
||
|
|
aircrafts = []
|
||
|
|
for aircraft_directory in os.listdir(directory):
|
||
|
|
for rootfile in os.listdir(os.path.join(directory, aircraft_directory)):
|
||
|
|
rootfile = os.path.join(os.path.join(directory, aircraft_directory), rootfile)
|
||
|
|
if not (os.path.isfile(rootfile) and rootfile.endswith("-set.xml")):
|
||
|
|
continue
|
||
|
|
# Houston, we found a set.xml. Let's read it. This implies an aircraft
|
||
|
|
tree = etree.parse(rootfile)
|
||
|
|
name = aircraft_directory
|
||
|
|
|
||
|
|
|
||
|
|
for descr in tree.iter('description'):
|
||
|
|
name = descr.text
|
||
|
|
break
|
||
|
|
aircraft = {
|
||
|
|
'sources' : [],
|
||
|
|
'name' : name,
|
||
|
|
'dir' : aircraft_directory,
|
||
|
|
'thumbnail': False,
|
||
|
|
'extra_documentation': False
|
||
|
|
}
|
||
|
|
# Check if the aircraft provides additional documentation
|
||
|
|
aircraft['extra_documentation'] = os.path.isdir(os.path.join(directory,
|
||
|
|
aircraft_directory
|
||
|
|
+ "/Docs/")) & os.path.isfile(os.path.join(directory, aircraft_directory
|
||
|
|
+ "/Docs/"
|
||
|
|
+ aircraft_directory
|
||
|
|
+ "_documentation.tex"))
|
||
|
|
# Check if there is a thumbnail
|
||
|
|
if os.path.isfile(os.path.join(directory, aircraft_directory + "/thumbnail.jpg")):
|
||
|
|
aircraft['thumbnail'] = True
|
||
|
|
|
||
|
|
for checklist in tree.iter('checklists'):
|
||
|
|
aircraft['sources'].append(os.path.join(directory, aircraft_directory, checklist.attrib['include']))
|
||
|
|
if 'include' in tree.getroot().attrib:
|
||
|
|
try:
|
||
|
|
tree = etree.parse(os.path.join(directory, aircraft_directory) + "/" + tree.getroot().attrib['include'])
|
||
|
|
for checklist in tree.iter('checklists'):
|
||
|
|
if 'include' in checklist.attrib:
|
||
|
|
aircraft['sources'].append(os.path.join(directory, aircraft_directory, checklist.attrib['include']))
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if(len(aircraft['sources']) > 0 or aircraft['extra_documentation']):
|
||
|
|
aircrafts.append(aircraft)
|
||
|
|
|
||
|
|
# Check if there are checklists that can be parsed
|
||
|
|
# for root, dirs, files in os.walk(os.path.join(directory, aircraft_directory), topdown=False):
|
||
|
|
# for name in files:
|
||
|
|
# if(name.endswith("checklists.xml")):
|
||
|
|
# aircraft['sources'].append(os.path.join(root,name))
|
||
|
|
|
||
|
|
return aircrafts
|
||
|
|
|
||
|
|
|
||
|
|
def compile_airplane_latex(source):
|
||
|
|
"""
|
||
|
|
Compile the procedures using pdflatex
|
||
|
|
"""
|
||
|
|
wd = os.getcwd()
|
||
|
|
root = os.path.join("procedures", source['name'])
|
||
|
|
os.chdir(root)
|
||
|
|
with open(os.devnull, "w") as fnull:
|
||
|
|
mainfile = source['dir'] + "_documentation.tex"
|
||
|
|
call(["pdflatex", "-interaction", "nonstopmode", mainfile], stdout=fnull,stderr=fnull)
|
||
|
|
call(["pdflatex", "-interaction", "nonstopmode", mainfile], stdout=fnull,stderr=fnull)
|
||
|
|
mainfile = source['dir'] + "_checklist.tex"
|
||
|
|
call(["pdflatex", "-interaction", "nonstopmode", mainfile], stdout=fnull,stderr=fnull)
|
||
|
|
call(["pdflatex", "-interaction", "nonstopmode", mainfile], stdout=fnull,stderr=fnull)
|
||
|
|
os.chdir(wd)
|
||
|
|
|
||
|
|
|
||
|
|
def generate_airplane_documentation():
|
||
|
|
# First generate the index of what should be included in the
|
||
|
|
# documentation
|
||
|
|
aircrafts = gather_aircraft_metadata("../Aircraft")
|
||
|
|
if(len(aircrafts) == 0 ):
|
||
|
|
print "No aircraft found; wrong directory?"
|
||
|
|
if(not os.path.isdir("procedures")):
|
||
|
|
os.mkdir("procedures")
|
||
|
|
|
||
|
|
# Then generate the documentation per airplane
|
||
|
|
for aircraft in aircrafts:
|
||
|
|
generate_airplane_latex(aircraft)
|
||
|
|
compile_airplane_latex(aircraft)
|
||
|
|
|
||
|
|
# The main procedure
|
||
|
|
generate_airplane_documentation()
|