modusa 0.3.72__py3-none-any.whl → 0.3.74__py3-none-any.whl

modusa/tools/plotter.py

@@ -95,6 +95,14 @@ class Fig:
 		
 		return curr_row
 	
+	def _get_prev_row(self):
+		"""
+		Get the prev row where you can add arrows.
+		"""
+		prev_pow = self._axs[self._curr_row_idx - 1]
+		
+		return prev_pow
+	
 	def _get_new_color(self):
 		"""
 		Get a new color for different lines.
@@ -272,7 +280,7 @@ class Fig:
 			curr_row[0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
 				
 		
-	def add_matrix(self, M, y=None, x=None, c="gray_r", o="upper", label=None, ylabel=None, ylim=None, yticks=None, yticklabels=None, xticks=None, xticklabels=None, cbar=True, grid=True, ax=None):
+	def add_matrix(self, M, y=None, x=None, c="viridis", o="upper", label=None, ylabel=None, ylim=None, yticks=None, yticklabels=None, xticks=None, xticklabels=None, cbar=True, grid=True, ax=None):
 		"""
 		Add matrix to the figure.
 			
@@ -529,6 +537,53 @@ class Fig:
 		
 		if grid is True:
 			curr_row[0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
+	
+	from matplotlib.patches import Rectangle
+	import matplotlib.pyplot as plt
+	
+	def add_arrow(self, xy, label, text_offset=(0, 0), c="r", fontsize=12, ax=None):
+		"""
+		Add an arrow pointing to a specific point with a boxed label at the tail.
+	
+		Parameters
+		----------
+		xy : tuple[float, float]
+			- Target point (x, y) for the arrow head.
+		label : str
+			- Text label at the arrow tail.
+		text_offset : tuple[float, float]
+			- Offset (dx, dy) for label position from arrow tail.
+		c : str
+			- Color for arrow and text.
+			- Default: "r" for red
+		fontsize : int
+			- Font size of the label text.
+			- Default: 12
+		ax : int | None
+			- Which specific axis to plot (1, 2, 3, ...).
+			- If None, uses the current row.
+	
+		Returns
+		-------
+		None
+		"""
+		curr_row = self._get_prev_row() if ax is None else self._axs[ax]
+		
+		arrowprops = dict(arrowstyle="->", color=c, lw=2)
+		bbox = dict(boxstyle="round,pad=0.3", fc="white", ec=c, lw=1.2)
+			
+		text_x, text_y = xy[0] + text_offset[0], xy[1] + text_offset[1]
+		
+		curr_row[0].annotate(
+			label,
+			xy=xy, xycoords="data",
+			xytext=(text_x, text_y), textcoords="data",
+			arrowprops=arrowprops,
+			fontsize=fontsize,
+			color=c,
+			ha="center", va="center",
+			bbox=bbox
+		)
 			
 	def add_legend(self, ypos=1.0):
 		"""

{modusa-0.3.72.dist-info → modusa-0.3.74.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: modusa
-Version: 0.3.72
+Version: 0.3.74
 Summary: A modular signal analysis python library.
 Author-Email: Ankit Anand <ankit0.anand0@gmail.com>
 License: MIT

{modusa-0.3.72.dist-info → modusa-0.3.74.dist-info}/RECORD

@@ -1,7 +1,7 @@
-modusa-0.3.72.dist-info/METADATA,sha256=Hj2E34RK_GRREkK_BI8FZjmp4Q3YfuKea14dvVSVi_0,1436
-modusa-0.3.72.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.72.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.72.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.74.dist-info/METADATA,sha256=CoBn0MvvxeiFUbBHS7bALJWTBvOm4osnsFg7Uew2XEg,1436
+modusa-0.3.74.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.74.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.74.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
 modusa/.DS_Store,sha256=_gm6qJREwfMi8dE7n5S89_RG46u5t3xHyD-smNhtNoM,6148
 modusa/__init__.py,sha256=RMIKAZZ27w1oh2UFMJvZQtQ-SluIQsac_3mSK_LaM30,277
 modusa/config.py,sha256=bTqK4t00FZqERVITrxW_q284aDDJAa9aMSfFknfR-oU,280
@@ -50,7 +50,7 @@ modusa/tools/audio_player.py,sha256=GP04TWW4jBwQBjANkfR_cJtEy7cIhvbu8RTwnf9hD6E,
 modusa/tools/audio_recorder.py,sha256=d2fVt0Sd2tlBdb2WlUs60K4N23zuxM3KUpQqX0ifPi8,2769
 modusa/tools/base.py,sha256=C0ESJ0mIfjjRlAkRbSetNtMoOfS6IrHBjexRp3l_Mh4,1293
 modusa/tools/math_ops.py,sha256=ZZ7U4DgqT7cOeE7_Lzi_Qq-48WYfwR9_osbZwTmE9eg,8690
-modusa/tools/plotter.py,sha256=NwyFMpLKNwiJ8wxYrX8wlaJKysMCGEXmPEq_jLnJinA,21557
+modusa/tools/plotter.py,sha256=A_1GA-toJe6MjXiF8adfD9awE2EDQlTniEpy91yc4c8,22944
 modusa/tools/youtube_downloader.py,sha256=hB_X8-7nOHXOlxg6vv3wyhBLoAsWyomrULP6_uCQL7s,1698
 modusa/utils/.DS_Store,sha256=nLXMwF7QJNuglLI_Gk74F7vl5Dyus2Wd74Mgowijmdo,6148
 modusa/utils/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
@@ -59,4 +59,4 @@ modusa/utils/excp.py,sha256=L9vhaGjKpv9viJYdmC9n5ndmk2GVbUBuFyZyhAQZmWY,906
 modusa/utils/logger.py,sha256=K0rsnObeNKCxlNeSnVnJeRhgfmob6riB2uyU7h3dDmA,571
 modusa/utils/np_func_cat.py,sha256=TyIFgRc6bARRMDnZxlVURO5Z0I-GWhxRONYyIv-Vwxs,1007
 modusa/utils/plot.py,sha256=s_vNdxvKfwxEngvJPgrF1PcmxZNnNaaXPViHWjyjJ-c,5335
-modusa-0.3.72.dist-info/RECORD,,
+modusa-0.3.74.dist-info/RECORD,,