j2k-cuda-runtime 0.7.2

CUDA codec engine and Driver API runtime for j2k device adapters
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

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

use super::{
    validate_transcode_pool_context, validate_transcode_pool_context_match,
    TRANSCODE_POOL_CONTEXT_MISMATCH,
};
use crate::{
    CudaBufferPool, CudaContext, CudaDwt97BatchGeometry, CudaDwt97BatchWithPoolRequest, CudaError,
    CudaHtj2k97CodeblockBatchWithPoolRequest, CudaHtj2k97I16CodeblockBatchWithPoolRequest,
    CudaHtj2k97QuantizeParams,
};

fn assert_transcode_pool_context_mismatch<T>(result: Result<T, CudaError>) {
    match result {
        Err(CudaError::InvalidArgument { message }) => {
            assert_eq!(message, TRANSCODE_POOL_CONTEXT_MISMATCH);
        }
        Err(error) => panic!("expected a CUDA transcode pool context mismatch, got {error}"),
        Ok(_) => panic!("expected a CUDA transcode pool context mismatch"),
    }
}

fn empty_geometry() -> CudaDwt97BatchGeometry {
    CudaDwt97BatchGeometry {
        item_count: 0,
        block_cols: 0,
        block_rows: 0,
        width: 0,
        height: 0,
    }
}

fn quantize_params() -> CudaHtj2k97QuantizeParams {
    CudaHtj2k97QuantizeParams {
        inv_delta_ll: 1.0,
        inv_delta_hl: 1.0,
        inv_delta_lh: 1.0,
        inv_delta_hh: 1.0,
        cb_width: 1,
        cb_height: 1,
    }
}

fn f32_request(pool: &CudaBufferPool) -> CudaHtj2k97CodeblockBatchWithPoolRequest<'_> {
    CudaHtj2k97CodeblockBatchWithPoolRequest {
        blocks: &[],
        geometry: empty_geometry(),
        params: quantize_params(),
        pool,
    }
}

#[test]
fn transcode_pool_context_validation_accepts_match_and_rejects_mismatch() {
    assert!(validate_transcode_pool_context_match(true).is_ok());
    assert_transcode_pool_context_mismatch(validate_transcode_pool_context_match(false));
}

#[test]
fn every_public_transcode_pool_api_rejects_a_foreign_pool_first() {
    if !j2k_test_support::cuda_runtime_gate(module_path!()) {
        return;
    }

    let context = CudaContext::system_default().expect("launch CUDA context");
    let foreign_context = CudaContext::system_default().expect("foreign CUDA context");
    let local_pool = context.buffer_pool();
    let foreign_pool = foreign_context.buffer_pool();
    assert!(validate_transcode_pool_context(&context, &local_pool).is_ok());

    assert_transcode_pool_context_mismatch(context.j2k_transcode_dwt97_batch_with_pool(
        CudaDwt97BatchWithPoolRequest {
            blocks: &[],
            geometry: empty_geometry(),
            pool: &foreign_pool,
        },
    ));
    assert_transcode_pool_context_mismatch(
        context.j2k_transcode_htj2k97_codeblock_batch_with_pool(f32_request(&foreign_pool)),
    );
    assert_transcode_pool_context_mismatch(
        context
            .j2k_transcode_htj2k97_codeblock_batch_resident_with_pool(f32_request(&foreign_pool)),
    );
    assert_transcode_pool_context_mismatch(
        context.j2k_transcode_htj2k97_codeblock_i16_batch_resident_with_pool(
            CudaHtj2k97I16CodeblockBatchWithPoolRequest {
                blocks: &[],
                geometry: empty_geometry(),
                params: quantize_params(),
                pool: &foreign_pool,
            },
        ),
    );
}