honeycrisp/aruminium/src/ffi/selectors.rs

rust
ฯ€ 0.0% 282 words ยท 2 min 
//! Cached ObjC selectors
//!
//! sel_registerName is idempotent: same C string -> same pointer.
//! We cache the result via AtomicPtr with Relaxed ordering โ€” on ARM64 this
//! compiles to a plain `ldr` (no memory barrier). The race on first init is
//! benign: sel_registerName always returns the same value for the same string.

use super::{sel_registerName, ObjcSel};

macro_rules! cached_sel {
    ($name:ident, $lit:expr) => {
        #[inline(always)]
        pub fn $name() -> ObjcSel {
            use std::sync::atomic::{AtomicPtr, Ordering};
            static CACHE: AtomicPtr<()> = AtomicPtr::new(std::ptr::null_mut());
            let p = CACHE.load(Ordering::Relaxed);
            if !p.is_null() {
                return p as ObjcSel;
            }
            let s = unsafe { sel_registerName($lit.as_ptr()) };
            CACHE.store(s as *mut (), Ordering::Relaxed);
            s
        }
    };
}

// Hot path selectors โ€” called on every command buffer dispatch
cached_sel!(SEL_commandBuffer, c"commandBuffer");
cached_sel!(SEL_computeCommandEncoder, c"computeCommandEncoder");
cached_sel!(SEL_setComputePipelineState, c"setComputePipelineState:");
cached_sel!(SEL_setBuffer_offset_atIndex, c"setBuffer:offset:atIndex:");
cached_sel!(SEL_setBytes_length_atIndex, c"setBytes:length:atIndex:");
cached_sel!(
    SEL_dispatchThreads,
    c"dispatchThreads:threadsPerThreadgroup:"
);
cached_sel!(
    SEL_dispatchThreadgroups,
    c"dispatchThreadgroups:threadsPerThreadgroup:"
);
cached_sel!(SEL_endEncoding, c"endEncoding");
cached_sel!(SEL_commit, c"commit");
cached_sel!(SEL_waitUntilCompleted, c"waitUntilCompleted");
cached_sel!(SEL_contents, c"contents");
cached_sel!(SEL_status, c"status");
cached_sel!(SEL_error, c"error");

// Fast command buffer (unretained references)
cached_sel!(
    SEL_commandBufferWithUnretainedReferences,
    c"commandBufferWithUnretainedReferences"
);

// Device selectors
cached_sel!(SEL_name, c"name");
cached_sel!(SEL_newCommandQueue, c"newCommandQueue");
cached_sel!(
    SEL_newBufferWithLength_options,
    c"newBufferWithLength:options:"
);
cached_sel!(
    SEL_newBufferWithBytes_length_options,
    c"newBufferWithBytes:length:options:"
);
cached_sel!(
    SEL_newBufferWithBytesNoCopy_length_options_deallocator,
    c"newBufferWithBytesNoCopy:length:options:deallocator:"
);
cached_sel!(
    SEL_newLibraryWithSource_options_error,
    c"newLibraryWithSource:options:error:"
);
cached_sel!(
    SEL_newComputePipelineStateWithFunction_error,
    c"newComputePipelineStateWithFunction:error:"
);
cached_sel!(SEL_hasUnifiedMemory, c"hasUnifiedMemory");
cached_sel!(SEL_maxBufferLength, c"maxBufferLength");
cached_sel!(SEL_maxThreadsPerThreadgroup, c"maxThreadsPerThreadgroup");
cached_sel!(
    SEL_recommendedMaxWorkingSetSize,
    c"recommendedMaxWorkingSetSize"
);
cached_sel!(SEL_newTextureWithDescriptor, c"newTextureWithDescriptor:");
cached_sel!(SEL_newFence, c"newFence");
cached_sel!(SEL_newEvent, c"newEvent");
cached_sel!(SEL_newSharedEvent, c"newSharedEvent");

// Library/function selectors
cached_sel!(SEL_newFunctionWithName, c"newFunctionWithName:");
cached_sel!(SEL_functionNames, c"functionNames");

// Pipeline selectors
cached_sel!(
    SEL_maxTotalThreadsPerThreadgroup,
    c"maxTotalThreadsPerThreadgroup"
);
cached_sel!(SEL_threadExecutionWidth, c"threadExecutionWidth");
cached_sel!(
    SEL_staticThreadgroupMemoryLength,
    c"staticThreadgroupMemoryLength"
);

// Command buffer timing selectors
cached_sel!(SEL_GPUStartTime, c"GPUStartTime");
cached_sel!(SEL_GPUEndTime, c"GPUEndTime");

// Collection selectors
cached_sel!(SEL_count, c"count");
cached_sel!(SEL_objectAtIndex, c"objectAtIndex:");

// Blit encoder
cached_sel!(SEL_blitCommandEncoder, c"blitCommandEncoder");
cached_sel!(
    SEL_copyFromBuffer,
    c"copyFromBuffer:sourceOffset:toBuffer:destinationOffset:size:"
);

// Texture
cached_sel!(SEL_width, c"width");
cached_sel!(SEL_height, c"height");
cached_sel!(SEL_depth, c"depth");
cached_sel!(SEL_pixelFormat, c"pixelFormat");
cached_sel!(
    SEL_replaceRegion,
    c"replaceRegion:mipmapLevel:withBytes:bytesPerRow:"
);
cached_sel!(
    SEL_getBytes,
    c"getBytes:bytesPerRow:fromRegion:mipmapLevel:"
);

// Sync
cached_sel!(SEL_signaledValue, c"signaledValue");

// NSString / NSError
cached_sel!(SEL_localizedDescription, c"localizedDescription");
cached_sel!(SEL_UTF8String, c"UTF8String");

Neighbours