? ? ?之前在自己的項目中,遇到了這樣一需求,項目里有聊天的功能,需要自己管理好友列表,而且好友是屬于有上下級的關系,所有就要做一個樹狀的UITabelView的結構.
首先說一下大概的思路:
1.根據我自己的項目來說,首先是請求接口,獲得聯系人列表數據,當然需要根據數據建立數據模型model,這個是我項目里的數據模型:
@interface ObjModel : NSObject
@property (nonatomic,copy)NSString * ID;//這個是該對象上一級的唯一標示ID
@property (nonatomic,copy)NSString * imCode;
@property (nonatomic,copy)NSString * name;
@property (nonatomic,copy)NSString * pId;//這個是該對象的唯一標示ID
@property (nonatomic,copy)NSString * type;//對象所屬的類型
@property (nonatomic,copy)NSString * level;//用來標示是樹狀結構中的哪一級
@property (nonatomic,copy)NSString * isOpened;//這里需要構建模型的時候,將isOpened設置為NO(我是用的KVC給數據模型賦值)
@property (nonatomic,retain)NSMutableArray *subItems;//該對象包含子對象,也就是樹狀的節點
@end
2.將模型建立好,根據模型來進行分組,整理出我們想要的數據結構,具體的方法,我自己寫的很繞,但是沒有找到好的循環方法....如下:
#pragma mark --- 對對象進行分組
- (NSMutableArray * )groupWithArray:(NSMutableArray *)array{
? ? ? ? NSMutableArray * dataArray = @[].mutableCopy;
? ? ? ? ?for (ObjModel * rootItem in array) {
? ? ? ? ? ? ? ? if ([rootItem.type isEqualToString:@"firstLevel"]) {//首先根據類型找出第一級
? ? ? ? ? ? ? ? ? ? ? rootItem.level = @"0";//第一級的level
? ? ? ? ? ? ? ? ? ? ? rootItem.subItems = @[].mutableCopy;
? ? ? ? ? ? ? ? ? ? ?for (ObjModel * firstItem in array) {
? ? ? ? ? ? ? ? ? ? ? ? ? if ([firstItem.type isEqualToString:@"secondLevel"]&&[firstItem.pId isEqualToString:rootItem.ID]) {//根據類別找出第二級,如果該對象的pId和第一級對象的ID一樣,那么它是第二級
? ? ? ? ? ? ? ? ? ? ? ? firstItem.level = @"1";//第二級的level
? ? ? ? ? ? ? ? ? ? ? ? ?firstItem.subItems = @[].mutableCopy;
? ? ? ? ? ? ? ? ? ? ? ? ?[rootItem.subItems addObject:firstItem];//將第二級加到第一級中
? ? ? ? ? ? ? ? ? ? ? ? ? ? ?for (ObjModel * thirdItem in array) {
? ? ? ? ? ? ? ? ? ? ? ? ? ? if ([thirdItem.type isEqualToString:@"thirdLevel"]&&[thirdItem.pId isEqualToString:firstItem.ID]) {//同理,找出第三級
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? thirdItem.level = @"2";
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? thirdItem.subItems = @[].mutableCopy;
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? [firstItem.subItems addObject:thirdItem];
? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?}
? ? ? ? ? ? ? ? ? ? ? ? ? ?}
? ? ? ? ? ? ? ? }
? ? ? ? ?}
? ? ? ?[dataArray addObject:rootItem];//把第一級加到dataArray中,也就是tableView的數據源中,此時的數據就是包含了樹狀結構的
? ? }
}
return dataArray;
}
3.數據結構整理好,那么初始化tableView,其中tableView要設置為UITableViewStylePlain的樣式,
最終要的地方:要實現這個方法的樹狀結構,最主要的是要用到tableView的兩個方法:
- (void)insertRowsAtIndexPaths:(NSArray*)indexPaths withRowAnimation:(UITableViewRowAnimation)animation;//根據indexPath向局部插入新的數據,刷新局部數據
- (void)deleteRowsAtIndexPaths:(NSArray*)indexPaths withRowAnimation:(UITableViewRowAnimation)animation;//根據indexPath向局部刪除數據,刷新局部數據
和- (void)reloadRowsAtIndexPaths:(NSArray*)indexPaths withRowAnimation:(UITableViewRowAnimation)animation很類似
我這里就是利用到了點擊tableViewCell,判斷點擊cell所屬對象的isOpened,如果是NO,就調用- (void)insertRowsAtIndexPaths:(NSArray*)indexPaths withRowAnimation:(UITableViewRowAnimation)animation
如果是YES,就調用- (void)reloadRowsAtIndexPaths:(NSArray*)indexPaths withRowAnimation:(UITableViewRowAnimation)animation
具體實現的方法:
- (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath{
? ? ? ? [tableView deselectRowAtIndexPath:indexPath animated:YES];
? ? ? ? ObjModel * item = self.dataArray[indexPath.row];//得到該cell對應的數據對象
? ? ? if ([item.isOpened isEqualToString:@"YES"]) {?
? ? ? ? ? ? ? ? //這個是樹狀展開的狀態
? ? ? ? ? ? ? NSMutableArray * arr = @[].mutableCopy;
? ? ? ? ? ? ? arr = [self deleteMenuIndexPaths:item];//讓樹狀收起來的方法
? ? ? ? ? ? ? if ([arr count] >0) {
? ? ? ? ? ? ? [self.tableView deleteRowsAtIndexPaths: arr ? ? ? ? ? withRowAnimation:UITableViewRowAnimationMiddle];
? ? ? ? ? }
}else if([item.isOpened isEqualToString:@"NO"]){
? ? ? ? //這個是樹狀收起來的狀態
? ? ? ? ?NSMutableArray * arr = @[].mutableCopy;
? ? ? ? ?arr = [self insertMenuIndexPaths:item]; ? ?//讓樹狀展開
? ? ? ? ?if ([arr count] >0) {
? ? ? ? ? ? ? [self.tableView insertRowsAtIndexPaths:arr withRowAnimation:UITableViewRowAnimationBottom];
? ? ? ?}
? }
}
#pragma mark --- insert
- (NSMutableArray *)insertMenuIndexPaths:(ObjModel *)item{
NSMutableArray * array = @[].mutableCopy;
[self.treeItemsToInsert removeAllObjects];//treeItemsToInsert為需要插入的數據模型數組
[self insertMenuObject:item];
array = [self insertIndexsOfMenuObject:self.treeItemsToInsert];
return array;
}
- (void)insertMenuObject:(ObjModel *)item{
if (item == nil) {
return;
}
NSIndexPath * path = [NSIndexPath indexPathForRow:[self.dataArray indexOfObject:item] inSection:0];
for (int i = 0; i < item.subItems.count; i ++) {
ObjModel * childItem = item.subItems[i];
[self.dataArray insertObject:childItem atIndex:path.row + i + 1];
[self.treeItemsToInsert addObject:childItem];
item.isOpened = @"YES";
}
}
- (NSMutableArray *)insertIndexsOfMenuObject:(NSMutableArray *)array{
NSMutableArray * tempArray = @[].mutableCopy;
for (ObjModel * item in array) {
NSIndexPath * path = [NSIndexPath indexPathForRow:[self.dataArray indexOfObject:item] inSection:0];
[tempArray addObject:path];
}
return tempArray;
}
#pragma mark --- delete
- (NSMutableArray *)deleteMenuIndexPaths:(ObjModel *)item{
NSMutableArray * array = @[].mutableCopy;
[self.treeItemsToRemove removeAllObjects];
[self deleteMenuObject:item];
array = [self deleteIndexsOfMenuObject:self.treeItemsToRemove];
return array;
}
- (void)deleteMenuObject:(ObjModel *)item{
if (item == nil)
{
return ;
}
for (int i = 0; i < item.subItems.count && [item.isOpened isEqualToString:@"YES"]; i ++) {
ObjModel * childItem = item.subItems[i];
[self deleteMenuObject:childItem];
[self.treeItemsToRemove addObject:childItem];
}
item.isOpened = @"NO";
}
- (NSMutableArray *)deleteIndexsOfMenuObject:(NSMutableArray *)array{
NSMutableArray * tempArray = @[].mutableCopy;
NSMutableIndexSet * set;
set = [NSMutableIndexSet indexSet];
for (ObjModel * item in array) {
NSIndexPath * path = [NSIndexPath indexPathForRow:[self.dataArray indexOfObject:item] inSection:0];
[tempArray addObject:path];
[set addIndex:path.row];
}
[self.dataArray removeObjectsAtIndexes:set];
return tempArray;
}
這樣,就可以實現tableview的樹狀結構了.
