modusa 0.4.10__py3-none-any.whl → 0.4.12__py3-none-any.whl

modusa/__init__.py

@@ -8,4 +8,4 @@ from modusa.tools import play, convert, record, save
 from modusa.tools import download
 from modusa.tools import load, load_ann
 
-__version__ = "0.4.10"
+__version__ = "0.4.12"

modusa/tools/ann_loader.py

@@ -6,16 +6,15 @@
 # Email: ankit0.anand0@gmail.com
 #---------------------------------
 
-from pathlib import Path
-
 def load_ann(path, clip=None):
 	"""
-	Load annotation from audatity label text file.
+	Load annotation from audatity label
+	text file and also ctm file.
 
 	Parameters
 	----------
 	path: str
-		- label text file path.
+		- label text/ctm 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).
@@ -25,12 +24,23 @@ def load_ann(path, clip=None):
 	-------
 	list[tuple, ...]
 		- annotation data structure
-		- [(start, end, tag), ...]
+		- [(start, end, label), ...]
 	"""
+	from pathlib import Path
 	
 	if not isinstance(path, (str, Path)):
 		raise ValueError(f"`path` must be one of (str, Path), got {type(path)}")
 	
+	# Convert to Path object
+	path = Path(path)
+	
+	# Check if the path exists
+	if not path.exists():
+		raise FileExistsError(f"{path} does not exist")
+		
+	ann = [] # This will store the annotation
+	
+	# Clipping the annotation to match with the clipped audio
 	if clip is not None:
 		# Map clip input to the right format
 		if isinstance(clip, int or float):
@@ -40,26 +50,56 @@ def load_ann(path, clip=None):
 		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)
+	if path.suffix == ".txt":
+		with open(str(path), "r") as f:
+			lines = [line.rstrip("\n") for line in f]
+			for line in lines:
+				start, end, label = 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, label))
+				else:
+					ann.append((start, end, label))
+					
+	elif path.suffix == ".ctm":
+		with open(str(path), "r") as f:
+			content = f.read().split("\n")
 			
-			# 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
+		for c in content:
+			c = c.strip()
+			if c == "": # Handle empty line usually at the end of the ctm file
+				continue
+			elif len(c.split(" ")) != 5:
+				warnings.warn(f" '{c}' is not a standard ctm line.")
+			if len(c.split(" ")) == 5:
+				_, _, start, dur, label = c.split(" ")
+				start, dur = float(start), float(dur)
+				end = start + dur
 				
-				# 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))
+				# 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, label))
+				else:
+					ann.append((start, end, label))
 				
+	else:
+		raise Exception(f"Unsupported file type {path.suffix}")
 	return ann

{modusa-0.4.10.dist-info → modusa-0.4.12.dist-info}/METADATA

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

{modusa-0.4.10.dist-info → modusa-0.4.12.dist-info}/RECORD

@@ -1,9 +1,9 @@
-modusa-0.4.10.dist-info/METADATA,sha256=TmKus5RBUh4HkCZ-1eK7GLmsMM2S00MMyJc5KMurV-o,1436
-modusa-0.4.10.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.4.10.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.4.10.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
-modusa/.DS_Store,sha256=_gm6qJREwfMi8dE7n5S89_RG46u5t3xHyD-smNhtNoM,6148
-modusa/__init__.py,sha256=PGHCD9XdADguJN6KWAqLZRORiKMNdIGkU18hCMlY57Q,318
+modusa-0.4.12.dist-info/METADATA,sha256=VrYqleweSm6-s9GwBfaR5k8Q6K4vdSDKKdMJT0WxfLc,1436
+modusa-0.4.12.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.4.12.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.4.12.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa/.DS_Store,sha256=b3Lw2RBGnDArqCyxps1EmN2oazEM0wnzg3BgurdE0_M,8196
+modusa/__init__.py,sha256=9qlTZXo5slwSmrlrdbAcvoc09imlI4fETfalfMCBpIM,318
 modusa/config.py,sha256=bTqK4t00FZqERVITrxW_q284aDDJAa9aMSfFknfR-oU,280
 modusa/decorators.py,sha256=8zeNX_wE37O6Vp0ysR4-WCZaEL8mq8dyCF_I5DHOzks,5905
 modusa/devtools/generate_docs_source.py,sha256=UDflHsk-Yh9-3YJTVBzKL32y8hcxiRgAlFEBTMiDqwM,3301
@@ -43,7 +43,7 @@ modusa/plugins/__init__.py,sha256=r1Bf5mnrVKRIwxboutY1iGzDy4EPQhqpk1kSW7iJj_Q,54
 modusa/plugins/base.py,sha256=Bh_1Bja7fOymFsCgwhXDbV6ys3D8muNrPwrfDrG_G_A,2382
 modusa/tools/__init__.py,sha256=q4d7J4xGraVLVLsk4h1OcWvi19FXd7DbJrGdUAOn5Es,336
 modusa/tools/_plotter_old.py,sha256=KGow7mihA2H1WNq7s5bpivhCgGo2qVIeDaO6iabpsrg,19294
-modusa/tools/ann_loader.py,sha256=GWiX7v4CxyAECthmtBAaSMMqsjo1YaVfRA9u-CtTTFM,1718
+modusa/tools/ann_loader.py,sha256=EMV6lTm1IeTjvWjvagDgwosux9IV6MZkVdUMzVft9Pc,3017
 modusa/tools/audio_converter.py,sha256=415qBoPm2sBIuBSI7m1XBKm0AbmVmPydIPPr-uO8D3c,1778
 modusa/tools/audio_loader.py,sha256=n9Q9t_GmlE8AtioVwRcXX3rnd6PkbGTx-hAoNgUnNOQ,2780
 modusa/tools/audio_player.py,sha256=pXKaNsDbqhCVbrMx5QRv7aC2IPMJFY5POAG60f-1LmU,2901
@@ -60,4 +60,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.4.10.dist-info/RECORD,,
+modusa-0.4.12.dist-info/RECORD,,