modusa 0.3.67__tar.gz → 0.3.69__tar.gz

{modusa-0.3.67 → modusa-0.3.69}/PKG-INFO

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

{modusa-0.3.67 → modusa-0.3.69}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "modusa"
-version = "0.3.67"
+version = "0.3.69"
 description = "A modular signal analysis python library."
 authors = [
     { name = "Ankit Anand", email = "ankit0.anand0@gmail.com" },

modusa-0.3.69/src/modusa/tools/ann_loader.py

@@ -0,0 +1,65 @@
+#!/usr/bin/env python3
+
+#---------------------------------
+# Author: Ankit Anand
+# Date: 12/08/25
+# Email: ankit0.anand0@gmail.com
+#---------------------------------
+
+from pathlib import Path
+
+def load_ann(path, clip=None):
+	"""
+	Load annotation from audatity label text file.
+
+	Parameters
+	----------
+	path: str
+		- label text file path.
+	clip: tuple[number, number] | number | None
+		- Incase you clipped the audio signal, this parameter will help clip the annotation.
+		- If you clip the audio, say from (10, 20), set the clip to (10, 20).
+		- Default: None
+
+	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)}")
+	
+	if clip is not None:
+		# Map clip input to the right format
+		if isinstance(clip, int or float):
+			clip = (0, clip)
+		elif isinstance(clip, tuple) and len(clip) > 1:
+			clip = (clip[0], clip[1])
+		else:
+			raise ValueError(f"Invalid clip type or length: {type(clip)}, len={len(clip)}")
+	
+	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)
+			
+			# Incase user has clipped the audio signal, we adjust the annotation
+			# to match the clipped audio
+			if clip is not None:
+				offset = clip[0]
+				# Clamp annotation to clip boundaries
+				new_start = max(start, clip[0]) - offset
+				new_end   = min(end, clip[1]) - offset
+				
+				# only keep if there's still overlap
+				if new_start < new_end:
+					ann.append((new_start, new_end, tag))
+			else:
+				ann.append((start, end, tag))
+				
+	return ann

{modusa-0.3.67 → modusa-0.3.69}/src/modusa/tools/audio_loader.py

@@ -10,7 +10,7 @@ from .youtube_downloader import download
 from .audio_converter import convert
 
 
-def load(path, sr=None):
+def load(path, sr=None, clip=None):
 	"""
 	Loads audio file from various sources.
 
@@ -19,15 +19,19 @@ def load(path, sr=None):
 		import modusa as ms
 		audio_fp = ms.load(
 			"https://www.youtube.com/watch?v=lIpw9-Y_N0g",
-			sr = None)
+			sr = None, clip=(5, 10))
 
 	Parameters
 	----------
 	path: str
 		- Path to the audio
 		- Youtube URL
-	sr: int
+	sr: int | None
 		- Sampling rate to load the audio in.
+	clip: number | tuple[number, number] | None
+		- Which segment of the audio you want.
+		- Eg., 10 => First 10 sec, (5, 10) => 5 to 10 second
+		- Default: None => Entire audio.
 
 	Return
 	------
@@ -86,5 +90,17 @@ def load(path, sr=None):
 				n_samples = int(len(audio_data) * sr / audio_sr)
 				audio_data = resample(audio_data, n_samples)
 				audio_sr = sr
+				
+	# Clip the audio signal as per needed
+	if clip is not None:
+		# Map clip input to the right format
+		if isinstance(clip, int or float):
+			clip = (0, clip)
+		elif isinstance(clip, tuple) and len(clip) > 1:
+			clip = (clip[0], clip[1])
+		else:
+			raise ValueError(f"Invalid clip type or length: {type(clip)}, len={len(clip)}")
+		
+		audio_data = audio_data[int(clip[0]*sr):int(clip[1]*sr)]
 	
 	return audio_data, audio_sr, title

{modusa-0.3.67 → modusa-0.3.69}/src/modusa/tools/plotter.py

@@ -167,14 +167,14 @@ 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='gray', alpha=0.7)
+				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='gray', alpha=0.7)
+				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='gray', alpha=0.7)
+				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])

modusa-0.3.67/src/modusa/tools/ann_loader.py

@@ -1,38 +0,0 @@
-#!/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