risk-network 0.0.6b1__py3-none-any.whl → 0.0.6b2__py3-none-any.whl

risk/__init__.py

@@ -7,4 +7,4 @@ RISK: RISK Infers Spatial Kinships
 
 from risk.risk import RISK
 
-__version__ = "0.0.6-beta.1"
+__version__ = "0.0.6-beta.2"

risk/network/graph.py

@@ -101,31 +101,32 @@ class NetworkGraph:
 
     def get_domain_colors(
         self,
+        cmap: str = "gist_rainbow",
+        color: Union[str, None] = None,
         min_scale: float = 0.8,
         max_scale: float = 1.0,
         scale_factor: float = 1.0,
         random_seed: int = 888,
-        **kwargs,
     ) -> np.ndarray:
-        """Generate composite colors for domains.
-
-        This method generates composite colors for nodes based on their enrichment scores and transforms
-        them to ensure proper alpha values and intensity. For nodes with alpha == 0, it assigns new colors
-        based on the closest valid neighbors within a specified distance.
+        """Generate composite colors for domains based on enrichment or specified colors.
 
         Args:
-            min_scale (float, optional): Minimum scale for color intensity. Defaults to 0.8.
-            max_scale (float, optional): Maximum scale for color intensity. Defaults to 1.0.
-            scale_factor (float, optional): Exponent for scaling, where values > 1 increase contrast by dimming small
-                values more. Defaults to 1.0.
-            random_seed (int, optional): Seed for random number generation. Defaults to 888.
-            **kwargs: Additional keyword arguments for color generation.
+            cmap (str, optional): Name of the colormap to use for generating domain colors. Defaults to "gist_rainbow".
+            color (str or None, optional): A specific color to use for all generated colors. Defaults to None.
+            min_scale (float, optional): Minimum intensity scale for the colors generated by the colormap.
+                Controls the dimmest colors. Defaults to 0.8.
+            max_scale (float, optional): Maximum intensity scale for the colors generated by the colormap.
+                Controls the brightest colors. Defaults to 1.0.
+            scale_factor (float, optional): Exponent for adjusting the color scaling based on enrichment scores.
+                A higher value increases contrast by dimming lower scores more. Defaults to 1.0.
+            random_seed (int, optional): Seed for random number generation to ensure reproducibility of color assignments.
+                Defaults to 888.
 
         Returns:
-            np.ndarray: Array of transformed colors.
+            np.ndarray: Array of RGBA colors generated for each domain, based on enrichment or the specified color.
         """
         # Get colors for each domain
-        domain_colors = self._get_domain_colors(random_seed=random_seed)
+        domain_colors = self._get_domain_colors(cmap=cmap, color=color, random_seed=random_seed)
         # Generate composite colors for nodes
         node_colors = self._get_composite_node_colors(domain_colors)
         # Transform colors to ensure proper alpha values and intensity
@@ -161,12 +162,16 @@ class NetworkGraph:
         return composite_colors
 
     def _get_domain_colors(
-        self, color: Union[str, None] = None, random_seed: int = 888
+        self,
+        cmap: str = "gist_rainbow",
+        color: Union[str, None] = None,
+        random_seed: int = 888,
     ) -> Dict[str, Any]:
         """Get colors for each domain.
 
         Args:
-            color (Union[str, None], optional): Specific color to use for all domains. If specified, it will overwrite the colormap.
+            cmap (str, optional): The name of the colormap to use. Defaults to "gist_rainbow".
+            color (str or None, optional): A specific color to use for all generated colors. Defaults to None.
             random_seed (int, optional): Seed for random number generation. Defaults to 888.
 
         Returns:
@@ -178,7 +183,7 @@ class NetworkGraph:
         ]
         domains = np.sort(numeric_domains)
         domain_colors = _get_colors(
-            num_colors_to_generate=len(domains), color=color, random_seed=random_seed
+            num_colors_to_generate=len(domains), cmap=cmap, color=color, random_seed=random_seed
         )
         return dict(zip(self.domain_to_nodes.keys(), domain_colors))
 
@@ -273,17 +278,17 @@ def _extract_node_coordinates(G: nx.Graph) -> np.ndarray:
 
 def _get_colors(
     num_colors_to_generate: int = 10,
-    cmap: str = "hsv",
-    random_seed: int = 888,
+    cmap: str = "gist_rainbow",
     color: Union[str, None] = None,
+    random_seed: int = 888,
 ) -> List[Tuple]:
     """Generate a list of RGBA colors from a specified colormap or use a direct color string.
 
     Args:
         num_colors_to_generate (int): The number of colors to generate. Defaults to 10.
-        cmap (str): The name of the colormap to use. Defaults to "hsv".
+        cmap (str, optional): The name of the colormap to use. Defaults to "gist_rainbow".
+        color (str or None, optional): A specific color to use for all generated colors.
         random_seed (int): Seed for random number generation. Defaults to 888.
-        color (str, optional): Specific color to use for all nodes. If specified, it will overwrite the colormap.
             Defaults to None.
 
     Returns:

risk/network/plot.py

@@ -223,7 +223,7 @@ class NetworkPlotter:
             node_color (str, list, tuple, or np.ndarray, optional): Color of the nodes. Can be a single color or an array of colors. Defaults to "white".
             node_edgecolor (str, list, tuple, or np.ndarray, optional): Color of the node edges. Defaults to "black".
             edge_color (str, list, tuple, or np.ndarray, optional): Color of the edges. Defaults to "black".
-            node_alpha (float, optional): Alpha value (transparency) for the nodes. Defaults to 1.0.
+            node_alpha (float, optional): Alpha value (transparency) for the nodes. Defaults to 1.0. Annotated node_color alphas will override this value.
             edge_alpha (float, optional): Alpha value (transparency) for the edges. Defaults to 1.0.
         """
         # Log the plotting parameters
@@ -522,7 +522,7 @@ class NetworkPlotter:
 
     def plot_labels(
         self,
-        perimeter_scale: float = 1.05,
+        scale: float = 1.05,
         offset: float = 0.10,
         font: str = "Arial",
         fontsize: int = 10,
@@ -541,7 +541,7 @@ class NetworkPlotter:
         """Annotate the network graph with labels for different domains, positioned around the network for clarity.
 
         Args:
-            perimeter_scale (float, optional): Scale factor for positioning labels around the perimeter. Defaults to 1.05.
+            scale (float, optional): Scale factor for positioning labels around the perimeter. Defaults to 1.05.
             offset (float, optional): Offset distance for labels from the perimeter. Defaults to 0.10.
             font (str, optional): Font name for the labels. Defaults to "Arial".
             fontsize (int, optional): Font size for the labels. Defaults to 10.
@@ -559,7 +559,7 @@ class NetworkPlotter:
         """
         # Log the plotting parameters
         params.log_plotter(
-            label_perimeter_scale=perimeter_scale,
+            label_perimeter_scale=scale,
             label_offset=offset,
             label_font=font,
             label_fontsize=fontsize,
@@ -637,9 +637,7 @@ class NetworkPlotter:
             valid_indices = [valid_indices[i] for i in selected_indices]
 
         # Calculate the bounding box around the network
-        center, radius = _calculate_bounding_box(
-            self.graph.node_coordinates, radius_margin=perimeter_scale
-        )
+        center, radius = _calculate_bounding_box(self.graph.node_coordinates, radius_margin=scale)
         # Calculate the best positions for labels around the perimeter
         best_label_positions = _calculate_best_label_positions(
             filtered_domain_centroids, center, radius, offset
@@ -666,7 +664,7 @@ class NetworkPlotter:
         nodes: List,
         label: str,
         radial_position: float = 0.0,
-        perimeter_scale: float = 1.05,
+        scale: float = 1.05,
         offset: float = 0.10,
         font: str = "Arial",
         fontsize: int = 10,
@@ -682,7 +680,7 @@ class NetworkPlotter:
             nodes (List[str]): List of node labels to be used for calculating the centroid.
             label (str): The label to be annotated on the network.
             radial_position (float, optional): Radial angle for positioning the label, in degrees (0-360). Defaults to 0.0.
-            perimeter_scale (float, optional): Scale factor for positioning the label around the perimeter. Defaults to 1.05.
+            scale (float, optional): Scale factor for positioning the label around the perimeter. Defaults to 1.05.
             offset (float, optional): Offset distance for the label from the perimeter. Defaults to 0.10.
             font (str, optional): Font name for the label. Defaults to "Arial".
             fontsize (int, optional): Font size for the label. Defaults to 10.
@@ -708,9 +706,7 @@ class NetworkPlotter:
         # Calculate the centroid of the provided nodes
         centroid = self._calculate_domain_centroid(node_ids)
         # Calculate the bounding box around the network
-        center, radius = _calculate_bounding_box(
-            self.graph.node_coordinates, radius_margin=perimeter_scale
-        )
+        center, radius = _calculate_bounding_box(self.graph.node_coordinates, radius_margin=scale)
         # Convert radial position to radians, adjusting for a 90-degree rotation
         radial_radians = np.deg2rad(radial_position - 90)
         label_position = (
@@ -755,26 +751,41 @@ class NetworkPlotter:
 
     def get_annotated_node_colors(
         self,
+        cmap: str = "gist_rainbow",
+        color: Union[str, None] = None,
+        min_scale: float = 0.8,
+        max_scale: float = 1.0,
+        scale_factor: float = 1.0,
         alpha: float = 1.0,
         nonenriched_color: Union[str, List, Tuple, np.ndarray] = "white",
         nonenriched_alpha: float = 1.0,
         random_seed: int = 888,
-        **kwargs,
     ) -> np.ndarray:
         """Adjust the colors of nodes in the network graph based on enrichment.
 
         Args:
+            cmap (str, optional): Colormap to use for coloring the nodes. Defaults to "gist_rainbow".
+            color (str or None, optional): Color to use for the nodes. If None, the colormap will be used. Defaults to None.
+            min_scale (float, optional): Minimum scale for color intensity. Defaults to 0.8.
+            max_scale (float, optional): Maximum scale for color intensity. Defaults to 1.0.
+            scale_factor (float, optional): Factor for adjusting the color scaling intensity. Defaults to 1.0.
             alpha (float, optional): Alpha value for enriched nodes. Defaults to 1.0.
             nonenriched_color (str, list, tuple, or np.ndarray, optional): Color for non-enriched nodes. Defaults to "white".
             nonenriched_alpha (float, optional): Alpha value for non-enriched nodes. Defaults to 1.0.
             random_seed (int, optional): Seed for random number generation. Defaults to 888.
-            **kwargs: Additional keyword arguments for `get_domain_colors`.
 
         Returns:
             np.ndarray: Array of RGBA colors adjusted for enrichment status.
         """
         # Get the initial domain colors for each node, which are returned as RGBA
-        network_colors = self.graph.get_domain_colors(**kwargs, random_seed=random_seed)
+        network_colors = self.graph.get_domain_colors(
+            cmap=cmap,
+            color=color,
+            min_scale=min_scale,
+            max_scale=max_scale,
+            scale_factor=scale_factor,
+            random_seed=random_seed,
+        )
         # Apply the alpha value for enriched nodes
         network_colors[:, 3] = alpha  # Apply the alpha value to the enriched nodes' A channel
         # Convert the non-enriched color to RGBA using the _to_rgba helper function
@@ -813,68 +824,120 @@ class NetworkPlotter:
 
         return node_sizes
 
-    def get_annotated_contour_colors(self, random_seed: int = 888, **kwargs) -> np.ndarray:
-        """Get colors for the contours based on node annotations.
+    def get_annotated_contour_colors(
+        self,
+        cmap: str = "gist_rainbow",
+        color: Union[str, None] = None,
+        min_scale: float = 0.8,
+        max_scale: float = 1.0,
+        scale_factor: float = 1.0,
+        random_seed: int = 888,
+    ) -> np.ndarray:
+        """Get colors for the contours based on node annotations or a specified colormap.
 
         Args:
-            random_seed (int, optional): Seed for random number generation. Defaults to 888.
-            **kwargs: Additional keyword arguments for `_get_annotated_domain_colors`.
+            cmap (str, optional): Name of the colormap to use for generating contour colors. Defaults to "gist_rainbow".
+            color (str or None, optional): Color to use for the contours. If None, the colormap will be used. Defaults to None.
+            min_scale (float, optional): Minimum intensity scale for the colors generated by the colormap.
+                Controls the dimmest colors. Defaults to 0.8.
+            max_scale (float, optional): Maximum intensity scale for the colors generated by the colormap.
+                Controls the brightest colors. Defaults to 1.0.
+            scale_factor (float, optional): Exponent for adjusting color scaling based on enrichment scores.
+                A higher value increases contrast by dimming lower scores more. Defaults to 1.0.
+            random_seed (int, optional): Seed for random number generation to ensure reproducibility. Defaults to 888.
 
         Returns:
             np.ndarray: Array of RGBA colors for contour annotations.
         """
-        return self._get_annotated_domain_colors(**kwargs, random_seed=random_seed)
+        return self._get_annotated_domain_colors(
+            cmap=cmap,
+            color=color,
+            min_scale=min_scale,
+            max_scale=max_scale,
+            scale_factor=scale_factor,
+            random_seed=random_seed,
+        )
 
-    def get_annotated_label_colors(self, random_seed: int = 888, **kwargs) -> np.ndarray:
-        """Get colors for the labels based on node annotations.
+    def get_annotated_label_colors(
+        self,
+        cmap: str = "gist_rainbow",
+        color: Union[str, None] = None,
+        min_scale: float = 0.8,
+        max_scale: float = 1.0,
+        scale_factor: float = 1.0,
+        random_seed: int = 888,
+    ) -> np.ndarray:
+        """Get colors for the labels based on node annotations or a specified colormap.
 
         Args:
-            random_seed (int, optional): Seed for random number generation. Defaults to 888.
-            **kwargs: Additional keyword arguments for `_get_annotated_domain_colors`.
+            cmap (str, optional): Name of the colormap to use for generating label colors. Defaults to "gist_rainbow".
+            color (str or None, optional): Color to use for the labels. If None, the colormap will be used. Defaults to None.
+            min_scale (float, optional): Minimum intensity scale for the colors generated by the colormap.
+                Controls the dimmest colors. Defaults to 0.8.
+            max_scale (float, optional): Maximum intensity scale for the colors generated by the colormap.
+                Controls the brightest colors. Defaults to 1.0.
+            scale_factor (float, optional): Exponent for adjusting color scaling based on enrichment scores.
+                A higher value increases contrast by dimming lower scores more. Defaults to 1.0.
+            random_seed (int, optional): Seed for random number generation to ensure reproducibility. Defaults to 888.
 
         Returns:
             np.ndarray: Array of RGBA colors for label annotations.
         """
-        return self._get_annotated_domain_colors(**kwargs, random_seed=random_seed)
+        return self._get_annotated_domain_colors(
+            cmap=cmap,
+            color=color,
+            min_scale=min_scale,
+            max_scale=max_scale,
+            scale_factor=scale_factor,
+            random_seed=random_seed,
+        )
 
     def _get_annotated_domain_colors(
         self,
-        color: Union[str, List, Tuple, np.ndarray, None] = None,
+        cmap: str = "gist_rainbow",
+        color: Union[str, None] = None,
+        min_scale: float = 0.8,
+        max_scale: float = 1.0,
+        scale_factor: float = 1.0,
         random_seed: int = 888,
-        **kwargs,
     ) -> np.ndarray:
-        """Get colors for the domains based on node annotations.
+        """Get colors for the domains based on node annotations, or use a specified color.
 
         Args:
-            color (str, list, tuple, np.ndarray, or None, optional): If provided, use this color or list of colors for domains. Defaults to None.
-            random_seed (int, optional): Seed for random number generation. Defaults to 888.
-            **kwargs: Additional keyword arguments for `get_domain_colors`.
+            cmap (str, optional): Colormap to use for generating domain colors. Defaults to "gist_rainbow".
+            color (str or None, optional): Color to use for the domains. If None, the colormap will be used. Defaults to None.
+            min_scale (float, optional): Minimum scale for color intensity when generating domain colors.
+                Defaults to 0.8.
+            max_scale (float, optional): Maximum scale for color intensity when generating domain colors.
+                Defaults to 1.0.
+            scale_factor (float, optional): Factor for adjusting the contrast in the colors generated based on
+                enrichment. Higher values increase the contrast. Defaults to 1.0.
+            random_seed (int, optional): Seed for random number generation to ensure reproducibility. Defaults to 888.
 
         Returns:
             np.ndarray: Array of RGBA colors for each domain.
         """
-        if color is not None:
-            # Convert color(s) to RGBA using _to_rgba helper
-            if isinstance(color, (list, tuple, np.ndarray)):
-                return np.array([_to_rgba(c) for c in color])
-            else:
-                rgba_color = _to_rgba(color)
-                return np.array([rgba_color for _ in self.graph.domain_to_nodes])
-
-        # Generate colors for each domain using provided arguments and random seed
-        node_colors = self.graph.get_domain_colors(**kwargs, random_seed=random_seed)
+        # Generate domain colors based on the enrichment data
+        node_colors = self.graph.get_domain_colors(
+            cmap=cmap,
+            color=color,
+            min_scale=min_scale,
+            max_scale=max_scale,
+            scale_factor=scale_factor,
+            random_seed=random_seed,
+        )
         annotated_colors = []
         for _, nodes in self.graph.domain_to_nodes.items():
             if len(nodes) > 1:
-                # For domains with multiple nodes, select the brightest color (sum of RGB values)
+                # For multi-node domains, choose the brightest color based on RGB sum
                 domain_colors = np.array([node_colors[node] for node in nodes])
                 brightest_color = domain_colors[
-                    np.argmax(domain_colors[:, :3].sum(axis=1))
-                ]  # Only consider RGB
+                    np.argmax(domain_colors[:, :3].sum(axis=1))  # Sum the RGB values
+                ]
                 annotated_colors.append(brightest_color)
             else:
-                # Assign a default color (white) for single-node domains
-                default_color = np.array([1.0, 1.0, 1.0, 1.0])  # RGBA for white
+                # Single-node domains default to white (RGBA)
+                default_color = np.array([1.0, 1.0, 1.0, 1.0])
                 annotated_colors.append(default_color)
 
         return np.array(annotated_colors)

{risk_network-0.0.6b1.dist-info → risk_network-0.0.6b2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: risk-network
-Version: 0.0.6b1
+Version: 0.0.6b2
 Summary: A Python package for biological network analysis
 Author: Ira Horecka
 Author-email: Ira Horecka <ira89@icloud.com>

{risk_network-0.0.6b1.dist-info → risk_network-0.0.6b2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-risk/__init__.py,sha256=7d7mnBPR928yjdJisfakH8QFv7SDN4BgrXse4YDx1w4,112
+risk/__init__.py,sha256=Gxy0lKfxRz2r__O9V8E-JuEVysqWPz04bTIkWmP48Oc,112
 risk/constants.py,sha256=XInRaH78Slnw_sWgAsBFbUHkyA0h0jL0DKGuQNbOvjM,550
 risk/risk.py,sha256=PONl5tzN5DSVUf4MgczfOvzGV-5JoAOLTQ6YWl10mZ8,20697
 risk/annotations/__init__.py,sha256=vUpVvMRE5if01Ic8QY6M2Ae3EFGJHdugEe9PdEkAW4Y,138
@@ -13,9 +13,9 @@ risk/neighborhoods/domains.py,sha256=HwuChmZH0RGD9eQOvk2-ezQDJRUHHn93vhVgHb-kX6I
 risk/neighborhoods/neighborhoods.py,sha256=sHmjFFl2U5qV9YbQCRbpbI36j7dS7IFfFwwRb1_-AuM,13945
 risk/network/__init__.py,sha256=iEPeJdZfqp0toxtbElryB8jbz9_t_k4QQ3iDvKE8C_0,126
 risk/network/geometry.py,sha256=H1yGVVqgbfpzBzJwEheDLfvGLSA284jGQQTn612L4Vc,6759
-risk/network/graph.py,sha256=DTjNRQwgQbt6jJYDBHm7sgrdbZUDLL4HiZpdixkSI7Y,12433
+risk/network/graph.py,sha256=DhfNw2cBD238Kz0X1OSr5VLTW9h1n4cXvqUObxB4CAs,12892
 risk/network/io.py,sha256=gG50kOknO-D3HkW1HsbHMkTMvjUtn3l4W4Jwd-rXNr8,21202
-risk/network/plot.py,sha256=C4uIk9zX5qOAtP57vGBXGkbjGbpCKmn08jddsgGMqi0,50563
+risk/network/plot.py,sha256=QItILQH9eUEofA4ui0WtuSaCtz63GNJgxwCGQwbd1lU,53867
 risk/stats/__init__.py,sha256=e-BE_Dr_jgiK6hKM-T-tlG4yvHnId8e5qjnM0pdwNVc,230
 risk/stats/fisher_exact.py,sha256=-bPwzu76-ob0HzrTV20mXUTot7v-MLuqFaAoab-QxPg,4966
 risk/stats/hypergeom.py,sha256=lrIFdhCWRjvM4apYw1MlOKqT_IY5OjtCwrjdtJdt6Tg,4954
@@ -23,8 +23,8 @@ risk/stats/stats.py,sha256=kvShov-94W6ffgDUTb522vB9hDJQSyTsYif_UIaFfSM,7059
 risk/stats/permutation/__init__.py,sha256=neJp7FENC-zg_CGOXqv-iIvz1r5XUKI9Ruxhmq7kDOI,105
 risk/stats/permutation/permutation.py,sha256=qLWdwxEY6nmkYPxpM8HLDcd2mbqYv9Qr7CKtJvhLqIM,9220
 risk/stats/permutation/test_functions.py,sha256=HuDIM-V1jkkfE1rlaIqrWWBSKZt3dQ1f-YEDjWpnLSE,2343
-risk_network-0.0.6b1.dist-info/LICENSE,sha256=jOtLnuWt7d5Hsx6XXB2QxzrSe2sWWh3NgMfFRetluQM,35147
-risk_network-0.0.6b1.dist-info/METADATA,sha256=3XAItYSzr6H-N_kyAIMdTX5GD0jUDL59AIAXm4056xA,43142
-risk_network-0.0.6b1.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-risk_network-0.0.6b1.dist-info/top_level.txt,sha256=NX7C2PFKTvC1JhVKv14DFlFAIFnKc6Lpsu1ZfxvQwVw,5
-risk_network-0.0.6b1.dist-info/RECORD,,
+risk_network-0.0.6b2.dist-info/LICENSE,sha256=jOtLnuWt7d5Hsx6XXB2QxzrSe2sWWh3NgMfFRetluQM,35147
+risk_network-0.0.6b2.dist-info/METADATA,sha256=dri96kn6JF3ZK04yUukO8km6vCMhs09S-vpAIYS8QQg,43142
+risk_network-0.0.6b2.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+risk_network-0.0.6b2.dist-info/top_level.txt,sha256=NX7C2PFKTvC1JhVKv14DFlFAIFnKc6Lpsu1ZfxvQwVw,5
+risk_network-0.0.6b2.dist-info/RECORD,,