#import "ASDisplayNodeExtras.h"

#import "ASDisplayNodeInternal.h"

ASDisplayNode *ASLayerToDisplayNode(CALayer *layer)

{

return layer.asyncdisplaykit_node;

}

ASDisplayNode *ASViewToDisplayNode(UIView *view)

{

return view.asyncdisplaykit_node;

}

id ASDisplayNodeFind(ASDisplayNode *node, BOOL (^block)(ASDisplayNode *node))

{

CALayer *layer = node.layer;

while (layer) {

node = ASLayerToDisplayNode(layer);

if (block(node)) {

return node;

}

layer = layer.superlayer;

}

return nil;

}

id ASDisplayNodeFindClass(ASDisplayNode *start, Class c)

{

return ASDisplayNodeFind(start, ^(ASDisplayNode *n) {

return [n isKindOfClass:c];

});

}

static void _ASCollectDisplayNodes(NSMutableArray *array, CALayer *layer)

{

ASDisplayNode *node = ASLayerToDisplayNode(layer);

if (nil != node) {

[array addObject:node];

}

for (CALayer *sublayer in layer.sublayers)

_ASCollectDisplayNodes(array, sublayer);

}

extern NSArray *ASCollectDisplayNodes(ASDisplayNode *node)

{

NSMutableArray *list = [NSMutableArray array];

for (CALayer *sublayer in node.layer.sublayers) {

_ASCollectDisplayNodes(list, sublayer);

}

return list;

}

#pragma mark - Find all subnodes

static void _ASDisplayNodeFindAllSubnodes(NSMutableArray *array, ASDisplayNode *node, BOOL (^block)(ASDisplayNode *node))

{

if (!node)

return;

for (ASDisplayNode *subnode in node.subnodes) {

if (block(subnode)) {

[array addObject:node];

}

_ASDisplayNodeFindAllSubnodes(array, subnode, block);

}

}

extern NSArray *ASDisplayNodeFindAllSubnodes(ASDisplayNode *start, BOOL (^block)(ASDisplayNode *node))

{

NSMutableArray *list = [NSMutableArray array];

_ASDisplayNodeFindAllSubnodes(list, start, block);

return list;

}

extern NSArray *ASDisplayNodeFindAllSubnodesOfClass(ASDisplayNode *start, Class c)

{

return ASDisplayNodeFindAllSubnodes(start, ^(ASDisplayNode *n) {

return [n isKindOfClass:c];

});

}

#pragma mark - Find first subnode

static ASDisplayNode *_ASDisplayNodeFindFirstSubnode(ASDisplayNode *startNode, BOOL includeStartNode, BOOL (^block)(ASDisplayNode *node))

{

for (ASDisplayNode *subnode in startNode.subnodes) {

ASDisplayNode *foundNode = _ASDisplayNodeFindFirstSubnode(subnode, YES, block);

if (foundNode) {

return foundNode;

}

}

if (includeStartNode && block(startNode))

return startNode;

return nil;

}

extern id ASDisplayNodeFindFirstSubnode(ASDisplayNode *startNode, BOOL (^block)(ASDisplayNode *node))

{

return _ASDisplayNodeFindFirstSubnode(startNode, NO, block);

}

extern id ASDisplayNodeFindFirstSubnodeOfClass(ASDisplayNode *start, Class c)

{

return ASDisplayNodeFindFirstSubnode(start, ^(ASDisplayNode *n) {

return [n isKindOfClass:c];

});

}

#pragma mark - Placeholders

UIColor *ASDisplayNodeDefaultPlaceholderColor()

{

return [UIColor colorWithWhite:0.95 alpha:1.0];

}

#pragma mark - Hierarchy Notifications

void ASDisplayNodeDisableHierarchyNotifications(ASDisplayNode *node)

{

[node __incrementVisibilityNotificationsDisabled];

}

void ASDisplayNodeEnableHierarchyNotifications(ASDisplayNode *node)

{

[node __decrementVisibilityNotificationsDisabled];

}