Merge pull request #137 from EmilyBourne/ebourne-133-equation-rendering

Author: JakubAndrysek

mkdoxy/markdown.py

@@ -224,3 +224,22 @@ def render(self, f: MdRenderer, indent: str):
                 f.write("|")
             is_first = False
         f.write("\n\n")
+
+
+class MdInlineEquation(Md):
+    def __init__(self, equation: str):
+        self.equation = equation
+
+    def render(self, f: MdRenderer, indent: str):
+        if self.equation:
+            f.write(rf"\({self.equation}\)")
+
+
+class MdBlockEquation(Md):
+    def __init__(self, equation: str):
+        self.equation = equation
+
+    def render(self, f: MdRenderer, indent: str):
+        f.write("\n")
+        f.write(rf"{indent}\[{self.equation}\]")
+        f.write("\n")

mkdoxy/xml_parser.py

@@ -6,10 +6,12 @@
     Code,
     Md,
     MdBlockQuote,
+    MdBlockEquation,
     MdBold,
     MdCodeBlock,
     MdHeader,
     MdImage,
+    MdInlineEquation,
     MdItalic,
     MdLink,
     MdList,
@@ -277,6 +279,12 @@ def paras(self, p: Element, italic: bool = False) -> [Md]:
 
             elif item.tag == "emphasis":
                 ret.append(MdItalic(self.paras(item)))
+            elif item.tag == "formula" and item.text:
+                equation = item.text.strip("$").strip()
+                if len(p) == 1 and not item.tail:
+                    ret.append(MdBlockEquation(equation))
+                else:
+                    ret.append(MdInlineEquation(equation))
 
             if item.tail:
                 if italic: