modusa 0.3.69__py3-none-any.whl → 0.3.71__py3-none-any.whl

modusa/tools/plotter.py

@@ -167,14 +167,11 @@ class Fig:
 				axs[i, 1].axis("off")
 			elif char == "a": # Remove ticks and labels from all the aux subplots
 				axs[i, 0].tick_params(left=False, bottom=False, labelleft=False, labelbottom=False)
-				axs[i, 0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
 				axs[i, 1].axis("off")
 			elif char == "s":
 				axs[i, 0].tick_params(bottom=False, labelbottom=False)
-				axs[i, 0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
 				axs[i, 1].axis("off")
 			elif char == "m":
-				axs[i, 0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
 				axs[i, 0].tick_params(bottom=False, labelbottom=False)
 				
 			axs[i, 0].sharex(axs[0, 0])
@@ -189,7 +186,7 @@ class Fig:
 		
 		return fig, axs
 	
-	def add_signal(self, y, x=None, c=None, ls=None, lw=None, m=None, ms=3, label=None, ylabel=None, ylim=None, yticks=None, yticklabels=None, xticks=None, xticklabels=None, ax=None):
+	def add_signal(self, y, x=None, c=None, ls=None, lw=None, m=None, ms=3, label=None, ylabel=None, ylim=None, yticks=None, yticklabels=None, xticks=None, xticklabels=None, grid=True, ax=None):
 		"""
 		Add signal to the figure.
 			
@@ -233,6 +230,9 @@ class Fig:
 			- Positions at which to place x-axis ticks.
 		xticklabels : list of str, optional
 			- Labels corresponding to `xticks`. Must be the same length as `xticks`.
+		grid: bool
+			- Do you want the grid?
+			- Default: True
 		ax: int
 			- Which specific axis to plot (1, 2, 3, ...)
 			- None
@@ -267,9 +267,12 @@ class Fig:
 			curr_row[0].set_xticks(xticks)
 			if xticklabels is not None:
 				curr_row[0].set_xticklabels(xticklabels)
+		
+		if grid is True:
+			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, ax=None):
+	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):
 		"""
 		Add matrix to the figure.
 			
@@ -309,6 +312,9 @@ class Fig:
 		cbar: bool
 			- Show colorbar
 			- Default: True
+		grid: bool
+			- Do you want the grid?
+			- Default: True
 		ax: int
 			- Which specific axis to plot (1, 2, 3, ...)
 			- None
@@ -349,8 +355,12 @@ class Fig:
 			if xticklabels is not None:
 				curr_row[0].set_xticklabels(xticklabels)
 				
+		if grid is True:
+			curr_row[0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
+			
+				
 				
-	def add_events(self, events, c=None, ls=None, lw=None, label=None, ax=None):
+	def add_events(self, events, c=None, ls=None, lw=None, label=None, grid=True, ax=None):
 		"""
 		Add events to the figure.
 		
@@ -371,6 +381,9 @@ class Fig:
 			- Label for the event type.
 			- This will appear in the legend.
 			- Default: None
+		grid: bool
+			- Do you want the grid?
+			- Default: True
 		ax: int
 			- Which specific axis to plot (1, 2, 3, ...)
 			- None
@@ -399,7 +412,10 @@ class Fig:
 				else:
 					curr_row[0].axvline(x=event, color=c, linestyle=ls, linewidth=lw)
 					
-	def add_annotation(self, ann, label=None, patterns=None, ax=None, ylim=(0, 1), text_loc="middle"):
+		if grid is True:
+			curr_row[0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
+					
+	def add_annotation(self, ann, label=None, patterns=None, ylim=(0, 1), text_loc="middle", grid=True, ax=None):
 		"""
 		Add annotation to the figure.
 		
@@ -416,15 +432,18 @@ class Fig:
 			- 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
 		ylim: tuple[number, number]
 			- Y-limit for the annotation.
 			- Default: (0, 1)
 		text_loc: str
 			- Location of text relative to the box. (b for bottom, m for middle, t for top)
 			- Default: "middle"
+		grid: bool
+			- Do you want the grid?
+			- Default: True
+		ax: int
+			- Which specific axis to plot (1, 2, 3, ...)
+			- None
 		Returns
 		-------
 		None
@@ -507,6 +526,9 @@ class Fig:
 			curr_row[0].set_ylabel(label, rotation=0, ha="center", va="center")
 			curr_row[0].yaxis.set_label_position("right")
 			curr_row[0].yaxis.set_label_coords(1.05, 0.75)
+		
+		if grid is True:
+			curr_row[0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
 			
 	def add_legend(self, ypos=1.0):
 		"""

{modusa-0.3.69.dist-info → modusa-0.3.71.dist-info}/METADATA

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

{modusa-0.3.69.dist-info → modusa-0.3.71.dist-info}/RECORD

@@ -1,7 +1,7 @@
-modusa-0.3.69.dist-info/METADATA,sha256=BVNPMjegkTKx-M2wB64UF5CaliXy-6J-eSTXs2mn7ck,1436
-modusa-0.3.69.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.69.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.69.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.71.dist-info/METADATA,sha256=LR0dA8ReDc2Ii-DPSI-LITAowNq0748ohej3DwlDm5E,1436
+modusa-0.3.71.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.71.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.71.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=2BotJit9on5R5pfDMMxE8w-cREAHUW6_QpVzoqtx8-s,21051
+modusa/tools/plotter.py,sha256=kLSZmzw4AHAQXTSb8u1F_fA6sWCR9G6xHOuhD6UC_DE,21518
 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.69.dist-info/RECORD,,
+modusa-0.3.71.dist-info/RECORD,,