@aks-dev/easyui 1.0.31 → 1.0.34

package/ios/libs/SDAutoLayout/UITableView+SDAutoTableViewCellHeight.m

@@ -0,0 +1,537 @@
+//
+//  UITableView+SDAutoTableViewCellHeight.m
+//  SDAutoLayout 测试 Demo
+//
+//  Created by aier on 15/11/1.
+//  Copyright © 2015年 gsd. All rights reserved.
+//
+
+/*
+ 
+ *********************************************************************************
+ *                                                                                *
+ * 在您使用此自动布局库的过程中如果出现bug请及时以以下任意一种方式联系我们，我们会及时修复bug并  *
+ * 帮您解决问题。                                                                    *
+ * QQ    : 2689718696(gsdios)                                                      *
+ * Email : gsdios@126.com                                                          *
+ * GitHub: https://github.com/gsdios                                               *
+ * 新浪微博:GSD_iOS                                                                 *
+ *                                                                                *
+ *********************************************************************************
+ 
+ */
+
+#import "UITableView+SDAutoTableViewCellHeight.h"
+#import <objc/runtime.h>
+
+@interface SDCellAutoHeightManager ()
+
+@property (nonatomic, weak) UITableView *modelTableview;
+
+@end
+
+@implementation SDCellAutoHeightManager
+{
+    NSMutableDictionary *_cacheDictionary;
+    NSMutableDictionary *_modelCellsDict;
+}
+
+- (instancetype)init
+{
+    if (self = [super init]) {
+        [self setup];
+    }
+    return self;
+}
+
+- (instancetype)initWithCellClass:(Class)cellClass tableView:(UITableView *)tableView
+{
+    if (self = [super init]) {
+        [self setup];
+        self.modelTableview = tableView;
+        [self registerCellWithCellClass:cellClass];
+    }
+    return self;
+}
+
+- (instancetype)initWithCellClasses:(NSArray *)cellClassArray tableView:(UITableView *)tableView
+{
+    if (self = [super init]) {
+        [self setup];
+        self.modelTableview = tableView;
+        [cellClassArray enumerateObjectsUsingBlock:^(Class obj, NSUInteger idx, BOOL *stop) {
+            [self registerCellWithCellClass:obj];
+        }];
+    }
+    return self;
+}
+
+- (void)setup
+{
+    _cacheDictionary = [NSMutableDictionary new];
+    _modelCellsDict = [NSMutableDictionary new];
+}
+
+- (void)registerCellWithCellClass:(Class)cellClass
+{
+    [_modelTableview registerClass:cellClass forCellReuseIdentifier:NSStringFromClass(cellClass)];
+    self.modelCell = [_modelTableview dequeueReusableCellWithIdentifier:NSStringFromClass(cellClass)];
+    
+    if (!self.modelCell.contentView.subviews.count) {
+        NSString *path = [[NSBundle mainBundle] pathForResource:[NSString stringWithFormat:@"%@.nib", NSStringFromClass(cellClass)] ofType:nil];
+        if (path) {
+            self.modelCell = nil;
+            [_modelTableview registerNib:[UINib nibWithNibName:NSStringFromClass(cellClass) bundle:nil] forCellReuseIdentifier:NSStringFromClass(cellClass)];
+            self.modelCell = [_modelTableview dequeueReusableCellWithIdentifier:NSStringFromClass(cellClass)];
+        }
+    }
+    if (self.modelCell) {
+        [_modelCellsDict setObject:self.modelCell forKey:NSStringFromClass(cellClass)];
+    }
+}
+
++ (instancetype)managerWithCellClass:(Class)cellClass tableView:(UITableView *)tableView
+{
+    SDCellAutoHeightManager *manager = [[self alloc] initWithCellClass:cellClass tableView:tableView];
+    return manager;
+}
+
+- (UITableViewCell *)modelCell
+{
+    if (_modelCell.contentView.tag != kSDModelCellTag) {
+        _modelCell.contentView.tag = kSDModelCellTag;
+    }
+    return _modelCell;
+}
+
+- (NSDictionary *)heightCacheDict
+{
+    return _cacheDictionary;
+}
+
+- (void)clearHeightCache
+{
+    [_cacheDictionary removeAllObjects];
+    [_subviewFrameCacheDict removeAllObjects];
+}
+
+- (NSString *)cacheKeyForIndexPath:(NSIndexPath *)indexPath
+{
+    return [NSString stringWithFormat:@"%ld-%ld", (long)indexPath.section, (long)indexPath.row];
+}
+
+- (void)clearHeightCacheOfIndexPaths:(NSArray *)indexPaths
+{
+    [indexPaths enumerateObjectsUsingBlock:^(NSIndexPath *indexPath, NSUInteger idx, BOOL *stop) {
+        NSString *cacheKey = [self cacheKeyForIndexPath:indexPath];
+        [_cacheDictionary removeObjectForKey:cacheKey];
+        [_subviewFrameCacheDict removeObjectForKey:cacheKey];
+    }];
+}
+
+- (void)deleteThenResetHeightCache:(NSIndexPath *)indexPathToDelete
+{
+
+    NSString *cacheKey = [self cacheKeyForIndexPath:indexPathToDelete];
+    [_cacheDictionary removeObjectForKey:cacheKey];
+    [_subviewFrameCacheDict removeObjectForKey:cacheKey];
+    
+    long sectionOfToDeleteItem = indexPathToDelete.section;
+    long rowOfToDeleteItem = indexPathToDelete.row;
+    NSMutableDictionary *tempHeightCacheDict = [NSMutableDictionary new];
+    NSMutableDictionary *tempFrameCacheDict = [NSMutableDictionary new];
+    for (NSString *key in _cacheDictionary.allKeys) {
+        NSArray *res = [key componentsSeparatedByString:@"-"];
+        long section = [res.firstObject integerValue];
+        long row = [res.lastObject integerValue];
+        if (section == sectionOfToDeleteItem && row > rowOfToDeleteItem) {
+            NSNumber *heightCache = _cacheDictionary[key];
+            NSArray *frameCache = _subviewFrameCacheDict[key];
+            NSString *newKey = [NSString stringWithFormat:@"%ld-%ld", section, (row - 1)];
+            [tempHeightCacheDict setValue:heightCache forKey:newKey];
+            [tempFrameCacheDict setValue:frameCache forKey:newKey];
+            [_cacheDictionary removeObjectForKey:key];
+            [_subviewFrameCacheDict removeObjectForKey:key];
+        }
+    }
+    [_cacheDictionary addEntriesFromDictionary:tempHeightCacheDict];
+    [_subviewFrameCacheDict addEntriesFromDictionary:tempFrameCacheDict];
+
+}
+
+- (void)insertNewDataAtTheBeginingOfSection:(NSInteger)section newDataCount:(NSInteger)count
+{
+    NSMutableDictionary *tempHeightCacheDict = [NSMutableDictionary new];
+    NSMutableDictionary *tempFrameCacheDict = [NSMutableDictionary new];
+    for (NSString *key in _cacheDictionary.allKeys) {
+        NSArray *res = [key componentsSeparatedByString:@"-"];
+        long originalSection = [res.firstObject integerValue];
+        long row = [res.lastObject integerValue];
+        if (originalSection == section) {
+            NSNumber *heightCache = _cacheDictionary[key];
+            NSArray *frameCache = _subviewFrameCacheDict[key];
+            NSString *newKey = [NSString stringWithFormat:@"%ld-%ld", originalSection, (row + count)];
+            [tempHeightCacheDict setValue:heightCache forKey:newKey];
+            [tempFrameCacheDict setValue:frameCache forKey:newKey];
+            [_cacheDictionary removeObjectForKey:key];
+            [_subviewFrameCacheDict removeObjectForKey:key];
+        }
+    }
+    [_cacheDictionary addEntriesFromDictionary:tempHeightCacheDict];
+    [_subviewFrameCacheDict addEntriesFromDictionary:tempFrameCacheDict];
+}
+
+- (void)insertNewDataAtIndexPaths:(NSArray *)indexPaths
+{
+    NSMutableDictionary *sectionsdict = [NSMutableDictionary new];
+    for (NSIndexPath *indexPath in indexPaths) {
+        NSString *sectionkey = [@(indexPath.section) stringValue];
+        if (![sectionsdict objectForKey:sectionkey]) {
+            [sectionsdict setValue:[NSMutableArray new] forKey:sectionkey];
+        }
+        NSMutableArray *arr = sectionsdict[sectionkey];
+        [arr addObject:indexPath];
+    }
+    for (NSString *sectionkey in sectionsdict.allKeys) {
+        NSMutableArray *tempHeightCaches = [NSMutableArray new];
+        NSMutableArray *tempFrameCaches = [NSMutableArray new];
+        NSInteger section = [sectionkey integerValue];
+        NSInteger rowCount = [self.modelTableview numberOfRowsInSection:section];
+        if (rowCount <= 0) {
+            continue;
+        } else {
+            for (int i = 0; i < rowCount; i++) {
+                [tempHeightCaches addObject:[NSNull null]];
+                [tempFrameCaches addObject:[NSNull null]];
+            }
+        }
+        
+        for (NSString *key in _cacheDictionary.allKeys) {
+            NSArray *res = [key componentsSeparatedByString:@"-"];
+            long originalSection = [res.firstObject integerValue];
+            long row = [res.lastObject integerValue];
+            if (originalSection == section) {
+                NSNumber *heightCache = _cacheDictionary[key];
+                NSArray *frameCache = _subviewFrameCacheDict[key];
+                [tempHeightCaches setObject:heightCache atIndexedSubscript:row];
+                [tempFrameCaches setObject:frameCache atIndexedSubscript:row];
+                [_cacheDictionary removeObjectForKey:key];
+                [_subviewFrameCacheDict removeObjectForKey:key];
+            }
+        }
+        NSMutableArray *objsToInsert = [NSMutableArray new];
+        NSMutableIndexSet *indexSet = [NSMutableIndexSet new];
+        NSArray *indexPaths = sectionsdict[sectionkey];
+        [indexPaths enumerateObjectsUsingBlock:^(NSIndexPath *obj, NSUInteger idx, BOOL *stop) {
+            [objsToInsert addObject:[NSNull null]];
+            [indexSet addIndex:obj.row];
+        }];
+        [tempHeightCaches insertObjects:objsToInsert atIndexes:indexSet];
+        [tempFrameCaches insertObjects:objsToInsert atIndexes:indexSet];
+        [tempHeightCaches enumerateObjectsUsingBlock:^(NSNumber *heightCache, NSUInteger idx, BOOL *stop) {
+            if (![heightCache isKindOfClass:[NSNull class]]) {
+                NSString *key = [NSString stringWithFormat:@"%zd-%zd", section, idx];
+                [_cacheDictionary setValue:heightCache forKey:key];
+                [_subviewFrameCacheDict setValue:[tempFrameCaches objectAtIndex:idx] forKey:key];
+            }
+        }];
+    }
+}
+
+- (NSNumber *)heightCacheForIndexPath:(NSIndexPath *)indexPath
+{
+    /*
+     如果程序卡在了这里很可能是由于你用了“dequeueReusableCellWithIdentifier:forIndexPath:”方法来重用cell，换成““dequeueReusableCellWithIdentifier:”（不带IndexPath）方法即可解决
+     */
+    NSString *cacheKey = [self cacheKeyForIndexPath:indexPath];
+    return (NSNumber *)[_cacheDictionary objectForKey:cacheKey];
+}
+
+- (CGFloat)cellHeightForIndexPath:(NSIndexPath *)indexPath model:(id)model keyPath:(NSString *)keyPath
+{
+    
+    NSNumber *cacheHeight = [self heightCacheForIndexPath:indexPath];
+    if (cacheHeight) {
+        return [cacheHeight floatValue];
+    } else {
+        if (!self.modelCell) {
+            return 0;
+        }
+        
+        if (self.modelTableview && self.modelTableview != self.modelCell.sd_tableView) {
+            self.modelCell.sd_tableView = self.modelTableview;
+        }
+        self.modelCell.sd_indexPath = indexPath;
+        
+        if (model && keyPath) {
+            [self.modelCell setValue:model forKey:keyPath];
+        } else if (self.cellDataSetting) {
+            self.cellDataSetting(self.modelCell);
+        }
+        
+        
+#ifdef SDDebugWithAssert
+        /*
+         如果程序卡在了这里说明你的cell还没有调用“setupAutoHeightWithBottomView:(UIView *)bottomView bottomMargin:(CGFloat)bottomMargin”方法或者你传递的bottomView为nil，请检查并修改。例：
+         
+         //注意：bottomView不能为nil
+         [cell setupAutoHeightWithBottomView:bottomView bottomMargin:bottomMargin];
+         */
+        NSAssert(self.modelCell.sd_bottomViewsArray.count, @">>>>>> 你的cell还没有调用“setupAutoHeightWithBottomView:(UIView *)bottomView bottomMargin:(CGFloat)bottomMargin”方法或者你传递的bottomView为nil，请检查并修改");
+        
+#endif
+        
+        [self.modelCell.contentView layoutSubviews];
+        NSString *cacheKey = [self cacheKeyForIndexPath:indexPath];
+        [_cacheDictionary setObject:@(self.modelCell.autoHeight) forKey:cacheKey];
+        
+        
+        if (self.modelCell.sd_indexPath && self.modelCell.sd_tableView) {
+            if (self.modelCell.contentView.shouldReadjustFrameBeforeStoreCache) {
+                self.modelCell.contentView.height_sd = self.modelCell.autoHeight;
+                [self.modelCell.contentView layoutSubviews];
+            }
+            [self.modelCell.contentView.autoLayoutModelsArray enumerateObjectsUsingBlock:^(SDAutoLayoutModel *model, NSUInteger idx, BOOL *stop) {
+                [self.modelTableview.cellAutoHeightManager setSubviewFrameCache:model.needsAutoResizeView.frame WithIndexPath:self.modelCell.sd_indexPath];
+            }];
+        }
+        
+        
+        return self.modelCell.autoHeight;
+    }
+}
+
+- (CGFloat)cellHeightForIndexPath:(NSIndexPath *)indexPath model:(id)model keyPath:(NSString *)keyPath cellClass:(Class)cellClass
+{
+    if (![self.modelCell isKindOfClass:cellClass]) {
+        self.modelCell = nil;
+        self.modelCell = [_modelCellsDict objectForKey:NSStringFromClass(cellClass)];
+        if (!self.modelCell) {
+            [self registerCellWithCellClass:cellClass];
+        }
+        _modelCell.contentView.tag = kSDModelCellTag;
+    }
+    if (self.modelCell.contentView.width_sd != self.contentViewWidth) {
+        _modelCell.contentView.width_sd = self.contentViewWidth;
+    }
+    return [self cellHeightForIndexPath:indexPath model:model keyPath:keyPath];
+}
+
+- (void)setContentViewWidth:(CGFloat)contentViewWidth
+{
+    if (_contentViewWidth == contentViewWidth) return;
+    
+    CGFloat lastContentViewWidth = _contentViewWidth;
+    _contentViewWidth = contentViewWidth;
+    
+    self.modelCell.contentView.width_sd = self.contentViewWidth;
+    
+    if (lastContentViewWidth > 0) {
+        [_subviewFrameCacheDict removeAllObjects];
+        dispatch_async(dispatch_get_main_queue(), ^{
+            [self clearHeightCache];
+            [self.modelTableview reloadData];
+        });
+    }
+}
+
+
+- (void)setSubviewFrameCache:(CGRect)rect WithIndexPath:(NSIndexPath *)indexPath
+{
+    if (!self.subviewFrameCacheDict) {
+        self.subviewFrameCacheDict = [NSMutableDictionary new];
+    }
+    NSString *cacheKey = [self cacheKeyForIndexPath:indexPath];
+    NSMutableArray *caches = [self.subviewFrameCacheDict objectForKey:cacheKey];
+    if (!caches) {
+        caches = [NSMutableArray new];
+        [self.subviewFrameCacheDict setValue:caches forKey:cacheKey];
+    }
+    [caches addObject:[NSValue valueWithCGRect:rect]];
+}
+
+- (NSMutableArray *)subviewFrameCachesWithIndexPath:(NSIndexPath *)indexPath
+{
+    NSString *cacheKey = [self cacheKeyForIndexPath:indexPath];
+    return [self.subviewFrameCacheDict valueForKey:cacheKey];
+}
+
+@end
+
+
+@implementation UITableView (SDAutoTableViewCellHeight)
+
++ (void)load {
+    static dispatch_once_t onceToken;
+    dispatch_once(&onceToken, ^{
+        
+        NSArray *selStringsArray = @[@"reloadData", @"reloadRowsAtIndexPaths:withRowAnimation:", @"deleteRowsAtIndexPaths:withRowAnimation:", @"insertRowsAtIndexPaths:withRowAnimation:"];
+        
+        [selStringsArray enumerateObjectsUsingBlock:^(NSString *selString, NSUInteger idx, BOOL *stop) {
+            NSString *mySelString = [@"sd_" stringByAppendingString:selString];
+            
+            Method originalMethod = class_getInstanceMethod(self, NSSelectorFromString(selString));
+            Method myMethod = class_getInstanceMethod(self, NSSelectorFromString(mySelString));
+            method_exchangeImplementations(originalMethod, myMethod);
+        }];
+    });
+}
+
+- (void)sd_reloadData
+{
+    if (!self.cellAutoHeightManager.shouldKeepHeightCacheWhenReloadingData) {
+        [self.cellAutoHeightManager clearHeightCache];
+    }
+    [self sd_reloadData];
+    self.cellAutoHeightManager.shouldKeepHeightCacheWhenReloadingData = NO;
+}
+
+- (void)sd_reloadRowsAtIndexPaths:(NSArray *)indexPaths withRowAnimation:(UITableViewRowAnimation)animation
+{
+    [self.cellAutoHeightManager clearHeightCacheOfIndexPaths:indexPaths];
+    [self sd_reloadRowsAtIndexPaths:indexPaths withRowAnimation:animation];
+}
+
+- (void)sd_deleteRowsAtIndexPaths:(NSArray *)indexPaths withRowAnimation:(UITableViewRowAnimation)animation
+{
+    for (NSIndexPath *indexPath in indexPaths) {
+        [self.cellAutoHeightManager deleteThenResetHeightCache:indexPath];
+    }
+    [self sd_deleteRowsAtIndexPaths:indexPaths withRowAnimation:animation];
+}
+
+
+- (void)sd_insertRowsAtIndexPaths:(NSArray *)indexPaths withRowAnimation:(UITableViewRowAnimation)animation
+{
+    [self.cellAutoHeightManager insertNewDataAtIndexPaths:indexPaths];
+    [self sd_insertRowsAtIndexPaths:indexPaths withRowAnimation:animation];
+}
+
+/*
+ * 下一步即将实现的功能
+ 
+ - (void)sd_moveRowAtIndexPath:(NSIndexPath *)indexPath toIndexPath:(NSIndexPath *)newIndexPath
+ {
+ [self sd_moveRowAtIndexPath:indexPath toIndexPath:newIndexPath];
+ }
+ 
+ */
+
+- (CGFloat)cellHeightForIndexPath:(NSIndexPath *)indexPath model:(id)model keyPath:(NSString *)keyPath cellClass:(Class)cellClass contentViewWidth:(CGFloat)contentViewWidth
+{
+    self.cellAutoHeightManager.modelTableview = self;
+    
+    self.cellAutoHeightManager.contentViewWidth = contentViewWidth;
+    
+    return [self.cellAutoHeightManager cellHeightForIndexPath:indexPath model:model keyPath:keyPath cellClass:cellClass];
+}
+
+- (CGFloat)cellHeightForIndexPath:(NSIndexPath *)indexPath cellClass:(__unsafe_unretained Class)cellClass cellContentViewWidth:(CGFloat)width cellDataSetting:(AutoCellHeightDataSettingBlock)cellDataSetting
+{
+
+    self.cellDataSetting = cellDataSetting;
+
+    return [self cellHeightForIndexPath:indexPath model:nil keyPath:nil cellClass:cellClass contentViewWidth:width];
+}
+
+- (void)reloadDataWithExistedHeightCache
+{
+    self.cellAutoHeightManager.shouldKeepHeightCacheWhenReloadingData = YES;
+    [self reloadData];
+}
+
+- (void)reloadDataWithInsertingDataAtTheBeginingOfSection:(NSInteger)section newDataCount:(NSInteger)count
+{
+    self.cellAutoHeightManager.shouldKeepHeightCacheWhenReloadingData = YES;
+    [self.cellAutoHeightManager insertNewDataAtTheBeginingOfSection:section newDataCount:count];
+    [self reloadData];
+}
+
+- (void)reloadDataWithInsertingDataAtTheBeginingOfSections:(NSArray *)sectionNumsArray newDataCounts:(NSArray *)dataCountsArray
+{
+    self.cellAutoHeightManager.shouldKeepHeightCacheWhenReloadingData = YES;
+    [sectionNumsArray enumerateObjectsUsingBlock:^(NSNumber *num, NSUInteger idx, BOOL *stop) {
+        int section = [num intValue];
+        int dataCountForSection = [dataCountsArray[idx] intValue];
+        [self.cellAutoHeightManager insertNewDataAtTheBeginingOfSection:section newDataCount:dataCountForSection];
+    }];
+    [self reloadData];
+}
+
+- (CGFloat)cellsTotalHeight
+{
+    CGFloat h = 0;
+    if (!self.cellAutoHeightManager.heightCacheDict.count) {
+        [self reloadData];
+    }
+    NSArray *values = [self.cellAutoHeightManager.heightCacheDict allValues];
+    for (NSNumber *number in values) {
+        h += [number floatValue];
+    }
+    return h;
+}
+
+- (SDCellAutoHeightManager *)cellAutoHeightManager
+{
+    
+    SDCellAutoHeightManager *cellAutoHeightManager = objc_getAssociatedObject(self, _cmd);
+    
+    if (!cellAutoHeightManager) {
+        
+        cellAutoHeightManager = [[SDCellAutoHeightManager alloc] init];
+        
+        [self setCellAutoHeightManager:cellAutoHeightManager];
+    }
+    
+    return cellAutoHeightManager;
+}
+
+- (void)setCellAutoHeightManager:(SDCellAutoHeightManager *)cellAutoHeightManager
+{
+    objc_setAssociatedObject(self, @selector(cellAutoHeightManager), cellAutoHeightManager, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
+}
+
+- (void)setCellDataSetting:(AutoCellHeightDataSettingBlock)cellDataSetting
+{
+    self.cellAutoHeightManager.cellDataSetting = cellDataSetting;
+}
+
+- (AutoCellHeightDataSettingBlock)cellDataSetting
+{
+    return self.cellAutoHeightManager.cellDataSetting;
+}
+
+@end
+
+
+@implementation UITableViewController (SDTableViewControllerAutoCellHeight)
+
+- (CGFloat)cellHeightForIndexPath:(NSIndexPath *)indexPath cellContentViewWidth:(CGFloat)width
+{
+    return [self cellHeightForIndexPath:indexPath cellContentViewWidth:width tableView:self.tableView];
+}
+
+@end
+
+@implementation NSObject (SDAnyObjectAutoCellHeight)
+
+- (CGFloat)cellHeightForIndexPath:(NSIndexPath *)indexPath cellContentViewWidth:(CGFloat)width tableView:(UITableView *)tableView
+{
+    tableView.cellAutoHeightManager.modelTableview = tableView;
+
+    if (tableView.cellAutoHeightManager.contentViewWidth != width) {
+        tableView.cellAutoHeightManager.contentViewWidth = width;
+    }
+    if ([tableView.cellAutoHeightManager heightCacheForIndexPath:indexPath]) {
+        return [[tableView.cellAutoHeightManager heightCacheForIndexPath:indexPath] floatValue];
+    }
+    UITableViewCell *cell = [tableView.dataSource tableView:tableView cellForRowAtIndexPath:indexPath];
+    tableView.cellAutoHeightManager.modelCell = cell;
+    if (cell.contentView.width_sd != width) {
+        cell.contentView.width_sd = width;
+    }
+    return [[tableView cellAutoHeightManager] cellHeightForIndexPath:indexPath model:nil keyPath:nil];
+}
+
+@end
+