modusa 0.3.22__py3-none-any.whl → 0.3.24__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 not 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

@@ -362,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.22.dist-info → modusa-0.3.24.dist-info}/METADATA

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

{modusa-0.3.22.dist-info → modusa-0.3.24.dist-info}/RECORD

@@ -1,12 +1,12 @@
-modusa-0.3.22.dist-info/METADATA,sha256=uKAnJIVUJhOqOl4eXWXtvytJ7dACKCk3Te-0bctAzCA,1369
-modusa-0.3.22.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.22.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.22.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.24.dist-info/METADATA,sha256=XroQ__wqEQdtRoR1y22KngZO7gROam3uHT5aEPyAuPM,1369
+modusa-0.3.24.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.24.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.24.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
@@ -40,13 +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/tools/__init__.py,sha256=jOFL7PJ1IztN6F7aMdrVKNMTid2yUmoSlSST-WFf0Cc,199
+modusa/tools/__init__.py,sha256=ixqujVIJwU5TNX2j64lVajGtZlXTspcxN-30X54GgcM,243
+modusa/tools/ann_loader.py,sha256=RePlwD4qG6gGrD4mOJ3RDR9q_gUscCY90_R9lgFU9lM,780
 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=pmB9l_CTXnxC25fML_6M4jiBV1l645GtpZP4Ewoeo4E,10735
+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
@@ -55,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.22.dist-info/RECORD,,
+modusa-0.3.24.dist-info/RECORD,,