2019-11-11 16:39:27 +04:00

73 lines
1.8 KiB
Mathematica

//
// ASAssert.m
// Texture
//
// Copyright (c) Pinterest, Inc. All rights reserved.
// Licensed under Apache 2.0: http://www.apache.org/licenses/LICENSE-2.0
//
#import <AsyncDisplayKit/ASAssert.h>
#import <AsyncDisplayKit/ASAvailability.h>
<<<<<<< HEAD
#ifndef MINIMAL_ASDK
static _Thread_local int tls_mainThreadAssertionsDisabledCount;
#endif
=======
#if AS_TLS_AVAILABLE
>>>>>>> 565da7d4935740d12fc204aa061faf093831da1e
static _Thread_local int tls_mainThreadAssertionsDisabledCount;
BOOL ASMainThreadAssertionsAreDisabled() {
#ifdef MINIMAL_ASDK
return false;
#else
return tls_mainThreadAssertionsDisabledCount > 0;
#endif
}
void ASPushMainThreadAssertionsDisabled() {
#ifndef MINIMAL_ASDK
tls_mainThreadAssertionsDisabledCount += 1;
#endif
}
void ASPopMainThreadAssertionsDisabled() {
#ifndef MINIMAL_ASDK
tls_mainThreadAssertionsDisabledCount -= 1;
ASDisplayNodeCAssert(tls_mainThreadAssertionsDisabledCount >= 0, @"Attempt to pop thread assertion-disabling without corresponding push.");
#endif
}
#else
#import <dispatch/once.h>
static pthread_key_t ASMainThreadAssertionsDisabledKey() {
static pthread_key_t k;
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
pthread_key_create(&k, NULL);
});
return k;
}
BOOL ASMainThreadAssertionsAreDisabled() {
return (pthread_getspecific(ASMainThreadAssertionsDisabledKey()) > 0);
}
void ASPushMainThreadAssertionsDisabled() {
let key = ASMainThreadAssertionsDisabledKey();
let oldVal = pthread_getspecific(key);
pthread_setspecific(key, oldVal + 1);
}
void ASPopMainThreadAssertionsDisabled() {
let key = ASMainThreadAssertionsDisabledKey();
let oldVal = pthread_getspecific(key);
pthread_setspecific(key, oldVal - 1);
ASDisplayNodeCAssert(oldVal > 0, @"Attempt to pop thread assertion-disabling without corresponding push.");
}
#endif // AS_TLS_AVAILABLE