Source code for feynml.connector

from dataclasses import dataclass, field
from typing import Optional

from smpl_util.util import withify

from feynml.labeled import Labeled
from feynml.momentum import Momentum
from feynml.pdgid import PDG
from feynml.styled import Styled
from feynml.vertex import Vertex




@withify()
@dataclass
class Connector(Labeled, Styled, PDG):
    momentum: Optional[Momentum] = field(
        default=None, metadata={"name": "momentum", "type": "Element"}
    )
    """Momentum of the connector"""



    def replace_vertex(self, old_vertex: Vertex, new_vertex: Vertex):
        """Replace the old vertex with the new vertex"""
        raise NotImplementedError


    def goes_into(self, vertex):
        raise NotImplementedError

    def goes_out_of(self, vertex):
        raise NotImplementedError



    def with_tension(self, tension: float):
        """Add tension to the connector"""
        return self.with_style_properties(tension=tension)




    def get_tension(self):
        """Get tension of the connector"""
        return self.get_style_property("tension")




    def with_length(self, length: float):
        """Add length to the connector"""
        return self.with_style_properties(length=length)




    def get_length(self):
        """Get length of the connector"""
        return self.get_style_property("length")