modusa 0.3.56__py3-none-any.whl → 0.3.58__py3-none-any.whl

modusa/tools/plotter.py

@@ -12,8 +12,8 @@ import matplotlib.font_manager as fm
 import matplotlib.pyplot as plt
 import matplotlib.gridspec as gridspec
 from matplotlib.patches import Rectangle
-
 import numpy as np
+import fnmatch
 
 #===== Loading Devanagari font ========
 def _load_devanagari_font():
@@ -259,7 +259,6 @@ class Fig:
 		curr_row = self._get_curr_row() if ax is None else self._axs[ax]
 		
 		extent = self._calculate_extent(x, y, o)
-		print(extent)
 		
 		im = curr_row[0].imshow(M, aspect="auto", origin=o, cmap=c, extent=extent)
 		
@@ -326,7 +325,7 @@ class Fig:
 				else:
 					curr_row[0].axvline(x=event, color=c, linestyle=ls, linewidth=lw)
 					
-	def add_annotation(self, ann, label=None, ax=None):
+	def add_annotation(self, ann, label=None, patterns=None, ax=None):
 		"""
 		Add annotation to the figure.
 		
@@ -339,6 +338,10 @@ class Fig:
 			- Label for the annotation type.
 			- This will appear to the right of the aux plot.
 			- Default: None
+		patterns: list[str]
+			- Patterns to group annotations
+			- E.g., "*R" or "<tag>*" or ["A*", "*B"]
+			- All elements in a group will have same color.
 		ax: int
 			- Which specific axis to plot (1, 2, 3, ...)
 			- None
@@ -350,7 +353,24 @@ class Fig:
 		
 		xlim = self._xlim
 		
-		for i, (start, end, tag) in enumerate(ann):
+		if isinstance(patterns, str): patterns = [patterns]
+		
+		if patterns is not None:
+			for i, (start, end, tag) in enumerate(ann):
+				for j, pattern in enumerate(patterns):
+					if fnmatch.fnmatch(tag, pattern):
+						ann[i] = (start, end, tag, j)
+						break
+					else:
+						ann[i] = (start, end, tag, None)
+		else:
+			for i, (start, end, tag) in enumerate(ann):
+				ann[i] = (start, end, tag, None)
+					
+		colors = plt.rcParams['axes.prop_cycle'].by_key()['color']
+		print(ann)
+		
+		for i, (start, end, tag, group) in enumerate(ann):
 			# We make sure that we only plot annotation that are within the x range of the current view
 			if xlim is not None:
 				if start >= xlim[1] or end <= xlim[0]:
@@ -360,8 +380,10 @@ class Fig:
 				start = max(start, xlim[0])
 				end = min(end, xlim[1])
 				
-				box_colors = ["gray", "lightgray"] # Alternates color between two
-				box_color = box_colors[i % 2]
+				if group is not None:
+					box_color = colors[group]
+				else:
+					box_color = "lightgray"
 				
 				width = end - start
 				rect = Rectangle((start, 0), width, 1, facecolor=box_color, edgecolor="black", alpha=0.7)
@@ -375,8 +397,10 @@ class Fig:
 				
 				text_obj.set_clip_path(rect)
 			else:
-				box_colors = ["gray", "lightgray"] # Alternates color between two
-				box_color = box_colors[i % 2]
+				if group is not None:
+					box_color = colors[group]
+				else:
+					box_color = "lightgray"
 				
 				width = end - start
 				rect = Rectangle((start, 0), width, 1, facecolor=box_color, edgecolor="black", alpha=0.7)

{modusa-0.3.56.dist-info → modusa-0.3.58.dist-info}/METADATA

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

{modusa-0.3.56.dist-info → modusa-0.3.58.dist-info}/RECORD

@@ -1,7 +1,7 @@
-modusa-0.3.56.dist-info/METADATA,sha256=n_8ejSivvs_MZXpNqtMcwyIqu94G6TnxcE9V49BLdHo,1436
-modusa-0.3.56.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.56.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.56.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.58.dist-info/METADATA,sha256=aMaG5ln4Uq8f-llp978sCAUA7sU9ZlgB5DTrR7YzyEk,1436
+modusa-0.3.58.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.58.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.58.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=wNiDmAawvrET9021YiZG_kbOACFu25TdzywtMLYhpa0,17232
+modusa/tools/plotter.py,sha256=gxAqWAoXTsXXoA8uRV5KtjZbbCxeMTxLztMuoywAJxc,17874
 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.56.dist-info/RECORD,,
+modusa-0.3.58.dist-info/RECORD,,