patch-torchcodec 0.1.0__tar.gz

patch_torchcodec-0.1.0/.gitignore

@@ -0,0 +1,228 @@
+# ============================================================================
+# Manually added items (project-specific)
+# ============================================================================
+
+# Version file (auto-generated by hatch-vcs)
+/src/mediaref/_version.py
+
+# ============================================================================
+# Below items are from GitHub's official Python .gitignore template
+# https://github.com/github/gitignore/blob/main/Python.gitignore
+# ============================================================================
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[codz]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#   Usually these files are written by a python script from a template
+#   before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py.cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+# Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+# poetry.lock
+# poetry.toml
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#   pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
+#   https://pdm-project.org/en/latest/usage/project/#working-with-version-control
+# pdm.lock
+# pdm.toml
+.pdm-python
+.pdm-build/
+
+# pixi
+#   Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
+# pixi.lock
+#   Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
+#   in the .venv directory. It is recommended not to include this directory in version control.
+.pixi
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# Redis
+*.rdb
+*.aof
+*.pid
+
+# RabbitMQ
+mnesia/
+rabbitmq/
+rabbitmq-data/
+
+# ActiveMQ
+activemq-data/
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#   JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#   be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#   and can be added to the global gitignore or merged into this file.  For a more nuclear
+#   option (not recommended) you can uncomment the following to ignore the entire idea folder.
+# .idea/
+
+# Abstra
+#   Abstra is an AI-powered process automation framework.
+#   Ignore directories containing user credentials, local state, and settings.
+#   Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#   Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore 
+#   that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#   and can be added to the global gitignore or merged into this file. However, if you prefer, 
+#   you could uncomment the following to ignore the entire vscode folder
+# .vscode/
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
+
+# Streamlit
+.streamlit/secrets.toml

patch_torchcodec-0.1.0/PKG-INFO

@@ -0,0 +1,87 @@
+Metadata-Version: 2.4
+Name: patch-torchcodec
+Version: 0.1.0
+Summary: Patch TorchCodec to use PyAV's bundled FFmpeg — one command, no system FFmpeg needed
+Project-URL: Homepage, https://github.com/open-world-agents/MediaRef
+Project-URL: Repository, https://github.com/open-world-agents/MediaRef
+Author: MediaRef Contributors
+License-Expression: MIT
+Keywords: decoder,ffmpeg,patchelf,pyav,torchcodec,video
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Multimedia :: Video
+Requires-Python: >=3.9
+Requires-Dist: av>=10.0.0
+Requires-Dist: patchelf
+Description-Content-Type: text/markdown
+
+# patch-torchcodec
+
+Patch TorchCodec to use PyAV's bundled FFmpeg — **one command, no system FFmpeg needed**.
+
+## Problem
+
+TorchCodec requires FFmpeg shared libraries (`libavcodec.so.62`, etc.), but installing FFmpeg system-wide can be complex and may cause version conflicts.
+
+## Solution
+
+PyAV already bundles FFmpeg. `patch-torchcodec` patches TorchCodec's RPATH to find them — no environment variables needed!
+
+## Quick Start
+
+```bash
+pip install torchcodec patch-torchcodec   # torchcodec + patcher (av & patchelf included)
+patch-torchcodec                          # patches RPATH — done!
+```
+
+That's it. TorchCodec now works:
+
+```python
+from torchcodec.decoders import VideoDecoder  # ✓ just works
+```
+
+## Command Line Options
+
+```bash
+patch-torchcodec               # Patch RPATH (default, recommended)
+patch-torchcodec --env-only    # Symlinks only (requires LD_LIBRARY_PATH)
+patch-torchcodec --status      # Check current setup status
+patch-torchcodec --verify      # Verify TorchCodec works
+patch-torchcodec --quiet       # Silent mode
+```
+
+## Python API
+
+```python
+from patch_torchcodec import setup_with_patchelf, verify_torchcodec, is_rpath_patched
+
+# Patch RPATH (recommended) — persists across sessions
+setup_with_patchelf(verbose=True)
+
+# Verify
+assert verify_torchcodec(require_env=False)
+```
+
+## How It Works
+
+1. **Finds PyAV's bundled FFmpeg** in `site-packages/av.libs/`
+2. **Creates symbolic links** with standard names (e.g., `libavcodec.so.62`)
+3. **Patches TorchCodec's `.so` files** with `patchelf` to include PyAV's library path in RPATH
+
+## Limitations
+
+- **Linux only** (RPATH is Linux-specific)
+- **Same virtualenv**: PyAV and TorchCodec must be in the same environment
+- **Re-run after reinstall**: If you reinstall TorchCodec, run `patch-torchcodec` again
+
+## License
+
+MIT
+

patch_torchcodec-0.1.0/README.md

@@ -0,0 +1,63 @@
+# patch-torchcodec
+
+Patch TorchCodec to use PyAV's bundled FFmpeg — **one command, no system FFmpeg needed**.
+
+## Problem
+
+TorchCodec requires FFmpeg shared libraries (`libavcodec.so.62`, etc.), but installing FFmpeg system-wide can be complex and may cause version conflicts.
+
+## Solution
+
+PyAV already bundles FFmpeg. `patch-torchcodec` patches TorchCodec's RPATH to find them — no environment variables needed!
+
+## Quick Start
+
+```bash
+pip install torchcodec patch-torchcodec   # torchcodec + patcher (av & patchelf included)
+patch-torchcodec                          # patches RPATH — done!
+```
+
+That's it. TorchCodec now works:
+
+```python
+from torchcodec.decoders import VideoDecoder  # ✓ just works
+```
+
+## Command Line Options
+
+```bash
+patch-torchcodec               # Patch RPATH (default, recommended)
+patch-torchcodec --env-only    # Symlinks only (requires LD_LIBRARY_PATH)
+patch-torchcodec --status      # Check current setup status
+patch-torchcodec --verify      # Verify TorchCodec works
+patch-torchcodec --quiet       # Silent mode
+```
+
+## Python API
+
+```python
+from patch_torchcodec import setup_with_patchelf, verify_torchcodec, is_rpath_patched
+
+# Patch RPATH (recommended) — persists across sessions
+setup_with_patchelf(verbose=True)
+
+# Verify
+assert verify_torchcodec(require_env=False)
+```
+
+## How It Works
+
+1. **Finds PyAV's bundled FFmpeg** in `site-packages/av.libs/`
+2. **Creates symbolic links** with standard names (e.g., `libavcodec.so.62`)
+3. **Patches TorchCodec's `.so` files** with `patchelf` to include PyAV's library path in RPATH
+
+## Limitations
+
+- **Linux only** (RPATH is Linux-specific)
+- **Same virtualenv**: PyAV and TorchCodec must be in the same environment
+- **Re-run after reinstall**: If you reinstall TorchCodec, run `patch-torchcodec` again
+
+## License
+
+MIT
+

patch_torchcodec-0.1.0/__init__.py

@@ -0,0 +1,49 @@
+"""Patch TorchCodec to use PyAV's bundled FFmpeg — no system FFmpeg needed.
+
+Usage:
+    pip install patch-torchcodec   # installs av + patchelf
+    patch-torchcodec               # patches RPATH — done!
+
+Python API:
+    from patch_torchcodec import setup_with_patchelf
+    setup_with_patchelf()  # One-time setup, persists across sessions
+
+Alternative (no binary patching):
+    patch-torchcodec --env-only
+    export LD_LIBRARY_PATH="<shown path>:$LD_LIBRARY_PATH"
+"""
+
+from __future__ import annotations
+
+__version__ = "0.2.0"
+
+from .core import (
+    create_all_symlinks,
+    create_symlinks,
+    find_av_libs_dir,
+    find_patchelf,
+    find_torchcodec_libs,
+    get_ld_library_path,
+    get_library_mappings,
+    is_rpath_patched,
+    patch_rpath,
+    setup,
+    setup_with_patchelf,
+    verify_torchcodec,
+)
+
+__all__ = [
+    "create_all_symlinks",
+    "create_symlinks",
+    "find_av_libs_dir",
+    "find_patchelf",
+    "find_torchcodec_libs",
+    "get_ld_library_path",
+    "get_library_mappings",
+    "is_rpath_patched",
+    "patch_rpath",
+    "setup",
+    "setup_with_patchelf",
+    "verify_torchcodec",
+]
+

patch_torchcodec-0.1.0/__main__.py

@@ -0,0 +1,223 @@
+#!/usr/bin/env python3
+"""Command-line interface for patch-torchcodec."""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from pathlib import Path
+
+from .core import (
+    create_all_symlinks,
+    create_symlinks,
+    find_av_libs_dir,
+    find_patchelf,
+    find_torchcodec_libs,
+    get_library_mappings,
+    is_rpath_patched,
+    setup_with_patchelf,
+    verify_torchcodec,
+)
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        prog="patch-torchcodec",
+        description="Patch TorchCodec to use PyAV's bundled FFmpeg libraries via RPATH.",
+        epilog="""
+Examples:
+  patch-torchcodec               # Patch RPATH (recommended, one-time)
+  patch-torchcodec --env-only    # Symlinks only (requires LD_LIBRARY_PATH)
+  patch-torchcodec --verify      # Verify TorchCodec works
+  patch-torchcodec --status      # Check current setup status
+        """,
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+    parser.add_argument(
+        "--env-only",
+        action="store_true",
+        help="Only create symlinks, don't patch RPATH (requires LD_LIBRARY_PATH)",
+    )
+    parser.add_argument(
+        "--print-env",
+        action="store_true",
+        help="Print shell command to set LD_LIBRARY_PATH",
+    )
+    parser.add_argument(
+        "--create-activate",
+        action="store_true",
+        help="Create an activation script (activate_torchcodec.sh)",
+    )
+    parser.add_argument(
+        "--verify", "--verify-only",
+        action="store_true",
+        dest="verify_only",
+        help="Only verify TorchCodec works, don't make changes",
+    )
+    parser.add_argument(
+        "--status",
+        action="store_true",
+        help="Show current setup status",
+    )
+    parser.add_argument(
+        "--quiet", "-q",
+        action="store_true",
+        help="Suppress verbose output",
+    )
+    args = parser.parse_args()
+    verbose = not args.quiet
+
+    # Find PyAV libraries
+    libs_dir = find_av_libs_dir()
+    if libs_dir is None:
+        print("Error: Could not find PyAV's av.libs directory.", file=sys.stderr)
+        print("Make sure PyAV is installed: pip install av", file=sys.stderr)
+        sys.exit(1)
+
+    # Status mode
+    if args.status:
+        print("patch-torchcodec: Status")
+        print("=" * 40)
+        print(f"PyAV av.libs: {libs_dir}")
+
+        torchcodec_libs = find_torchcodec_libs()
+        if torchcodec_libs:
+            print(f"TorchCodec libraries: {len(torchcodec_libs)} found")
+        else:
+            print("TorchCodec: NOT INSTALLED")
+
+        patchelf = find_patchelf()
+        print(f"patchelf: {'found at ' + str(patchelf) if patchelf else 'NOT FOUND'}")
+
+        rpath_patched = is_rpath_patched()
+        print(f"RPATH patched: {'YES' if rpath_patched else 'NO'}")
+
+        print()
+        if verify_torchcodec(libs_dir, require_env=False):
+            print("✓ TorchCodec works WITHOUT LD_LIBRARY_PATH")
+        elif verify_torchcodec(libs_dir, require_env=True):
+            print("⚠ TorchCodec works only WITH LD_LIBRARY_PATH")
+        else:
+            print("✗ TorchCodec is NOT working")
+        sys.exit(0)
+
+    # Verify only mode
+    if args.verify_only:
+        if verbose:
+            print("Verifying TorchCodec...")
+
+        # Try without LD_LIBRARY_PATH first
+        if verify_torchcodec(libs_dir, require_env=False):
+            if verbose:
+                print("✓ TorchCodec works without LD_LIBRARY_PATH (RPATH patched)")
+            sys.exit(0)
+        elif verify_torchcodec(libs_dir, require_env=True):
+            if verbose:
+                print("✓ TorchCodec works with LD_LIBRARY_PATH")
+                print(f"  Run: export LD_LIBRARY_PATH=\"{libs_dir}:$LD_LIBRARY_PATH\"")
+            sys.exit(0)
+        else:
+            if verbose:
+                print("✗ TorchCodec verification failed.")
+            sys.exit(1)
+
+    if verbose:
+        print("patch-torchcodec: Patching")
+        print("=" * 40)
+        print(f"PyAV av.libs: {libs_dir}")
+
+    # Decide setup method
+    if args.env_only:
+        # LD_LIBRARY_PATH only mode
+        if verbose:
+            print("\nSetup mode: LD_LIBRARY_PATH (symlinks only)")
+            print("\nCreating symbolic links...")
+
+        mappings = get_library_mappings(libs_dir)
+        if not mappings:
+            print("Error: No FFmpeg libraries found in PyAV bundle.", file=sys.stderr)
+            sys.exit(1)
+
+        created = create_symlinks(libs_dir, mappings)
+        # Also create symlinks for all other libraries (libdrm, etc.)
+        all_created = create_all_symlinks(libs_dir)
+
+        if verbose:
+            print(f"  Created {len(created) + len(all_created)} symbolic links")
+            print(f"\n✓ Symlinks created")
+            print(f"\nTo use TorchCodec, set LD_LIBRARY_PATH:")
+            print(f'  export LD_LIBRARY_PATH="{libs_dir}:$LD_LIBRARY_PATH"')
+
+        if args.print_env:
+            print(f'\nexport LD_LIBRARY_PATH="{libs_dir}:$LD_LIBRARY_PATH"')
+    else:
+        # RPATH patching mode (default)
+        if verbose:
+            print("\nSetup mode: RPATH patching (recommended)")
+
+        patchelf = find_patchelf()
+        if patchelf is None:
+            print("\nError: patchelf not found.", file=sys.stderr)
+            print("Install with: pip install patchelf", file=sys.stderr)
+            print("\nAlternatively, use --env-only to setup with LD_LIBRARY_PATH.", file=sys.stderr)
+            sys.exit(1)
+
+        torchcodec_libs = find_torchcodec_libs()
+        if not torchcodec_libs:
+            print("\nError: TorchCodec not found.", file=sys.stderr)
+            print("Install with: pip install torchcodec", file=sys.stderr)
+            sys.exit(1)
+
+        if verbose:
+            print(f"  patchelf: {patchelf}")
+            print(f"  TorchCodec libraries: {len(torchcodec_libs)}")
+
+        success = setup_with_patchelf(verbose=verbose)
+
+        if not success:
+            print("\nError: RPATH patching failed.", file=sys.stderr)
+            sys.exit(1)
+
+        if verbose:
+            print("\n✓ RPATH patching complete!")
+            print("  TorchCodec will now automatically find FFmpeg libraries.")
+            print("  No LD_LIBRARY_PATH needed!")
+
+    # Create activation script if requested
+    if args.create_activate:
+        script_path = Path("activate_torchcodec.sh")
+        script_content = f"""#!/bin/bash
+# Activation script for TorchCodec FFmpeg libraries
+# Source this file: source {script_path}
+
+export LD_LIBRARY_PATH="{libs_dir}:$LD_LIBRARY_PATH"
+echo "TorchCodec FFmpeg libraries activated"
+"""
+        script_path.write_text(script_content)
+        script_path.chmod(0o755)
+        if verbose:
+            print(f"\n✓ Created activation script: {script_path}")
+            print(f"  Usage: source {script_path}")
+
+    # Verify
+    if verbose:
+        print("\nVerifying TorchCodec...")
+
+    require_env = args.env_only
+    if verify_torchcodec(libs_dir, require_env=require_env):
+        if verbose:
+            if require_env:
+                print("✓ TorchCodec works (with LD_LIBRARY_PATH)")
+            else:
+                print("✓ TorchCodec works (without LD_LIBRARY_PATH)")
+    else:
+        if verbose:
+            print("⚠ TorchCodec verification failed.")
+            if require_env:
+                print("  Make sure to set LD_LIBRARY_PATH before running your code.")
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()
+

patch_torchcodec-0.1.0/core.py

@@ -0,0 +1,361 @@
+"""Core functionality for setting up FFmpeg libraries for TorchCodec."""
+
+from __future__ import annotations
+
+import os
+import re
+import shutil
+import subprocess
+import sys
+from pathlib import Path
+
+
+def find_av_libs_dir() -> Path | None:
+    """Find PyAV's bundled FFmpeg libraries directory.
+    
+    Returns:
+        Path to av.libs directory, or None if not found.
+    """
+    try:
+        import av
+        av_path = Path(av.__file__).parent
+        libs_dir = av_path.parent / "av.libs"
+        if libs_dir.exists():
+            return libs_dir
+    except ImportError:
+        pass
+    return None
+
+
+def get_library_mappings(libs_dir: Path) -> dict[str, Path]:
+    """Get mapping from standard library names to actual PyAV library files.
+    
+    Args:
+        libs_dir: Path to PyAV's av.libs directory.
+    
+    Returns:
+        Dict mapping standard names (e.g., 'libavcodec.so.62') to actual file paths.
+    """
+    required_libs = [
+        "libavcodec",
+        "libavformat", 
+        "libavutil",
+        "libswscale",
+        "libswresample",
+        "libavdevice",
+        "libavfilter",
+    ]
+    
+    mappings = {}
+    
+    for lib_name in required_libs:
+        # Find the actual library file (e.g., libavcodec-e57b519c.so.62.11.100)
+        pattern = re.compile(rf"^{lib_name}-[a-f0-9]+\.so\.(\d+)\.\d+\.\d+$")
+        
+        for file in libs_dir.iterdir():
+            match = pattern.match(file.name)
+            if match:
+                major_version = match.group(1)
+                standard_name = f"{lib_name}.so.{major_version}"
+                mappings[standard_name] = file
+                break
+    
+    return mappings
+
+
+def create_symlinks(libs_dir: Path, mappings: dict[str, Path] | None = None) -> list[Path]:
+    """Create symbolic links for FFmpeg libraries.
+    
+    Args:
+        libs_dir: Path to PyAV's av.libs directory.
+        mappings: Optional pre-computed library mappings.
+    
+    Returns:
+        List of created symlink paths.
+    """
+    if mappings is None:
+        mappings = get_library_mappings(libs_dir)
+    
+    created = []
+    
+    for standard_name, actual_file in mappings.items():
+        symlink_path = libs_dir / standard_name
+        
+        # Remove existing symlink if present
+        if symlink_path.is_symlink():
+            symlink_path.unlink()
+        elif symlink_path.exists():
+            continue
+        
+        # Create relative symlink
+        symlink_path.symlink_to(actual_file.name)
+        created.append(symlink_path)
+    
+    return created
+
+
+def get_ld_library_path() -> str | None:
+    """Get the LD_LIBRARY_PATH value needed for TorchCodec.
+    
+    Returns:
+        Path string to add to LD_LIBRARY_PATH, or None if PyAV not found.
+    """
+    libs_dir = find_av_libs_dir()
+    if libs_dir is None:
+        return None
+    return str(libs_dir)
+
+
+def setup(verbose: bool = False) -> bool:
+    """Setup FFmpeg libraries for TorchCodec.
+    
+    Creates symbolic links from PyAV's bundled FFmpeg libraries to standard
+    library names that TorchCodec expects.
+    
+    Args:
+        verbose: If True, print progress messages.
+    
+    Returns:
+        True if setup was successful, False otherwise.
+    """
+    libs_dir = find_av_libs_dir()
+    if libs_dir is None:
+        if verbose:
+            print("Error: PyAV not found. Install with: pip install av", file=sys.stderr)
+        return False
+    
+    mappings = get_library_mappings(libs_dir)
+    if not mappings:
+        if verbose:
+            print("Error: No FFmpeg libraries found in PyAV bundle.", file=sys.stderr)
+        return False
+    
+    created = create_symlinks(libs_dir, mappings)
+    
+    if verbose:
+        print(f"Created {len(created)} symbolic links in {libs_dir}")
+        print(f"\nTo use TorchCodec, set LD_LIBRARY_PATH:")
+        print(f'  export LD_LIBRARY_PATH="{libs_dir}:$LD_LIBRARY_PATH"')
+    
+    return True
+
+
+def find_torchcodec_libs() -> list[Path]:
+    """Find TorchCodec shared libraries.
+
+    Returns:
+        List of paths to TorchCodec .so files.
+    """
+    try:
+        import torchcodec
+        torchcodec_dir = Path(torchcodec.__file__).parent
+        return list(torchcodec_dir.glob("libtorchcodec_*.so"))
+    except ImportError:
+        return []
+
+
+def find_patchelf() -> Path | None:
+    """Find patchelf executable.
+
+    Returns:
+        Path to patchelf executable, or None if not found.
+    """
+    # Check in current Python environment first
+    try:
+        import site
+        for sp in site.getsitepackages():
+            patchelf = Path(sp).parent.parent / "bin" / "patchelf"
+            if patchelf.exists():
+                return patchelf
+    except Exception:
+        pass
+
+    # Check system PATH
+    patchelf = shutil.which("patchelf")
+    if patchelf:
+        return Path(patchelf)
+
+    return None
+
+
+def patch_rpath(library_path: Path, rpath: str, patchelf: Path | None = None) -> bool:
+    """Set RPATH on a shared library using patchelf.
+
+    Args:
+        library_path: Path to the .so file to patch.
+        rpath: RPATH value to set.
+        patchelf: Path to patchelf executable. Auto-detected if None.
+
+    Returns:
+        True if successful, False otherwise.
+    """
+    if patchelf is None:
+        patchelf = find_patchelf()
+    if patchelf is None:
+        return False
+
+    try:
+        subprocess.run(
+            [str(patchelf), "--force-rpath", "--set-rpath", rpath, str(library_path)],
+            check=True,
+            capture_output=True,
+        )
+        return True
+    except (subprocess.CalledProcessError, Exception):
+        return False
+
+
+def setup_with_patchelf(verbose: bool = False) -> bool:
+    """Setup TorchCodec to use PyAV's FFmpeg libraries via RPATH patching.
+
+    This method patches TorchCodec's shared libraries to include PyAV's av.libs
+    directory in their RPATH, eliminating the need for LD_LIBRARY_PATH.
+
+    Requires: patchelf (install with: pip install patchelf)
+
+    Args:
+        verbose: If True, print progress messages.
+
+    Returns:
+        True if setup was successful, False otherwise.
+    """
+    # Find dependencies
+    libs_dir = find_av_libs_dir()
+    if libs_dir is None:
+        if verbose:
+            print("Error: PyAV not found. Install with: pip install av", file=sys.stderr)
+        return False
+
+    torchcodec_libs = find_torchcodec_libs()
+    if not torchcodec_libs:
+        if verbose:
+            print("Error: TorchCodec not found. Install with: pip install torchcodec", file=sys.stderr)
+        return False
+
+    patchelf = find_patchelf()
+    if patchelf is None:
+        if verbose:
+            print("Error: patchelf not found. Install with: pip install patchelf", file=sys.stderr)
+        return False
+
+    # Create symlinks for all libraries in av.libs
+    if verbose:
+        print(f"Creating symbolic links in {libs_dir}...")
+
+    created_symlinks = create_all_symlinks(libs_dir)
+    if verbose:
+        print(f"  Created {len(created_symlinks)} symbolic links")
+
+    # Patch TorchCodec libraries
+    if verbose:
+        print(f"\nPatching TorchCodec libraries with RPATH: {libs_dir}")
+
+    rpath = str(libs_dir.resolve())
+    patched = 0
+    for lib in torchcodec_libs:
+        if patch_rpath(lib, rpath, patchelf):
+            patched += 1
+            if verbose:
+                print(f"  Patched: {lib.name}")
+        elif verbose:
+            print(f"  Failed: {lib.name}", file=sys.stderr)
+
+    if verbose:
+        print(f"\nPatched {patched}/{len(torchcodec_libs)} libraries")
+
+    return patched == len(torchcodec_libs)
+
+
+def create_all_symlinks(libs_dir: Path) -> list[Path]:
+    """Create symbolic links for ALL libraries in av.libs.
+
+    This creates symlinks from hash-included names to standard names,
+    e.g., libdrm-b0291a67.so.2.4.0 -> libdrm.so.2
+
+    Args:
+        libs_dir: Path to PyAV's av.libs directory.
+
+    Returns:
+        List of created symlink paths.
+    """
+    created = []
+
+    # Pattern: libname-hash.so.major.minor.patch
+    pattern = re.compile(r"^(.+)-[a-f0-9]+\.so\.(\d+)(?:\.\d+)*$")
+
+    for file in libs_dir.iterdir():
+        if file.is_symlink():
+            continue
+
+        match = pattern.match(file.name)
+        if match:
+            lib_name = match.group(1)
+            major_version = match.group(2)
+            standard_name = f"{lib_name}.so.{major_version}"
+            symlink_path = libs_dir / standard_name
+
+            if not symlink_path.exists():
+                symlink_path.symlink_to(file.name)
+                created.append(symlink_path)
+
+    return created
+
+
+def verify_torchcodec(libs_dir: Path | None = None, require_env: bool = True) -> bool:
+    """Verify TorchCodec can load with the configured libraries.
+
+    Args:
+        libs_dir: Path to av.libs directory. If None, will be auto-detected.
+        require_env: If True, set LD_LIBRARY_PATH for verification.
+                     If False, test without setting LD_LIBRARY_PATH (for RPATH-patched installs).
+
+    Returns:
+        True if TorchCodec loads successfully, False otherwise.
+    """
+    if libs_dir is None:
+        libs_dir = find_av_libs_dir()
+
+    env = os.environ.copy()
+    if require_env and libs_dir is not None:
+        env["LD_LIBRARY_PATH"] = f"{libs_dir}:{env.get('LD_LIBRARY_PATH', '')}"
+
+    try:
+        result = subprocess.run(
+            [sys.executable, "-c", "from torchcodec.decoders import VideoDecoder; print('OK')"],
+            env=env,
+            capture_output=True,
+            text=True,
+        )
+        return result.returncode == 0 and "OK" in result.stdout
+    except Exception:
+        return False
+
+
+def is_rpath_patched() -> bool:
+    """Check if TorchCodec libraries have been RPATH-patched.
+
+    Returns:
+        True if RPATH is set to av.libs directory.
+    """
+    libs_dir = find_av_libs_dir()
+    if libs_dir is None:
+        return False
+
+    torchcodec_libs = find_torchcodec_libs()
+    if not torchcodec_libs:
+        return False
+
+    # Check RPATH of one library
+    try:
+        result = subprocess.run(
+            ["readelf", "-d", str(torchcodec_libs[0])],
+            capture_output=True,
+            text=True,
+        )
+        if result.returncode == 0:
+            return str(libs_dir) in result.stdout
+    except Exception:
+        pass
+
+    return False
+

patch_torchcodec-0.1.0/pyproject.toml

@@ -0,0 +1,42 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "patch-torchcodec"
+version = "0.1.0"
+description = "Patch TorchCodec to use PyAV's bundled FFmpeg — one command, no system FFmpeg needed"
+readme = "README.md"
+license = "MIT"
+requires-python = ">=3.9"
+authors = [
+    { name = "MediaRef Contributors" }
+]
+keywords = ["torchcodec", "ffmpeg", "pyav", "video", "decoder", "patchelf"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: POSIX :: Linux",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Topic :: Multimedia :: Video",
+]
+dependencies = [
+    "av>=10.0.0",
+    "patchelf",
+]
+
+[project.scripts]
+patch-torchcodec = "patch_torchcodec.__main__:main"
+
+[project.urls]
+Homepage = "https://github.com/open-world-agents/MediaRef"
+Repository = "https://github.com/open-world-agents/MediaRef"
+
+[tool.hatch.build.targets.wheel]
+packages = ["patch_torchcodec"]
+