o
    hl                     @   s.   d Z dgZddlZejdddd	ddZdS )
z
Ego graph.
	ego_graph    NT)preserve_all_attrsreturns_graph   Fc           	      C   s   |r!|durt j|  |||d\}}n'tt j|  ||d}n|dur1t j| |||d\}}n
tt j| ||d}| | }|sI|| |S )a  Returns induced subgraph of neighbors centered at node n within
    a given radius.

    Parameters
    ----------
    G : graph
      A NetworkX Graph or DiGraph

    n : node
      A single node

    radius : number, optional
      Include all neighbors of distance<=radius from n.

    center : bool, optional
      If False, do not include center node in graph

    undirected : bool, optional
      If True use both in- and out-neighbors of directed graphs.

    distance : key, optional
      Use specified edge data key as distance.  For example, setting
      distance='weight' will use the edge weight to measure the
      distance from the node n.

    Notes
    -----
    For directed graphs D this produces the "out" neighborhood
    or successors.  If you want the neighborhood of predecessors
    first reverse the graph with D.reverse().  If you want both
    directions use the keyword argument undirected=True.

    Node, edge, and graph attributes are copied to the returned subgraph.
    N)cutoffweight)r   )nxsingle_source_dijkstrato_undirecteddict"single_source_shortest_path_lengthsubgraphcopyremove_node)	Gnradiuscenter
undirecteddistancesp_H r   k/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/networkx/generators/ego.pyr   
   s"   $

)r   TFN)__doc____all__networkxr   _dispatchabler   r   r   r   r   <module>   s
    