modusa 0.3.31__tar.gz → 0.3.41__tar.gz

{modusa-0.3.31 → modusa-0.3.41}/PKG-INFO

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

{modusa-0.3.31 → modusa-0.3.41}/pyproject.toml

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

{modusa-0.3.31 → modusa-0.3.41}/src/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,7 +161,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])
+						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]
@@ -142,15 +175,22 @@ 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, label=legend[i])
+						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: