Source code for pm4py.algo.discovery.inductive.base_case.single_activity

from pm4py.algo.discovery.inductive.base_case.abc import BaseCase
from pm4py.algo.discovery.inductive.dtypes.im_ds import (
    IMDataStructureUVCL,
    IMDataStructureDFG,
)
from pm4py.objects.process_tree.obj import ProcessTree
from typing import Optional, Dict, Any




class SingleActivityBaseCaseUVCL(BaseCase[IMDataStructureUVCL]):


    @classmethod
    def holds(
        cls,
        obj=IMDataStructureUVCL,
        parameters: Optional[Dict[str, Any]] = None,
    ) -> bool:
        if len(obj.data_structure.keys()) != 1:
            return False
        if len(list(obj.data_structure.keys())[0]) > 1:
            return False
        return True




    @classmethod
    def leaf(
        cls,
        obj=IMDataStructureUVCL,
        parameters: Optional[Dict[str, Any]] = None,
    ) -> ProcessTree:
        for t in obj.data_structure:
            if t:
                return ProcessTree(label=t[0])
            else:
                return ProcessTree()






class SingleActivityBaseCaseDFG(BaseCase[IMDataStructureDFG]):



    @classmethod
    def holds(
        cls,
        obj=IMDataStructureDFG,
        parameters: Optional[Dict[str, Any]] = None,
    ) -> bool:
        return (
            len(obj.dfg.graph) == 0
            and len(
                set(obj.dfg.start_activities).union(obj.dfg.end_activities)
            )
            == 1
        )




    @classmethod
    def leaf(
        cls,
        obj=IMDataStructureDFG,
        parameters: Optional[Dict[str, Any]] = None,
    ) -> ProcessTree:
        return ProcessTree(label=list(obj.dfg.start_activities)[0])