update: partly handle hypotheses

Author: ozekik

docs/usage/natural-deduction.md

@@ -88,17 +88,17 @@ print(deriv.latex())
 
 $$
 \begin{prooftree}
-\AxiomC{$A$}
+\AxiomC{$[A]$}
 \RightLabel{$\lor$I}
 \UnaryInfC{$A \lor C$}
-\AxiomC{$B$}
-\RightLabel{$\to$E}
 \AxiomC{$B \to C$}
+\AxiomC{$[B]$}
+\RightLabel{$\to$E}
 \BinaryInfC{$C$}
 \RightLabel{$\lor$I}
 \UnaryInfC{$A \lor C$}
-\RightLabel{$\lor$E}
 \AxiomC{$A \lor B$}
+\RightLabel{$\lor$E}
 \TrinaryInfC{$A \lor C$}
 \end{prooftree}
 $$

mathesis/deduction/natural_deduction/natural_deduction.py

@@ -114,7 +114,11 @@ def latex(self, number=False, arrow=r"\Rightarrow"):
                 tmpl = r"\BinaryInfC{{${}$}}"
             elif len(node.children) == 3:
                 tmpl = r"\TrinaryInfC{{${}$}}"
-            output += label_part + tmpl.format(node.name.fml.latex()) + "\n"
+
+            fml = str(node.name.fml.latex())
+            if hasattr(node, "hyp") and node.hyp:
+                fml = f"[{fml}]"
+            output += label_part + tmpl.format(fml) + "\n"
 
         return """\\begin{{prooftree}}\n{}\\end{{prooftree}}""".format(output)