core_video/
opengl_texture_cache.rs

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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
use std::ptr::{null, null_mut};

use core_foundation::{
    base::{kCFAllocatorDefault, CFAllocatorRef, CFType, CFTypeID, TCFType},
    dictionary::{CFDictionary, CFDictionaryRef},
    string::{CFString, CFStringRef},
};
use libc::c_void;

use crate::{
    base::CVOptionFlags,
    image_buffer::{CVImageBuffer, CVImageBufferRef},
    opengl_texture::{CVOpenGLTexture, CVOpenGLTextureRef},
    r#return::{kCVReturnSuccess, CVReturn},
    CGLContextObj, CGLPixelFormatObj,
};

#[repr(C)]
pub struct __CVOpenGLTextureCache(c_void);

pub type CVOpenGLTextureCacheRef = *mut __CVOpenGLTextureCache;

extern "C" {
    pub static kCVOpenGLTextureCacheChromaSamplingModeKey: CFStringRef;
    pub static kCVOpenGLTextureCacheChromaSamplingModeAutomatic: CFStringRef;
    pub static kCVOpenGLTextureCacheChromaSamplingModeHighestQuality: CFStringRef;
    pub static kCVOpenGLTextureCacheChromaSamplingModeBestPerformance: CFStringRef;

    pub fn CVOpenGLTextureCacheGetTypeID() -> CFTypeID;
    pub fn CVOpenGLTextureCacheRetain(textureCache: CVOpenGLTextureCacheRef) -> CVOpenGLTextureCacheRef;
    pub fn CVOpenGLTextureCacheRelease(textureCache: CVOpenGLTextureCacheRef);
    pub fn CVOpenGLTextureCacheCreate(
        allocator: CFAllocatorRef,
        cacheAttributes: CFDictionaryRef,
        cglContext: CGLContextObj,
        cglPixelFormat: CGLPixelFormatObj,
        textureAttributes: CFDictionaryRef,
        cacheOut: *mut CVOpenGLTextureCacheRef,
    ) -> CVReturn;
    pub fn CVOpenGLTextureCacheCreateTextureFromImage(
        allocator: CFAllocatorRef,
        textureCache: CVOpenGLTextureCacheRef,
        sourceImage: CVImageBufferRef,
        attributes: CFDictionaryRef,
        textureOut: *mut CVOpenGLTextureRef,
    ) -> CVReturn;
    pub fn CVOpenGLTextureCacheFlush(textureCache: CVOpenGLTextureCacheRef, options: CVOptionFlags);
}

pub enum CVOpenGLTextureCacheKeys {
    ChromaSamplingMode,
}

impl From<CVOpenGLTextureCacheKeys> for CFStringRef {
    fn from(key: CVOpenGLTextureCacheKeys) -> Self {
        unsafe {
            match key {
                CVOpenGLTextureCacheKeys::ChromaSamplingMode => kCVOpenGLTextureCacheChromaSamplingModeKey,
            }
        }
    }
}

impl From<CVOpenGLTextureCacheKeys> for CFString {
    fn from(key: CVOpenGLTextureCacheKeys) -> Self {
        unsafe { CFString::wrap_under_get_rule(CFStringRef::from(key)) }
    }
}

pub enum CVOpenGLTextureCacheChromaSamplingMode {
    Automatic,
    HighestQuality,
    BestPerformance,
}

impl From<CVOpenGLTextureCacheChromaSamplingMode> for CFStringRef {
    fn from(mode: CVOpenGLTextureCacheChromaSamplingMode) -> Self {
        unsafe {
            match mode {
                CVOpenGLTextureCacheChromaSamplingMode::Automatic => kCVOpenGLTextureCacheChromaSamplingModeAutomatic,
                CVOpenGLTextureCacheChromaSamplingMode::HighestQuality => kCVOpenGLTextureCacheChromaSamplingModeHighestQuality,
                CVOpenGLTextureCacheChromaSamplingMode::BestPerformance => kCVOpenGLTextureCacheChromaSamplingModeBestPerformance,
            }
        }
    }
}

impl From<CVOpenGLTextureCacheChromaSamplingMode> for CFString {
    fn from(mode: CVOpenGLTextureCacheChromaSamplingMode) -> Self {
        unsafe { CFString::wrap_under_get_rule(CFStringRef::from(mode)) }
    }
}

pub struct CVOpenGLTextureCache(CVOpenGLTextureCacheRef);

impl Drop for CVOpenGLTextureCache {
    fn drop(&mut self) {
        unsafe { CVOpenGLTextureCacheRelease(self.0) }
    }
}

impl_TCFType!(CVOpenGLTextureCache, CVOpenGLTextureCacheRef, CVOpenGLTextureCacheGetTypeID);
impl_CFTypeDescription!(CVOpenGLTextureCache);

impl CVOpenGLTextureCache {
    #[inline]
    pub unsafe fn new(
        cache_attributes: Option<&CFDictionary<CFString, CFString>>,
        cgl_context: CGLContextObj,
        cgl_pixel_format: CGLPixelFormatObj,
        texture_attributes: Option<&CFDictionary<CFString, CFType>>,
    ) -> Result<CVOpenGLTextureCache, CVReturn> {
        let mut cache: CVOpenGLTextureCacheRef = null_mut();
        let status = unsafe {
            CVOpenGLTextureCacheCreate(
                kCFAllocatorDefault,
                cache_attributes.map_or(null(), |attrs| attrs.as_concrete_TypeRef()),
                cgl_context,
                cgl_pixel_format,
                texture_attributes.map_or(null(), |attrs| attrs.as_concrete_TypeRef()),
                &mut cache,
            )
        };
        if status == kCVReturnSuccess {
            Ok(unsafe { TCFType::wrap_under_create_rule(cache) })
        } else {
            Err(status)
        }
    }

    #[inline]
    pub fn create_texture_from_image(
        &self,
        source_image: &CVImageBuffer,
        attributes: Option<&CFDictionary<CFString, CFType>>,
    ) -> Result<CVOpenGLTexture, CVReturn> {
        let mut texture: CVOpenGLTextureRef = null_mut();
        let status = unsafe {
            CVOpenGLTextureCacheCreateTextureFromImage(
                kCFAllocatorDefault,
                self.as_concrete_TypeRef(),
                source_image.as_concrete_TypeRef(),
                attributes.map_or(null(), |attrs| attrs.as_concrete_TypeRef()),
                &mut texture,
            )
        };
        if status == kCVReturnSuccess {
            Ok(unsafe { TCFType::wrap_under_create_rule(texture) })
        } else {
            Err(status)
        }
    }

    #[inline]
    pub fn flush(&self, options: CVOptionFlags) {
        unsafe { CVOpenGLTextureCacheFlush(self.as_concrete_TypeRef(), options) }
    }
}