fas_graph.py

from abc import ABC, abstractmethod
from typing import Iterator, Self

Node = int
Edge = tuple[Node, Node]


class FASGraph(ABC):
    @abstractmethod
    def get_node_labels(self) -> list[str]:
        raise NotImplementedError

    @abstractmethod
    def get_nodes(self) -> list[Node]:
        raise NotImplementedError

    @abstractmethod
    def get_num_nodes(self) -> int:
        raise NotImplementedError

    @abstractmethod
    def get_num_edges(self) -> int:
        raise NotImplementedError

    @abstractmethod
    def get_out_degree(self, node: Node) -> int:
        raise NotImplementedError

    @abstractmethod
    def get_in_degree(self, node: Node) -> int:
        raise NotImplementedError

    @abstractmethod
    def iter_out_neighbors(self, node: Node) -> Iterator[Node]:
        raise NotImplementedError

    @abstractmethod
    def iter_in_neighbors(self, node: Node) -> Iterator[Node]:
        raise NotImplementedError

    @abstractmethod
    def iter_components(self) -> Iterator[Self]:
        raise NotImplementedError

    @abstractmethod
    def remove_runs(self) -> dict[Edge, Edge]:
        raise NotImplementedError

    @abstractmethod
    def remove_2cycles(self) -> list[Edge]:
        raise NotImplementedError

    @abstractmethod
    def remove_self_loops(self) -> list[Edge]:
        raise NotImplementedError

    @abstractmethod
    def is_acyclic(self):
        raise NotImplementedError

    @abstractmethod
    def get_edge_weight(self, source: Node, target: Node) -> int:
        raise NotImplementedError

    @abstractmethod
    def edge_between_components(self, source: Node, target: Node) -> bool:
        raise NotImplementedError

    @abstractmethod
    def add_edges(self, edges: list[Edge]):
        raise NotImplementedError

    @abstractmethod
    def add_edge(self, source: Node, target: Node):
        raise NotImplementedError

    @abstractmethod
    def remove_edge(self, source: Node, target: Node):
        raise NotImplementedError

    @abstractmethod
    def remove_edges(self, edges: list[Edge]):
        raise NotImplementedError

    @classmethod
    @abstractmethod
    def load_from_edge_list(cls, filename: str) -> tuple[Self, dict[str, Node]]:
        raise NotImplementedError

    @classmethod
    @abstractmethod
    def load_from_dot(cls, filename: str) -> tuple[Self, dict[str, Node]]:
        raise NotImplementedError

    @classmethod
    @abstractmethod
    def load_from_adjacency_list(
        cls, filename: str
    ) -> tuple[Self, dict[str, Node]]:
        raise NotImplementedError