lightning-pose-app 1.8.1a2__py3-none-any.whl → 1.8.1a4__py3-none-any.whl

litpose_app/transcode_fine.py

@@ -0,0 +1,175 @@
+import shutil
+import subprocess
+from multiprocessing import Pool, cpu_count
+from pathlib import Path
+
+# --- Configuration ---
+TARGET_SUFFIX = "sec.mp4"
+OUTPUT_SUFFIX_ADDITION = ".fine"
+MAX_CONCURRENCY = 6
+# FFmpeg options for transcoding:
+# -g 1: Intra frame for every frame (Group of Pictures size 1)
+# -c:v libx264: Use libx264 encoder
+# -preset medium: A balance between encoding speed and compression.
+#                 Options: ultrafast, superfast, veryfast, faster, fast, medium, slow, slower, veryslow.
+# -crf 23: Constant Rate Factor. Lower values mean better quality and larger files (0-51, default 23).
+# -c:a copy: Copy audio stream without re-encoding. If audio re-encoding is needed, change this.
+# -y: Overwrite output files without asking.
+FFMPEG_OPTIONS = [
+    "-c:v",
+    "libx264",
+    "-g",
+    "1",
+    "-preset",
+    "medium",
+    "-crf",
+    "23",
+    "-c:a",
+    "copy",
+]
+
+
+def check_dependencies():
+    """Checks if ffmpeg and ffprobe are installed and in PATH."""
+    if shutil.which("ffmpeg") is None:
+        print("Error: ffmpeg is not installed or not found in PATH.")
+        return False
+    if shutil.which("ffprobe") is None:
+        print("Error: ffprobe is not installed or not found in PATH.")
+        return False
+    return True
+
+
+def transcode_file(
+    input_file_path: Path,
+    output_file_path: Path,
+) -> tuple[bool, str, Path | None]:
+    """
+    Transcodes a single video file to have an intra frame for every frame.
+    The output file will be named by inserting ".fine" before the final ".mp4"
+    and placed in the specified output_dir.
+    Example: "video.sec.mp4" -> "video.sec.fine.mp4"
+    Returns a tuple: (success_status: bool, message: str, output_path: Path | None)
+    """
+    try:
+
+        if output_file_path.exists():
+            print(
+                f"Output file '{output_file_path.name}' already exists. Skipping transcoding."
+            )
+            return True, f"Skipped (exists): {output_file_path.name}", output_file_path
+
+        import sys
+
+        print(f"Processing: {input_file_path.name} -> {output_file_path.name}")
+
+        ffmpeg_cmd = [
+            "ffmpeg",
+            "-i",
+            str(input_file_path),
+            *FFMPEG_OPTIONS,
+            "-y",  # Overwrite output without asking (though we check existence above)
+            str(output_file_path),
+        ]
+
+        process = subprocess.Popen(
+            ffmpeg_cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True
+        )
+
+        stdout, stderr = process.communicate()
+
+        if process.returncode == 0:
+            print(f"Successfully transcoded: {output_file_path.name}")
+            return True, f"Success: {output_file_path.name}", output_file_path
+        else:
+            print(f"Error transcoding '{input_file_path.name}':")
+            print(f"FFmpeg stdout:\n{stdout}")
+            print(f"FFmpeg stderr:\n{stderr}")
+            # Clean up partially created file on error
+            if output_file_path.exists():
+                try:
+                    output_file_path.unlink()
+                except OSError as e:
+                    print(
+                        f"Could not remove partially created file '{output_file_path}': {e}"
+                    )
+            return (
+                False,
+                f"Error: {input_file_path.name} - FFmpeg failed (code {process.returncode})",
+                None,
+            )
+
+    except Exception as e:
+        print(f"Error processing '{input_file_path.name}': {e}")
+        return False, f"Error: {input_file_path.name} - Exception: {e}", None
+
+
+def main():
+    """
+    Main function to find and transcode videos.
+    """
+    if not check_dependencies():
+        return
+
+    script_dir = Path(__file__).parent  # Process files in the script's directory
+    # To process files in the current working directory instead:
+    # current_dir = Path.cwd()
+
+    print(
+        f"Scanning for '*{TARGET_SUFFIX}' H.264 files in '{script_dir}' and its subdirectories..."
+    )
+
+    # Find all files ending with TARGET_SUFFIX recursively
+    files_to_check = list(script_dir.rglob(f"*{TARGET_SUFFIX}"))
+
+    if not files_to_check:
+        print(f"No files found ending with '{TARGET_SUFFIX}'.")
+        return
+
+    print(f"Found {len(files_to_check)} potential files. Checking H.264 codec...")
+
+    valid_files_to_transcode = []
+    for f_path in files_to_check:
+        # Ensure it's not an already processed file
+        if OUTPUT_SUFFIX_ADDITION + ".mp4" in f_path.name:
+            continue
+        valid_files_to_transcode.append(f_path)
+
+    if not valid_files_to_transcode:
+        print("No H.264 files matching the criteria need transcoding.")
+        return
+
+    print(f"\nFound {len(valid_files_to_transcode)} H.264 files to transcode:")
+    for f in valid_files_to_transcode:
+        print(f"  - {f.name}")
+
+    # Determine number of processes
+    num_processes = min(MAX_CONCURRENCY, cpu_count(), len(valid_files_to_transcode))
+    print(f"\nStarting transcoding with up to {num_processes} parallel processes...\n")
+
+    output_file_paths = []
+    for f in valid_files_to_transcode:
+        base_name = f.name[: -len(TARGET_SUFFIX)]
+        output_file_name = f"{base_name}{TARGET_SUFFIX.replace('.mp4', '')}{OUTPUT_SUFFIX_ADDITION}.mp4"
+        output_file_path = f.parent / output_file_name
+        output_file_paths.append(output_file_path)
+    # In this main function, output_dir is still the parent of the input file
+    # For RPC, we will specify FINE_VIDEO_DIR as output_dir
+    with Pool(processes=num_processes) as pool:
+        # A dummy output_dir for the script's main function; not used by RPC.
+        # It ensures compatibility if this script were run standalone.
+        # In the RPC, we'll pass FINE_VIDEO_DIR explicitly.
+        results = pool.starmap(
+            transcode_file,
+            [(f, of) for f, of in zip(valid_files_to_transcode, output_file_paths)],
+        )
+
+    print("\n--- Transcoding Summary ---")
+    for result in results:
+        print(result)
+    print("--------------------------")
+    print("All tasks completed.")
+
+
+if __name__ == "__main__":
+    main()

lightning_pose_app-1.8.1a2.dist-info/METADATA

@@ -1,15 +0,0 @@
-Metadata-Version: 2.3
-Name: lightning-pose-app
-Version: 1.8.1a2
-Summary: 
-Requires-Python: >=3.10
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: fastapi
-Requires-Dist: tomli
-Requires-Dist: tomli_w
-Requires-Dist: uvicorn[standard]
-Requires-Dist: wcmatch

lightning_pose_app-1.8.1a2.dist-info/RECORD

@@ -1,20 +0,0 @@
-litpose_app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-litpose_app/main.py,sha256=UQ9WIxjHDDckhcjSwFqiuKj84tO7Ohvf6tP_OiiMIpE,9737
-litpose_app/ngdist/ng_app/3rdpartylicenses.txt,sha256=k34Q8jZikvFWsgFXPlQxFJVcqMXDEiKKzC_VIEK8jNs,25366
-litpose_app/ngdist/ng_app/app.component-IZ5OUDH2.css.map,sha256=e6lXWzmK3xppKK3tXHUccK1yGZqd1kzyTpDH0F1nC2g,344
-litpose_app/ngdist/ng_app/error-dialog.component-HYLQSJEP.css.map,sha256=zJuF-LfB994Y1IrnIz38mariDFb8yucffbWPXgHGbvw,355
-litpose_app/ngdist/ng_app/favicon.ico,sha256=QtbXVfx3HI-WqWh_kkBIQyYzJsDmLw_3Y4_UN7pBepE,15406
-litpose_app/ngdist/ng_app/index.html,sha256=FA0fsUwQVubXOgASPFXVR4cHlWgcKXezeXR4QX5wYdg,22954
-litpose_app/ngdist/ng_app/main-WFYIUX2C.js,sha256=ZzmVXvxTaXNe0akOR1LXkJNTF9cPinKAUc5MNEM0MMY,2719071
-litpose_app/ngdist/ng_app/main-WFYIUX2C.js.map,sha256=iN_kYNo-dAPFZSLWZ-EFFkKYhOnTTpYfSMOMPQ9PF3g,5553036
-litpose_app/ngdist/ng_app/prerendered-routes.json,sha256=p53cyKEVGQ6cGUee02kUdBp9HbdPChFTUp78gHJVBf4,18
-litpose_app/ngdist/ng_app/project-settings.component-BXKZMYM3.css.map,sha256=v5tyba9p8ec3ZbHYyyUGTEFdEAsqT0l7JqtGRGjki6w,371
-litpose_app/ngdist/ng_app/styles-AJ6NQDUD.css,sha256=IFn3SaSSlwBudgsGOQj_GirVrrJ0moGfEYouc9FLj6g,68857
-litpose_app/ngdist/ng_app/styles-AJ6NQDUD.css.map,sha256=hC5GJkT1Xw6TZqjnb9c5tCUOW17HNwqirjMpJaLKw50,74649
-litpose_app/ngdist/ng_app/video-player-controls.component-C4JZHYJ2.css.map,sha256=vX-dgeDCUCPLiea4Qy9O_EBm6IzzwB7R_uSBa0qU5Go,771
-litpose_app/ngdist/ng_app/viewer-page.component-KIYG73MW.css.map,sha256=Uf1FgoCiF_qJpD4Ggk34Dq7kM73Q7i7NT6h3m30tbaY,211
-litpose_app/run_ffprobe.py,sha256=VlZhCzaqqn7xy7Opb3GrBsWdWX_q_5JoF7n_Dg5l3Fc,4843
-litpose_app/super_rglob.py,sha256=uk5u2WKQ10L4sWvQ0aTxHiKD-yXRnwxbwAg9ci3JlYI,1571
-lightning_pose_app-1.8.1a2.dist-info/METADATA,sha256=zVkJb7nbH8IAHWf8BbYn8sUYnLXG1RfGohk7DqBLgNQ,473
-lightning_pose_app-1.8.1a2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-lightning_pose_app-1.8.1a2.dist-info/RECORD,,