Vendor things

third-party/vendor/objc2/benches/autorelease.rs

@@ -0,0 +1,165 @@
+use core::ffi::c_void;
+use std::mem::ManuallyDrop;
+
+use objc2::rc::{autoreleasepool, Id, Shared};
+use objc2::runtime::{Class, Object, Sel};
+use objc2::{class, msg_send, sel};
+
+const BYTES: &[u8] = &[1, 2, 3, 4, 5, 6, 7, 8, 9, 10];
+
+fn empty() {
+    #[cfg(feature = "gnustep-1-7")]
+    unsafe {
+        objc2::__gnustep_hack::get_class_to_force_linkage()
+    };
+}
+
+fn pool_cleanup() {
+    autoreleasepool(|_| {})
+}
+
+fn class() -> &'static Class {
+    class!(NSObject)
+}
+
+fn sel() -> Sel {
+    sel!(alloc)
+}
+
+fn send_message() -> &'static Class {
+    unsafe { msg_send![class!(NSObject), class] }
+}
+
+fn alloc_nsobject() -> *mut Object {
+    unsafe { msg_send![class!(NSObject), alloc] }
+}
+
+fn new_nsobject() -> Id<Object, Shared> {
+    let obj = alloc_nsobject();
+    let obj: *mut Object = unsafe { msg_send![obj, init] };
+    unsafe { Id::new(obj).unwrap_unchecked() }
+}
+
+fn new_nsdata() -> Id<Object, Shared> {
+    let bytes_ptr = BYTES.as_ptr() as *const c_void;
+    let obj: *mut Object = unsafe { msg_send![class!(NSData), alloc] };
+    let obj: *mut Object = unsafe {
+        msg_send![
+            obj,
+            initWithBytes: bytes_ptr,
+            length: BYTES.len(),
+        ]
+    };
+    unsafe { Id::new(obj).unwrap_unchecked() }
+}
+
+fn new_leaked_nsdata() -> *const Object {
+    Id::as_ptr(&*ManuallyDrop::new(new_nsdata()))
+}
+
+fn autoreleased_nsdata() -> *const Object {
+    // let bytes_ptr = BYTES.as_ptr() as *const c_void;
+    // unsafe {
+    //     msg_send![
+    //         class!(NSData),
+    //         dataWithBytes: bytes_ptr,
+    //         length: BYTES.len(),
+    //     ]
+    // }
+    unsafe { msg_send![new_leaked_nsdata(), autorelease] }
+}
+
+fn new_nsstring() -> Id<Object, Shared> {
+    let obj: *mut Object = unsafe { msg_send![class!(NSString), alloc] };
+    let obj: *mut Object = unsafe { msg_send![obj, init] };
+    unsafe { Id::new(obj).unwrap_unchecked() }
+}
+
+fn new_leaked_nsstring() -> *const Object {
+    Id::as_ptr(&*ManuallyDrop::new(new_nsstring()))
+}
+
+fn autoreleased_nsstring() -> *const Object {
+    // unsafe { msg_send![class!(NSString), string] }
+    unsafe { msg_send![new_leaked_nsstring(), autorelease] }
+}
+
+fn retain_autoreleased(obj: *const Object) -> Id<Object, Shared> {
+    unsafe { Id::retain_autoreleased((obj as *mut Object).cast()).unwrap_unchecked() }
+}
+
+fn autoreleased_nsdata_pool_cleanup() -> *const Object {
+    autoreleasepool(|_| autoreleased_nsdata())
+}
+
+fn autoreleased_nsdata_fast_caller_cleanup() -> Id<Object, Shared> {
+    retain_autoreleased(autoreleased_nsdata())
+}
+
+fn autoreleased_nsdata_fast_caller_cleanup_pool_cleanup() -> Id<Object, Shared> {
+    autoreleasepool(|_| retain_autoreleased(autoreleased_nsdata()))
+}
+
+fn autoreleased_nsstring_pool_cleanup() -> *const Object {
+    autoreleasepool(|_| autoreleased_nsstring())
+}
+
+fn autoreleased_nsstring_fast_caller_cleanup() -> Id<Object, Shared> {
+    retain_autoreleased(autoreleased_nsstring())
+}
+
+fn autoreleased_nsstring_fast_caller_cleanup_pool_cleanup() -> Id<Object, Shared> {
+    autoreleasepool(|_| retain_autoreleased(autoreleased_nsstring()))
+}
+
+macro_rules! main_with_warmup {
+    ($($f:ident,)+) => {
+        mod warmup_fns {
+            $(
+                #[inline(never)]
+                pub fn $f() {
+                    let _ = iai::black_box(super::$f());
+                }
+            )+
+        }
+
+        // Required to get DYLD to resolve the stubs on x86_64
+        fn warmup() {
+            $(
+                warmup_fns::$f();
+            )+
+        }
+
+        iai::main! {
+            warmup,
+            $(
+                $f,
+            )+
+        }
+    };
+}
+
+main_with_warmup! {
+    // Baseline
+    empty,
+    pool_cleanup,
+    class,
+    sel,
+    send_message,
+    alloc_nsobject,
+    new_nsobject,
+    // NSData
+    new_nsdata,
+    new_leaked_nsdata,
+    autoreleased_nsdata,
+    autoreleased_nsdata_pool_cleanup,
+    autoreleased_nsdata_fast_caller_cleanup,
+    autoreleased_nsdata_fast_caller_cleanup_pool_cleanup,
+    // NSString
+    new_nsstring,
+    new_leaked_nsstring,
+    autoreleased_nsstring,
+    autoreleased_nsstring_pool_cleanup,
+    autoreleased_nsstring_fast_caller_cleanup,
+    autoreleased_nsstring_fast_caller_cleanup_pool_cleanup,
+}