33c745d997
I moved book_format.py to the root directory so that all of the notebooks do not need to modify the system path to import it. It modifies the path on import so that all of the code in ./code can then be accessed. Altered links to nbviewer to account for no longer using subdirectories.
82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
from __future__ import print_function
|
|
import io
|
|
from IPython.nbformat import current
|
|
import sys
|
|
|
|
|
|
def remove_formatting(nb):
|
|
w = nb['worksheets']
|
|
node = w[0]
|
|
c = node['cells']
|
|
for i in range (len(c)):
|
|
if 'input' in c[i].keys():
|
|
if c[i]['input'][0:16] == '#format the book':
|
|
del c[i]
|
|
return
|
|
|
|
|
|
def remove_links(nb):
|
|
w = nb['worksheets']
|
|
node = w[0]
|
|
c = node['cells']
|
|
for i in range (len(c)):
|
|
if 'source' in c[i].keys():
|
|
if c[i]['source'][0:19] == '[Table of Contents]':
|
|
del c[i]
|
|
return
|
|
|
|
|
|
def remove_links_add_appendix(nb):
|
|
w = nb['worksheets']
|
|
node = w[0]
|
|
c = node['cells']
|
|
for i in range (len(c)):
|
|
if 'source' in c[i].keys():
|
|
if c[i]['source'][0:19] == '[Table of Contents]':
|
|
c[i]['source'] = '\\appendix'
|
|
return
|
|
|
|
|
|
def merge_notebooks(filenames):
|
|
merged = None
|
|
added_appendix = False
|
|
for fname in filenames:
|
|
with io.open(fname, 'r', encoding='utf-8') as f:
|
|
nb = current.read(f, u'json')
|
|
remove_formatting(nb)
|
|
if not added_appendix and fname[0:8] == 'Appendix':
|
|
remove_links_add_appendix(nb)
|
|
added_appendix = True
|
|
else:
|
|
remove_links(nb)
|
|
if merged is None:
|
|
merged = nb
|
|
else:
|
|
merged.worksheets[0].cells.extend(nb.worksheets[0].cells)
|
|
merged.metadata.name += "_merged"
|
|
|
|
print(current.writes(merged, u'json'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
#merge_notebooks(sys.argv[1:])
|
|
merge_notebooks(
|
|
['../Preface.ipynb',
|
|
'../01_g-h_filter.ipynb',
|
|
'../02_Discrete_Bayes.ipynb',
|
|
'../03_Least_Squares_Filters.ipynb',
|
|
'../04_Gaussians.ipynb',
|
|
'../05_Kalman_Filters.ipynb',
|
|
'../06_Multivariate_Kalman_Filters.ipynb',
|
|
'../07_Kalman_Filter_Math.ipynb',
|
|
'../08_Designing_Kalman_Filters.ipynb',
|
|
'../09_Extended_Kalman_Filters.ipynb',
|
|
'../10_Unscented_Kalman_Filter.ipynb',
|
|
'../11_Ensemble_Kalman_Filters.ipynb',
|
|
'../12_Designing_Nonlinear_Kalman_Filters.ipynb',
|
|
'../13_HInfinity_Filters.ipynb',
|
|
'../14_Smoothing.ipynb',
|
|
'../15_Adaptive_Filtering.ipynb',
|
|
'../Appendix_A_Installation.ipynb',
|
|
'../Appendix_B_Symbols_and_Notations.ipynb'])
|