bmtool 0.5.9__py3-none-any.whl → 0.5.9.5__py3-none-any.whl

bmtool/bmplot.py

@@ -405,6 +405,76 @@ def connection_histogram(config=None,nodes=None,edges=None,sources=[],targets=[]
         tids = []
     util.relation_matrix(config,nodes,edges,sources,targets,sids,tids,prepend_pop,relation_func=connection_pair_histogram,synaptic_info=synaptic_info)
 
+def connection_distance(config: str,source: str,target: str,
+                        source_cell_id: int,target_id_type: str) -> None:
+    """
+    Plots the 3D spatial distribution of target nodes relative to a source node
+    and a histogram of distances from the source node to each target node.
+
+    Parameters:
+    ----------
+    config: (str) A BMTK simulation config 
+    sources: (str) network name(s) to plot
+    targets: (str) network name(s) to plot
+    source_cell_id : (int) ID of the source cell for calculating distances to target nodes.
+    target_id_type : (str) A string to filter target nodes based off the target_query.
+
+    """
+    if not config:
+        raise Exception("config not defined")
+    if not source or not target:
+        raise Exception("Sources or targets not defined")
+    #if source != target:
+        #raise Exception("Code is setup for source and target to be the same! Look at source code for function to add feature")
+    
+    # Load nodes and edges based on config file
+    nodes, edges = util.load_nodes_edges_from_config(config)
+    
+    edge_network = source + "_to_" + target
+    node_network = source
+
+    # Filter edges to obtain connections originating from the source node
+    edge = edges[edge_network]
+    edge = edge[edge['source_node_id'] == source_cell_id]
+    if target_id_type:
+        edge = edge[edge['target_query'].str.contains(target_id_type, na=False)]
+
+    target_node_ids = edge['target_node_id']
+
+    # Filter nodes to obtain only the target and source nodes
+    node = nodes[node_network]
+    target_nodes = node.loc[node.index.isin(target_node_ids)]
+    source_node = node.loc[node.index == source_cell_id]
+
+    # Calculate distances between source node and each target node
+    target_positions = target_nodes[['pos_x', 'pos_y', 'pos_z']].values
+    source_position = np.array([source_node['pos_x'], source_node['pos_y'], source_node['pos_z']]).ravel()  # Ensure 1D shape
+    distances = np.linalg.norm(target_positions - source_position, axis=1)
+
+    # Plot positions of source and target nodes in 3D space
+    fig = plt.figure(figsize=(8, 6)) 
+    ax = fig.add_subplot(111, projection='3d')
+
+    ax.scatter(target_nodes['pos_x'], target_nodes['pos_y'], target_nodes['pos_z'], c='blue', label="target cells")
+    ax.scatter(source_node['pos_x'], source_node['pos_y'], source_node['pos_z'], c='red', label="source cell")
+
+    # Optional: Add text annotations for distances
+    # for i, distance in enumerate(distances):
+    #     ax.text(target_nodes['pos_x'].iloc[i], target_nodes['pos_y'].iloc[i], target_nodes['pos_z'].iloc[i],
+    #             f'{distance:.2f}', color='black', fontsize=8, ha='center')
+
+    plt.legend()
+    plt.show()
+
+    # Plot distances in a separate 2D plot
+    plt.figure(figsize=(8, 6)) 
+    plt.hist(distances, bins=20, color='blue', edgecolor='black')
+    plt.xlabel("Distance")
+    plt.ylabel("Count")
+    plt.title("Distance from Source Node to Each Target Node")
+    plt.grid(True)
+    plt.show()
+
 def edge_histogram_matrix(config=None,sources = None,targets=None,sids=None,tids=None,no_prepend_pop=None,edge_property = None,time = None,time_compare = None,report=None,title=None,save_file=None):
     """
     write about function here

{bmtool-0.5.9.dist-info → bmtool-0.5.9.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bmtool
-Version: 0.5.9
+Version: 0.5.9.5
 Summary: BMTool
 Home-page: https://github.com/cyneuro/bmtool
 Download-URL: 

{bmtool-0.5.9.dist-info → bmtool-0.5.9.5.dist-info}/RECORD

@@ -1,7 +1,7 @@
 bmtool/SLURM.py,sha256=AX5MKV7dD-XwS8SROnW1IyesZ3jDwMf0txO6mHxTbuw,13694
 bmtool/__init__.py,sha256=ZStTNkAJHJxG7Pwiy5UgCzC4KlhMS5pUNPtUJZVwL_Y,136
 bmtool/__main__.py,sha256=TmFkmDxjZ6250nYD4cgGhn-tbJeEm0u-EMz2ajAN9vE,650
-bmtool/bmplot.py,sha256=XN7XPhhVFv6352KciBiPN59wU9x0hoNQ67caLn1ZTZk,47931
+bmtool/bmplot.py,sha256=pdwj1BG-NjVKb3sb-Sya2_9AjkqMm_iooQbEOuy5l3w,50931
 bmtool/connectors.py,sha256=8FbQXewdI9Q0Dlx6cVbM3hPKt9ead4WEDkG22tyKa2o,72187
 bmtool/graphs.py,sha256=K8BiughRUeXFVvAgo8UzrwpSClIVg7UfmIcvtEsEsk0,6020
 bmtool/manage.py,sha256=_lCU0qBQZ4jSxjzAJUd09JEetb--cud7KZgxQFbLGSY,657
@@ -16,9 +16,9 @@ bmtool/util/commands.py,sha256=zJF-fiLk0b8LyzHDfvewUyS7iumOxVnj33IkJDzux4M,64396
 bmtool/util/util.py,sha256=24E5rUoDU86nqypDF4uZJkuJKO1BrwrQE8lZzAxu1kw,56770
 bmtool/util/neuron/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bmtool/util/neuron/celltuner.py,sha256=xSRpRN6DhPFz4q5buq_W8UmsD7BbUrkzYBEbKVloYss,87194
-bmtool-0.5.9.dist-info/LICENSE,sha256=qrXg2jj6kz5d0EnN11hllcQt2fcWVNumx0xNbV05nyM,1068
-bmtool-0.5.9.dist-info/METADATA,sha256=uqQ0aU9ChnJupHjVXUpTzu0IEfesh6rfY2etN5qg4zM,24129
-bmtool-0.5.9.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-bmtool-0.5.9.dist-info/entry_points.txt,sha256=0-BHZ6nUnh0twWw9SXNTiRmKjDnb1VO2DfG_-oprhAc,45
-bmtool-0.5.9.dist-info/top_level.txt,sha256=gpd2Sj-L9tWbuJEd5E8C8S8XkNm5yUE76klUYcM-eWM,7
-bmtool-0.5.9.dist-info/RECORD,,
+bmtool-0.5.9.5.dist-info/LICENSE,sha256=qrXg2jj6kz5d0EnN11hllcQt2fcWVNumx0xNbV05nyM,1068
+bmtool-0.5.9.5.dist-info/METADATA,sha256=v8IKa7wbwiu3s8oA-VxmYBR1xhkicRoAgakcotSpg0E,24131
+bmtool-0.5.9.5.dist-info/WHEEL,sha256=a7TGlA-5DaHMRrarXjVbQagU3Man_dCnGIWMJr5kRWo,91
+bmtool-0.5.9.5.dist-info/entry_points.txt,sha256=0-BHZ6nUnh0twWw9SXNTiRmKjDnb1VO2DfG_-oprhAc,45
+bmtool-0.5.9.5.dist-info/top_level.txt,sha256=gpd2Sj-L9tWbuJEd5E8C8S8XkNm5yUE76klUYcM-eWM,7
+bmtool-0.5.9.5.dist-info/RECORD,,

{bmtool-0.5.9.dist-info → bmtool-0.5.9.5.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.3.0)
+Generator: setuptools (75.4.0)
 Root-Is-Purelib: true
 Tag: py3-none-any