longpipe 0.0.14 → 0.0.16

package/dist/model.d.cts

@@ -0,0 +1,155 @@
+import { T as Tensor, M as MLBuffer, b as DataView_, B as Backend, D as Dtype, R as RenderTarget, c as ModelWeights, O as Op, U as UNetWrapperWeights, I as InputOp, d as ImageSource } from './index-BctoEDhZ.cjs';
+export { A as Activation, e as Conv2DWeights, C as Conv2dParams, f as DecoderBlockWeights, a as DepthwiseParams, g as DepthwiseWeights, F as FaceTouchupStyle, h as FaceWeights, i as FlowWeights, L as LandmarkWeights, j as MBConvWeights, P as Presenter, k as UpsampleParams, W as WebGPUBackend } from './index-BctoEDhZ.cjs';
+
+interface WebGLTensor extends Tensor {
+    readonly texture: WebGLTexture;
+    readonly texW: number;
+    readonly texH: number;
+}
+interface WebGLMLBuffer extends MLBuffer {
+    readonly data: DataView_;
+}
+
+interface WebGLBackendOptions {
+    canvas: HTMLCanvasElement | OffscreenCanvas;
+    dtype?: Dtype;
+}
+interface TextureFormat {
+    internalFormat: GLenum;
+    format: GLenum;
+    type: GLenum;
+    bytesPerElement: 2 | 4;
+}
+declare class WebGLBackend implements Backend {
+    readonly gl: WebGL2RenderingContext;
+    readonly canvas: HTMLCanvasElement | OffscreenCanvas;
+    readonly dtype: Dtype;
+    readonly ops: Backend['ops'];
+    readonly presenters: Backend['presenters'];
+    readonly fbo: WebGLFramebuffer;
+    readonly textureFormat: TextureFormat;
+    private constructor();
+    static isAvailable(): boolean;
+    static create(opts: WebGLBackendOptions): WebGLBackend;
+    attachCanvas(name: RenderTarget): void;
+    bindDisplayFramebuffer(): void;
+    toTextureView(data: DataView_ | null): Float32Array | Uint16Array | null;
+    tensor(h: number, w: number, c: number, data?: DataView_): WebGLTensor;
+    upload(data: DataView_): WebGLMLBuffer;
+    readback(tensor: WebGLTensor): Promise<Float32Array>;
+    copyTensor(src: WebGLTensor, dst: WebGLTensor): void;
+    sync(): Promise<void>;
+    destroy(): void;
+}
+
+type UNetVariant = 'A' | 'B' | 'E' | 'D';
+interface UNetWrapperParams {
+    variant: UNetVariant;
+    cHigh: number;
+    cLow: number;
+    cUp: number;
+}
+
+interface BaseNetwork {
+    readonly featLowRes: Tensor;
+    readonly encoderTaps: Tensor[];
+    readonly halfTap?: Tensor;
+    run(): void;
+}
+type BaseNetworkCtor = new (backend: Backend, input: Tensor, w: ModelWeights) => BaseNetwork;
+declare class TierModel implements Op {
+    readonly inputs: Tensor[];
+    readonly output: Tensor;
+    readonly encoderTaps: Tensor[];
+    readonly halfTap?: Tensor;
+    private readonly steps;
+    constructor(backend: Backend, x_hr: Tensor, baseWeights: ModelWeights, wrapperWeights: UNetWrapperWeights, params: UNetWrapperParams, BaseCtor: BaseNetworkCtor);
+    run(): void;
+}
+
+interface Res {
+    w: number;
+    h: number;
+}
+interface TierConfig {
+    base: BaseNetworkCtor;
+    wrapper: UNetWrapperParams;
+    canvasRes: Res;
+    baseRes: Res;
+    flowFuseStem?: boolean;
+}
+declare const TIER_CONFIG: Record<string, TierConfig>;
+
+declare function loadWeightsFromBinary(buf: ArrayBuffer): ModelWeights;
+
+type UpscalerMode = 'bilinear' | 'bicubic';
+type BackgroundConfig = {
+    mode: 'solid';
+    color: [number, number, number];
+} | {
+    mode: 'image';
+    image: Tensor;
+} | {
+    mode: 'blur';
+    sigma: number;
+};
+type CompositeSpec = BackgroundConfig | {
+    mode: 'passthrough';
+} | {
+    mode: 'fg-passthrough';
+};
+interface NetworkLike {
+    readonly output: Tensor;
+    run(): void;
+}
+interface EffectStage {
+    readonly output: Tensor;
+    run(): void;
+}
+interface RenderOptions {
+    upscaler: UpscalerMode;
+    background: BackgroundConfig;
+}
+declare class RenderOp<N extends NetworkLike = NetworkLike> {
+    private readonly backend;
+    private readonly displayInput;
+    private readonly image;
+    private network;
+    private networkInput;
+    private upscaler;
+    private upscalerMode;
+    private bgConfig;
+    private effectChain;
+    private viewRect;
+    private viewFg;
+    private viewAlpha;
+    private displayDirty;
+    private compositors;
+    constructor(backend: Backend);
+    attachNetwork(network: N, networkInput: InputOp, options: RenderOptions): void;
+    hasNetwork(): boolean;
+    setSource(src: ImageSource): void;
+    setUpscaler(mode: UpscalerMode): void;
+    setBackground(config: BackgroundConfig): void;
+    get displayImage(): Tensor;
+    private fgImage;
+    private chainTail;
+    private alphaImage;
+    setViewTransform(rect: Tensor | null): void;
+    private rebuildView;
+    setEffectChain(stages: EffectStage[]): void;
+    runEffects(): void;
+    run(): void;
+    runDisplay(): void;
+    runFgPassthrough(): void;
+    runPassthrough(): void;
+    compositeMain(mode: 'effect' | 'fg' | 'raw'): void;
+    refreshDisplayInput(): void;
+    compositeTo(target: RenderTarget, spec: CompositeSpec): void;
+    applyAlpha(alpha: Tensor): void;
+    runModel(): void;
+    private makeUpscaler;
+    private buildCompositor;
+}
+
+export { Backend, type BackgroundConfig, type BaseNetwork, type BaseNetworkCtor, type CompositeSpec, Dtype, ImageSource, InputOp, ModelWeights, Op, RenderOp, type RenderOptions, RenderTarget, type Res, TIER_CONFIG, Tensor, type TierConfig, TierModel, UNetWrapperWeights, type UpscalerMode, WebGLBackend, loadWeightsFromBinary };

package/dist/model.d.ts

@@ -0,0 +1,155 @@
+import { T as Tensor, M as MLBuffer, b as DataView_, B as Backend, D as Dtype, R as RenderTarget, c as ModelWeights, O as Op, U as UNetWrapperWeights, I as InputOp, d as ImageSource } from './index-BctoEDhZ.js';
+export { A as Activation, e as Conv2DWeights, C as Conv2dParams, f as DecoderBlockWeights, a as DepthwiseParams, g as DepthwiseWeights, F as FaceTouchupStyle, h as FaceWeights, i as FlowWeights, L as LandmarkWeights, j as MBConvWeights, P as Presenter, k as UpsampleParams, W as WebGPUBackend } from './index-BctoEDhZ.js';
+
+interface WebGLTensor extends Tensor {
+    readonly texture: WebGLTexture;
+    readonly texW: number;
+    readonly texH: number;
+}
+interface WebGLMLBuffer extends MLBuffer {
+    readonly data: DataView_;
+}
+
+interface WebGLBackendOptions {
+    canvas: HTMLCanvasElement | OffscreenCanvas;
+    dtype?: Dtype;
+}
+interface TextureFormat {
+    internalFormat: GLenum;
+    format: GLenum;
+    type: GLenum;
+    bytesPerElement: 2 | 4;
+}
+declare class WebGLBackend implements Backend {
+    readonly gl: WebGL2RenderingContext;
+    readonly canvas: HTMLCanvasElement | OffscreenCanvas;
+    readonly dtype: Dtype;
+    readonly ops: Backend['ops'];
+    readonly presenters: Backend['presenters'];
+    readonly fbo: WebGLFramebuffer;
+    readonly textureFormat: TextureFormat;
+    private constructor();
+    static isAvailable(): boolean;
+    static create(opts: WebGLBackendOptions): WebGLBackend;
+    attachCanvas(name: RenderTarget): void;
+    bindDisplayFramebuffer(): void;
+    toTextureView(data: DataView_ | null): Float32Array | Uint16Array | null;
+    tensor(h: number, w: number, c: number, data?: DataView_): WebGLTensor;
+    upload(data: DataView_): WebGLMLBuffer;
+    readback(tensor: WebGLTensor): Promise<Float32Array>;
+    copyTensor(src: WebGLTensor, dst: WebGLTensor): void;
+    sync(): Promise<void>;
+    destroy(): void;
+}
+
+type UNetVariant = 'A' | 'B' | 'E' | 'D';
+interface UNetWrapperParams {
+    variant: UNetVariant;
+    cHigh: number;
+    cLow: number;
+    cUp: number;
+}
+
+interface BaseNetwork {
+    readonly featLowRes: Tensor;
+    readonly encoderTaps: Tensor[];
+    readonly halfTap?: Tensor;
+    run(): void;
+}
+type BaseNetworkCtor = new (backend: Backend, input: Tensor, w: ModelWeights) => BaseNetwork;
+declare class TierModel implements Op {
+    readonly inputs: Tensor[];
+    readonly output: Tensor;
+    readonly encoderTaps: Tensor[];
+    readonly halfTap?: Tensor;
+    private readonly steps;
+    constructor(backend: Backend, x_hr: Tensor, baseWeights: ModelWeights, wrapperWeights: UNetWrapperWeights, params: UNetWrapperParams, BaseCtor: BaseNetworkCtor);
+    run(): void;
+}
+
+interface Res {
+    w: number;
+    h: number;
+}
+interface TierConfig {
+    base: BaseNetworkCtor;
+    wrapper: UNetWrapperParams;
+    canvasRes: Res;
+    baseRes: Res;
+    flowFuseStem?: boolean;
+}
+declare const TIER_CONFIG: Record<string, TierConfig>;
+
+declare function loadWeightsFromBinary(buf: ArrayBuffer): ModelWeights;
+
+type UpscalerMode = 'bilinear' | 'bicubic';
+type BackgroundConfig = {
+    mode: 'solid';
+    color: [number, number, number];
+} | {
+    mode: 'image';
+    image: Tensor;
+} | {
+    mode: 'blur';
+    sigma: number;
+};
+type CompositeSpec = BackgroundConfig | {
+    mode: 'passthrough';
+} | {
+    mode: 'fg-passthrough';
+};
+interface NetworkLike {
+    readonly output: Tensor;
+    run(): void;
+}
+interface EffectStage {
+    readonly output: Tensor;
+    run(): void;
+}
+interface RenderOptions {
+    upscaler: UpscalerMode;
+    background: BackgroundConfig;
+}
+declare class RenderOp<N extends NetworkLike = NetworkLike> {
+    private readonly backend;
+    private readonly displayInput;
+    private readonly image;
+    private network;
+    private networkInput;
+    private upscaler;
+    private upscalerMode;
+    private bgConfig;
+    private effectChain;
+    private viewRect;
+    private viewFg;
+    private viewAlpha;
+    private displayDirty;
+    private compositors;
+    constructor(backend: Backend);
+    attachNetwork(network: N, networkInput: InputOp, options: RenderOptions): void;
+    hasNetwork(): boolean;
+    setSource(src: ImageSource): void;
+    setUpscaler(mode: UpscalerMode): void;
+    setBackground(config: BackgroundConfig): void;
+    get displayImage(): Tensor;
+    private fgImage;
+    private chainTail;
+    private alphaImage;
+    setViewTransform(rect: Tensor | null): void;
+    private rebuildView;
+    setEffectChain(stages: EffectStage[]): void;
+    runEffects(): void;
+    run(): void;
+    runDisplay(): void;
+    runFgPassthrough(): void;
+    runPassthrough(): void;
+    compositeMain(mode: 'effect' | 'fg' | 'raw'): void;
+    refreshDisplayInput(): void;
+    compositeTo(target: RenderTarget, spec: CompositeSpec): void;
+    applyAlpha(alpha: Tensor): void;
+    runModel(): void;
+    private makeUpscaler;
+    private buildCompositor;
+}
+
+export { Backend, type BackgroundConfig, type BaseNetwork, type BaseNetworkCtor, type CompositeSpec, Dtype, ImageSource, InputOp, ModelWeights, Op, RenderOp, type RenderOptions, RenderTarget, type Res, TIER_CONFIG, Tensor, type TierConfig, TierModel, UNetWrapperWeights, type UpscalerMode, WebGLBackend, loadWeightsFromBinary };