JKExpandTableView.m | searchcode

/ExpandTableView/JKExpandTableView.m

https://gitlab.com/gaurav1981/JKExpandTableView
Objective C | 391 lines | 290 code | 64 blank | 37 comment | 57 complexity | c34dff1fb924747ded73e571eccb7641 MD5 | raw file

//
//  JKExpandTableView.m
//  ExpandTableView
//
//  Created by Jack Kwok on 7/5/13.
//  Copyright (c) 2013 Jack Kwok. All rights reserved.
//

#import "JKExpandTableView.h"
#import "JKParentTableViewCell.h"
#import "JKMultiSelectSubTableViewCell.h"
#import "JKSingleSelectSubTableViewCell.h"

@implementation JKExpandTableView
@synthesize tableViewDelegate, expansionStates;

#define HEIGHT_FOR_CELL 44.0

- (id)initWithFrame:(CGRect)frame dataSource:dataDelegate tableViewDelegate:tableDelegate {
    self = [super initWithFrame:frame style:UITableViewStylePlain];
    //self = [super initWithFrame:frame style:UITableViewStyleGrouped];
    if (self) {
        // Initialization code
        [self initialize];
    }
    return self;
}

-(id)initWithCoder:(NSCoder *)coder {
    self = [super initWithCoder:coder];
    if (self) {
        // Initialization code
        [self initialize];
    }
    return self;
}

/* not working. override animation for insert and delete for custom animation
- (void)insertRowsAtIndexPaths:(NSArray *)indexPaths withRowAnimation:(UITableViewRowAnimation)animation
{
    for (NSIndexPath *indexPath in indexPaths)
    {
        UITableViewCell *cell = [self tableView:self cellForRowAtIndexPath:indexPath];
        [cell setFrame:CGRectMake(320, cell.frame.origin.y, cell.frame.size.width, cell.frame.size.height)];
        
        [UIView beginAnimations:NULL context:nil];
        [UIView setAnimationDuration:1];
        [cell setFrame:CGRectMake(0, cell.frame.origin.y, cell.frame.size.width, cell.frame.size.height)];
        [UIView commitAnimations];
    }
}
*/

- (void) initialize {
    [self setDataSource:self];
    [self setDelegate:self];
    self.separatorStyle = UITableViewCellSeparatorStyleSingleLine;
    // Trick to hide UITableView Empty Cell Separator Lines (stuff below last nonempty cell)
    UIView *footer = [[UIView alloc] initWithFrame:CGRectZero];
    self.tableFooterView = footer;
}

- (id) getDataSourceDelegate {
    return dataSourceDelegate;
}

- (void) setDataSourceDelegate:(id) deleg {
    dataSourceDelegate = deleg;
    [self initExpansionStates];
}

- (void) initExpansionStates
{
    // all collapsed initially
    expansionStates = [[NSMutableArray alloc] initWithCapacity:[self.dataSourceDelegate numberOfParentCells]];
    for(int i = 0; i<[self.dataSourceDelegate numberOfParentCells]; i++) {
        [expansionStates addObject:@"NO"];
    }
}

- (void) expandForParentAtRow: (NSInteger) row {
    NSUInteger parentIndex = [self parentIndexForRow:row];
    
    if ([[self.expansionStates objectAtIndex:parentIndex] boolValue]) {
        return;
    }
    // update expansionStates so backing data is ready before calling insertRowsAtIndexPaths
    [self.expansionStates replaceObjectAtIndex:parentIndex withObject:@"YES"];

    [self insertRowsAtIndexPaths:[NSArray arrayWithObject:[NSIndexPath indexPathForRow:(row + 1) inSection:0]] withRowAnimation:UITableViewRowAnimationFade];
}

- (void) collapseForParentAtRow: (NSInteger) row {
    NSUInteger parentIndex = [self parentIndexForRow:row];
    
    if (![[self.expansionStates objectAtIndex:parentIndex] boolValue]) {
        return;
    }
    // update expansionStates so backing data is ready before calling deleteRowsAtIndexPaths
    [self.expansionStates replaceObjectAtIndex:parentIndex withObject:@"NO"];
    
    [self deleteRowsAtIndexPaths:[NSArray arrayWithObject:[NSIndexPath indexPathForRow:(row + 1) inSection:0]] withRowAnimation:UITableViewRowAnimationFade];
}

- (void) animateParentCellIconExpand: (BOOL) expand forCell: (JKParentTableViewCell *) cell {
    if ([self.dataSourceDelegate respondsToSelector:@selector(shouldRotateIconForParentOnToggle)] &&
        ([self.dataSourceDelegate shouldRotateIconForParentOnToggle] == NO)) {
    } else {
        if (expand) {
            [cell rotateIconToExpanded];
        } else {
            [cell rotateIconToCollapsed];
        }
    }
}

- (NSUInteger) rowForParentIndex:(NSUInteger) parentIndex {
    NSUInteger row = 0;
    NSUInteger currentParentIndex = 0;
    
    if (parentIndex == 0) {
        return 0;
    }
    
    while (currentParentIndex < parentIndex) {
        BOOL expanded = [[self.expansionStates objectAtIndex:currentParentIndex] boolValue];
        if (expanded) {
            row++;
        }
        currentParentIndex++;
        row++;
    }
    return row;
}

- (NSUInteger) parentIndexForRow:(NSUInteger) row {
    NSUInteger parentIndex = -1;

    NSUInteger i = 0;
    
    while (i <= row) {
        parentIndex ++;
        i++;
        if ([[self.expansionStates objectAtIndex:parentIndex] boolValue]) {
            i++;
        }
    }
    NSLog(@"parentIndexForRow row: %ld parentIndex: %ld", (long)row, (long)parentIndex);
    return parentIndex;
}

- (BOOL) isExpansionCell:(NSUInteger) row {
    if (row < 1) {
        return NO;
    }
    NSUInteger parentIndex = [self parentIndexForRow:row];
    NSUInteger parentIndex2 = [self parentIndexForRow:(row-1)];
    return (parentIndex == parentIndex2);
}

#pragma mark - UITableViewDataSource
- (NSInteger)numberOfSectionsInTableView:(UITableView *)tableView
{
    // Return the number of sections.
    return 1;
}

- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section
{
    // returns sum of parent cells and expanded cells
    NSInteger parentCount = [self.dataSourceDelegate numberOfParentCells];
    NSCountedSet * countedSet = [[NSCountedSet alloc] initWithArray:self.expansionStates];
    NSUInteger expandedParentCount = [countedSet countForObject:@"YES"];
    
    return parentCount + expandedParentCount;
}

- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
{
    static NSString *CellIdentifier_Parent = @"CellReuseId_Parent";
    static NSString *CellIdentifier_MultiSelect = @"CellReuseId_MultiSelectExpand";
    static NSString *CellIdentifier_SingleSelect = @"CellReuseId_SingleSelectExpand";
    
    NSInteger row = indexPath.row;
    NSUInteger parentIndex = [self parentIndexForRow:row];
    BOOL isExpansionCell = [self isExpansionCell:row];
    
    if (isExpansionCell) {
        BOOL isMultiSelect = [self.tableViewDelegate shouldSupportMultipleSelectableChildrenAtParentIndex:parentIndex];
        if (isMultiSelect) {
            JKMultiSelectSubTableViewCell *cell = (JKMultiSelectSubTableViewCell *)[self dequeueReusableCellWithIdentifier:CellIdentifier_MultiSelect];
            if (cell == nil) {
                cell = [[JKMultiSelectSubTableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:CellIdentifier_MultiSelect];
            } else {
                NSLog(@"reusing existing JKMultiSelectSubTableViewCell");
            }
            
            if ([self.tableViewDelegate respondsToSelector:@selector(backgroundColor)]) {
                UIColor * bgColor = [self.tableViewDelegate backgroundColor];
                [cell setSubTableBackgroundColor:bgColor];
            }
            
            if ([self.tableViewDelegate respondsToSelector:@selector(foregroundColor)]) {
                UIColor * fgColor = [self.tableViewDelegate foregroundColor];
                [cell setSubTableForegroundColor:fgColor];
            }
            
            if ([self.tableViewDelegate respondsToSelector:@selector(selectionIndicatorIcon)]) {
                [cell setSelectionIndicatorImg:[self.tableViewDelegate selectionIndicatorIcon]];
            }
            
            NSLog(@"cellForRowAtIndexPath MultiSelect parentIndex: %ld", (long)parentIndex);
            [cell setParentIndex:parentIndex];
            [cell setDelegate:self];
            [cell reload];
            return cell;
        } else {
            JKSingleSelectSubTableViewCell *cell = (JKSingleSelectSubTableViewCell *)[self dequeueReusableCellWithIdentifier:CellIdentifier_SingleSelect];
            if (cell == nil) {
                cell = [[JKSingleSelectSubTableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:CellIdentifier_SingleSelect];
            } else {
                NSLog(@"reusing existing JKSingleSelectSubTableViewCell");
            }
            
            if ([self.tableViewDelegate respondsToSelector:@selector(backgroundColor)]) {
                UIColor * bgColor = [self.tableViewDelegate backgroundColor];
                [cell setSubTableBackgroundColor:bgColor];
            }
            
            if ([self.tableViewDelegate respondsToSelector:@selector(foregroundColor)]) {
                UIColor * fgColor = [self.tableViewDelegate foregroundColor];
                [cell setSubTableForegroundColor:fgColor];
            }
            
            if ([self.tableViewDelegate respondsToSelector:@selector(selectionIndicatorIcon)]) {
                [cell setSelectionIndicatorImg:[self.tableViewDelegate selectionIndicatorIcon]];
            }
            
            if ([self.tableViewDelegate respondsToSelector:@selector(fontForChildren)]) {
                UIFont *font = [self.tableViewDelegate fontForChildren];
                [cell setSubTableFont:font];
            }
            
            NSLog(@"cellForRowAtIndexPath SingleSelect parentIndex: %ld", (long)parentIndex);
            [cell setParentIndex:parentIndex];
            [cell setDelegate:self];
            [cell reload];
            return cell;
        }
    } else {
        // regular parent cell
        JKParentTableViewCell *cell = (JKParentTableViewCell *)[self dequeueReusableCellWithIdentifier:CellIdentifier_Parent];
        if (cell == nil) {
            cell = [[JKParentTableViewCell alloc] initWithReuseIdentifier:CellIdentifier_Parent];
        } else {
            NSLog(@"reusing existing JKParentTableViewCell");
        }
        
        if ([self.tableViewDelegate respondsToSelector:@selector(backgroundColor)]) {
            UIColor * bgColor = [self.tableViewDelegate backgroundColor];
            [cell setCellBackgroundColor:bgColor];
        }
        
        if ([self.tableViewDelegate respondsToSelector:@selector(foregroundColor)]) {
            UIColor * fgColor = [self.tableViewDelegate foregroundColor];
            [cell setCellForegroundColor:fgColor];
        }
        
        if ([self.tableViewDelegate respondsToSelector:@selector(selectionIndicatorIcon)]) {
            [cell setSelectionIndicatorImg:[self.tableViewDelegate selectionIndicatorIcon]];
        }
        
        if ([self.tableViewDelegate respondsToSelector:@selector(fontForParents)]) {
            UIFont * font = [self.tableViewDelegate fontForParents];
            [cell.label setFont:font];
        }
        
        NSString * labelStr = [self.dataSourceDelegate labelForParentCellAtIndex:parentIndex];
        [[cell label] setText:labelStr];
        
        if ([self.dataSourceDelegate respondsToSelector:@selector(iconForParentCellAtIndex:)]) {
            UIImage *icon = [self.dataSourceDelegate iconForParentCellAtIndex:parentIndex];
            [[cell iconImage] setImage:icon];
        }
        
        [cell setParentIndex:parentIndex];
        [cell selectionIndicatorState:[self hasSelectedChild:parentIndex]];
        //[cell setupDisplay];
        
        return cell;
    }
}

#pragma mark - Table view delegate
-(CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath {
    NSInteger row = indexPath.row;
    NSLog(@"heightForRowAtIndexPath row: %ld", (long)row);
    // if cell is expanded, the cell height would be a multiple of the number of child cells
    BOOL isExpansionCell = [self isExpansionCell:row];
    if (isExpansionCell) {
        NSInteger parentIndex = [self parentIndexForRow:row];
        NSInteger numberOfChildren = [self.dataSourceDelegate numberOfChildCellsUnderParentIndex:parentIndex];
        return HEIGHT_FOR_CELL * numberOfChildren;
    } else {
        return HEIGHT_FOR_CELL;
    }
}

- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
{
    // if parent , expand/collpase then notify delegate (always check respond to selector)
    UITableViewCell *selectedCell = [tableView cellForRowAtIndexPath:indexPath];
    if ([selectedCell isKindOfClass:[JKParentTableViewCell class]]) {
        
        JKParentTableViewCell * pCell = (JKParentTableViewCell *) selectedCell;
        
        if ([[self.expansionStates objectAtIndex:[pCell parentIndex]] boolValue]) {
            [self collapseForParentAtRow:indexPath.row];
            [self animateParentCellIconExpand:NO forCell:pCell];  // TODO handle the case where there is no child.
        } else {
            [self expandForParentAtRow:indexPath.row];
            [self animateParentCellIconExpand:YES forCell:pCell];
        }
        
        if ([self.tableViewDelegate respondsToSelector:@selector(tableView:didSelectParentCellAtIndex:)]) {
            [self.tableViewDelegate tableView:tableView didSelectParentCellAtIndex:[pCell parentIndex]];
        }
        
    } else {
        // ignore clicks on child because the sub table should handle it.
    }
}

#pragma mark - JKMultiSelectSubTableViewCellDelegate
- (NSInteger) numberOfChildrenUnderParentIndex:(NSInteger)parentIndex {
    return [self.dataSourceDelegate numberOfChildCellsUnderParentIndex:parentIndex];
}

- (BOOL) isSelectedForChildIndex:(NSInteger)childIndex underParentIndex:(NSInteger)parentIndex {
    return [self.dataSourceDelegate shouldDisplaySelectedStateForCellAtChildIndex:childIndex withinParentCellIndex:parentIndex];
}

- (void) didSelectRowAtChildIndex:(NSInteger)childIndex
                         selected:(BOOL)isSwitchedOn
                 underParentIndex:(NSInteger)parentIndex {
    
    // check if at least one child is selected.  if yes, set the parent checkmark to indicate at least one chlid selected
    
    if (isSwitchedOn &&
        [self.tableViewDelegate respondsToSelector:@selector(tableView:didSelectCellAtChildIndex:withInParentCellIndex:)]) {
        [self.tableViewDelegate tableView:self didSelectCellAtChildIndex:childIndex withInParentCellIndex:parentIndex];
    }
    
    if (!isSwitchedOn &&
        [self.tableViewDelegate respondsToSelector:@selector(tableView:didDeselectCellAtChildIndex:withInParentCellIndex:)]) {
        [self.tableViewDelegate tableView:self didDeselectCellAtChildIndex:childIndex withInParentCellIndex:parentIndex];
    }
    
    NSIndexPath * indexPath = [NSIndexPath indexPathForRow:[self rowForParentIndex:parentIndex] inSection:0];
    UITableViewCell *selectedCell = [self cellForRowAtIndexPath:indexPath];
    if ([selectedCell isKindOfClass:[JKParentTableViewCell class]]) {
        JKParentTableViewCell * pCell = (JKParentTableViewCell *) selectedCell;
        [pCell selectionIndicatorState:[self hasSelectedChild:parentIndex]];
    }
}

- (NSString *) labelForChildIndex:(NSInteger)childIndex underParentIndex:(NSInteger)parentIndex {
    return [self.dataSourceDelegate labelForCellAtChildIndex:childIndex withinParentCellIndex:parentIndex];
}

- (UIImage *) iconForChildIndex:(NSInteger)childIndex underParentIndex:(NSInteger)parentIndex {
    if ([self.dataSourceDelegate respondsToSelector:@selector(iconForCellAtChildIndex:withinParentCellIndex:)]) {
        return [self.dataSourceDelegate iconForCellAtChildIndex:childIndex withinParentCellIndex:parentIndex];
    } else {
        return nil;
    }
}

- (BOOL) hasSelectedChild:(NSUInteger) parentIndex {
    NSInteger numberOfChildren = [self.dataSourceDelegate numberOfChildCellsUnderParentIndex:parentIndex];
    BOOL result = NO;
    for (int i = 0; i < numberOfChildren ; i++) {
        if ([self.dataSourceDelegate shouldDisplaySelectedStateForCellAtChildIndex:i withinParentCellIndex:parentIndex]) {
            result = YES;
            break;
        }
    }
    return result;
}

@end