prpy 0.2.2__tar.gz

prpy-0.2.2/.github/workflows/main.yml

@@ -0,0 +1,40 @@
+name: Tests
+
+on:
+  push:
+    branches: ["main"]
+  pull_request:
+    branches: ["main"]
+
+jobs:
+  build:
+
+    runs-on: [ubuntu-latest]
+    strategy:
+      matrix:
+        python-version: ["3.8", "3.9", "3.10"]
+
+    steps:
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install ffmpeg
+        uses: Iamshankhadeep/setup-ffmpeg@v1.2
+        with:
+          version: "4.4"
+          token: ${{ secrets.GITHUB_TOKEN }}
+      - name: Checkout prpy
+        uses: actions/checkout@v3
+        with:
+          token: ${{ secrets.GITHUB_TOKEN }}
+      - name: Install prpy and dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install ".[ffmpeg,numpy,tensorflow,torch,test]"
+      - name: Lint with flake8
+        run: |
+          flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+      - name: Test with pytest
+        run: |
+          pytest

prpy-0.2.2/.gitignore

@@ -0,0 +1,132 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+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/
+
+# 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
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.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
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.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/
+
+# OS X .DS_Store
+**/.DS_Store

prpy-0.2.2/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2024 Philipp Rouast
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

prpy-0.2.2/MANIFEST.in

@@ -0,0 +1,7 @@
+include README.md
+recursive-include prpy/numpy *
+recursive-include prpy/tensorflow *
+recursive-include prpy/torch *
+recursive-include prpy/ffmpeg *
+recursive-include tests *
+prune **/__pycache__ 

prpy-0.2.2/PKG-INFO

@@ -0,0 +1,75 @@
+Metadata-Version: 2.1
+Name: prpy
+Version: 0.2.2
+Summary: Collection of Python utils for signal, image, and video processing
+Author-email: Philipp Rouast <philipp@rouast.com>
+License: MIT License
+Project-URL: Repository, https://github.com/prouast/prpy.git
+Project-URL: Issues, https://github.com/prouast/prpy/issues
+Keywords: python,numpy,ffmpeg,tensorflow,torch
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Classifier: Development Status :: 4 - Beta
+Classifier: License :: OSI Approved :: MIT License
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Provides-Extra: ffmpeg
+Requires-Dist: ffmpeg-python; extra == "ffmpeg"
+Requires-Dist: numpy; extra == "ffmpeg"
+Provides-Extra: numpy
+Requires-Dist: numpy; extra == "numpy"
+Requires-Dist: scipy; extra == "numpy"
+Requires-Dist: Pillow; extra == "numpy"
+Requires-Dist: opencv-python; extra == "numpy"
+Requires-Dist: scikit-learn; extra == "numpy"
+Provides-Extra: numpy-min
+Requires-Dist: numpy; extra == "numpy-min"
+Requires-Dist: scipy; extra == "numpy-min"
+Requires-Dist: Pillow; extra == "numpy-min"
+Provides-Extra: tensorflow
+Requires-Dist: tensorflow==2.12.1; extra == "tensorflow"
+Requires-Dist: numpy; extra == "tensorflow"
+Provides-Extra: torch
+Requires-Dist: torch==2.1.0; extra == "torch"
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: flake8; extra == "test"
+
+[![Tests](https://github.com/prouast/prpy/actions/workflows/main.yml/badge.svg)](https://github.com/prouast/prpy/actions/workflows/main.yml)
+
+# prpy
+
+A collection of Python utilities for signal, image, and video processing.
+It contains subpackages for working with `numpy`, `ffmpeg`, `tensorflow`, and `torch`.
+
+## Installation
+
+General prerequisites are `python>=3.8` and `ffmpeg` installed and accessible via the `$PATH` environment variable.
+
+To install `prpy` and its dependencies:
+
+```
+git clone https://github.com/prouast/prpy.git
+pip install "./prpy[ffmpeg,numpy,tensorflow,torch,test]"
+```
+
+The above runs a full install of all dependencies.
+It is possible to customize the install of the dependencies by only listing the desired subpackages out of `ffmpeg`, `numpy`, `tensorflow`, `torch`, and `test` in the square brackets above.
+
+## Linting and tests
+
+To lint and run tests:
+
+```
+flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+pytest
+```
+
+## Build
+
+To build:
+
+```
+python -m build
+```

prpy-0.2.2/README.md

@@ -0,0 +1,37 @@
+[![Tests](https://github.com/prouast/prpy/actions/workflows/main.yml/badge.svg)](https://github.com/prouast/prpy/actions/workflows/main.yml)
+
+# prpy
+
+A collection of Python utilities for signal, image, and video processing.
+It contains subpackages for working with `numpy`, `ffmpeg`, `tensorflow`, and `torch`.
+
+## Installation
+
+General prerequisites are `python>=3.8` and `ffmpeg` installed and accessible via the `$PATH` environment variable.
+
+To install `prpy` and its dependencies:
+
+```
+git clone https://github.com/prouast/prpy.git
+pip install "./prpy[ffmpeg,numpy,tensorflow,torch,test]"
+```
+
+The above runs a full install of all dependencies.
+It is possible to customize the install of the dependencies by only listing the desired subpackages out of `ffmpeg`, `numpy`, `tensorflow`, `torch`, and `test` in the square brackets above.
+
+## Linting and tests
+
+To lint and run tests:
+
+```
+flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+pytest
+```
+
+## Build
+
+To build:
+
+```
+python -m build
+```

prpy-0.2.2/prpy/__init__.py

@@ -0,0 +1,19 @@
+# Copyright (c) 2024 Philipp Rouast
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.

prpy-0.2.2/prpy/constants.py

@@ -0,0 +1,24 @@
+# Copyright (c) 2024 Philipp Rouast
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+NANOS_PER_SECOND = 1000000000.0
+MICROS_PER_SECOND = 1000000.0
+MILLIS_PER_SECOND = 1000.0
+SECONDS_PER_MINUTE = 60.0

prpy-0.2.2/prpy/ffmpeg/__init__.py

@@ -0,0 +1,19 @@
+# Copyright (c) 2024 Philipp Rouast
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.

prpy-0.2.2/prpy/ffmpeg/probe.py

@@ -0,0 +1,86 @@
+# Copyright (c) 2024 Philipp Rouast
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+import ffmpeg
+from fractions import Fraction
+import logging
+from typing import Tuple
+import os
+
+def probe_video(
+    path: str
+  ) -> Tuple[float, int, int, int, str, float, int]:
+  """Probe a video file for metadata.
+
+  Args:
+    path: The path of the video.
+  Returns:
+    fps: The frame rate of the video
+    total_frames: The total number of frames
+    width: The width dimension of the video
+    height: The height dimension of the video
+    codec: The codec of the video
+    bitrate: The bitrate of the video
+    rotation: The rotation of the video
+  """
+  # Check if file exists
+  assert isinstance(path, str)
+  if not os.path.exists(path):
+    raise FileNotFoundError("File {} does not exist".format(path))
+  # ffprobe -show_streams -count_frames -pretty video.mp4
+  try:
+    probe = ffmpeg.probe(filename=path)
+  except Exception as e:
+    # The exception returned by `ffprobe` is in bytes
+    logging.warn("Exception probing video: {}".format(e))
+  else:
+    video_stream = next(
+      (
+        stream
+        for stream in probe["streams"]
+        if stream["codec_type"] == "video"
+      ),
+      None,
+    )
+    try:
+      fps = float(Fraction(video_stream["avg_frame_rate"]))
+    except Exception as e:
+      fps = 0
+    try:
+      total_frames = int(video_stream["nb_frames"])
+    except Exception as e:
+      duration = float(video_stream['duration'])
+      total_frames = int(duration*fps)
+    width = video_stream["width"]
+    height = video_stream["height"]
+    codec = video_stream["codec_name"]
+    try:
+      bitrate = float(video_stream["bit_rate"])/1000.0
+    except Exception as e:
+      bitrate = 0.0
+    rotation = 0
+    if 'tags' in video_stream and 'rotate' in video_stream['tags']:
+      # Regular
+      rotation = int(video_stream['tags']['rotate'])
+    elif 'side_data_list' in video_stream and 'rotation' in video_stream['side_data_list'][0]:
+      # iPhone
+      rotation = int(video_stream['side_data_list'][0]['rotation'])
+    return fps, total_frames, width, height, codec, bitrate, rotation
+