halib 0.2.14__py3-none-any.whl → 0.2.16__py3-none-any.whl

halib/exp/perf/profiler.py

@@ -12,6 +12,8 @@ from plotly.subplots import make_subplots
 import plotly.graph_objects as go
 import plotly.express as px  # for dynamic color scales
 
+from contextlib import contextmanager
+
 from ...common.common import ConsoleLog
 from ...system.path import *
 
@@ -43,6 +45,20 @@ def check_enabled(func):
 # ==========================================
 # 2. The Class
 # ==========================================
+class ContextScope:
+    """Helper to remember the current context name so you don't have to repeat it."""
+
+    def __init__(self, profiler, ctx_name):
+        self.profiler = profiler
+        self.ctx_name = ctx_name
+
+    @contextmanager
+    def step(self, step_name):
+        # Automatically passes the stored ctx_name + the new step_name
+        with self.profiler.measure(self.ctx_name, step_name):
+            yield
+
+
 class zProfiler:
     """A singleton profiler to measure execution time of contexts and steps.
 
@@ -52,13 +68,23 @@ class zProfiler:
         output_file (str): Path to save the profiling report.
         report_format (str): Output format for reports ("json" or "csv").
 
-    Example:
+    Example (using context manager):
+        prof = zProfiler()
+        with prof.measure("my_context") as ctx:
+            with ctx.step("step1"):
+                time.sleep(0.1)
+            with ctx.step("step2"):
+                time.sleep(0.2)
+
+    Example (using raw methods):
         prof = zProfiler()
         prof.ctx_start("my_context")
         prof.step_start("my_context", "step1")
         time.sleep(0.1)
         prof.step_end("my_context", "step1")
         prof.ctx_end("my_context")
+
+
     """
 
     _instance = None
@@ -127,6 +153,24 @@ class zProfiler:
             return
         self.time_dict[ctx_name]["step_dict"][step_name][-1].append(time.perf_counter())
 
+    @contextmanager
+    def measure(self, ctx_name, step_name=None):
+        if step_name is None:
+            # --- Context Mode ---
+            self.ctx_start(ctx_name)
+            try:
+                # Yield the helper object initialized with the current context name
+                yield ContextScope(self, ctx_name)
+            finally:
+                self.ctx_end(ctx_name)
+        else:
+            # --- Step Mode ---
+            self.step_start(ctx_name, step_name)
+            try:
+                yield
+            finally:
+                self.step_end(ctx_name, step_name)
+
     def _step_dict_to_detail(self, ctx_step_dict):
         """
                 'ctx_step_dict': {

halib/filetype/videofile.py

@@ -96,6 +96,34 @@ class VideoUtils:
         return dfmk["video_meta"].copy()
 
 
+    # -----------------------------
+    # FFmpeg Horizontal Stack
+    # -----------------------------
+    @staticmethod
+    def video_hstack(video_files, output_file):
+        """Horizontally stack multiple videos using FFmpeg."""
+        tmp_file = "video_list.txt"
+        try:
+            with open(tmp_file, "w") as f:
+                for video in video_files:
+                    f.write(f"file '{video}'\n")
+
+            ffmpeg_cmd = (
+                f"ffmpeg -f concat -safe 0 -i {tmp_file} "
+                f'-filter_complex "[0:v][1:v][2:v]hstack=inputs={len(video_files)}[v]" '
+                f'-map "[v]" -c:v libx264 -preset fast -crf 22 {output_file}'
+            )
+
+            os.system(ffmpeg_cmd)
+            print(f"[INFO] Video stacked successfully: {output_file}")
+
+        except Exception as e:
+            print(f"[ERROR] Video stacking failed: {e}")
+        finally:
+            if os.path.exists(tmp_file):
+                os.remove(tmp_file)
+
+
 class VideoResolution(Enum):
     VR480p = "720x480"
     VR576p = "1280x720"
@@ -146,7 +174,6 @@ def trim_video(source, destination, start_time, end_time):
 
 progress_bar = None
 
-
 def on_progress(bytes_done, total_bytes):
     global progress_bar
     if progress_bar is None:

{halib-0.2.14.dist-info → halib-0.2.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.2.14
+Version: 0.2.16
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -41,6 +41,7 @@ Requires-Dist: tqdm
 Requires-Dist: tube_dl
 Requires-Dist: wandb
 Requires-Dist: ipynbname
+Requires-Dist: typed-argument-parser
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier
@@ -53,8 +54,8 @@ Dynamic: summary
 
 # Helper package for coding and automation
 
-**Version 0.2.14**
-+ update `exp.perf.profiler`: add `enabled` flag to enable/disable profiling dynamically
+**Version 0.2.16**
++ update `exp.perf.profiler`: add `enabled` flag to enable/disable profiling dynamically, also add `measure` context manager to simplify measuring code blocks.
 
 **Version 0.2.13**
 + reorganize packages with most changes in `research` package; also rename `research` to `exp` (package for experiment management and utilities)

{halib-0.2.14.dist-info → halib-0.2.16.dist-info}/RECORD

@@ -35,7 +35,7 @@ halib/exp/perf/gpu_mon.py,sha256=vD41_ZnmPLKguuq9X44SB_vwd9JrblO4BDzHLXZhhFY,223
 halib/exp/perf/perfcalc.py,sha256=p7rhVShiie7DT_s50lbvbGftVCkrWE0tQGFLUEmTXi0,18326
 halib/exp/perf/perfmetrics.py,sha256=qRiNiCKGUSTLY7gPMVMuVHGAAyeosfGWup2eM4490aw,5485
 halib/exp/perf/perftb.py,sha256=IWElg3OB5dmhfxnY8pMZvkL2y_EnvLmEx3gJlpUR1Fs,31066
-halib/exp/perf/profiler.py,sha256=57OO_BEC6CWKDCmLNyoYyz5nAvy-q8Lyc61afJkD0f4,13451
+halib/exp/perf/profiler.py,sha256=jWcOu4tujac-8eoRMmqLtOv71qIx-FAi9Ri_35Jj0W8,14817
 halib/exp/viz/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 halib/exp/viz/plot.py,sha256=51FhD1mH4nthTrY3K4mrO5pxI5AzvHXpZy5_ToqkYHs,28580
 halib/filetype/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -43,7 +43,7 @@ halib/filetype/csvfile.py,sha256=dwWM1LglQr1SaSWq2KDl93fYzQn1638JGo_l1GbBhbk,646
 halib/filetype/ipynb.py,sha256=pd2LgmPa7ZbF0YlQJbeQZEsl6jHQUSoyVtkCT7WhU5Q,1657
 halib/filetype/jsonfile.py,sha256=2HcBqXYjLNvqFok3PHOgH59vlhDCZLZpt7ezvgx1TFM,474
 halib/filetype/textfile.py,sha256=3koEFyVme1SEHdud7TnjArHndoiqfMGfMdYY3NIFegM,397
-halib/filetype/videofile.py,sha256=wDyZp7Dh0ZuNgQUvt8gLTpy3Flx1jDr-QsO4-jzriGE,8104
+halib/filetype/videofile.py,sha256=aYPWiCw-rVv2eVXe-3ycP6R9hwhvd-rLAdwXjB-EcOw,9103
 halib/filetype/yamlfile.py,sha256=buvj016n3wmEBDfeDX9lbljqDk5R5OlKHuXDKSfIKK0,2411
 halib/online/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 halib/online/gdrive.py,sha256=RmF4y6UPxektkKIctmfT-pKWZsBM9FVUeld6zZmJkp0,7787
@@ -102,8 +102,8 @@ halib/utils/list.py,sha256=BM-8sRhYyqF7bh4p7TQtV7P_gnFruUCA6DTUOombaZg,337
 halib/utils/listop.py,sha256=Vpa8_2fI0wySpB2-8sfTBkyi_A4FhoFVVvFiuvW8N64,339
 halib/utils/tele_noti.py,sha256=-4WXZelCA4W9BroapkRyIdUu9cUVrcJJhegnMs_WpGU,5928
 halib/utils/video.py,sha256=zLoj5EHk4SmP9OnoHjO8mLbzPdtq6gQPzTQisOEDdO8,3261
-halib-0.2.14.dist-info/licenses/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
-halib-0.2.14.dist-info/METADATA,sha256=jEayj2DXN6lkHvIzsVFf1cix3zp0aOmVT7y09H6UhFE,6950
-halib-0.2.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-halib-0.2.14.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
-halib-0.2.14.dist-info/RECORD,,
+halib-0.2.16.dist-info/licenses/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
+halib-0.2.16.dist-info/METADATA,sha256=AmEmCz0eQJgwhd3rtube-XXu67XcqZOBzhbLQt5Xwp0,7059
+halib-0.2.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+halib-0.2.16.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
+halib-0.2.16.dist-info/RECORD,,