nettracer3d 0.7.3__py3-none-any.whl → 0.7.5__py3-none-any.whl

nettracer3d/network_analysis.py

@@ -4,7 +4,6 @@ import json
 import tifffile
 import numpy as np
 from networkx.algorithms import community
-from community import community_louvain
 from scipy.ndimage import zoom
 from scipy import ndimage
 from . import node_draw
@@ -239,183 +238,6 @@ def weighted_network(excel_file_path):
 
     return G, edge_weights
 
-def community_partition_simple(nodes, network, centroids = None, down_factor = None, color_code = False, directory = None):
-
-    G = network
-
-    communities = list(nx.community.label_propagation_communities(G))
-    partition = {}
-    for i, community in enumerate(communities):
-        for node in community:
-            partition[node] = i + 1
-    print(partition)
-
-
-    print("Generating excel notebook containing community of each node...")
-    if type(nodes) == str:
-        nodes = tifffile.imread(nodes)
-
-        if len(np.unique(nodes)) < 3:
-        
-            structure_3d = np.ones((3, 3, 3), dtype=int)
-            nodes, num_nodes = ndimage.label(nodes, structure=structure_3d)
-
-    # Convert dictionary to DataFrame with keys as index and values as a column
-    df = pd.DataFrame.from_dict(partition, orient='index', columns=['CommunityID'])
-
-    # Rename the index to 'Node ID'
-    df.index.name = 'Node ID'
-
-    if directory is None:
-
-        # Save DataFrame to Excel file
-        df.to_excel('communities.xlsx', engine='openpyxl')
-        print("Community info saved to communities.xls")
-    else:
-        df.to_excel(f'{directory}/communities.xlsx', engine='openpyxl')
-        print(f"Community info saved to {directory}/communities.xls")
-
-
-    print("Drawing overlay containing community labels for each node...")
-
-    if centroids is None:
-        centroids = _find_centroids(nodes, down_factor = down_factor)
-
-    labels = node_draw.degree_draw(partition, centroids, nodes)
-
-    if directory is None:
-        tifffile.imwrite("community_labels.tif", labels)
-        print("Community labels saved to community_labels.tif")
-    else:
-        tifffile.imwrite(f"{directory}/community_labels.tif", labels)
-        print(f"Community labels saved to {directory}/community_labels.tif")
-
-    print("Drawing overlay containing grayscale community labels for each node...")
-
-    masks = node_draw.degree_infect(partition, nodes)
-
-    if color_code:
-        print("And drawing color coded community labels...")
-        colored_masks = _color_code(masks)
-
-        if directory is None:
-
-            tifffile.imwrite("community_labels_colorcoded.tif", colored_masks)
-            print("Color coded communities saved to community_labels_colorcoded.tif")
-
-        else:
-            tifffile.imwrite(f"{directory}/community_labels_colorcoded.tif", colored_masks)
-            print(f"Color coded communities saved to {directory}/community_labels_colorcoded.tif")
-
-
-    if directory is None:
-
-        tifffile.imwrite("community_labels_grayscale.tif", masks)
-        print("Grayscale labeled communities saved to community_labels_grayscale.tif")
-
-    else:
-        tifffile.imwrite(f"{directory}/community_labels_grayscale.tif", masks)
-        print(f"Grayscale labeled communities saved to {directory}/community_labels_grayscale.tif")
-
-    return partition 
-
-
-def community_partition(nodes, network, centroids = None, down_factor = None, color_code = False, directory = None):
-
-    G, edge_weights = weighted_network(network)
-
-    G = nx.Graph()
-
-    # Find the maximum and minimum edge weights
-    max_weight = max(weight for edge, weight in edge_weights.items())
-    min_weight = min(weight for edge, weight in edge_weights.items())
-
-    if max_weight > 1:
-        # Normalize edge weights to the range [0.1, 1.0]
-        normalized_weights = {edge: 0.1 + 0.9 * ((weight - min_weight) / (max_weight - min_weight)) for edge, weight in edge_weights.items()}
-    else:
-        normalized_weights = {edge: 0.1 for edge, weight in edge_weights.items()}
-
-    # Add edges to the graph with normalized weights
-    for edge, normalized_weight in normalized_weights.items():
-        G.add_edge(edge[0], edge[1], weight=normalized_weight)
-
-    # Perform Louvain community detection
-    partition = community_louvain.best_partition(G)
-
-    for key in partition.keys():
-       partition[key] = partition[key] + 1 
-
-
-    print("Generating excel notebook containing community of each node...")
-    if type(nodes) == str:
-        nodes = tifffile.imread(nodes)
-
-        if len(np.unique(nodes)) < 3:
-        
-            structure_3d = np.ones((3, 3, 3), dtype=int)
-            nodes, num_nodes = ndimage.label(nodes, structure=structure_3d)
-
-    # Convert dictionary to DataFrame with keys as index and values as a column
-    df = pd.DataFrame.from_dict(partition, orient='index', columns=['CommunityID'])
-
-    # Rename the index to 'Node ID'
-    df.index.name = 'Node ID'
-
-    if directory is None:
-
-        # Save DataFrame to Excel file
-        df.to_excel('communities.xlsx', engine='openpyxl')
-        print("Community info saved to communities.xls")
-    else:
-        df.to_excel(f'{directory}/communities.xlsx', engine='openpyxl')
-        print(f"Community info saved to {directory}/communities.xls")
-
-
-    print("Drawing overlay containing community labels for each node...")
-
-    if centroids is None:
-        centroids = _find_centroids(nodes, down_factor = down_factor)
-
-    labels = node_draw.degree_draw(partition, centroids, nodes)
-
-    if directory is None:
-        tifffile.imwrite("community_labels.tif", labels)
-        print("Community labels saved to community_labels.tif")
-    else:
-        tifffile.imwrite(f"{directory}/community_labels.tif", labels)
-        print(f"Community labels saved to {directory}/community_labels.tif")
-
-    print("Drawing overlay containing grayscale community labels for each node...")
-
-    masks = node_draw.degree_infect(partition, nodes)
-
-    if color_code:
-        print("And drawing color coded community labels...")
-        colored_masks = _color_code(masks)
-
-        if directory is None:
-
-            tifffile.imwrite("community_labels_colorcoded.tif", colored_masks)
-            print("Color coded communities saved to community_labels_colorcoded.tif")
-
-        else:
-            tifffile.imwrite(f"{directory}/community_labels_colorcoded.tif", colored_masks)
-            print(f"Color coded communities saved to {directory}/community_labels_colorcoded.tif")
-
-
-    if directory is None:
-
-        tifffile.imwrite("community_labels_grayscale.tif", masks)
-        print("Grayscale labeled communities saved to community_labels_grayscale.tif")
-
-    else:
-        tifffile.imwrite(f"{directory}/community_labels_grayscale.tif", masks)
-        print(f"Grayscale labeled communities saved to {directory}/community_labels_grayscale.tif")
-
-    return partition
-
-
 
 def _color_code(grayscale_image):
     """Color code a grayscale array. Currently expects linearly ascending grayscale labels, will crash if there are gaps. (Main use case is grayscale anyway)"""