Источник данных для таблицы с экспандерами

Описание: создание источника данных для электронной таблицы.

Файл «DataSource.h»

#import <NuGridView/NuGridCellCache.h>
#import <NuGridView/NuGridDataSource.h>
#import "GridRow.h"
@interface DataSource : NSObject<NuGridDataSource> {
    GridRow *columnHeader; // Заголовки столбцов
    NSMutableArray *observers; // Массив наблюдателей
    id<NuGridCellCache> cellCacheProvider; // Поставщик кэша для таблицы
}
@property(assign) NSMutableArray *data; // Массив данных
- (id) init;
// Возвращает признак того, развёрнута ли указанная строка
- (BOOL)rowIsExpanded:(NSInteger)row;
// Разворачивает указанную строку
- (void) expandRow:(NSInteger) row;
// Сворачивает указанную строку
- (void) collapseRow:(NSInteger) row;
@end

Файл «DataSource.m»

#import "DataSource.h"
#import <NuGridView/NuGridView.h>
#import <NuGridView/NuGridCell.h>
#import <NuGridView/NuGridExpandableCell.h>
#import <NuGridView/NuTextGridCell.h>
#import <NuGridView/NuGridUpdateObserver.h>
#import <NuGridView/NuGridExpandableCell.h>
@implementation DataSource
@synthesize data = _data;
- (id) init {
    self=[super init];
    if(self)
    {
        // Инциализируем массивы
        observers = [NSMutableArray new];
        _data = [NSMutableArray new];
        columnHeader = [[GridRow alloc] initWithHeader:@"Table"
        andData:[NSArray arrayWithObjects:@"A",
        @"B",
        @"C",
        @"D",
        nil]];
        // Создаём данные для таблицы
        NSMutableArray *tmp;
        for (int i=0; i<10; i++) {
            tmp = [NSMutableArray new];
            for (int j=0; j<4; j++) {
                [tmp addObject:[NSString stringWithFormat:@"%d", i + j]];
            }
            GridRow *row = [[GridRow alloc] initWithHeader:[NSString stringWithFormat:@"%d", i] andData:tmp];
            [_data addObject:row];
            [tmp release];
            if (i == 1) {
                for (int k = 0; k < 4; k++) {
                    tmp = [NSMutableArray new];
                    i++;
                    for (int l=0; l<4; l++) {
                        [tmp addObject:[NSString stringWithFormat:@"%d", i + l]];
                    }
                    GridRow *subRow = [[GridRow alloc] initWithHeader:[NSString stringWithFormat:@"%d", i] andData:tmp andParent:row];
                    if (k == 2) {
                        for (int m = 0; m < 2; m++) {
                            tmp = [NSMutableArray new];
                            i++;
                            for (int n = 0; n < 4; n++) {
                                [tmp addObject:[NSString stringWithFormat:@"%d", i + n]];
                            }
                            [[GridRow alloc] initWithHeader:[NSString stringWithFormat:@"%d", i] andData:tmp andParent:subRow];
                            [tmp release];
                        }
                    }
                }
            };
            if(i == 3) {
                for (int k = 0; k < 3; k++) {
                    tmp = [NSMutableArray new];
                    i++;
                    for (int l=0; l<4; l++)
                    [tmp addObject:[NSString stringWithFormat:@"%d", i + l]];
                    GridRow *subRow = [[GridRow alloc] initWithHeader:[NSString stringWithFormat:@"%d", i] andData:tmp andParent:row];
                    if (k == 2) {
                        for (int m = 0; m < 2; m ++) {
                            tmp = [NSMutableArray new];
                            i++;
                            for (int n=0; n < 4; n++) {
                                [tmp addObject:[NSString stringWithFormat:@"%d", i + n]];
                            }
                            [[GridRow alloc] initWithHeader:[NSString stringWithFormat:@"%d", i] andData:tmp andParent:subRow];
                            [tmp release];
                        }
                    }
                    if (k == 3) {
                        for (int m = 0; m < 4; m++) {
                            tmp = [NSMutableArray new];
                            i++;
                            for (int n=0; n < 4; n++) {
                                [tmp addObject:[NSString stringWithFormat:@"%d", i + n]];
                            }
                            [[GridRow alloc] initWithHeader:[NSString stringWithFormat:@"%d", i] andData:tmp andParent:subRow];
                            [tmp release];
                        }
                    }
                    [tmp release];
                }
            }
        }
        return self;
    }
    return nil;
}
// Возвращает ячейку по строке заголовка с установкой данных
- (NuGridCell *)gridView:(NuGridView *)gridView headerForRow:(NSInteger)row number:(NSInteger)number {
    // Создаём ячейку с идентификатором
    NuGridExpandableCell *cell =  (cellCacheProvider) ? (NuGridExpandableCell *)[cellCacheProvider dequeueReusableCellWithID:@"row_header"] : nil;
    if (!cell) {
        cell = [[[NuGridExpandableCell alloc] initWithReuseIdentifier:@"header"] autorelease];
    }
    // Получаем значение для ячейки
    NSObject *value = [self gridView:gridView valueForHeaderForRow:row number:number];
    GridRow *gridRow = [_data objectAtIndex:row];
    // Устанавливаем значение для ячейки
    cell.value = value;
    NSMutableString *text = [NSMutableString stringWithString:[gridRow title]];
    for (int i = 0; i <= [gridRow level]; i ++) {
        [text insertString:@"   " atIndex:0];
    };
    // Устанавливаем значение, отображаемое в ячейке
    cell.text = [NSString stringWithString:text];
    // Указываем номер строки
    cell.rowNumber = row;
    // Указываем номер столбца
    cell.columnNumber = number;
    // Устанавливаем индекс ячейки для столбца
    cell.columnIndex = 0;
    // Устанавливаем индекс ячейки для строки
    cell.rowIndex = -1;
    // Указываем, что ячейка не является заголовком столбца
    cell.isColumnHeader = NO;
    // Указываем, что ячейка не является заголовком строки
    cell.isRowHeader = YES;
    // Привязываем ячейку к таблице, в которой она будет находиться
    cell.parentGrid = gridView;
    return cell;
}
// Возвращает признак того, развёрнута ли указанная строка
- (BOOL)rowIsExpanded:(NSInteger)row {
    if (row < [_data count]) {
        GridRow *gridRow = [_data objectAtIndex:row];
        return [gridRow isExpanded];
    }
    else
    return NO;
}
// Возвращает ячейку по столбцу заголовка с установкой данных
- (NuGridCell *)gridView:(NuGridView *)gridView headerForColumn:(NSInteger)column number:(NSInteger)number
{
    NuTextGridCell *cell =  (cellCacheProvider) ?
    (NuTextGridCell *)[cellCacheProvider dequeueReusableCellWithID:@"column_header"] : nil;
    if (!cell) {
        cell = [[[NuTextGridCell alloc] initWithReuseIdentifier:@"column_header"] autorelease];
    }
    NSObject *value = [self gridView:gridView valueForHeaderForColumn:column number:number];
    cell.value = value;
    cell.text = [columnHeader.values objectAtIndex:column];
    cell.rowNumber = number;
    cell.columnNumber = column;
    cell.columnIndex = -1;
    cell.rowIndex = 0;
    cell.isColumnHeader = YES;
    cell.isRowHeader = NO;
    cell.parentGrid = gridView;
    return cell;
}
// Возвращает ячейку по столбцу и строке с установкой данных
- (NuGridCell *)gridView:(NuGridView *)gridView cellInRow:(NSInteger)row inColumn:(NSInteger)column
{
    NuTextGridCell *cell =  (cellCacheProvider) ?
    (NuTextGridCell *)[cellCacheProvider dequeueReusableCellWithID:@"cell"] : nil;
    if (!cell) {
        cell = [[[NuTextGridCell alloc] initWithReuseIdentifier:@"cell"] autorelease];
    }
    NSObject *value = [self gridView:gridView valueForCellInRow:row inColumn:column];
    cell.value = value;
    cell.text = [NSString stringWithFormat:@"%@", value];
    cell.rowNumber = row;
    cell.columnNumber = column;
    cell.columnIndex = 0;
    cell.rowIndex = 0;
    cell.isColumnHeader = NO;
    cell.isRowHeader = NO;
    cell.parentGrid = gridView;
    return cell;
}
// Возвращает количество строк в таблице
- (NSInteger)gridViewRowCount:(NuGridView *)gridView
{
    return [_data count];
}
// Возвращает количество столбцов в таблице
- (NSInteger)gridViewColumnCount:(NuGridView *)gridView
{
    GridRow *row = [_data objectAtIndex:0];
    return [row.values count];
}
// Определяет, есть ли в таблице строки с заголовками
- (BOOL)gridViewHasRowHeader:(NuGridView *)gridView
{
    return YES;
}
// Определяет, есть ли в таблице столбцы с заголовками
- (BOOL)gridViewHasColumnHeader:(NuGridView *)gridView
{
    return YES;
}
// Возвращает количество строк заголовка
-(NSInteger)gridViewRowHeaderCount:(NuGridView *)gridView
{
    return 1;
}
// Возвращает количество столбцов заголовка
-(NSInteger)gridViewColumnHeaderCount:(NuGridView *)gridView
{
    return 1;
}
// Возвращает значение, хранимое в заголовке строки таблицы
- (NSObject *)gridView:(NuGridView *)gridView valueForHeaderForRow:(NSInteger)row number:(NSInteger)number
{
    if (row < [_data count]) {
        GridRow *gridRow = [_data objectAtIndex:row];
        return [gridRow title];
    }
    else
    return nil;
}
// Возвращает значение, хранимое в заголовке столбца таблицы
- (NSObject *)gridView:(NuGridView *)gridView valueForHeaderForColumn:(NSInteger)column number:(NSInteger)number
{
    return [columnHeader.values objectAtIndex:column];
}
// Возвращает значение, хранимое в ячейке таблицы
- (NSObject *)gridView:(NuGridView *)gridView valueForCellInRow:(NSInteger)row inColumn:(NSInteger)column
{
    if (row < [_data count]) {
        GridRow *gridRow = [_data objectAtIndex:row];
        return [gridRow.values objectAtIndex:column];
    }
    else
    return nil;
}
// Возвращает левую верхнюю ячейку таблицы
- (NuGridCell *) gridView:(NuGridView *) gridView cornerCellInRowNumber:(NSInteger)rowNumber inColumnNumber:(NSInteger)columnNumber
{
    NuTextGridCell *cell = [[NuTextGridCell alloc] initWithReuseIdentifier:@"corner"];
    cell.value = [self gridView:gridView valueForCornerCellInRowNumber:rowNumber inColumnNumber:columnNumber];
    cell.text = [NSString stringWithFormat:@"%@", [cell value]];
    cell.rowNumber = rowNumber;
    cell.columnNumber = columnNumber;
    cell.rowIndex = -1;
    cell.columnIndex = -1;
    cell.isRowHeader = YES;
    cell.isColumnHeader = YES;
    cell.parentGrid = gridView;
    cell.alignment = UITextAlignmentRight;
    return cell;
}
// Возвращает значение, хранимое в левой верхней ячейке таблицы
- (NSObject *)gridView:(NuGridView *)gridView valueForCornerCellInRowNumber:(NSInteger)rowNumber inColumnNumber:(NSInteger)columnNumber {
    return @"Prognoz";
}
// Добавляет наблюдателя в соответствующий массив наблюдателей
- (void)addObserver:(id<NuGridUpdateObserver>)observer
{
    [observers addObject:observer];
}
// Удаляет наблюдателя из соответствующего массива наблюдателей
- (void)removeObserver:(id<NuGridUpdateObserver>)observer
{
    [observers removeObject:observer];
}
// Разворачивает указанную строку
- (void)expandRow:(NSInteger)row {
    if (row < [_data count] && [self gridView:nil parentRowForRow:row] > 0) {
        GridRow *gridRow = [_data objectAtIndex:row];
        [gridRow setIsExpanded:YES];
        for (int i = 0; i < [gridRow childrenCount]; i++) {
            [_data insertObject:[gridRow childAtIndex:i] atIndex:row + i + 1];
        }
    }
}
// Сворачивает указанную строку
- (void) collapseRow:(NSInteger)row {
    if (row < [_data count] && [self gridView:nil parentRowForRow:row] > 0) {
        GridRow *gridRow = [_data objectAtIndex:row];
        [gridRow setIsExpanded:NO];
        for (int i = 0; i < [gridRow childrenCount]; i++) {
            GridRow *subRow = [_data objectAtIndex:row + 1];
            if ([subRow isExpanded]) {
                [subRow setIsExpanded:NO];
                [self collapseRow:row + 1];
            };
            [_data removeObject:subRow];
        }
    }
}
// Возвращает номер строки, являющейся родителем для строки с указанным номером
- (NSInteger)gridView:(NuGridView *)gridView parentRowForRow:(NSInteger)rowNumber {
    if(rowNumber < [_data count]) {
        GridRow *gridRow = [_data objectAtIndex:rowNumber];
        return [gridRow childrenCount];
    } else
    return -1;
}
// Возвращает номер столбца, являющяющегося родителем для столбца с указанным номером
- (NSInteger)gridView:(NuGridView *)gridView parentColumnForColumn:(NSInteger)columnNumber{
    return -1;
}
// Установим поставщика кэша таблицы
- (void)registerCellCacheProvider:(id<NuGridCellCache>)cacheProvider {
    cellCacheProvider = cacheProvider;
}
@end

См. также:

Создание таблицы с экспандерами