Small fixes on notebook generator

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

notebook/gen_latex.py

@@ -111,6 +111,7 @@ class LatexGenerator:
 
         # Writes own header to LaTeX output.
         def output_header(self, write):
+            if len(self.header) > 0:
                 write('\\begin{adjustwidth}{5pt}{5pt}\n')
 
             if 'Description' in self.header:
@@ -119,34 +120,28 @@ class LatexGenerator:
                 write('\\\\\n')
                 write('~\\\\\n')
 
-            if 'Time' in self.header:
-                write('\\textbf{\\footnotesize Time: }')
-                if len(self.header['Time']) > 1:
+            def output_complexity(comp):
+                if comp in self.header:
+                    write('\\textbf{\\footnotesize %s: }' % comp)
+                    if len(self.header[comp]) > 1:
                         write('\n')
                         write('\\begin{compactitem}\n')
-                    for i in self.header['Time']:
+                        for i in self.header[comp]:
                             line = i[1:].split(':')
                             write('\\item{\\footnotesize{%s: $%s$}}\n' % 
                                     (line[0].strip().replace('_', '\\_'), 
                                      line[1].strip()))
                         write('\\end{compactitem}\n')
                     else:
-                    write('\\footnotesize{$%s$}\\\\\n' % self.header['Time'][0])
+                        write('\\footnotesize{$%s$}\n' % self.header[comp][0])
+                    return True
+                return False
 
-            if 'Space' in self.header:
-                write('\\textbf{\\footnotesize Space: }')
-                if len(self.header['Space']) > 1:
-                    write('\n')
-                    write('\\begin{compactitem}\n')
-                    for i in self.header['Space']:
-                        line = i[1:].split(':')
-                        write('\\item{\\footnotesize{%s: $%s$}}\n' % 
-                                (line[0].strip().replace('_', '\\_'), 
-                                 line[1].strip()))
-                    write('\\end{compactitem}\n')
-                else:
-                    write('\\footnotesize{$%s$}\n' % self.header['Space'][0])
+            if output_complexity('Time'):
+                write('~\\\\\n')
+            output_complexity('Space')
 
+            if len(self.header) > 0:
                 write('\\end{adjustwidth}\n')
 
     def __init__(self, tree, output, header):

notebook/header.tex

@@ -52,7 +52,6 @@
     basicstyle=\scriptsize\ttfamily,
     tabsize=2,
     backgroundcolor=\color{light-gray},
-%    frame=t,
     breaklines=true,
     morekeywords={int32\_t, ll}
 }
@@ -61,7 +60,7 @@
 \lstdefinestyle{customvim}{
     basicstyle=\scriptsize\ttfamily,
     tabsize=2,
-%    frame=t,
+    backgroundcolor=\color{light-gray},
     breaklines=true,
     morekeywords={set}
 }