move ps to feynml

docs/source/interfaces/madgraph/index.rst

@@ -1,7 +1,20 @@
+=======================================
+MadGraph Post-Script Reader
+=======================================
+
+Transforms a MadGraph postscript file into a FeynML file.
+
+.. code-block:: python
+
+    from feynml.interface.madgraph.ps import ps_to_feynml
+    fml = ps_to_feynml("tests/interface/lo.ps")
+
 =======================================
 MadGraph Export Plugin Integration
 =======================================
 
+Export FeynML files directly from MadGraph by following these steps:
+
 .. code-block:: bash
 
    pip install --upgrade feynml[interfaces]

pyfeyn2/render/all.py

@@ -9,13 +9,13 @@
 from matplotlib import pyplot as plt
 from pylatex import Figure, NoEscape, SubFigure
 
-from pyfeyn2.render.eps.madgraph import MadGraphRender
 from pyfeyn2.render.js.mermaid import MermaidRender
 from pyfeyn2.render.latex.dot import DotRender
 from pyfeyn2.render.latex.feynmp import FeynmpRender
 from pyfeyn2.render.latex.latex import LatexRender
 from pyfeyn2.render.latex.tikzfeynman import TikzFeynmanRender
 from pyfeyn2.render.mpl.feynmanrender import FeynmanRender
+from pyfeyn2.render.ps.madgraph import MadGraphRender
 from pyfeyn2.render.pyx.pyxrender import PyxRender
 from pyfeyn2.render.text.ascii import ASCIIRender
 from pyfeyn2.render.text.asciipdf import ASCIIPDFRender

pyfeyn2/render/latex/latex.py

@@ -76,8 +76,8 @@ def render(
             # os.rename(tfile + ".pdf", file)
         # if delete:
         #    os.remove(tfile + ".pdf")
-        if clean_up and temp_dir:
-            temp_dir.cleanup()
         if show:
             display(wi)
+        if clean_up and temp_dir:
+            temp_dir.cleanup()
         return wi

pyfeyn2/render/latex/tikzfeynman.py

@@ -16,6 +16,7 @@
 type_map = {
     "gluon": ["gluon"],
     "ghost": ["ghost"],
+    "anti ghost": ["ghost"],
     "photon": ["photon"],
     "boson": ["photon"],
     "fermion": ["fermion"],

tests/interface/test_ps.py

@@ -0,0 +1,43 @@
+from feynml.interface.madgraph.ps import ps_to_feynml
+
+
+def test_load_ps_lo():
+    fml = ps_to_feynml("tests/interface/lo.ps")
+    assert fml != None
+    assert len(fml.diagrams) == 2
+    fml.render()
+
+
+def test_load_ps_born():
+    fml = ps_to_feynml("tests/interface/born.ps")
+    assert fml != None
+    assert len(fml.diagrams) == 3
+    fml.render()
+
+
+def test_load_ps_born_matrix():
+    fml = ps_to_feynml("tests/interface/born_matrix.ps")
+    assert fml != None
+    assert len(fml.diagrams) == 1
+    fml.render()
+
+
+def test_load_ps_matrix_1():
+    fml = ps_to_feynml("tests/interface/matrix_1.ps")
+    assert fml != None
+    assert len(fml.diagrams) == 6 + 6 + 4
+    fml.render()
+
+
+def test_load_ps_matrix_2():
+    fml = ps_to_feynml("tests/interface/matrix_2.ps")
+    assert fml != None
+    assert len(fml.diagrams) == 5
+    fml.render()
+
+
+def test_load_ps_loop_matrix():
+    fml = ps_to_feynml("tests/interface/loop_matrix.ps")
+    assert fml != None
+    assert len(fml.diagrams) == 6 + 5
+    fml.render()