mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-12-29 09:20:32 +00:00
1775bf29f5
* Build ASElementMap and use it * Keep building! * Clean up * Restrict ASDataController subclassing * Clean up more * More rearranging * Dear lord it's time for bed * Make things clearer * Sadly remove subclassing restriction for ASDataController * Remove dead initializer
70 lines
1.9 KiB
Plaintext
70 lines
1.9 KiB
Plaintext
//
|
|
// ASCollectionElement.mm
|
|
// AsyncDisplayKit
|
|
//
|
|
// Created by Huy Nguyen on 2/28/16.
|
|
//
|
|
// Copyright (c) 2014-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree. An additional grant
|
|
// of patent rights can be found in the PATENTS file in the same directory.
|
|
//
|
|
|
|
#import <AsyncDisplayKit/ASCollectionElement.h>
|
|
#import <AsyncDisplayKit/ASCellNode+Internal.h>
|
|
#import <mutex>
|
|
|
|
@interface ASCollectionElement ()
|
|
|
|
/// Required node block used to allocate a cell node. Nil after the first execution.
|
|
@property (nonatomic, strong) ASCellNodeBlock nodeBlock;
|
|
|
|
@end
|
|
|
|
@implementation ASCollectionElement {
|
|
std::mutex _lock;
|
|
ASCellNode *_node;
|
|
}
|
|
|
|
- (instancetype)initWithNodeBlock:(ASCellNodeBlock)nodeBlock
|
|
supplementaryElementKind:(nullable NSString *)supplementaryElementKind
|
|
constrainedSize:(ASSizeRange)constrainedSize
|
|
environment:(id<ASTraitEnvironment>)environment
|
|
{
|
|
NSAssert(nodeBlock != nil, @"Node block must not be nil");
|
|
self = [super init];
|
|
if (self) {
|
|
_nodeBlock = nodeBlock;
|
|
_supplementaryElementKind = [supplementaryElementKind copy];
|
|
_constrainedSize = constrainedSize;
|
|
_traitEnvironment = environment;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (ASCellNode *)node
|
|
{
|
|
std::lock_guard<std::mutex> l(_lock);
|
|
if (_nodeBlock != nil) {
|
|
ASCellNode *node = _nodeBlock();
|
|
_nodeBlock = nil;
|
|
if (node == nil) {
|
|
ASDisplayNodeFailAssert(@"Node block returned nil node!");
|
|
node = [[ASCellNode alloc] init];
|
|
}
|
|
node.owningNode = (ASDisplayNode *)_traitEnvironment;
|
|
node.collectionElement = self;
|
|
ASTraitCollectionPropagateDown(node, [_traitEnvironment primitiveTraitCollection]);
|
|
_node = node;
|
|
}
|
|
return _node;
|
|
}
|
|
|
|
- (ASCellNode *)nodeIfAllocated
|
|
{
|
|
std::lock_guard<std::mutex> l(_lock);
|
|
return _node;
|
|
}
|
|
|
|
@end
|