override setTextColor to adapt attribution stages properly accordin…

…g to new text color.
Austinate · Nov 11, 2014 · 28ba19c · 28ba19c
1 parent 0475dd0
commit 28ba19c
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/Classes/TOMSMorphingLabel.m b/Classes/TOMSMorphingLabel.m
@@ -99,6 +99,21 @@ - (void)numberOfAttributionStagesShouldChange
     }
 }
 
+- (void)setTextColor:(UIColor *)textColor
+{
+    [super setTextColor:textColor];
+
+    @synchronized(self) {
+        for (NSMutableDictionary *attributionStage in self.attributionStages) {
+            if ([attributionStage[NSForegroundColorAttributeName] isKindOfClass:[UIColor class]]) {
+                UIColor *attributionStageColor = attributionStage[NSForegroundColorAttributeName];
+                CGFloat attributionStageAlpha = CGColorGetAlpha(attributionStageColor.CGColor);
+                attributionStage[NSForegroundColorAttributeName] = [self textColorWithAlpha:attributionStageAlpha];
+            }
+        }
+    }
+}
+
 - (void)setTextWithoutMorphing:(NSString *)text
 {
     BOOL wasMorphingEnabled = self.isMorphingEnabled;
@@ -316,7 +331,6 @@ - (void)applyAttributionStage:(NSInteger)stage toString:(NSString *)aString
             attributionIndex = MIN(self.numberOfAttributionStages - 1, MAX(0, attributionIndex));
 
             NSMutableDictionary *attributionStage = self.attributionStages[attributionIndex];
-            attributionStage[NSForegroundColorAttributeName] = self.textColor;
             CGFloat kernFactor = [attributionStage[kTOMSKernFactorAttributeName] floatValue];
             NSString *character = [aString substringWithRange:range];
             CGSize characterSize = CGSizeZero;