modusa 0.3.28__py3-none-any.whl → 0.3.30__py3-none-any.whl

modusa/tools/plotter.py

@@ -27,7 +27,7 @@ def _calculate_extent(x, y):
 	]
 
 #======== 1D ===========
-def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylabel=None, title=None, legend=None, show_grid=False):
+def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylabel=None, title=None, legend=None, show_grid=False, show_stem=False):
 		"""
 		Plots a 1D signal using matplotlib.
 
@@ -75,6 +75,9 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		show_grid: bool
 			- If you want to show the grid.
 			- Default: False
+		show_stem: bool:
+			- If you want stem plot.
+			- Default: False
 	
 		Returns
 		-------
@@ -85,6 +88,7 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		for arg in args:
 			if len(arg) not in [1, 2]: # 1 if it just provides values, 2 if it provided axis as well
 				raise ValueError(f"1D signal needs to have max 2 arrays (y, x) or simply (y, )")
+			
 		if isinstance(legend, str): legend = (legend, )
 		
 		if legend is not None:
@@ -111,27 +115,52 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		for i, signal in enumerate(args):
 			if len(signal) == 1:
 				y = signal[0]
+				x = np.arange(y.size)
 				if legend is not None:
-					signal_ax.plot(y, label=legend[i])
+					if show_stem is True:
+						markerline, stemlines, baseline = signal_ax.stem(x, y, label=legend[i])
+						markerline.set_color(colors[i])
+						stemlines.set_color(colors[i])
+						baseline.set_color("k")
+					else:
+						signal_ax.plot(x, y, color=colors[i], label=legend[i])
 				else:
-					signal_ax.plot(y)
+					if show_stem is True:
+						markerline, stemlines, baseline = signal_ax.stem(x, y)
+						markerline.set_color(colors[i])
+						stemlines.set_color(colors[i])
+						baseline.set_color("k")
+					else:
+						signal_ax.plot(x, y, color=colors[i])
+						
 			elif len(signal) == 2:
 				y, x = signal[0], signal[1]
 				if legend is not None:
-					signal_ax.plot(x, y, label=legend[i])
+					if show_stem is True:
+						signal_ax.stem(x, y, linefmt=f"{colors[i]}-", markerfmt=f"{colors[i]}o", basefmt="k-", label=legend[i])
+					else:
+						signal_ax.plot(x, y, color=colors[i], label=legend[i])
 				else:
-					signal_ax.plot(x, y)
+					if show_stem is True:
+						signal_ax.stem(x, y, linefmt=f"{colors[i]}-", markerfmt=f"{colors[i]}o", basefmt="k-")
+					else:
+						signal_ax.plot(x, y, color=colors[i])
 		
 		# Add annotations
 		if ann is not None:
-			annotation_ax.set_ylim(0, 1)
+			annotation_ax.set_ylim(0, 1) # For consistent layout
+			# Determine visible x-range
+			x_view_min = xlim[0] if xlim is not None else np.min(x)
+			x_view_max = xlim[1] if xlim is not None else np.max(x)
+			
 			for i, (start, end, tag) in enumerate(ann):
-				if xlim is not None:
-					if end < xlim[0] or start > xlim[1]:
-						continue  # Skip out-of-view regions
-					# Clip boundaries to xlim
-					start = max(start, xlim[0])
-					end = min(end, xlim[1])
+				# We make sure that we only plot annotation that are within the x range of the current view
+				if start >= x_view_max or end <= x_view_min:
+					continue
+				
+				# Clip boundaries to xlim
+				start = max(start, x_view_min)
+				end = min(end, x_view_max)
 					
 				color = colors[i % len(colors)]
 				width = end - start
@@ -159,7 +188,7 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		# Add legend
 		if legend is not None:
 			handles, labels = signal_ax.get_legend_handles_labels()
-			fig.legend(handles, labels, loc='upper right', bbox_to_anchor=(0.9, 1.2), ncol=len(legend), frameon=False)
+			fig.legend(handles, labels, loc='upper right', bbox_to_anchor=(0.9, 1.2), ncol=len(legend), frameon=True)
 			
 		# Set title, labels
 		if title is not None:
@@ -315,14 +344,19 @@ def plot2d(*args, ann=None, events=None, xlim=None, ylim=None, origin="lower", M
 	
 	# Add annotations
 	if ann is not None:
-		annotation_ax.set_ylim(0, 1)
+		annotation_ax.set_ylim(0, 1) # For consistent layout
+		# Determine visible x-range
+		x_view_min = xlim[0] if xlim is not None else np.min(x)
+		x_view_max = xlim[1] if xlim is not None else np.max(x)
+		
 		for i, (start, end, tag) in enumerate(ann):
-			if xlim is not None:
-				if end < xlim[0] or start > xlim[1]:
-					continue  # Skip out-of-view regions
-				# Clip boundaries to xlim
-				start = max(start, xlim[0])
-				end = min(end, xlim[1])
+			# We make sure that we only plot annotation that are within the x range of the current view
+			if start >= x_view_max or end <= x_view_min:
+				continue
+			
+			# Clip boundaries to xlim
+			start = max(start, x_view_min)
+			end = min(end, x_view_max)
 				
 			color = colors[i % len(colors)]
 			width = end - start
@@ -451,11 +485,11 @@ def plot_dist(*args, ann=None, xlim=None, ylim=None, ylabel=None, xlabel=None, t
 		from scipy.stats import gaussian_kde
 	
 		if isinstance(legend, str):
-				legend = (legend, )
+			legend = (legend, )
 			
 		if legend is not None:
-				if len(legend) < len(args):
-						raise ValueError(f"Legend should be provided for each signal.")
+			if len(legend) < len(args):
+				raise ValueError(f"Legend should be provided for each signal.")
 					
 		# Create figure
 		fig = plt.figure(figsize=(16, 4))
@@ -468,66 +502,70 @@ def plot_dist(*args, ann=None, xlim=None, ylim=None, ylabel=None, xlabel=None, t
 	
 		# Set limits
 		if xlim is not None:
-				dist_ax.set_xlim(xlim)
+			dist_ax.set_xlim(xlim)
 			
 		if ylim is not None:
-				dist_ax.set_ylim(ylim)
+			dist_ax.set_ylim(ylim)
 			
 		# Add plot
 		for i, data in enumerate(args):
-				# Fit gaussian to the data
-				kde = gaussian_kde(data)
-			
-				# Create points to evaluate KDE
-				x_vals = np.linspace(min(data), max(data), npoints)
-				y_vals = kde(x_vals)
-			
-				if legend is not None:
-						dist_ax.plot(x_vals, y_vals, color=colors[i], label=legend[i])
-						if show_hist is True:
-								dist_ax.hist(data, bins=bins, density=True, alpha=0.3, facecolor=colors[i], edgecolor='black', label=legend[i])
-				else:
-						dist_ax.plot(x_vals, y_vals, color=colors[i])
-						if show_hist is True:
-								dist_ax.hist(data, bins=bins, density=True, alpha=0.3, facecolor=colors[i], edgecolor='black')
+			# Fit gaussian to the data
+			kde = gaussian_kde(data)
+		
+			# Create points to evaluate KDE
+			x = np.linspace(np.min(data), np.max(data), npoints)
+			y = kde(x)
+		
+			if legend is not None:
+				dist_ax.plot(x, y, color=colors[i], label=legend[i])
+				if show_hist is True:
+					dist_ax.hist(data, bins=bins, density=True, alpha=0.3, facecolor=colors[i], edgecolor='black', label=legend[i])
+			else:
+				dist_ax.plot(x, y, color=colors[i])
+				if show_hist is True:
+					dist_ax.hist(data, bins=bins, density=True, alpha=0.3, facecolor=colors[i], edgecolor='black')
 							
 		# Add annotations
 		if ann is not None:
-				annotation_ax.set_ylim(0, 1)
-				for i, (start, end, tag) in enumerate(ann):
-						if xlim is not None:
-								if end < xlim[0] or start > xlim[1]:
-										continue  # Skip out-of-view regions
-								# Clip boundaries to xlim
-								start = max(start, xlim[0])
-								end = min(end, xlim[1])
-							
-						color = colors[i % len(colors)]
-						width = end - start
-						rect = Rectangle((start, 0), width, 1, color=color, alpha=0.7)
-						annotation_ax.add_patch(rect)
+			annotation_ax.set_ylim(0, 1) # For consistent layout
+			# Determine visible x-range
+			x_view_min = xlim[0] if xlim is not None else np.min(x)
+			x_view_max = xlim[1] if xlim is not None else np.max(x)
+			for i, (start, end, tag) in enumerate(ann):
+				# We make sure that we only plot annotation that are within the x range of the current view
+				if start >= x_view_max or end <= x_view_min:
+					continue
+				
+				# Clip boundaries to xlim
+				start = max(start, x_view_min)
+				end = min(end, x_view_max)
 					
-						text_obj = annotation_ax.text((start + end) / 2, 0.5, tag, ha='center', va='center', fontsize=10, color='white', fontweight='bold', zorder=10, clip_on=True)
-						text_obj.set_clip_path(rect)
+				color = colors[i % len(colors)]
+				width = end - start
+				rect = Rectangle((start, 0), width, 1, color=color, alpha=0.7)
+				annotation_ax.add_patch(rect)
+			
+				text_obj = annotation_ax.text((start + end) / 2, 0.5, tag, ha='center', va='center', fontsize=10, color='white', fontweight='bold', zorder=10, clip_on=True)
+				text_obj.set_clip_path(rect)
 					
 		# Add legend
 		if legend is not None:
-				handles, labels = dist_ax.get_legend_handles_labels()
-				fig.legend(handles, labels, loc='upper right', bbox_to_anchor=(0.9, 1.1), ncol=len(legend), frameon=True)
+			handles, labels = dist_ax.get_legend_handles_labels()
+			fig.legend(handles, labels, loc='upper right', bbox_to_anchor=(0.9, 1.1), ncol=len(legend), frameon=True)
 			
 		# Set title, labels
 		if title is not None:
-				annotation_ax.set_title(title, pad=10, size=11)
+			annotation_ax.set_title(title, pad=10, size=11)
 		if xlabel is not None:
-				dist_ax.set_xlabel(xlabel)
+			dist_ax.set_xlabel(xlabel)
 		if ylabel is not None:
-				dist_ax.set_ylabel(ylabel)
+			dist_ax.set_ylabel(ylabel)
 			
 		# Remove the boundaries and ticks from annotation axis
 		if ann is not None:
-				annotation_ax.tick_params(left=False, bottom=False, labelleft=False, labelbottom=False)
+			annotation_ax.tick_params(left=False, bottom=False, labelleft=False, labelbottom=False)
 		else:
-				annotation_ax.axis("off")
+			annotation_ax.axis("off")
 			
 		fig.subplots_adjust(hspace=0.01, wspace=0.05)
 		plt.close()

{modusa-0.3.28.dist-info → modusa-0.3.30.dist-info}/METADATA

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

{modusa-0.3.28.dist-info → modusa-0.3.30.dist-info}/RECORD

@@ -1,7 +1,7 @@
-modusa-0.3.28.dist-info/METADATA,sha256=wNR4dM9jMye2V4s7H9zhtD6k9w6B8kUdlzGQdxbINvc,1369
-modusa-0.3.28.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.28.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.28.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.30.dist-info/METADATA,sha256=ue8QCA3HR90-JBEZkGmfM-up5D6tFqn88u2QqGrdHd8,1369
+modusa-0.3.30.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.30.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.30.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
 modusa/.DS_Store,sha256=_gm6qJREwfMi8dE7n5S89_RG46u5t3xHyD-smNhtNoM,6148
 modusa/__init__.py,sha256=uq6kORFFAODiCMGmOLWO0shE8-dVFWf5gmV8wxekmnk,280
 modusa/config.py,sha256=bTqK4t00FZqERVITrxW_q284aDDJAa9aMSfFknfR-oU,280
@@ -47,7 +47,7 @@ modusa/tools/audio_loader.py,sha256=DrCzq0pdiQrUDIG-deLJGcu8EaylO5yRtwT4lr8WSf8,
 modusa/tools/audio_player.py,sha256=GP04TWW4jBwQBjANkfR_cJtEy7cIhvbu8RTwnf9hD6E,2817
 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=Pfj24jl-WHgdfPI-f4J_XLpMyed6OHkwAvtuIBKoAEw,16016
+modusa/tools/plotter.py,sha256=rIv5TmKtr9uliRcLNRjmejwvIixZqpUf4_dtqSlx2Hg,17510
 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
@@ -56,4 +56,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.28.dist-info/RECORD,,
+modusa-0.3.30.dist-info/RECORD,,