j2k-metal 0.6.1

Apple Metal GPU adapter for Rust JPEG 2000 and HTJ2K decode/encode paths
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

// SPDX-License-Identifier: Apache-2.0

use std::{
    collections::HashMap,
    sync::{Arc, Mutex},
};

use super::PreparedDirectGrayscalePlan;
use crate::Error;

#[derive(Clone, Copy, PartialEq, Eq, Hash)]
struct CpuTier1CoefficientCacheKey {
    step_idx: usize,
    output_len: usize,
}

#[derive(Default)]
pub(super) struct CpuTier1CoefficientCache {
    entries: Mutex<HashMap<CpuTier1CoefficientCacheKey, Arc<[f32]>>>,
}

impl PreparedDirectGrayscalePlan {
    pub(super) fn cached_cpu_tier1_coefficients(
        &self,
        step_idx: usize,
        output_len: usize,
    ) -> Result<Option<Vec<f32>>, Error> {
        let key = CpuTier1CoefficientCacheKey {
            step_idx,
            output_len,
        };
        let entries = self
            .cpu_tier1_cache
            .entries
            .lock()
            .map_err(|_| Error::MetalKernel {
                message: "J2K MetalDirect hybrid CPU Tier-1 cache lock is poisoned".to_string(),
            })?;
        Ok(entries.get(&key).map(|coefficients| coefficients.to_vec()))
    }

    pub(super) fn store_cpu_tier1_coefficients(
        &self,
        step_idx: usize,
        output_len: usize,
        coefficients: Vec<f32>,
    ) -> Result<Vec<f32>, Error> {
        let key = CpuTier1CoefficientCacheKey {
            step_idx,
            output_len,
        };
        let cached = Arc::<[f32]>::from(coefficients.clone());
        let mut entries = self
            .cpu_tier1_cache
            .entries
            .lock()
            .map_err(|_| Error::MetalKernel {
                message: "J2K MetalDirect hybrid CPU Tier-1 cache lock is poisoned".to_string(),
            })?;
        entries.insert(key, cached);
        Ok(coefficients)
    }

    pub(super) fn clear_cpu_tier1_cache(&self) -> Result<(), Error> {
        let mut entries = self
            .cpu_tier1_cache
            .entries
            .lock()
            .map_err(|_| Error::MetalKernel {
                message: "J2K MetalDirect hybrid CPU Tier-1 cache lock is poisoned".to_string(),
            })?;
        entries.clear();
        Ok(())
    }
}