diff --git a/XLPagerTabStrip/XL/Controllers/XLButtonBarPagerTabStripViewController.m b/XLPagerTabStrip/XL/Controllers/XLButtonBarPagerTabStripViewController.m index abbc7d9..d1ef5cf 100644 --- a/XLPagerTabStrip/XL/Controllers/XLButtonBarPagerTabStripViewController.m +++ b/XLPagerTabStrip/XL/Controllers/XLButtonBarPagerTabStripViewController.m @@ -207,6 +207,15 @@ [buttonBarCell.label setText:[childController titleForPagerTabStripViewController:self]]; + if ([childController respondsToSelector:@selector(imageForPagerTabStripViewController:)]) { + UIImage *image = [childController imageForPagerTabStripViewController:self]; + buttonBarCell.imageView.image = image; + } + + if ([childController respondsToSelector:@selector(highlightedImageForPagerTabStripViewController:)]) { + UIImage *image = [childController highlightedImageForPagerTabStripViewController:self]; + buttonBarCell.imageView.highlightedImage = image; + } if (self.isProgressiveIndicator) { if (self.changeCurrentIndexProgressiveBlock) { diff --git a/XLPagerTabStrip/XL/Controllers/XLPagerTabStripViewController.h b/XLPagerTabStrip/XL/Controllers/XLPagerTabStripViewController.h index b76a1ef..0880d24 100644 --- a/XLPagerTabStrip/XL/Controllers/XLPagerTabStripViewController.h +++ b/XLPagerTabStrip/XL/Controllers/XLPagerTabStripViewController.h @@ -40,6 +40,7 @@ @optional - (UIImage *)imageForPagerTabStripViewController:(XLPagerTabStripViewController *)pagerTabStripViewController; +- (UIImage *)highlightedImageForPagerTabStripViewController:(XLPagerTabStripViewController *)pagerTabStripViewController; - (UIColor *)colorForPagerTabStripViewController:(XLPagerTabStripViewController *)pagerTabStripViewController; @end