modusa 0.3.91__py3-none-any.whl → 0.3.93__py3-none-any.whl

modusa/__init__.py

@@ -1,11 +1,11 @@
 from modusa.utils import excp, config
 
 #=====Giving access to plot functions to plot multiple signals.=====
-from modusa.tools import plot_dist, fig
+from modusa.tools import dist_plot, hill_plot, fig
 #=====
 
 from modusa.tools import play, convert, record, save
 from modusa.tools import download
 from modusa.tools import load, load_ann
 
-__version__ = "0.3.91"
+__version__ = "0.3.93"

modusa/tools/__init__.py

@@ -9,4 +9,4 @@ from .ann_loader import load_ann
 from .audio_recorder import record
 
 from .plotter import Fig as fig
-from .plotter import plot_dist
+from .plotter import dist_plot, hill_plot

modusa/tools/plotter.py

@@ -564,9 +564,6 @@ class Fig:
 		if grid is True:
 			curr_row[0].grid(True, linestyle='--', linewidth=0.7, color="lightgray" ,alpha=0.6)
 	
-	from matplotlib.patches import Rectangle
-	import matplotlib.pyplot as plt
-	
 	def add_arrows(self, xys, labels, text_offset=(0, 0), c="r", fontsize=12, ax=None):
 		"""
 		Add multiple arrows pointing to specific points with boxed labels at the tails.
@@ -715,10 +712,153 @@ class Fig:
 		fig = self._fig
 		fig.savefig(path, bbox_inches="tight")
 		
+#======== Distribution ==========
+def hill_plot(*args, labels=None, xlabel=None, ylabel=None, title=None, widths=0.7, bw_method=0.3, jitter_amount=0.1, side='upper', show_stats=True, ax=None):
+		"""
+		A plot to see distribution of different groups
+		along with statistical markers.
+		
+		Parameters
+		----------
+		*args : array-like
+				- Data arrays for each group.
+		labels : list of str, optional
+				- Labels for each group (y-axis).
+		side : str, 'upper' or 'lower'
+				Which half of the violin to draw (upper or lower relative to y-axis).
+		show_stats : bool, optional
+				Whether to show mean and median markers.
+		xlabel : str, optional
+				Label for x-axis.
+		ylabel : str, optional
+				Label for y-axis.
+		ax : matplotlib axes, optional
+				Axes to plot on.
+		title : str, optional
+				Plot title.
+		widths : float, optional
+				Width of violins.
+		bw_method : float, optional
+				Bandwidth method for KDE.
+		jitter_amount : float, optional
+				Amount of vertical jitter for strip points.
+		"""
+	
+		created_fig = False
+		if ax is None:
+			fig, ax = plt.subplots(figsize=(8, len(args) * 1.5))
+			created_fig = True
+			
+		n = len(args)
+	
+		# Default labels/colors
+		if labels is None:
+			labels = [f"Group {i}" for i in range(1, n+1)]
+			
+		colors = plt.cm.tab10.colors
+		if len(colors) < n:
+			colors = [colors[i % len(colors)] for i in range(n)] # Repeat colors
+			
+		# --- Half-violin ---
+		parts = ax.violinplot(args, vert=False, showmeans=False, showmedians=False, widths=widths, bw_method=bw_method)
+	
+		# Remove the default bar lines from violin plot
+		for key in ['cbars', 'cmins', 'cmaxes', 'cmedians']:
+			if key in parts:
+				parts[key].set_visible(False)
+					
+		# Clip violin bodies to show only upper or lower half
+		for i, pc in enumerate(parts['bodies']):
+			verts = pc.get_paths()[0].vertices
+			y_center = i + 1  # Center y-position for this violin
+		
+			if side == 'upper':
+					verts[:, 1] = np.maximum(verts[:, 1], y_center)
+			else:  # 'lower'
+					verts[:, 1] = np.minimum(verts[:, 1], y_center)
+		
+			pc.set_facecolor(colors[i])
+			pc.set_edgecolor(colors[i])
+			pc.set_linewidth(1.5)
+			pc.set_alpha(0.3)
+			
+		# --- Strip points with jitter ---
+		for i, x in enumerate(args, start=1):
+			x = np.array(x)
+			jitter = (np.random.rand(len(x)) - 0.5) * jitter_amount
+			y_positions = np.full(len(x), i) + jitter
+			ax.scatter(x, y_positions, color=colors[i-1], alpha=0.6, s=25, edgecolor="white", linewidth=0.8, zorder=2)
+			
+		# --- Statistical markers on violin distribution curve ---
+		if show_stats:
+			for i, (pc, x) in enumerate(zip(parts['bodies'], args), start=1):
+				x = np.array(x)
+				median_val = np.median(x)
+				mean_val = np.mean(x)
+			
+				# Get the violin curve vertices
+				verts = pc.get_paths()[0].vertices
+			
+				# Find y-position on violin curve for median
+				median_mask = np.abs(verts[:, 0] - median_val) < (np.ptp(x) * 0.01)
+				if median_mask.any():
+					median_y = np.max(verts[median_mask, 1]) if side == 'upper' else np.min(verts[median_mask, 1])
+				else:
+					median_y = i + widths/2 if side == 'upper' else i - widths/2
+					
+				# Find y-position on violin curve for mean
+				mean_mask = np.abs(verts[:, 0] - mean_val) < (np.ptp(x) * 0.01)
+				if mean_mask.any():
+					mean_y = np.max(verts[mean_mask, 1]) if side == 'upper' else np.min(verts[mean_mask, 1])
+				else:
+					mean_y = i + widths/2 if side == 'upper' else i - widths/2
+					
+				# Triangle offset from curve
+				triangle_offset = 0.05
+			
+				# Mean marker - triangle below curve pointing up
+				ax.scatter(mean_val, mean_y - triangle_offset, marker='^', s=30, 
+									facecolor=colors[i-1], edgecolor='none', 
+									linewidth=0, zorder=6,
+									label='Mean' if i == 1 else '')
+			
+				# Mean value text - below the triangle
+				ax.text(mean_val, mean_y - triangle_offset - 0.05, f'mean: {mean_val:.2f}', ha='center', va='top', fontsize=8, fontweight='bold', color=colors[i-1], zorder=7)
+			
+				# Median marker - triangle above curve pointing down
+				ax.scatter(median_val, median_y + triangle_offset, marker='v', s=30, 
+									facecolor=colors[i-1], edgecolor='none', 
+									linewidth=0, zorder=6,
+									label='Median' if i == 1 else '')
+			
+				# Median value text - above the triangle
+				ax.text(median_val, median_y + triangle_offset + 0.05, f'median: {median_val:.2f}', ha='center', va='bottom', fontsize=8, fontweight='bold', color=colors[i-1], zorder=7)
+					
+		# --- Labels & formatting ---
+		ax.set_yticks(range(1, n + 1))
+		ax.set_yticklabels(labels, fontsize=10)
+		if side == "lower":
+			ax.set_ylim(0.2, n + 0.5)
+		else:
+			ax.set_ylim(0.5, n + 0.5) 
+			
+		# Style improvements
+		ax.spines['top'].set_visible(False)
+		ax.spines['right'].set_visible(False)
+		ax.grid(axis='x', alpha=0.3, linestyle='--', linewidth=0.5)
+	
+		if xlabel:
+			ax.set_xlabel(xlabel, fontsize=11, fontweight='bold')
+		if ylabel:
+			ax.set_ylabel(ylabel, fontsize=11, fontweight='bold')
+		if title:
+			ax.set_title(title, fontsize=13, fontweight='bold', pad=15)
+			
+		plt.tight_layout()
+		return fig
 
 
-#======== Plot distribution ===========
-def plot_dist(*args, ann=None, xlim=None, ylim=None, ylabel=None, xlabel=None, title=None, legend=None, show_hist=True, npoints=200, bins=30):
+def dist_plot(*args, ann=None, xlim=None, ylim=None, ylabel=None, xlabel=None, title=None, legend=None, show_hist=True, npoints=200, bins=30):
 		"""
 		Plot distribution.
 

{modusa-0.3.91.dist-info → modusa-0.3.93.dist-info}/METADATA

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

{modusa-0.3.91.dist-info → modusa-0.3.93.dist-info}/RECORD

@@ -1,9 +1,9 @@
-modusa-0.3.91.dist-info/METADATA,sha256=3wLMqIRUHqEe23OD1j9OqnBXV3RBqM6YaU380XhXkZ8,1436
-modusa-0.3.91.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.3.91.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.3.91.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa-0.3.93.dist-info/METADATA,sha256=RDrVzEQDd_gSKmqp5zJb3sEXW1qf8ly0pUr_mMoT3ds,1436
+modusa-0.3.93.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.3.93.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.3.93.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
 modusa/.DS_Store,sha256=_gm6qJREwfMi8dE7n5S89_RG46u5t3xHyD-smNhtNoM,6148
-modusa/__init__.py,sha256=4PuyritLFoTqhN66PQKiIJXIiM2jz_f4LSjfJDnxa2I,307
+modusa/__init__.py,sha256=NlYjCzGVEesvEr3X2fIpNC4ZmNoEcoB3uUo-9VeI-8A,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
@@ -41,7 +41,7 @@ modusa/models/t_ax.py,sha256=ZUhvZPUW1TkdZYuUd6Ucm-vsv0JqtZ9yEe3ab67Ma6w,8022
 modusa/models/tds.py,sha256=FAGfibjyyE_lkEuQp-vSCuqQnopOjmy_IXqUjRlg9kc,11677
 modusa/plugins/__init__.py,sha256=r1Bf5mnrVKRIwxboutY1iGzDy4EPQhqpk1kSW7iJj_Q,54
 modusa/plugins/base.py,sha256=Bh_1Bja7fOymFsCgwhXDbV6ys3D8muNrPwrfDrG_G_A,2382
-modusa/tools/__init__.py,sha256=V2Zr75ZsKLQAokWUsxgK2nfuzYR4HLzIRLMHpCnu7_4,325
+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/audio_converter.py,sha256=415qBoPm2sBIuBSI7m1XBKm0AbmVmPydIPPr-uO8D3c,1778
@@ -51,7 +51,7 @@ modusa/tools/audio_recorder.py,sha256=K_LGqsPdjTdf3figEZTSQLmgMzYWgz18HTO8C1j5fE
 modusa/tools/audio_saver.py,sha256=ldzfr_AydsHTnKbxmBLJblN-hLzTmOlppOm306xI4Ug,510
 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=zxt2ceFdVtLA0IQ43EtHQnwslfDvjt1d8zwNvlBzePw,24871
+modusa/tools/plotter.py,sha256=QUeFYJLP0u88VM_HJVRDzLRU0d8jTndgIvz8-OzUGuI,29786
 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
@@ -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.3.91.dist-info/RECORD,,
+modusa-0.3.93.dist-info/RECORD,,