j2k-metal 0.8.1

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
94
95
96
97
98

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

use super::super::{
    PreparedDirectColorPlan, PreparedDirectGrayscalePlan, PreparedDirectGrayscaleStep,
};
use std::sync::Mutex;

pub(super) static HYBRID_COUNTER_TEST_LOCK: Mutex<()> = Mutex::new(());

pub(super) fn prepared_direct_color_tier1_input_count(plan: &PreparedDirectColorPlan) -> usize {
    plan.component_plans
        .iter()
        .map(prepared_direct_component_tier1_input_count)
        .sum()
}

fn prepared_direct_component_tier1_input_count(plan: &PreparedDirectGrayscalePlan) -> usize {
    let mut count = 0;
    let mut step_idx = 0;
    while step_idx < plan.steps.len() {
        if let Some(group) = plan.classic_group_starting_at(step_idx) {
            count += 1;
            step_idx = group.end_step;
            continue;
        }
        if let Some(group) = plan.ht_group_starting_at(step_idx) {
            count += 1;
            step_idx = group.end_step;
            continue;
        }
        if matches!(
            &plan.steps[step_idx],
            PreparedDirectGrayscaleStep::ClassicSubBand(_)
                | PreparedDirectGrayscaleStep::HtSubBand(_)
        ) {
            count += 1;
        }
        step_idx += 1;
    }
    count
}

pub(super) fn cached_direct_color_tier1_input_count(plan: &PreparedDirectColorPlan) -> usize {
    plan.component_plans
        .iter()
        .map(cached_direct_component_tier1_input_count)
        .sum()
}

fn cached_direct_component_tier1_input_count(plan: &PreparedDirectGrayscalePlan) -> usize {
    let mut count = 0;
    let mut step_idx = 0;
    while step_idx < plan.steps.len() {
        if let Some(group) = plan.classic_group_starting_at(step_idx) {
            if has_cached_cpu_tier1_coefficients(plan, step_idx, group.total_coefficients) {
                count += 1;
            }
            step_idx = group.end_step;
            continue;
        }
        if let Some(group) = plan.ht_group_starting_at(step_idx) {
            if has_cached_cpu_tier1_coefficients(plan, step_idx, group.total_coefficients) {
                count += 1;
            }
            step_idx = group.end_step;
            continue;
        }
        match &plan.steps[step_idx] {
            PreparedDirectGrayscaleStep::ClassicSubBand(sub_band) => {
                let output_len = sub_band.width as usize * sub_band.height as usize;
                if has_cached_cpu_tier1_coefficients(plan, step_idx, output_len) {
                    count += 1;
                }
            }
            PreparedDirectGrayscaleStep::HtSubBand(sub_band) => {
                let output_len = sub_band.width as usize * sub_band.height as usize;
                if has_cached_cpu_tier1_coefficients(plan, step_idx, output_len) {
                    count += 1;
                }
            }
            PreparedDirectGrayscaleStep::Idwt(_) | PreparedDirectGrayscaleStep::Store(_) => {}
        }
        step_idx += 1;
    }
    count
}

fn has_cached_cpu_tier1_coefficients(
    plan: &PreparedDirectGrayscalePlan,
    step_idx: usize,
    output_len: usize,
) -> bool {
    let mut budget =
        crate::batch_allocation::BatchMetadataBudget::new("J2K Metal test CPU Tier-1 cache lookup");
    plan.cached_cpu_tier1_coefficients(&mut budget, step_idx, output_len)
        .expect("CPU Tier-1 cache lookup")
        .is_some()
}