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

bmtool/bmplot.py

@@ -480,8 +480,14 @@ def plot_connection_info(text, num, source_labels,target_labels, title, syn_info
     # Loop over data dimensions and create text annotations.
     for i in range(num_source):
         for j in range(num_target):
-            edge_info = text[i,j]
-            graph_dict[str(source_labels[i])+'2'+str(target_labels[j])] = edge_info
+            edge_info = text[i, j]
+            
+            # Initialize the dictionary for the source node if not already done
+            if source_labels[i] not in graph_dict:
+                graph_dict[source_labels[i]] = {}
+            
+            # Add edge info for the target node
+            graph_dict[source_labels[i]][target_labels[j]] = edge_info
             if syn_info =='2' or syn_info =='3':
                 if num_source > 8 and num_source <20:
                     fig_text = ax1.text(j, i, edge_info,
@@ -726,71 +732,101 @@ def plot_spikes(nodes, spikes_file,save_file=None):
     
     return
     
-def plot_3d_positions(config=None,populations_list=None,group_by=None,title=None,save_file=None):
+def plot_3d_positions(config=None, populations_list=None, group_by=None, title=None, save_file=None, subset=None):
     """
-    plots a 3D graph of all cells with x,y,z location
-    config: A BMTK simulation config 
-    populations_list: Which network(s) to plot 
-    group_by: How to name cell groups
-    title: plot title
-    save_file: If plot should be saved
+    Plots a 3D graph of all cells with x, y, z location.
+    
+    Parameters:
+    - config: A BMTK simulation config 
+    - populations_list: Which network(s) to plot 
+    - group_by: How to name cell groups
+    - title: Plot title
+    - save_file: If plot should be saved
+    - subset: Take every Nth row. This will make plotting large network graphs easier to see.
     """
     
     if not config:
         raise Exception("config not defined")
-    if populations_list == None:
+    
+    if populations_list is None:
         populations_list = "all"
+    
+    # Set group keys (e.g., node types)
     group_keys = group_by
-    if title == None:
+    if title is None:
         title = "3D positions"
 
+    # Load nodes from the configuration
     nodes = util.load_nodes_from_config(config)
     
+    # Get the list of populations to plot
     if 'all' in populations_list:
         populations = list(nodes)
     else:
         populations = populations_list.split(",")
-
+    
+    # Split group_by into list 
     group_keys = group_keys.split(",")
-    group_keys += (len(populations)-len(group_keys)) * ["node_type_id"] #Extend the array to default values if not enough given
-    fig = plt.figure(figsize=(10,10))
+    group_keys += (len(populations) - len(group_keys)) * ["node_type_id"]  # Extend the array to default values if not enough given
+    if len(group_keys) > 1:
+        raise Exception("Only one group by is supported currently!")
+    
+    fig = plt.figure(figsize=(10, 10))
     ax = fig.add_subplot(projection='3d')
     handles = []
-    for nodes_key,group_key in zip(list(nodes),group_keys):
-        if 'all' not in populations and nodes_key not in populations:
-            continue
-            
-        nodes_df = nodes[nodes_key]
 
+    for pop in (list(nodes)):
+        
+        if 'all' not in populations and pop not in populations:
+            continue
+        
+        nodes_df = nodes[pop]
+        group_key = group_keys[0] 
+        
+        # If group_key is provided, ensure the column exists in the dataframe
         if group_key is not None:
             if group_key not in nodes_df:
-                raise Exception('Could not find column {}'.format(group_key))
+                raise Exception(f"Could not find column '{group_key}' in {pop}")
+            
             groupings = nodes_df.groupby(group_key)
-
             n_colors = nodes_df[group_key].nunique()
-            color_norm = colors.Normalize(vmin=0, vmax=(n_colors-1))
+            color_norm = colors.Normalize(vmin=0, vmax=(n_colors - 1))
             scalar_map = cmx.ScalarMappable(norm=color_norm, cmap='hsv')
-            color_map = [scalar_map.to_rgba(i) for i in range(0, n_colors)]
+            color_map = [scalar_map.to_rgba(i) for i in range(n_colors)]
         else:
             groupings = [(None, nodes_df)]
             color_map = ['blue']
 
+        # Loop over groupings and plot
         for color, (group_name, group_df) in zip(color_map, groupings):
-            if "pos_x" not in group_df: #could also check model type == virtual
-                continue #can't plot them if there isn't an xy coordinate (may be virtual)
-            h = ax.scatter(group_df["pos_x"],group_df["pos_y"],group_df["pos_z"],color=color,label=group_name)
+            if "pos_x" not in group_df or "pos_y" not in group_df or "pos_z" not in group_df:
+                print(f"Warning: Missing position columns in group '{group_name}' for {pop}. Skipping this group.")
+                continue  # Skip if position columns are missing
+
+            # Subset the dataframe by taking every Nth row if subset is provided
+            if subset is not None:
+                group_df = group_df.iloc[::subset]
+
+            h = ax.scatter(group_df["pos_x"], group_df["pos_y"], group_df["pos_z"], color=color, label=group_name)
             handles.append(h)
+    
     if not handles:
+        print("No data to plot.")
         return
+    
+    # Set plot title and legend
     plt.title(title)
     plt.legend(handles=handles)
     
+    # Draw the plot
     plt.draw()
 
+    # Save the plot if save_file is provided
     if save_file:
         plt.savefig(save_file)
-    notebook = is_notebook
-    if notebook == False:
+
+    # Show the plot if running outside of a notebook
+    if not is_notebook:
         plt.show()
 
     return

bmtool/connectors.py

@@ -580,9 +580,11 @@ class ReciprocalConnector(AbstractConnector):
         self.source = source
         self.target = target
         if self.source is None or len(self.source) == 0:
-            raise ValueError("Source nodes do not exists")
+            src_str, trg_str = self.get_nodes_info()
+            raise ValueError(f"{src_str} nodes do not exists")
         if self.target is None or len(self.target) == 0:
-            raise ValueError("Target nodes do not exists")
+            src_str, trg_str = self.get_nodes_info()
+            raise ValueError(f"{trg_str} nodes do not exists")
 
         # Setup nodes
         self.recurrent = is_same_pop(self.source, self.target, quick=self.quick)
@@ -1130,9 +1132,11 @@ class UnidirectionConnector(AbstractConnector):
         self.source = source
         self.target = target
         if self.source is None or len(self.source) == 0:
-            raise ValueError("Source nodes do not exists")
+            src_str, trg_str = self.get_nodes_info()
+            raise ValueError(f"{src_str} nodes do not exists")
         if self.target is None or len(self.target) == 0:
-            raise ValueError("Target nodes do not exists")
+            src_str, trg_str = self.get_nodes_info()
+            raise ValueError(f"{trg_str} nodes do not exists")
         self.n_pair = len(self.source) * len(self.target)
 
     def edge_params(self):
@@ -1179,6 +1183,7 @@ class UnidirectionConnector(AbstractConnector):
                       + src_str + "\n  to " + trg_str,flush=True)
 
         # Make random connections
+
         p_arg = self.p_arg(source, target)
         p = self.p(p_arg)
         possible = p > 0
@@ -1274,8 +1279,9 @@ class GapJunction(UnidirectionConnector):
     def setup_nodes(self, source=None, target=None):
         super().setup_nodes(source=source, target=target)
         if len(self.source) != len(self.target):
-            raise ValueError("Source and target must be the same for "
-                             "gap junction.")
+            src_str, trg_str = self.get_nodes_info()
+            raise ValueError(f"Source and target must be the same for "
+                             f"gap junction. Nodes are {src_str} and {trg_str}")
         self.n_source = len(self.source)
 
     def make_connection(self, source, target, *args, **kwargs):
@@ -1500,8 +1506,9 @@ class OneToOneSequentialConnector(AbstractConnector):
             source, target = target, source
         if self.target_count == 0:
             if source is None or len(source) == 0:
-                raise ValueError(("Target" if self.partition_source else
-                                  "Source") + " nodes do not exists")
+                src_str, trg_str = self.get_nodes_info()
+                raise ValueError((f"{trg_str}" if self.partition_source else
+                                  f"{src_str}") + " nodes do not exists")
             self.source = source
             self.n_source = len(source)
         if target is None or len(target) == 0:

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

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

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

@@ -1,8 +1,8 @@
 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=C0zlEtUWtP7mSYXYqRjvLhKqMp8wRKe241aHUSnep30,46687
-bmtool/connectors.py,sha256=d5Pt66f9UHm8YY2qJ0mtYbASimczOzfviFxZQWkazjs,71804
+bmtool/bmplot.py,sha256=XN7XPhhVFv6352KciBiPN59wU9x0hoNQ67caLn1ZTZk,47931
+bmtool/connectors.py,sha256=8FbQXewdI9Q0Dlx6cVbM3hPKt9ead4WEDkG22tyKa2o,72187
 bmtool/graphs.py,sha256=K8BiughRUeXFVvAgo8UzrwpSClIVg7UfmIcvtEsEsk0,6020
 bmtool/manage.py,sha256=_lCU0qBQZ4jSxjzAJUd09JEetb--cud7KZgxQFbLGSY,657
 bmtool/plot_commands.py,sha256=Tqujyf0c0u8olhiHOMwgUSJXIIE1hgjv6otb25G9cA0,12298
@@ -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.7.2.dist-info/LICENSE,sha256=qrXg2jj6kz5d0EnN11hllcQt2fcWVNumx0xNbV05nyM,1068
-bmtool-0.5.7.2.dist-info/METADATA,sha256=7zLPFl5uVwrzMwlXm_D6iJMrwQUk2oXNXIGqvyzgapM,24131
-bmtool-0.5.7.2.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-bmtool-0.5.7.2.dist-info/entry_points.txt,sha256=0-BHZ6nUnh0twWw9SXNTiRmKjDnb1VO2DfG_-oprhAc,45
-bmtool-0.5.7.2.dist-info/top_level.txt,sha256=gpd2Sj-L9tWbuJEd5E8C8S8XkNm5yUE76klUYcM-eWM,7
-bmtool-0.5.7.2.dist-info/RECORD,,
+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.7.2.dist-info → bmtool-0.5.9.dist-info}/WHEEL

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