modusa 0.3.30__py3-none-any.whl → 0.3.41__py3-none-any.whl

modusa/tools/plotter.py

@@ -25,9 +25,31 @@ def _calculate_extent(x, y):
 		y[0] - dy / 2,
 		y[-1] + dy / 2
 	]
+	
+# Helper to load fonts (devnagri)
+def set_default_hindi_font():
+	"""
+	Hindi fonts works for both english and hindi.
+	"""
+	from pathlib import Path
+	import matplotlib as mpl
+	import matplotlib.font_manager as fm
+	# Path to your bundled font
+	font_path = Path(__file__).resolve().parents[1] / "fonts" / "NotoSansDevanagari-Regular.ttf"
+	
+	# Register the font with matplotlib
+	fm.fontManager.addfont(str(font_path))
+	
+	# Get the font family name from the file
+	hindi_font = fm.FontProperties(fname=str(font_path))
+	
+	# Set as default rcParam
+	mpl.rcParams['font.family'] = hindi_font.get_name()
+
+set_default_hindi_font()
 
 #======== 1D ===========
-def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylabel=None, title=None, legend=None, show_grid=False, show_stem=False):
+def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylabel=None, title=None, legend=None, fmt=None, show_grid=False, show_stem=False):
 		"""
 		Plots a 1D signal using matplotlib.
 
@@ -72,6 +94,9 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		legend : list[str] | None
 			- List of legend labels corresponding to each signal if plotting multiple lines.
 			- Default: None
+		fmt: list[str] | None
+			- linefmt for different line plots.
+			- Default: None
 		show_grid: bool
 			- If you want to show the grid.
 			- Default: False
@@ -84,21 +109,24 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		plt.Figure
 			Matplolib figure.
 		"""
-
 		for arg in args:
 			if len(arg) not in [1, 2]: # 1 if it just provides values, 2 if it provided axis as well
 				raise ValueError(f"1D signal needs to have max 2 arrays (y, x) or simply (y, )")
 			
 		if isinstance(legend, str): legend = (legend, )
-		
 		if legend is not None:
 			if len(legend) < len(args):
-				raise ValueError(f"Legend should be provided for each signal.")
+				raise ValueError(f"`legend` should be provided for each signal.")
+		
+		if isinstance(fmt, str): fmt = [fmt]
+		if fmt is not None:
+			if len(fmt) < len(args):
+				raise ValueError(f"`fmt` should be provided for each signal.")
+
+		colors = plt.get_cmap('tab10').colors
 
 		fig = plt.figure(figsize=(16, 2))
 		gs = gridspec.GridSpec(2, 1, height_ratios=[0.2, 1])
-			
-		colors = plt.get_cmap('tab10').colors
 		
 		signal_ax = fig.add_subplot(gs[1, 0])
 		annotation_ax = fig.add_subplot(gs[0, 0], sharex=signal_ax)
@@ -109,7 +137,6 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 		
 		if ylim is not None:
 			signal_ax.set_ylim(ylim)
-		
 			
 		# Add signal plot
 		for i, signal in enumerate(args):
@@ -123,7 +150,10 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 						stemlines.set_color(colors[i])
 						baseline.set_color("k")
 					else:
-						signal_ax.plot(x, y, color=colors[i], label=legend[i])
+						if fmt is not None:
+							signal_ax.plot(x, y, fmt[i], markersize=4, label=legend[i])
+						else:
+							signal_ax.plot(x, y, color=colors[i], label=legend[i])
 				else:
 					if show_stem is True:
 						markerline, stemlines, baseline = signal_ax.stem(x, y)
@@ -131,20 +161,36 @@ def plot1d(*args, ann=None, events=None, xlim=None, ylim=None, xlabel=None, ylab
 						stemlines.set_color(colors[i])
 						baseline.set_color("k")
 					else:
-						signal_ax.plot(x, y, color=colors[i])
+						if fmt is not None:
+							signal_ax.plot(x, y, fmt[i], markersize=4)
+						else:
+							signal_ax.plot(x, y, color=colors[i])
 						
 			elif len(signal) == 2:
 				y, x = signal[0], signal[1]
 				if legend is not None:
 					if show_stem is True:
-						signal_ax.stem(x, y, linefmt=f"{colors[i]}-", markerfmt=f"{colors[i]}o", basefmt="k-", label=legend[i])
+						markerline, stemlines, baseline = signal_ax.stem(x, y, label=legend[i])
+						markerline.set_color(colors[i])
+						stemlines.set_color(colors[i])
+						baseline.set_color("k")
 					else:
-						signal_ax.plot(x, y, color=colors[i], label=legend[i])
+						if fmt is not None:
+							signal_ax.plot(x, y, fmt[i], markersize=4, label=legend[i])
+						else:
+							signal_ax.plot(x, y, color=colors[i], label=legend[i])
 				else:
 					if show_stem is True:
-						signal_ax.stem(x, y, linefmt=f"{colors[i]}-", markerfmt=f"{colors[i]}o", basefmt="k-")
+						markerline, stemlines, baseline = signal_ax.stem(x, y)
+						markerline.set_color(colors[i])
+						stemlines.set_color(colors[i])
+						baseline.set_color("k")
 					else:
-						signal_ax.plot(x, y, color=colors[i])
+						if fmt is not None:
+							signal_ax.plot(x, y, fmt[i], markersize=4)
+						else:
+							signal_ax.plot(x, y, color=colors[i])
+							
 		
 		# Add annotations
 		if ann is not None:

{modusa-0.3.30.dist-info → modusa-0.3.41.dist-info}/METADATA

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

{modusa-0.3.30.dist-info → modusa-0.3.41.dist-info}/RECORD

@@ -1,7 +1,7 @@
-modusa-0.3.30.dist-info/METADATA,sha256=ue8QCA3HR90-JBEZkGmfM-up5D6tFqn88u2QqGrdHd8,1369
-modusa-0.3.30.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.30.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.30.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.41.dist-info/METADATA,sha256=h6zQe0lZ2sqn5JHLbqAw9wqexEXjWW28gNqmXV8OdCk,1369
+modusa-0.3.41.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.41.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.41.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
 modusa/.DS_Store,sha256=_gm6qJREwfMi8dE7n5S89_RG46u5t3xHyD-smNhtNoM,6148
 modusa/__init__.py,sha256=uq6kORFFAODiCMGmOLWO0shE8-dVFWf5gmV8wxekmnk,280
 modusa/config.py,sha256=bTqK4t00FZqERVITrxW_q284aDDJAa9aMSfFknfR-oU,280
@@ -17,6 +17,7 @@ modusa/devtools/templates/model.py,sha256=iwmXadLMA4IAMgOEktdCWKZsceLgaaZkTWcJHq
 modusa/devtools/templates/plugin.py,sha256=MWqrxBxfRKVw2zjEgfAG0RYUAMjuazhqC5PXIb0gJi4,868
 modusa/devtools/templates/test.py,sha256=CjZ_oqwJFhltNYzqfK1pcz1ShgeJXZLOnYpmOawfASs,244
 modusa/devtools/templates/tool.py,sha256=2iQfHbFKtTELPvNoX42rHr9_5lXRrXpwxb9z5OL96Xc,435
+modusa/fonts/NotoSansDevanagari-Regular.ttf,sha256=CEqU2J61Sq-5OgVuFUJcNP2Fn2NCh1Fl0wSDezvPwtI,221084
 modusa/generators/__init__.py,sha256=xQUsOObRhTEPzdZsBGd4nT_9AWaL8uE_bisFAN_Wp9Q,236
 modusa/generators/audio.py,sha256=91CnBDPZhagfJOx-cDoBBj3NrE92m70PmFmv7T8aQco,4477
 modusa/generators/audio_waveforms.py,sha256=x-r4_kDi2xieQNBCo11qY2uk2bFcvzeNOppfJnj2oYs,6105
@@ -47,7 +48,7 @@ modusa/tools/audio_loader.py,sha256=DrCzq0pdiQrUDIG-deLJGcu8EaylO5yRtwT4lr8WSf8,
 modusa/tools/audio_player.py,sha256=GP04TWW4jBwQBjANkfR_cJtEy7cIhvbu8RTwnf9hD6E,2817
 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=rIv5TmKtr9uliRcLNRjmejwvIixZqpUf4_dtqSlx2Hg,17510
+modusa/tools/plotter.py,sha256=e1NKFTFm8tG7yd-9VLmxmI9h9mTcUl1vaK1XDMvyoOw,18917
 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
@@ -56,4 +57,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.30.dist-info/RECORD,,
+modusa-0.3.41.dist-info/RECORD,,