Fix a lot of stuff

Signed-off-by: Bruno Freitas Tissei <bft15@inf.ufpr.br>

gen_notebook

 #!/bin/bash
 
-#tex_file=$(mktemp)
-python3 notebook/gen_latex.py # --header=notebook/header.tex --output=$tex_file
+if [ "$1" = "DEBUG" ]; then
+  tex_file=result.tex
+  python3 notebook/gen_latex.py --header=notebook/header.tex --output=$tex_file
+else
+  tex_file=$(mktemp)
+  python3 notebook/gen_latex.py --header=notebook/header.tex --output=$tex_file
 
-#pdflatex $tex_file -output-directory . && 
-#pdflatex $tex_file -output-directory .
+  pdflatex $tex_file -output-directory . && 
+  pdflatex $tex_file -output-directory .
 
-#mv tmp.pdf caderno.pdf
-#rm tmp*
+  mv tmp.pdf caderno.pdf
+  rm tmp*
+fi

misc/template.cpp

+/// Template
 #include <bits/stdc++.h>
 
 #define EPS 1e-6

notebook/gen_latex.py

+import argparse
 from pathlib import Path
 
 class Tree:
-
-    # Constructor.
-    # @param dirs list of directories to build Latex over
     def __init__(self, dirs):
         self.dirs = dirs;
         self.tree = {}
         self.build()
 
-    # Sorts the lists on tree leaves.
-    def sort(self, sub):
-        if type(sub) == list:
-            return sorted(sub)
-        else:
-            for i in sub:
-                sub[i] = self.sort(sub[i])
-            return sub
-
     # Builds a tree represented as dict, with structure of files
     # and directories.
     def build(self):
         for di in self.dirs:
-            path_list = Path(di).glob('**/*.cpp')
+            path_list = sorted(list(Path(di).glob('**/*.cpp')))
 
             for path in path_list:
                 branch = str(path).split('/')
-                curr = self.tree
+                node = self.tree
 
                 for i in branch[:-2]:
-                    if not i in curr:
-                        curr[i] = {}
-                    curr = curr[i]
+                    if i not in node:
+                        node[i] = {}
+                    node = node[i]
 
-                if not branch[-2] in curr:
-                    curr[branch[-2]] = []
-                curr[branch[-2]].append(str(branch[-1]))
-
-        self.tree = self.sort(self.tree)
+                if branch[-2] not in node:
+                    node[branch[-2]] = []
+                node[branch[-2]].append(str(branch[-1]))
 
+    # Allows access to tree dict from instance of class.
     def __getitem__(self, arg):
         return self.tree[arg]
 
+    # Allows iteration on tree dict from instance of class.
     def __iter__(self):
         return iter(self.tree)
 
 
 class LatexGenerator:
-    def __init__(self, tree):
-        # TODO: Create output file, add header and finish code parsing and latex
+    def __init__(self, tree, output, header):
+        self.output = open(output, 'w')
+        self.header = open(header, 'r')
+
         self.tree = tree
-        self.hierarchy = ['chapter'] + [i*'sub' + 'section' for i in range(3)]
+        self.files = {}
+        self.hierarchy = ['part', 'chapter'] + [i*'sub' + 'section' for i in range(3)]
+
+        self._add_header()
         self.gen_latex(tree)
+        self.output.write('\end{document}')
+
+    # Adds Latex header to the output.
+    def _add_header(self):
+        lines = self.header.readlines()
+        for i in lines:
+            self.output.write(i)
+        self.output.write('\\newpage\n')
+        self.output.write('\n')
 
     # Prints section title in Latex format.
-    # TODO: Instead of parsing format, get name of section through header of 
-    #       the algorithm/problem file. Capitalization should be used only for
-    #       directories.
     def gen_title_latex(self, content, depth):
-        text = ' '.join(list(map(lambda x : x.capitalize(), 
-                content.split('.')[0].split('_'))))
-        print('\\' + self.hierarchy[depth] + '{' + text + '}')
+        self.output.write('\\' + self.hierarchy[depth] + '{' + content + '}' + '\n')
 
     # Prints code in Latex format.
     def gen_code_latex(self, path):
-        # TODO: Add code parsing funcion and code to latex generation
-        pass
+        # TODO: Add code parsing funcion to extract text
+        self.output.write('\\begin{lstlisting}[style=customcpp]\n')
+
+        with open(path) as f:
+            lines = f.readlines()
+        for i in lines:
+            self.output.write(i)
+
+        self.output.write('\\end{lstlisting}\n')
+        self.output.write('\n')
 
     # Generates Latex for entire tree recursively
     def gen_latex(self, sub, path = '', depth = 0):
         if type(sub) == list:
             for i in sub:
-                self.gen_title_latex(i, depth)
+                self.gen_title_latex(self._parse_file_name(path + i), depth)
                 self.gen_code_latex(path + i)
         else:
             for i in sub:
-                self.gen_title_latex(i, depth)
+                self.gen_title_latex(self._parse_dir_name(i), depth)
                 self.gen_latex(sub[i], path + i + '/', depth + 1)
 
+        #self.output.write('\\newpage\n')
+        self.output.write('\n')
+
+    # Parses name of the section (capitalize)
+    def _parse_dir_name(self, name):
+        return ' '.join(list(map(lambda x : x.capitalize(), name.split('_'))))
+
+    # Parses name of file by getting the first line of the source code.
+    def _parse_file_name(self, name):
+        with open(name) as f:
+            result = f.readline()
+        return result[4:-1]
+
+
+def get_args():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('--header', 
+            action='store', 
+            type=str, 
+            help='Header of Latex file')
+
+    parser.add_argument('--output', 
+            action='store', 
+            type=str, 
+            help='Output Latex file')
+
+    return parser.parse_args()
 
 def main():
-    dirs = [ 'algorithms', 'misc' ]
-    tree = Tree(dirs)
-    tex = LatexGenerator(tree)
+    args = get_args()
+
+    tree = Tree(['algorithms', 'misc'])
+    tex = LatexGenerator(tree, args.output, args.header)
 
 if __name__ == "__main__":
     main()

notebook/header.tex

-\documentclass{article}
+\documentclass[oneside]{book}
 \usepackage{listings}
 \usepackage{titlesec}
 \usepackage[a4paper, total={6in, 8in}]{geometry}