dfttest2 10.dev0__tar.gz

dfttest2-10.dev0/.gitignore

@@ -0,0 +1,289 @@
+# 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/
+# Temporary file for partial code execution
+tempCodeRunnerFile.py
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
+
+# Streamlit
+.streamlit/secrets.toml
+
+
+# Prerequisites
+*.d
+
+# Compiled Object files
+*.slo
+*.lo
+*.o
+*.obj
+
+# Precompiled Headers
+*.gch
+*.pch
+
+# Linker files
+*.ilk
+
+# Debugger Files
+*.pdb
+
+# Compiled Dynamic libraries
+*.so
+*.dylib
+*.dll
+*.so.*
+
+
+# Fortran module files
+*.mod
+*.smod
+
+# Compiled Static libraries
+*.lai
+*.la
+*.a
+*.lib
+
+# Executables
+*.exe
+*.out
+*.app
+
+# Build directories
+build/
+Build/
+build-*/
+
+# CMake generated files
+CMakeFiles/
+CMakeCache.txt
+cmake_install.cmake
+Makefile
+install_manifest.txt
+compile_commands.json
+
+# Temporary files
+*.tmp
+*.log
+*.bak
+*.swp
+
+# vcpkg
+vcpkg_installed/
+
+# debug information files
+*.dwo
+
+# test output & cache
+Testing/
+.cache/

dfttest2-10.dev0/PKG-INFO

@@ -0,0 +1,87 @@
+Metadata-Version: 2.4
+Name: dfttest2
+Version: 10.dev0
+Summary: DFTTest re-implemetation
+Project-URL: Source Code, https://github.com/AmusementClub/vs-dfttest2
+Project-URL: Bug Tracker, https://github.com/AmusementClub/vs-dfttest2/issues
+Project-URL: Repository, https://github.com/Jaded-Encoding-Thaumaturgy/vs-wheels
+Author: WolframRhodium
+Maintainer-email: Vardë <ichunjo.le.terrible@gmail.com>
+License-Expression: GPL-3.0-or-later
+License-File: vs-dfttest2/LICENSE
+Requires-Python: >=3.12
+Requires-Dist: vapoursynth>=75
+Provides-Extra: gcc
+Requires-Dist: vapoursynth-dfttest2-gcc; extra == 'gcc'
+Description-Content-Type: text/markdown
+
+# dfttest2
+
+A VapourSynth re-implementation of DFTTest.
+
+This package serves as a **meta-package** providing:
+
+1.  A high-level **Python interface** for `dfttest2`.
+2.  Optional **backend dependencies** via extras to simplify installation.
+
+## Backends
+
+### **GPU Backends**
+
+| Backend  | Platform                                | Implementation Type        |
+| -------- | --------------------------------------- | -------------------------- |
+| `nvrtc`  | Windows, Linux (x86-64, aarch64)        | JIT Compilation            |
+| `hiprtc` | Windows, Linux (x86-64)                 | JIT Compilation            |
+| `cufft`  | Windows, Linux (x86-64, aarch64)        | Library                    |
+| `hipfft` | Windows, Linux (x86-64)                 | Library                    |
+| `cpu`    | Windows, Linux (x86-64), macOS (x86-64) | VCL                        |
+| `gcc`    | All Platforms                           | Compiler Vector Extensions |
+
+### **Which one to use?**
+
+| Backend | Requirement                         |
+| ------- | ----------------------------------- |
+| NVRTC   | `sbsize=16` and CUDA-enabled system |
+| HIPRTC  | `sbsize=16` and HIP-enabled system  |
+| cuFFT   | CUDA-enabled system                 |
+| hipFFT  | HIP-enabled system                  |
+| CPU     | `sbsize=16`                         |
+| GCC     | `sbsize=16`                         |
+
+**Approximate Speed:** NVRTC == HIPRTC >> cuFFT > hipFFT > CPU == GCC
+
+## Installation
+
+For example, to install the NVRTC and CPU backends:
+
+```bash
+pip install dfttest2[nvrtc,cpu]
+```
+
+---
+
+README.md from parent project
+
+---
+# vs-dfttest2
+DFTTest re-implemetation (CUDA and x86)
+
+## Usage
+```python3
+from dfttest2 import DFTTest
+output = DFTTest(input)
+```
+
+See also [VapourSynth-DFTTest](https://github.com/HomeOfVapourSynthEvolution/VapourSynth-DFTTest)
+
+## Compilation
+```bash
+# additional options: -D ENABLE_CUDA=ON -D ENABLE_CPU=ON
+cmake -S . -B build
+
+cmake --build build
+
+cmake --install build
+```
+
+If the vapoursynth library cannot be found by pkg-config, then the cmake variable `VS_INCLUDE_DIR` should be set.

dfttest2-10.dev0/dfttest2/__init__.py

@@ -0,0 +1,17 @@
+from ._dfttest2 import (  # type: ignore
+    FREQ,
+    SIGMA,
+    Backend,
+    DFTTest,
+    DFTTest2,
+    flatten,
+    get_location,
+    get_window,
+    get_window_value,
+    init_backend,
+    normalize,
+    select_backend,
+    to_func,
+)
+
+__all__ = ["Backend", "DFTTest", "DFTTest2"]

dfttest2-10.dev0/pyproject.toml

@@ -0,0 +1,63 @@
+[build-system]
+requires = ["hatchling>=1.29.0", "hatch-vcs", "hatch-fancy-pypi-readme"]
+build-backend = "hatchling.build"
+
+[project]
+name = "dfttest2"
+dynamic = ["version", "readme"]
+description = "DFTTest re-implemetation"
+license = "GPL-3.0-or-later"
+license-files = ["vs-dfttest2/LICENSE"]
+requires-python = ">=3.12"
+authors = [{ name = "WolframRhodium" }]
+maintainers = [{ name = "Vardë", email = "ichunjo.le.terrible@gmail.com" }]
+dependencies = ["vapoursynth>=75"]
+
+[project.optional-dependencies]
+# cuda = ["vapoursynth-dfttest2_cuda", "vapoursynth-dfttest2_nvrtc"]
+# hip = ["vapoursynth-dfttest2_hip", "vapoursynth-dfttest2_hiprtc"]
+# cufft = ["vapoursynth-dfttest2_cuda"]
+# nvrtc = ["vapoursynth-dfttest2_nvrtc"]
+# hipfft = ["vapoursynth-dfttest2_hip"]
+# hiprtc = ["vapoursynth-dfttest2_hiprtc"]
+# cpu = ["vapoursynth-dfttest2_cpu"]
+gcc = ["vapoursynth-dfttest2-gcc"]
+
+# windows = ["dfttest2[cuda,hip,gcc,cpu]"]
+# linux = [
+#     "dfttest2[cuda,hip,gcc,cpu] ; platform_machine == 'x86_64'",
+#     "dfttest2[cuda,gcc] ; platform_machine == 'aarch64'",
+# ]
+# macos = [
+#     "dfttest2[cpu,gcc] ; platform_machine == 'x86_64'",
+#     "dfttest2[gcc] ; platform_machine == 'arm64'",
+# ]
+
+[project.urls]
+"Source Code" = "https://github.com/AmusementClub/vs-dfttest2"
+"Bug Tracker" = "https://github.com/AmusementClub/vs-dfttest2/issues"
+"Repository" = "https://github.com/Jaded-Encoding-Thaumaturgy/vs-wheels"
+
+[tool.hatch.version]
+source = "vcs"
+tag-pattern = '^v(?P<version>.*)$'
+
+[tool.hatch.version.raw-options]
+root = "vs-dfttest2"
+local_scheme = "no-local-version"
+
+[tool.hatch.build.targets.sdist]
+include = ["dfttest2", "vs-dfttest2"]
+
+[tool.hatch.build.targets.wheel]
+include = ["dfttest2"]
+
+[tool.hatch.build.targets.wheel.force-include]
+"vs-dfttest2/dfttest2.py" = "dfttest2/_dfttest2.py"
+
+[tool.hatch.metadata.hooks.fancy-pypi-readme]
+content-type = "text/markdown"
+fragments = [{ path = "README.md" }, { path = "vs-dfttest2/README.md" }]
+
+[tool.uv.sources]
+vapoursynth-dfttest2-gcc = { workspace = true }

dfttest2-10.dev0/vs-dfttest2/.git

@@ -0,0 +1 @@
+gitdir: ../../../.git/modules/dfttest2/dfttest2/vs-dfttest2

dfttest2-10.dev0/vs-dfttest2/.github/workflows/linux.yml

@@ -0,0 +1,86 @@
+name: Build (Linux)
+
+on:
+  push:
+    paths:
+      - 'CMakeLists.txt'
+      - 'cpu_source/*'
+      - 'cuda_source/*'
+      - 'nvrtc_source/*'
+      - '.github/workflows/linux.yml'
+  workflow_dispatch:
+
+jobs:
+  build-linux:
+    runs-on: ubuntu-24.04
+    steps:
+    - name: Checkout repo
+      uses: actions/checkout@v4
+      with:
+        submodules: true
+
+    - name: Setup Ninja
+      run: pip3 install ninja
+
+    - name: Setup CUDA
+      run: |
+        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
+        sudo dpkg -i cuda-keyring_1.1-1_all.deb
+        sudo apt-get update
+        sudo apt-get install -y cuda-nvcc-13-0 cuda-cudart-dev-13-0 cuda-nvrtc-dev-13-0 libcufft-dev-13-0
+        echo "PATH=/usr/local/cuda/bin${PATH:+:${PATH}}" >> $GITHUB_ENV
+        echo "CUDA_PATH=/usr/local/cuda" >> $GITHUB_ENV
+        echo "LD_LIBRARY_PATH=/usr/local/cuda/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}" >> $GITHUB_ENV
+
+    - name: Download VapourSynth headers
+      run: |
+        wget -q -O vs.zip https://github.com/vapoursynth/vapoursynth/archive/refs/tags/R57.zip
+        unzip -q vs.zip
+        mv vapoursynth*/ vapoursynth
+
+    - name: Configure
+      run: cmake -S . -B build -G Ninja
+        -D VS_INCLUDE_DIR="`pwd`/vapoursynth/include"
+        -D ENABLE_CUDA=ON
+        -D USE_NVRTC_STATIC=ON
+        -D ENABLE_CPU=ON
+        -D CMAKE_BUILD_TYPE=Release
+        -D CMAKE_CXX_COMPILER=g++
+        -D CMAKE_CXX_FLAGS="-Wall -ffast-math"
+
+    - name: Build
+      run: cmake --build build --config Release --verbose
+
+    - name: Install
+      run: cmake --install build --prefix artifact
+
+    - name: Setup HIP
+      run: |
+        wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | gpg --dearmor | sudo tee /etc/apt/keyrings/rocm.gpg > /dev/null
+        echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/7.0 noble main" | sudo tee --append /etc/apt/sources.list.d/rocm.list
+        echo -e 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600' | sudo tee /etc/apt/preferences.d/rocm-pin-600
+        sudo apt update
+        sudo apt install hip-runtime-amd rocm-device-libs hipfft-dev hip-dev
+
+    - name: Configure (HIP)
+      run: cmake -S . -B build_hip -G Ninja
+        -D VS_INCLUDE_DIR="`pwd`/vapoursynth/include"
+        -D ENABLE_CUDA=OFF
+        -D ENABLE_CPU=OFF
+        -D ENABLE_HIP=ON
+        -D CMAKE_BUILD_TYPE=Release
+        -D CMAKE_CXX_FLAGS="-Wall -ffast-math"
+        -D CMAKE_PREFIX_PATH=/opt/rocm
+        -D CMAKE_POLICY_VERSION_MINIMUM=3.5
+
+    - name: Build (HIP)
+      run: cmake --build build_hip --config Release --verbose
+
+    - name: Install (HIP)
+      run: cmake --install build_hip --prefix artifact
+
+    - name: Upload
+      uses: actions/upload-artifact@v4
+      with:
+        name: vs-dfttest2-Linux
+        path: artifact

dfttest2-10.dev0/vs-dfttest2/.github/workflows/linux_arm64.yml

@@ -0,0 +1,67 @@
+name: Build (Linux, ARM64)
+
+on:
+  push:
+    paths:
+      - 'CMakeLists.txt'
+      - 'gcc_source/*'
+      - 'cuda_source/*'
+      - 'nvrtc_source/*'
+      - '.github/workflows/linux_arm64.yml'
+  workflow_dispatch:
+
+jobs:
+  build-linux:
+    runs-on: ubuntu-24.04-arm
+    steps:
+    - name: Checkout repo
+      uses: actions/checkout@v4
+      with:
+        submodules: true
+
+    - name: Setup clang
+      run: |
+        wget https://apt.llvm.org/llvm.sh
+        chmod +x llvm.sh
+        sudo ./llvm.sh all
+
+    - name: Setup Ninja
+      run: pip3 install ninja
+
+    - name: Setup CUDA
+      run: |
+        wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/sbsa/cuda-keyring_1.1-1_all.deb
+        sudo dpkg -i cuda-keyring_1.1-1_all.deb
+        sudo apt-get update
+        sudo apt-get install -y cuda-nvcc-13-0 cuda-cudart-dev-13-0 cuda-nvrtc-dev-13-0 libcufft-dev-13-0
+        echo "PATH=/usr/local/cuda/bin${PATH:+:${PATH}}" >> $GITHUB_ENV
+        echo "CUDA_PATH=/usr/local/cuda" >> $GITHUB_ENV
+        echo "LD_LIBRARY_PATH=/usr/local/cuda/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}" >> $GITHUB_ENV
+
+    - name: Download VapourSynth headers
+      run: |
+        wget -q -O vs.zip https://github.com/vapoursynth/vapoursynth/archive/refs/tags/R57.zip
+        unzip -q vs.zip
+        mv vapoursynth*/ vapoursynth
+
+    - name: Configure
+      run: cmake -S . -B build -G Ninja
+        -D VS_INCLUDE_DIR="`pwd`/vapoursynth/include"
+        -D ENABLE_CUDA=ON
+        -D ENABLE_CPU=OFF
+        -D ENABLE_GCC=ON
+        -D CMAKE_BUILD_TYPE=Release
+        -D CMAKE_CXX_COMPILER=clang++
+        -D CMAKE_CXX_FLAGS="-Wall -ffast-math"
+
+    - name: Build
+      run: cmake --build build --config Release --verbose
+
+    - name: Install
+      run: cmake --install build --prefix artifact
+
+    - name: Upload
+      uses: actions/upload-artifact@v4
+      with:
+        name: vs-dfttest2-Linux
+        path: artifact