j2k-metal 0.8.0

Metal decoder and encode-stage adapter for j2k
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

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

use std::mem::size_of;

use j2k_metal_support::{dispatch_1d_pipeline, dispatch_2d_pipeline, dispatch_3d_pipeline};

use crate::profile_env::{
    hybrid_stage_signpost, label_compute_encoder, SIGNPOST_DECODE_HYBRID_IDWT_COMMAND_ENCODE,
    SIGNPOST_DECODE_HYBRID_MCT_PACK_COMMAND_ENCODE, SIGNPOST_DECODE_HYBRID_STORE_COMMAND_ENCODE,
};

use super::abi::{
    J2kClassicCleanupBatchJob, J2kClassicRepeatedBatchParams, J2kClassicSegment, J2kClassicStatus,
    J2kGrayStoreParams, J2kHtCleanupBatchJob, J2kHtRepeatedBatchParams, J2kHtStatus,
    J2kIdwt97StepParams, J2kIdwtSingleDecompositionParams, J2kInverseMctParams,
    J2kRepeatedGrayStoreParams, J2kRepeatedIdwtSingleDecompositionParams, J2kRepeatedStoreParams,
    J2kStoreParams, J2K_CLASSIC_MAX_HEIGHT, J2K_CLASSIC_MAX_WIDTH, J2K_CLASSIC_STATUS_OK,
};
use super::{
    checked_buffer_slice, commit_and_wait_metal, copied_slice_buffer, decode_classic_status_error,
    j2k_u32_param, new_command_buffer, new_compute_command_encoder, new_shared_buffer,
    take_classic_coefficients_scratch_buffer, take_classic_states_scratch_buffer, with_runtime,
    zeroed_shared_buffer, Buffer, CommandBufferRef, ComputeCommandEncoderRef,
    DirectIdwtCommandBuffers, DirectScratchBuffer, DirectStatusCheck, Error, HtCodeBlockDecodeJob,
    J2kInverseMctJob, J2kSingleDecompositionIdwtJob, J2kStoreComponentJob, J2kWaveletTransform,
    MTLSize, MetalRuntime, PixelFormat, PreparedClassicSubBand, PreparedClassicSubBandGroup,
    PreparedHtSubBand, PreparedHtSubBandGroup, Surface,
};

mod classic_cleanup;
mod classic_subband;
mod ht_chunks;
mod ht_distinct;
mod ht_subband;
pub(in crate::compute) mod idwt;
pub(in crate::compute) mod mct;
pub(in crate::compute) mod store;

pub(in crate::compute) use self::classic_cleanup::{
    classic_batch_is_plain_arithmetic, classic_batch_uses_plain_fast_path,
    classic_repeated_uses_plain_fast_path, dispatch_classic_cleanup_batched,
    dispatch_classic_cleanup_batched_in_encoder,
    dispatch_classic_cleanup_plain_dev_repeated_batched_in_command_buffer,
    dispatch_classic_cleanup_repeated_batched_in_command_buffer,
    dispatch_classic_store_repeated_batched_in_command_buffer,
    encode_distinct_classic_sub_band_groups_to_buffer_in_command_buffer,
    encode_distinct_classic_sub_band_groups_to_buffer_in_encoder,
    encode_distinct_classic_sub_bands_to_buffer_in_command_buffer,
    encode_distinct_classic_sub_bands_to_buffer_in_encoder, ClassicCleanupBatchDispatch,
    ClassicPlainDevRepeatedCleanupDispatch, ClassicRepeatedCleanupDispatch,
    ClassicRepeatedStoreDispatch,
};
pub(in crate::compute) use self::classic_subband::{
    encode_prepared_classic_sub_band_group_to_buffer_in_encoder,
    encode_prepared_classic_sub_band_to_buffer_in_encoder,
    encode_repeated_classic_sub_band_group_to_buffer_in_command_buffer,
    encode_repeated_classic_sub_band_to_buffer_in_command_buffer,
};
pub(in crate::compute) use self::ht_chunks::{
    default_metal_ht_chunk_limits, encode_metal_ht_batches_in_encoder,
    encode_repeated_metal_ht_batch_in_command_buffer, HtBatchInput, HtPayloadSource,
    MetalHtPipelineKind, PreparedMetalHtExecutionCache,
};
pub(in crate::compute) use self::ht_distinct::{
    encode_distinct_ht_sub_band_groups_to_buffer_in_command_buffer,
    encode_distinct_ht_sub_band_groups_to_buffer_in_encoder,
    encode_distinct_ht_sub_bands_to_buffer_in_command_buffer,
    encode_distinct_ht_sub_bands_to_buffer_in_encoder,
};
pub(in crate::compute) use self::ht_subband::{
    dispatch_zero_u32_buffer_in_encoder, encode_prepared_ht_sub_band_group_to_buffer_in_encoder,
    encode_prepared_ht_sub_band_to_buffer_in_encoder,
    encode_repeated_ht_sub_band_group_to_buffer_in_command_buffer,
    encode_repeated_ht_sub_band_to_buffer_in_command_buffer, ht_batch_output_word_count,
    ht_output_word_count, required_ht_output_len,
};
pub(in crate::compute) use self::idwt::{
    dispatch_irreversible97_single_decomposition_buffers_in_command_buffer_with_offsets,
    dispatch_irreversible97_single_decomposition_buffers_in_encoder_with_offsets,
    dispatch_reversible53_repeated_buffers_in_command_buffer_with_offsets,
    dispatch_reversible53_repeated_buffers_in_encoder_with_offsets,
    dispatch_reversible53_single_decomposition_buffers_in_command_buffer_with_offsets,
    dispatch_reversible53_single_decomposition_buffers_in_encoder_with_offsets, IdwtSubBandBuffers,
    RepeatedIdwtDispatch, SingleIdwtDispatch,
};
pub(in crate::compute) use self::store::{
    dispatch_store_component_buffer_in_command_buffer_with_offsets,
    dispatch_store_component_buffer_in_encoder_with_offsets,
    dispatch_store_component_repeated_in_command_buffer,
    dispatch_store_component_repeated_in_encoder, encode_gray_store_to_destination_in_encoder,
    encode_gray_store_to_surface_in_encoder,
    encode_repeated_gray_store_to_surfaces_in_command_buffer, GrayStoreDestinationRequest,
};