廢話不多說,直接上程式碼!
為了不影響系統自帶功能,所以新增一個UITabBarItem
的分類:
// .h 檔案
@interface UITabBarItem (Custom)
// 設定角標值時,替換系統的 'setBadgeValue:'方法
- (void)my_setBadgeValue:(NSString *)badgeValue;
@end
複製程式碼
// .m 檔案
@implementation UITabBarItem (Custom)
- (void)my_setBadgeValue:(NSString *)badgeValue
{
// 先設定角標值
[self setBadgeValue:badgeValue];
#if __IPHONE_OS_VERSION_MIN_REQUIRED >= __IPHONE_10_0
// 如果系統是iOS10 以上的就使用系統方法修改
UIColor *badgeColor = [UIColor blueColor];
[self setBadgeColor:badgeColor];
#else
// 這裡替換角標顏色的圖片,需要注意的時:這個圖片size=(36px,36px),圓的
UIImage *badgeImage = [UIImage imageNamed:@"blueBadge"];
[self customBadgeColorWith:badgeImage];
#endif
}
- (void)customBadgeColorWith:(UIImage *)badgeImage
{
UIView *tabBarButton = (UIView *)[self performSelector:@selector(view)];
// iOS10以下的版本 角標其實是一張圖片,所以我們一直找下去這個圖片,然後替換他
for(UIView *subview in tabBarButton.subviews) {
NSString *classString = NSStringFromClass([subview class]);
if([classString rangeOfString:@"UIBadgeView"].location != NSNotFound) {
for(UIView *badgeSubview in subview.subviews) {
NSString *badgeSubviewClassString = NSStringFromClass([badgeSubview class]);
if([badgeSubviewClassString rangeOfString:@"BadgeBackground"].location != NSNotFound) {
[badgeSubview setValue:badgeImage forKey:@"image"];
}
}
}
}
}
複製程式碼