diff --git a/DTInfiniteGridView/DTInfiniteGridView.m b/DTInfiniteGridView/DTInfiniteGridView.m index b93ccab..567a772 100644 --- a/DTInfiniteGridView/DTInfiniteGridView.m +++ b/DTInfiniteGridView/DTInfiniteGridView.m @@ -13,16 +13,33 @@ @implementation DTInfiniteGridView @synthesize infiniteVerticalScrolling, infiniteHorizontalScrolling; +- (void) setup +{ + numberOfColumns = [[NSMutableDictionary alloc] init]; + self.showsHorizontalScrollIndicator = NO; + self.bounces = NO; +} + - (id)initWithFrame:(CGRect)frame { if (!(self = [super initWithFrame:frame])) return nil; - numberOfColumns = [[NSMutableDictionary alloc] init]; - self.showsHorizontalScrollIndicator = NO; - self.bounces = NO; + [self setup]; + return self; } +- (id)initWithCoder:(NSCoder *)aDecoder +{ + self = [super initWithCoder:aDecoder]; + + if (self) { + [self setup]; + } + + return self; +} + - (NSInteger)realRowNumber:(NSInteger)row { if (row >= fakeNumberOfRows) return (row % fakeNumberOfRows);