j2k-cuda 0.8.0

CUDA adapter for resident HTJ2K decode/encode and shared JPEG 2000 stages
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

// SPDX-License-Identifier: MIT OR Apache-2.0

mod facade;
mod htj2k;
mod packetization;
mod resident;
mod resident_session;
mod resident_tiles;
mod routing;
mod transforms;

#[cfg(feature = "cuda-runtime")]
use super::cuda_htj2k_encode_tables;
use super::packetization::{
    cuda_ht_segment_lengths, flatten_cuda_htj2k_packetization_job, CudaHtj2kPacketizationPlanError,
    CudaHtj2kPacketizationPlanTagNodeState,
};
#[cfg(feature = "cuda-runtime")]
use super::stage::cuda_dwt53_output_to_j2k;
use super::stage::cuda_packetization_plan_fallback_reason;
#[cfg(not(feature = "cuda-runtime"))]
use super::CudaEncodeFallbackReason;
#[cfg(feature = "cuda-runtime")]
use super::{
    cuda_resident_input_error, encode_lossless_from_cuda_buffer_to_cuda_buffer_with_report,
    encode_lossless_from_cuda_buffers_to_cuda_buffers_with_report, CudaLosslessEncodeTile,
};
use super::{
    encode_j2k_lossless_with_cuda, encode_j2k_lossless_with_cuda_and_profile,
    CudaEncodeStageAccelerator, CudaLosslessEncoder,
};
#[cfg(feature = "cuda-runtime")]
use crate::CudaSession;
#[cfg(feature = "cuda-runtime")]
use j2k::{encode_j2k_lossy_with_accelerator, J2kLossyEncodeOptions, J2kLossySamples};
use j2k::{
    EncodeBackendPreference, J2kBlockCodingMode, J2kEncodeValidation, J2kLosslessEncodeOptions,
    J2kLosslessSamples,
};
#[cfg(feature = "cuda-runtime")]
use j2k::{J2kDeinterleaveToF32Job, J2kHtCodeBlockEncodeJob, J2kResidentEncodeInputError};
use j2k::{
    J2kEncodeStageAccelerator, J2kEncodeStageError, J2kHtSubbandEncodeJob,
    J2kPacketizationBlockCodingMode, J2kPacketizationCodeBlock, J2kPacketizationEncodeJob,
    J2kPacketizationPacketDescriptor, J2kPacketizationProgressionOrder, J2kPacketizationResolution,
    J2kPacketizationSubband, J2kQuantizeSubbandJob,
};
#[cfg(feature = "cuda-runtime")]
use j2k_core::PixelFormat;
use j2k_core::{BackendKind, CodecError};
#[cfg(feature = "cuda-runtime")]
use j2k_cuda_runtime::{
    CudaContext, CudaHtj2kEncodeCodeBlockJob, CudaHtj2kEncodeCodeBlockRegionJob, CudaJ2kQuantizeJob,
};
#[cfg(feature = "cuda-runtime")]
use j2k_native::forward_dwt53_reference;
use j2k_native::{
    encode_with_accelerator as encode_with_native_accelerator, DecodeSettings, EncodeOptions,
    EncodeResult, Image,
};

fn assert_strict_cuda_classic_tier1_error<E: CodecError + ?Sized>(err: &E, context: &str) {
    assert!(err.is_unsupported());
    let message = err.to_string();
    assert!(
        message.contains("tier1_code_block") || message.contains("deinterleave"),
        "expected {context} error to mention either the missing classic tier-1 stage or unavailable CUDA deinterleave, got {message}"
	        );
}

#[cfg(feature = "cuda-runtime")]
fn strict_cuda_resident_lossless_options() -> J2kLosslessEncodeOptions {
    J2kLosslessEncodeOptions::default()
        .with_backend(EncodeBackendPreference::RequireDevice)
        .with_block_coding_mode(J2kBlockCodingMode::HighThroughput)
        .with_max_decomposition_levels(Some(0))
        .with_validation(J2kEncodeValidation::External)
}

struct CudaTestEncodeRequest<'a> {
    pixels: &'a [u8],
    width: u32,
    height: u32,
    components: u8,
    bit_depth: u8,
    signed: bool,
    options: &'a EncodeOptions,
    accelerator: &'a mut CudaEncodeStageAccelerator,
}

fn encode_with_cuda_test_accelerator(request: CudaTestEncodeRequest<'_>) -> EncodeResult<Vec<u8>> {
    let CudaTestEncodeRequest {
        pixels,
        width,
        height,
        components,
        bit_depth,
        signed,
        options,
        accelerator,
    } = request;
    encode_with_native_accelerator(
        pixels,
        width,
        height,
        u16::from(components),
        bit_depth,
        signed,
        options,
        accelerator,
    )
}