modusa 0.3.21__py3-none-any.whl → 0.3.23__py3-none-any.whl

modusa/__init__.py

@@ -1,8 +1,9 @@
 from modusa.utils import excp, config
 
 #=====Giving access to plot functions to plot multiple signals.=====
-from modusa.tools import plot1d, plot2d
+from modusa.tools import plot1d, plot2d, plot_dist
+#=====
+
 from modusa.tools import play, convert
 from modusa.tools import download
-from modusa.tools import load
-#=====
+from modusa.tools import load, load_ann

modusa/devtools/generate_docs_source.py

@@ -9,17 +9,15 @@ from collections import defaultdict
 # === Configuration ===
 BASE_MODULES = [
     'modusa.tools',
-    'modusa.models',
-    'modusa.generators',
-    'modusa.plugins',
-#   'modusa.io'
+#   'modusa.models',
+#   'modusa.generators',
+#   'modusa.plugins',
 ]
 OUTPUT_DIRS = [
     Path('docs/source/tools'),
-    Path('docs/source/models'),
-    Path('docs/source/generators'),
-    Path('docs/source/plugins'),
-#   Path('docs/source/io')
+#   Path('docs/source/models'),
+#   Path('docs/source/generators'),
+#   Path('docs/source/plugins'),
 ]
 
 # Ensure output directories exist
@@ -89,8 +87,6 @@ def generate_docs_source():
             write_module_rst_file(module_path, class_list, output_dir)
     
         section_name = base_module.split('.')[-1].capitalize()
-        if section_name == "Io":
-            section_name = "IO"
     
         write_index_rst_file(module_class_map, output_dir, section_name=section_name)
         print(f"✅ Documentation generated for {base_module} in {output_dir}")

modusa/tools/__init__.py

@@ -1,7 +1,8 @@
 #!/usr/bin/env python3
 
-from .plotter import plot1d, plot2d
+from .plotter import plot1d, plot2d, plot_dist
 from .audio_player import play
 from .audio_converter import convert
 from .youtube_downloader import download
-from .audio_loader import load
+from .audio_loader import load
+from .ann_loader import load_ann

modusa/tools/ann_loader.py

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+
+#---------------------------------
+# Author: Ankit Anand
+# Date: 12/08/25
+# Email: ankit0.anand0@gmail.com
+#---------------------------------
+
+from pathlib import Path
+
+def load_ann(path):
+	"""
+	Load annotation from audatity label text file.
+
+	Parameters
+	----------
+	path: str
+		label text file path.
+
+	Returns
+	-------
+	list[tuple, ...]
+		- annotation data structure
+		- [(start, end, tag), ...]
+	"""
+	
+	if isinstance(path, (str, Path)):
+		raise ValueError(f"`path` must be one of (str, Path), got {type(path)}")
+	
+	ann = []
+	with open(str(path), "r") as f:
+		lines = [line.rstrip("\n") for line in f]
+		for line in lines:
+			start, end, tag = line.split("\t")
+			start, end = float(start), float(end)
+			ann.append((start, end, tag))
+			
+	return ann

modusa/tools/plotter.py

@@ -59,7 +59,7 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		plt.Figure
 			Matplolib figure.
 		"""
-	
+
 		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, )")
@@ -115,9 +115,16 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 				width = end - start
 				rect = Rectangle((start, 0), width, 1, color=color, alpha=0.7)
 				annotation_ax.add_patch(rect)
-				annotation_ax.text((start + end) / 2, 0.5, tag,
-									ha='center', va='center',
-									fontsize=10, color='white', fontweight='bold', zorder=10)
+				
+				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 vlines
 		if events is not None:
 			for xpos in events:
@@ -140,7 +147,7 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		if ylabel is not None:
 			signal_ax.set_ylabel(ylabel)
 		
-		# Decorating annotation axis thicker
+		# Remove the boundaries and ticks from an axis
 		if ann is not None:
 			annotation_ax.tick_params(left=False, bottom=False, labelleft=False, labelbottom=False)
 		else:
@@ -296,9 +303,14 @@ def plot2d(*args, ann=None, events=None, xlim=None, ylim=None, origin="lower", M
 			width = end - start
 			rect = Rectangle((start, 0), width, 1, color=color, alpha=0.7)
 			annotation_ax.add_patch(rect)
-			annotation_ax.text((start + end) / 2, 0.5, tag,
-								ha='center', va='center',
-								fontsize=10, color='white', fontweight='bold', zorder=10)
+			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 vlines
 	if events is not None:
 		for xpos in events:
@@ -350,4 +362,142 @@ def plot2d(*args, ann=None, events=None, xlim=None, ylim=None, origin="lower", M
 
 	fig.subplots_adjust(hspace=0.01, wspace=0.05)
 	plt.close()
-	return fig
+	return fig
+
+#======== Plot distribution ===========
+def plot_dist(*args, ann=None, xlim=None, ylim=None, ylabel=None, xlabel=None, title=None, legend=None, show_hist=True, npoints=200):
+		"""
+		Plot distribution.
+
+		.. code-block:: python
+			
+			import modusa as ms
+			import numpy as np
+			np.random.seed(42)
+			data = np.random.normal(loc=1, scale=1, size=1000)
+			ms.plot_dist(data, data+5, data-10, ann=[(0, 1, "A")], legend=("D1", "D2", "D3"), ylim=(0, 1), xlabel="X", ylabel="Counts", title="Distribution")
+
+		Parameters
+		----------
+		*args: ndarray
+			- Data arrays for which distribution needs to be plotted.
+			- Arrays will be flattened.
+		ann : list[tuple[Number, Number, str] | None
+			- A list of annotations to mark specific points. Each tuple should be of the form (start, end, label).
+			- Default: None => No annotation.
+		events : list[Number] | None
+			- A list of x-values where vertical lines (event markers) will be drawn.
+			- Default: None
+		xlim : tuple[Number, Number] | None
+			- Limits for the x-axis as (xmin, xmax).
+			- Default: None
+		ylim : tuple[Number, Number] | None
+			- Limits for the y-axis as (ymin, ymax).
+			- Default: None
+		xlabel : str | None
+			- Label for the x-axis.
+			- - Default: None
+		ylabel : str | None
+			- Label for the y-axis.
+			- Default: None
+		title : str | None
+			- Title of the plot.
+			- Default: None
+		legend : list[str] | None
+			- List of legend labels corresponding to each signal if plotting multiple distributions.
+			- Default: None
+		show_hist: bool
+			- Want to show histogram as well.
+		npoints: int
+			- Number of points for which gaussian needs to be computed between min and max.
+			- Higher value means more points are evaluated with the fitted gaussian, thereby higher resolution.
+
+		Returns
+		-------
+		plt.Figure
+			- Matplotlib figure.
+		"""
+		from scipy.stats import gaussian_kde
+	
+		if isinstance(legend, str):
+				legend = (legend, )
+			
+		if legend is not None:
+				if len(legend) < len(args):
+						raise ValueError(f"Legend should be provided for each signal.")
+					
+		# Create figure
+		fig = plt.figure(figsize=(16, 4))
+		gs = gridspec.GridSpec(2, 1, height_ratios=[0.1, 1])
+	
+		colors = plt.get_cmap('tab10').colors
+	
+		dist_ax = fig.add_subplot(gs[1, 0])
+		annotation_ax = fig.add_subplot(gs[0, 0], sharex=dist_ax)
+	
+		# Set limits
+		if xlim is not None:
+				dist_ax.set_xlim(xlim)
+			
+		if ylim is not None:
+				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=30, 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=30, density=True, alpha=0.3, facecolor=colors[i], edgecolor='black', label=legeng[i])
+							
+		# 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)
+					
+						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)
+			
+		# Set title, labels
+		if title is not None:
+				annotation_ax.set_title(title, pad=10, size=11)
+		if xlabel is not None:
+				dist_ax.set_xlabel(xlabel)
+		if ylabel is not None:
+				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)
+		else:
+				annotation_ax.axis("off")
+			
+		fig.subplots_adjust(hspace=0.01, wspace=0.05)
+		plt.close()
+		return fig

{modusa-0.3.21.dist-info → modusa-0.3.23.dist-info}/METADATA

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

{modusa-0.3.21.dist-info → modusa-0.3.23.dist-info}/RECORD

@@ -1,12 +1,12 @@
-modusa-0.3.21.dist-info/METADATA,sha256=4eI0lipMc4DDbHET_5se-4HUd6bo4u-fuKBu-YyAi6c,1369
-modusa-0.3.21.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.21.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.21.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.23.dist-info/METADATA,sha256=4WtEHA0q3OuyzT_zEwzH_c4N7jjfrtD2b2AxqGNh9Eo,1369
+modusa-0.3.23.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.23.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.23.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
 modusa/.DS_Store,sha256=_gm6qJREwfMi8dE7n5S89_RG46u5t3xHyD-smNhtNoM,6148
-modusa/__init__.py,sha256=AALG_17vibpDZbo0MymvGoQroaLp3uVm_onZaESONQA,257
+modusa/__init__.py,sha256=uq6kORFFAODiCMGmOLWO0shE8-dVFWf5gmV8wxekmnk,280
 modusa/config.py,sha256=bTqK4t00FZqERVITrxW_q284aDDJAa9aMSfFknfR-oU,280
 modusa/decorators.py,sha256=8zeNX_wE37O6Vp0ysR4-WCZaEL8mq8dyCF_I5DHOzks,5905
-modusa/devtools/generate_docs_source.py,sha256=qqxr3UgTeM4iotarQGRN_q2QXwAzGH7ha9bVjs8D4GQ,3409
+modusa/devtools/generate_docs_source.py,sha256=UDflHsk-Yh9-3YJTVBzKL32y8hcxiRgAlFEBTMiDqwM,3301
 modusa/devtools/generate_template.py,sha256=0sT_5-SjMLXyvePRDaIxr1RCz_9v6zFH9OoJqpuD2zI,5204
 modusa/devtools/list_authors.py,sha256=FWBQKOLznVthvMYMASrx2Gw5lqKHEOccQpBisDZ53Dw,24
 modusa/devtools/list_plugins.py,sha256=g-R5hoaCzHfClY_Sfa788IQ9CAKzQGTfyqmthXDZQqw,1729
@@ -27,7 +27,6 @@ modusa/generators/s2d.py,sha256=kU67dZj4tdIDSUFJeheXm_JKbyHpZZOGmW5jya6w0wo,6874
 modusa/generators/s_ax.py,sha256=4CTFp_KwYnl4HDno-ATpTXOfKR9WEVOV45nwuk2OoVk,2221
 modusa/generators/t_ax.py,sha256=X-XOJuGWw99dPAVsCVzNTfBFr81Vw56aZlGDmcl5q3k,1478
 modusa/generators/tds.py,sha256=eGvzcjXyWaw5NVzM3D098r3xkoMcX8Ma9YoDUdL30Mo,5960
-modusa/main.py,sha256=v9oOhnPcbw8ULfkKRoantEfHw7MqOETJwXC-odx3dnA,25
 modusa/models/__init__.py,sha256=G8sNOnBnTKqPmC4Z46ximBhc_pfBOF_G6AIfxT8DFzw,271
 modusa/models/__pycache__/signal1D.cpython-312.pyc.4443461152,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 modusa/models/audio.py,sha256=IcNx3h8tb5Jt6KZug_TQKM5iufVk3i7Ug37iKG4gsJ0,2411
@@ -41,14 +40,14 @@ modusa/models/t_ax.py,sha256=ZUhvZPUW1TkdZYuUd6Ucm-vsv0JqtZ9yEe3ab67Ma6w,8022
 modusa/models/tds.py,sha256=FAGfibjyyE_lkEuQp-vSCuqQnopOjmy_IXqUjRlg9kc,11677
 modusa/plugins/__init__.py,sha256=r1Bf5mnrVKRIwxboutY1iGzDy4EPQhqpk1kSW7iJj_Q,54
 modusa/plugins/base.py,sha256=Bh_1Bja7fOymFsCgwhXDbV6ys3D8muNrPwrfDrG_G_A,2382
-modusa/tmp.py,sha256=zPwgFVk5Oh6Rp6GUCbWJdkmwo6vYvhSqI35QVhW5xQM,3180
-modusa/tools/__init__.py,sha256=jOFL7PJ1IztN6F7aMdrVKNMTid2yUmoSlSST-WFf0Cc,199
+modusa/tools/__init__.py,sha256=ixqujVIJwU5TNX2j64lVajGtZlXTspcxN-30X54GgcM,243
+modusa/tools/ann_loader.py,sha256=hsSpjXmAJOCsgKX4j202t8lPzgmBkAyU0278sSbBriM,776
 modusa/tools/audio_converter.py,sha256=415qBoPm2sBIuBSI7m1XBKm0AbmVmPydIPPr-uO8D3c,1778
 modusa/tools/audio_loader.py,sha256=DrCzq0pdiQrUDIG-deLJGcu8EaylO5yRtwT4lr8WSf8,2166
 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=lxc6T9H8gKYU3pbFZlQYHXb7p5rGNzJol73ClElCZ_g,10579
+modusa/tools/plotter.py,sha256=U-Xkm8Br7Ba5C60xBuApd1qf9AScJ1llH_ichQksrxY,15295
 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
@@ -57,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.21.dist-info/RECORD,,
+modusa-0.3.23.dist-info/RECORD,,

modusa/main.py

@@ -1,3 +0,0 @@
-#!/usr/bin/env python3
-
-

modusa/tmp.py

@@ -1,98 +0,0 @@
-#def autocorr(self) -> Self:
-#	"""
-#	
-#	"""
-#	raise NotImplementedError
-#	r = np.correlate(self.data, self.data, mode="full")
-#	r = r[self.data.shape[0] - 1:]
-#	r_signal = self.__class__(data=r, sr=self.sr, t0=self.t0, title=self.title + " [Autocorr]")
-#	return r_signal
-
-#	#----------------------------
-#	# To different signals
-#	#----------------------------
-#	def to_audio_signal(self) -> "AudioSignal":
-#		"""
-#		Moves TimeDomainSignal to AudioSignal
-#		"""
-#		raise NotImplementedError
-#		from modusa.signals.audio_signal import AudioSignal
-#		
-#		return AudioSignal(data=self.data, sr=self.sr, t0=self.t0, title=self.title)
-#	
-#	def to_spectrogram(
-#		self,
-#		n_fft: int = 2048,
-#		hop_length: int = 512,
-#		win_length: int | None = None,
-#		window: str = "hann"
-#	) -> "Spectrogram":
-#		"""
-#		Compute the Short-Time Fourier Transform (STFT) and return a Spectrogram object.
-#		
-#		Parameters
-#		----------
-#		n_fft : int
-#			FFT size.
-#		win_length : int or None
-#			Window length. Defaults to `n_fft` if None.
-#		hop_length : int
-#			Hop length between frames.
-#		window : str
-#			Type of window function to use (e.g., 'hann', 'hamming').
-#		
-#		Returns
-#		-------
-#		Spectrogram
-#			Spectrogram object containing S (complex STFT), t (time bins), and f (frequency bins).
-#		"""
-#		raise NotImplementedError
-#		import warnings
-#		warnings.filterwarnings("ignore", category=UserWarning, module="librosa.core.intervals")
-#		
-#		from modusa.signals.feature_time_domain_signal import FeatureTimeDomainSignal
-#		import librosa
-#		
-#		S = librosa.stft(self.data, n_fft=n_fft, win_length=win_length, hop_length=hop_length, window=window)
-#		f = librosa.fft_frequencies(sr=self.sr, n_fft=n_fft)
-#		t = librosa.frames_to_time(np.arange(S.shape[1]), sr=self.sr, hop_length=hop_length)
-#		frame_rate = self.sr / hop_length
-#		spec = FeatureTimeDomainSignal(data=S, feature=f, feature_label="Freq (Hz)", frame_rate=frame_rate, t0=self.t0, time_label="Time (sec)", title=self.title)
-#		if self.title != self._name: # Means title of the audio was reset so we pass that info to spec
-#			spec = spec.set_meta_info(title=self.title)
-#		
-#		return spec
-#	#=====================================
-	
-	#=====================================
-	
-	#--------------------------
-	# Other signal ops
-	#--------------------------
-	
-#	def interpolate(self, to: TimeDomainSignal, kind: str = "linear", fill_value: str | float = "extrapolate") -> TimeDomainSignal:
-#		"""
-#		Interpolate the current signal to match the time axis of `to`.
-#	
-#		Parameters:
-#			to (TimeDomainSignal): The signal whose time axis will be used.
-#			kind (str): Interpolation method ('linear', 'nearest', etc.)
-#			fill_value (str or float): Value used to fill out-of-bounds.
-#	
-#		Returns:
-#			TimeDomainSignal: A new signal with values interpolated at `to.t`.
-#		"""
-#		assert self.y.ndim == 1, "Only 1D signals supported for interpolation"
-#		
-#		interpolator = interp1d(
-#			self.t,
-#			self.y,
-#			kind=kind,
-#			fill_value=fill_value,
-#			bounds_error=False,
-#			assume_sorted=True
-#		)
-#		
-#		y_interp = interpolator(to.y)
-	
-#		return self.__class__(y=y_interp, sr=to.sr, t0=to.t0, title=f"{self.title} → interpolated")