Updates

Author: APN-Pucky

docs/source/gallery/feynman/feynman.ipynb

@@ -268,9 +268,7 @@
      ]
     }
    ],
-   "source": [
-    "print(fd)"
-   ]
+   "source": []
   },
   {
    "cell_type": "code",

docs/source/gallery/pyfeyn/pyfeyn.ipynb

@@ -55,6 +55,10 @@
     "empty": "empty dot",
     "cross": "crossed dot",
     "blob": "blob",
+    "star": "star, star points=5, fill",
+    "pentagon": "regular pentagon, pentagon points=5, fill",
+    "ellipse": "ellipse, fill",
+    "diamond": "diamond, fill",
 }
 
 
@@ -88,16 +92,30 @@ def stylize_connect(fd: FeynmanDiagram, c: Connector):
                 warnings.warn(f"Unknown label-side {label_side}")
                 ret += ",edge label=" + c.label
         if get_property_value(style, "momentum-arrow", None) == "true":
-            mas = get_property_value(style, "momentum-arrow-sense", None)
+            dir = get_property_value(style, "momentum-arrow-sense", "1")
+            rev = ""
+            if dir == "-1":
+                rev = "reversed "
+            elif dir == "1":
+                rev = ""
+            else:
+                warn(
+                    "momentum-arrow=true but momentum-arrow-sense is not 1 or -1, ignoring momentum-arrow-sense"
+                )
+
+            mas = get_property_value(style, "momentum-arrow-side", "1")
             if mas == "-1":
-                ret += ",momentum'=" + c.momentum.name
+                ret += f",{rev}momentum'=" + (
+                    c.momentum.name if c.momentum is not None else ""
+                )
             elif mas == "0":
                 warn(
-                    "momentum-arrow=true but momentum-arrow-sense=0, ignoring momentum-arrow"
+                    "momentum-arrow=true but momentum-arrow-side=0, ignoring momentum-arrow"
                 )
-                pass
             else:
-                ret += ",momentum=" + c.momentum.name
+                ret += f",{rev}momentum=" + (
+                    c.momentum.name if c.momentum is not None else ""
+                )
 
         if style.opacity is not None and style.opacity != "":
             ret += ",opacity=" + str(style.opacity)
@@ -254,6 +272,7 @@ def valid_styles(cls) -> bool:
             "bend-min-distance",
             "momentum-arrow",
             "momentum-arrow-sense",
+            "momentum-arrow-side",
             "double-distance",
             "label-side",
         ]

pyfeyn2/render/latex/tikzfeynman.py

@@ -134,8 +134,8 @@ def render(
             if l.y > maxy:
                 maxy = l.y
 
-        scalex = 1.0 / (maxx - minx) * buffer
-        scaley = 1.0 / (maxy - miny) * buffer
+        scalex = 1.0 / (maxx - minx) * buffer if maxx - minx != 0 else 1.0
+        scaley = 1.0 / (maxy - miny) * buffer if maxy - miny != 0 else 1.0
 
         kickx = -minx + 1 / scalex * (1 - buffer) / 2.0
         kicky = -miny + 1 / scaley * (1 - buffer) / 2.0

pyfeyn2/render/mpl/feynmanrender.py

@@ -243,4 +243,6 @@ def valid_styles(cls):
             "arrow-size",
             "arrow-angle",
             "arrow-constrict",
+            "momentum-arrow",
+            "momentum-arrow-sense",
         ]

pyfeyn2/render/pyx/pyxrender.py

@@ -68,39 +68,3 @@ def render(
         # Then convert ROOT to LaTeX
         self.set_src_diag(NoEscape(root_to_latex(self.get_src_root())))
         LatexRender.render(self, file, show, resolution, width, height, clean_up)
-
-    @classmethod
-    def valid_styles(cls) -> bool:
-        return super().valid_styles() + [
-            "line",
-            "symbol",
-            "symbol-size",
-            "color",
-            "opacity",
-            "bend-direction",
-            "bend-in",
-            "bend-out",
-            "bend-loop",
-            "bend-min-distance",
-            "momentum-arrow",
-            "momentum-arrow-sense",
-            "double-distance",
-            "label-side",
-        ]
-
-    @classmethod
-    def valid_attributes(cls) -> List[str]:
-        return super().valid_attributes() + [
-            "x",
-            "y",
-            "label",
-            "style",
-        ]
-
-    @classmethod
-    def valid_types(cls) -> List[str]:
-        return super().valid_types() + []
-
-    @classmethod
-    def valid_shapes(cls) -> List[str]:
-        return super().valid_types() + []