modusa 0.3.68__tar.gz → 0.3.70__tar.gz

{modusa-0.3.68 → modusa-0.3.70}/PKG-INFO

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

{modusa-0.3.68 → modusa-0.3.70}/pyproject.toml

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

modusa-0.3.70/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.68 → modusa-0.3.70}/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.68 → modusa-0.3.70}/src/modusa/tools/plotter.py

@@ -60,9 +60,12 @@ class Fig:
 	dark_mode: bool
 		- Do you want dark mode?
 		- Default: True
+	grid: bool
+		- Do you want the grid?
+		- Default: True
 	"""
 	
-	def __init__(self, arrangement="asm", xlim=None, width=16, dark_mode=True):
+	def __init__(self, arrangement="asm", xlim=None, width=16, dark_mode=True, grid=True):
 		
 		if dark_mode:
 			plt.style.use("dark_background")
@@ -80,6 +83,7 @@ class Fig:
 		self._xlim = xlim
 		self._curr_row_idx = 1 # Starting from 1 because row 0 is reserved for reference subplot
 		self._curr_color_idx = 0 # So that we have different color across all the subplots to avoid legend confusion
+		self._grid = grid
 		
 		# Subplot setup
 		self._fig, self._axs = self._generate_subplots(arrangement, width) # This will fill in the all the above variables
@@ -134,6 +138,7 @@ class Fig:
 		"""
 		
 		xlim = self._xlim
+		grid: bool = self._grid
 		fig_width = width
 		
 		n_aux_sp = arrangement.count("a")
@@ -167,14 +172,17 @@ 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)
+				if grid:
+					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)
+				if grid:
+					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)
+				if grid:
+					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.68/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