modusa 0.3.68__py3-none-any.whl → 0.3.70__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/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.dist-info → modusa-0.3.70.dist-info}/METADATA

@@ -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.dist-info → modusa-0.3.70.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.70.dist-info/METADATA,sha256=CRoFdZMzJT8bW68fBL5GpMsirp_zJDXFRqFdSnHyPfs,1436
+modusa-0.3.70.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.70.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.70.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,14 +43,14 @@ 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
 modusa/tools/math_ops.py,sha256=ZZ7U4DgqT7cOeE7_Lzi_Qq-48WYfwR9_osbZwTmE9eg,8690
-modusa/tools/plotter.py,sha256=2BotJit9on5R5pfDMMxE8w-cREAHUW6_QpVzoqtx8-s,21051
+modusa/tools/plotter.py,sha256=aWHjZ0nbPj7pQPrBV8urw9HUIlFoLwzIsE4cGm-BMKI,21206
 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
@@ -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.70.dist-info/RECORD,,