diff --git a/RETableViewManager/Cells/RETableViewInlineDatePickerCell.m b/RETableViewManager/Cells/RETableViewInlineDatePickerCell.m index 4039d1d..8290d7c 100644 --- a/RETableViewManager/Cells/RETableViewInlineDatePickerCell.m +++ b/RETableViewManager/Cells/RETableViewInlineDatePickerCell.m @@ -7,6 +7,7 @@ // #import "RETableViewInlineDatePickerCell.h" +#import "RETableViewManager.h" #import "REDateTimeItem.h" @interface RETableViewInlineDatePickerCell () @@ -44,6 +45,15 @@ self.datePicker.minuteInterval = self.item.dateTimeItem.minuteInterval; } +- (void)layoutSubviews +{ + [super layoutSubviews]; + self.datePicker.frame = self.bounds; + + if ([self.tableViewManager.delegate respondsToSelector:@selector(tableView:willLayoutCellSubviews:forRowAtIndexPath:)]) + [self.tableViewManager.delegate tableView:self.tableViewManager.tableView willLayoutCellSubviews:self forRowAtIndexPath:[self.tableViewManager.tableView indexPathForCell:self]]; +} + #pragma mark - #pragma mark Date picker value diff --git a/RETableViewManager/Cells/RETableViewInlinePickerCell.m b/RETableViewManager/Cells/RETableViewInlinePickerCell.m index 21e2f18..ade4691 100644 --- a/RETableViewManager/Cells/RETableViewInlinePickerCell.m +++ b/RETableViewManager/Cells/RETableViewInlinePickerCell.m @@ -7,6 +7,7 @@ // #import "RETableViewInlinePickerCell.h" +#import "RETableViewManager.h" #import "REPickerItem.h" @interface RETableViewInlinePickerCell () @@ -42,6 +43,15 @@ [self.pickerView reloadAllComponents]; } +- (void)layoutSubviews +{ + [super layoutSubviews]; + self.pickerView.frame = self.bounds; + + if ([self.tableViewManager.delegate respondsToSelector:@selector(tableView:willLayoutCellSubviews:forRowAtIndexPath:)]) + [self.tableViewManager.delegate tableView:self.tableViewManager.tableView willLayoutCellSubviews:self forRowAtIndexPath:[self.tableViewManager.tableView indexPathForCell:self]]; +} + #pragma mark - #pragma mark UIPickerViewDataSource