Files
punktfunk/crates/pyrowave-sys/vendor/pyrowave/pyrowave_encoder.hpp
T
enricobuehler 4c3b11445c feat(host): vendor PyroWave + minimal Granite subset as crates/pyrowave-sys
Phase 0 of design/pyrowave-codec-plan.md — the opt-in wired-LAN ultra-low-
latency codec. Vendored at upstream 509e4f88 (API 0.4.0, Granite 44362775,
volk + vulkan-headers pins in PUNKTFUNK-VENDOR.txt), pruned to the 6.6 MB
the standalone no-renderer build needs; scripts/vendor-pyrowave.sh
reproduces the tree (a pin bump is protocol-affecting, plan §4.2).

build.rs drives the wrapper CMakeLists (static archives incl. a static
C-API lib upstream only ships shared) + bindgen over pyrowave.h; Linux and
Windows only, empty stub elsewhere (Apple gets a native Metal port, §4.7).
Offline-safe by construction: no network, no system lib, vendored Vulkan
headers — same model as the opus dep (flatpak builder has no network).

Phase-0 validation on .21 (RTX 5070 Ti, driver 610.43.03):
- upstream pyrowave-c-test + interop test (incl. dmabuf/DRM-modifier
  Vulkan<->Vulkan) pass, from the pristine AND the pruned tree
- GPU kernel times at ~1.6 bpp noise: encode/decode 0.090/0.042 ms @800p,
  0.146/0.067 @1080p, 0.226/0.103 @1440p, 0.477/0.201 @4K — order of
  magnitude under NVENC's 1-2 ms retrieve, CBR lands within ~100 B of
  target
- cargo test -p pyrowave-sys green (static link + API-version pin check)

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-07-15 00:35:10 +02:00

64 lines
1.3 KiB
C++

// Copyright (c) 2025 Hans-Kristian Arntzen
// SPDX-License-Identifier: MIT
#pragma once
#include <memory>
#include <stddef.h>
#include <stdint.h>
#include "pyrowave_config.hpp"
namespace Vulkan
{
class Device;
class Buffer;
class ImageView;
class CommandBuffer;
}
namespace PyroWave
{
class Encoder
{
public:
Encoder();
~Encoder();
struct BitstreamBuffers
{
struct
{
const Vulkan::Buffer *buffer;
uint64_t offset;
uint64_t size;
} meta, bitstream;
size_t target_size;
};
bool init(Vulkan::Device *device, int width, int height, ChromaSubsampling chroma);
bool encode(Vulkan::CommandBuffer &cmd, const ViewBuffers &views, const BitstreamBuffers &buffers);
// Debug hackery
const Vulkan::ImageView &get_wavelet_band(int component, int level);
bool encode_pre_transformed(Vulkan::CommandBuffer &cmd, const BitstreamBuffers &buffers, float quant_scale);
//
uint64_t get_meta_required_size() const;
struct Packet
{
size_t offset;
size_t size;
};
size_t compute_num_packets(const void *mapped_meta, size_t packet_boundary) const;
size_t packetize(Packet *packets, size_t packet_boundary,
void *bitstream, size_t size,
const void *mapped_meta, const void *mapped_bitstream) const;
void report_stats(const void *mapped_meta, const void *mapped_bitstream) const;
private:
struct Impl;
std::unique_ptr<Impl> impl;
};
}