mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-09-03 19:30:09 +00:00
117 lines
4.2 KiB
Objective-C
117 lines
4.2 KiB
Objective-C
//
|
||
// ASTwoDimensionalArrayUtils.m
|
||
// Texture
|
||
//
|
||
// 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 /ASDK-Licenses directory of this source tree. An additional
|
||
// grant of patent rights can be found in the PATENTS file in the same directory.
|
||
//
|
||
// Modifications to this file made after 4/13/2017 are: Copyright (c) 2017-present,
|
||
// Pinterest, Inc. Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
|
||
#import <AsyncDisplayKit/ASAssert.h>
|
||
#import <AsyncDisplayKit/ASInternalHelpers.h>
|
||
#import <AsyncDisplayKit/ASTwoDimensionalArrayUtils.h>
|
||
|
||
// Import UIKit to get [NSIndexPath indexPathForItem:inSection:] which uses
|
||
// tagged pointers.
|
||
#import <UIKit/UIKit.h>
|
||
|
||
#pragma mark - Public Methods
|
||
|
||
NSMutableArray<NSMutableArray *> *ASTwoDimensionalArrayDeepMutableCopy(NSArray<NSArray *> *array)
|
||
{
|
||
NSMutableArray *newArray = [NSMutableArray arrayWithCapacity:array.count];
|
||
NSInteger i = 0;
|
||
for (NSArray *subarray in array) {
|
||
ASDisplayNodeCAssert([subarray isKindOfClass:[NSArray class]], @"This function expects NSArray<NSArray *> *");
|
||
newArray[i++] = [subarray mutableCopy];
|
||
}
|
||
return newArray;
|
||
}
|
||
|
||
void ASDeleteElementsInTwoDimensionalArrayAtIndexPaths(NSMutableArray *mutableArray, NSArray<NSIndexPath *> *indexPaths)
|
||
{
|
||
if (indexPaths.count == 0) {
|
||
return;
|
||
}
|
||
|
||
#if ASDISPLAYNODE_ASSERTIONS_ENABLED
|
||
NSArray *sortedIndexPaths = [indexPaths sortedArrayUsingSelector:@selector(asdk_inverseCompare:)];
|
||
ASDisplayNodeCAssert([sortedIndexPaths isEqualToArray:indexPaths], @"Expected array of index paths to be sorted in descending order.");
|
||
#endif
|
||
|
||
/**
|
||
* It is tempting to do something clever here and collect indexes into ranges or NSIndexSets
|
||
* but deep down, __NSArrayM only implements removeObjectAtIndex: and so doing all that extra
|
||
* work ends up running the same code.
|
||
*/
|
||
for (NSIndexPath *indexPath in indexPaths) {
|
||
NSInteger section = indexPath.section;
|
||
if (section >= mutableArray.count) {
|
||
ASDisplayNodeCFailAssert(@"Invalid section index %ld – only %ld sections", (long)section, (long)mutableArray.count);
|
||
continue;
|
||
}
|
||
|
||
NSMutableArray *subarray = mutableArray[section];
|
||
NSInteger item = indexPath.item;
|
||
if (item >= subarray.count) {
|
||
ASDisplayNodeCFailAssert(@"Invalid item index %ld – only %ld items in section %ld", (long)item, (long)subarray.count, (long)section);
|
||
continue;
|
||
}
|
||
[subarray removeObjectAtIndex:item];
|
||
}
|
||
}
|
||
|
||
NSArray<NSIndexPath *> *ASIndexPathsForTwoDimensionalArray(NSArray <NSArray *>* twoDimensionalArray)
|
||
{
|
||
NSMutableArray *result = [[NSMutableArray alloc] init];
|
||
NSInteger section = 0;
|
||
NSInteger i = 0;
|
||
for (NSArray *subarray in twoDimensionalArray) {
|
||
ASDisplayNodeCAssert([subarray isKindOfClass:[NSArray class]], @"This function expects NSArray<NSArray *> *");
|
||
NSInteger itemCount = subarray.count;
|
||
for (NSInteger item = 0; item < itemCount; item++) {
|
||
result[i++] = [NSIndexPath indexPathForItem:item inSection:section];
|
||
}
|
||
section++;
|
||
}
|
||
return result;
|
||
}
|
||
|
||
NSArray *ASElementsInTwoDimensionalArray(NSArray <NSArray *>* twoDimensionalArray)
|
||
{
|
||
NSMutableArray *result = [[NSMutableArray alloc] init];
|
||
NSInteger i = 0;
|
||
for (NSArray *subarray in twoDimensionalArray) {
|
||
ASDisplayNodeCAssert([subarray isKindOfClass:[NSArray class]], @"This function expects NSArray<NSArray *> *");
|
||
for (id element in subarray) {
|
||
result[i++] = element;
|
||
}
|
||
}
|
||
return result;
|
||
}
|
||
|
||
id ASGetElementInTwoDimensionalArray(NSArray *array, NSIndexPath *indexPath)
|
||
{
|
||
ASDisplayNodeCAssertNotNil(indexPath, @"Expected non-nil index path");
|
||
ASDisplayNodeCAssert(indexPath.length == 2, @"Expected index path of length 2. Index path: %@", indexPath);
|
||
NSInteger section = indexPath.section;
|
||
if (array.count <= section) {
|
||
return nil;
|
||
}
|
||
|
||
NSArray *innerArray = array[section];
|
||
NSInteger item = indexPath.item;
|
||
if (innerArray.count <= item) {
|
||
return nil;
|
||
}
|
||
return innerArray[item];
|
||
}
|