PhotoHandoff/AAPLCell.m

/*
 Copyright (C) 2014 Apple Inc. All Rights Reserved.
 See LICENSE.txt for this sample’s licensing information
 
 */
 
#import "AAPLCell.h"
#import "AAPLCustomCellBackground.h"
 
@interface AAPLCell ()
@property (nonatomic, strong) UIColor *labelColor;
@end
 
 
#pragma mark -
 
@implementation AAPLCell
 
- (instancetype)initWithCoder:(NSCoder *)aDecoder {
    
    self = [super initWithCoder:aDecoder];
    if (self) {
        self.selectedBackgroundView = [[AAPLCustomCellBackground customCellBackground] init];
    }
    return self;
}
 
- (void)setSelected:(BOOL)selected {
    
    [super setSelected:selected];
    if (selected) {
        _labelColor = self.label.textColor;
        self.label.textColor = [UIColor blackColor];
        [self setNeedsDisplay];
    }
    else {
        if (self.labelColor)
            self.label.textColor = self.labelColor;
    }
}
 
- (void)setHighlighted:(BOOL)highlighted {
    
    [super setHighlighted:highlighted];
    if (highlighted) {
        _labelColor = self.label.textColor;
        self.label.textColor = [UIColor blackColor];
        [self setNeedsDisplay];
    }
    else {
        if (self.labelColor)
            self.label.textColor = self.labelColor;
    }
}
 
@end