modusa 0.3.68__py3-none-any.whl → 0.3.69__py3-none-any.whl

modusa/tools/ann_loader.py

@@ -8,14 +8,18 @@
 
 from pathlib import Path
 
-def load_ann(path):
+def load_ann(path, clip=None):
 	"""
 	Load annotation from audatity label text file.
 
 	Parameters
 	----------
 	path: str
-		label text file path.
+		- 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
 	-------
@@ -27,12 +31,35 @@ def load_ann(path):
 	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)
-			ann.append((start, end, tag))
 			
+			# 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/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.dist-info → modusa-0.3.69.dist-info}/METADATA

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

{modusa-0.3.68.dist-info → modusa-0.3.69.dist-info}/RECORD

@@ -1,7 +1,7 @@
-modusa-0.3.68.dist-info/METADATA,sha256=kOUJdmGTDXd5CHrTUbg_b7pBR0jZgTcuAx3lK_Ttkls,1436
-modusa-0.3.68.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.68.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.68.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.69.dist-info/METADATA,sha256=BVNPMjegkTKx-M2wB64UF5CaliXy-6J-eSTXs2mn7ck,1436
+modusa-0.3.69.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.69.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.69.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
 modusa/.DS_Store,sha256=_gm6qJREwfMi8dE7n5S89_RG46u5t3xHyD-smNhtNoM,6148
 modusa/__init__.py,sha256=RMIKAZZ27w1oh2UFMJvZQtQ-SluIQsac_3mSK_LaM30,277
 modusa/config.py,sha256=bTqK4t00FZqERVITrxW_q284aDDJAa9aMSfFknfR-oU,280
@@ -43,9 +43,9 @@ modusa/plugins/__init__.py,sha256=r1Bf5mnrVKRIwxboutY1iGzDy4EPQhqpk1kSW7iJj_Q,54
 modusa/plugins/base.py,sha256=Bh_1Bja7fOymFsCgwhXDbV6ys3D8muNrPwrfDrG_G_A,2382
 modusa/tools/__init__.py,sha256=TfIlnbSRVEk8C9YY6H5y0qiaKttFV_X3m6nI2qPYUH0,295
 modusa/tools/_plotter_old.py,sha256=KGow7mihA2H1WNq7s5bpivhCgGo2qVIeDaO6iabpsrg,19294
-modusa/tools/ann_loader.py,sha256=RePlwD4qG6gGrD4mOJ3RDR9q_gUscCY90_R9lgFU9lM,780
+modusa/tools/ann_loader.py,sha256=GWiX7v4CxyAECthmtBAaSMMqsjo1YaVfRA9u-CtTTFM,1718
 modusa/tools/audio_converter.py,sha256=415qBoPm2sBIuBSI7m1XBKm0AbmVmPydIPPr-uO8D3c,1778
-modusa/tools/audio_loader.py,sha256=DrCzq0pdiQrUDIG-deLJGcu8EaylO5yRtwT4lr8WSf8,2166
+modusa/tools/audio_loader.py,sha256=LEknkmc09SoUGN80-Uy59sVyeC6_55tD3a6ICbvaqWQ,2768
 modusa/tools/audio_player.py,sha256=GP04TWW4jBwQBjANkfR_cJtEy7cIhvbu8RTwnf9hD6E,2817
 modusa/tools/audio_recorder.py,sha256=d2fVt0Sd2tlBdb2WlUs60K4N23zuxM3KUpQqX0ifPi8,2769
 modusa/tools/base.py,sha256=C0ESJ0mIfjjRlAkRbSetNtMoOfS6IrHBjexRp3l_Mh4,1293
@@ -59,4 +59,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.68.dist-info/RECORD,,
+modusa-0.3.69.dist-info/RECORD,,