#import <UIKit/UIKit.h>typedef NS_ENUM(NSInteger, GradientType) { GradientFromTopToBottom = 1,   //從上到下 GradientFromLeftToRight,    //從做到右 GradientFromLeftTopToRightBottom,  //從上到下 GradientFromLeftBottomToRightTop  //從上到下};@interface UIImage (Gradient)/** * 根據給定的顏色，生成漸變色的圖片 * @param imageSize  要生成的圖片的大小 * @param colorArr   漸變顏色的數組 * @param percents   漸變顏色的占比數組 * @param gradientType  漸變色的類型 */- (UIImage *)createImageWithSize:(CGSize)imageSize gradientColors:(NSArray *)colorArr percentage:(NSArray *)percents gradientType:(GradientType)gradientType;@end#import <UIKit/UIKit.h>#import "UIImage+Gradient.h"@interface UIButton (Gradient)/** * 根據給定的顏色，設置按鈕的顏色 * @param btnSize 這里要求手動設置下生成圖片的大小，防止coder使用第三方layout,沒有設置大小 * @param clrs  漸變顏色的數組 * @param percent 漸變顏色的占比數組 * @param type  漸變色的類型 */- (UIButton *)gradientButtonWithSize:(CGSize)btnSize colorArray:(NSArray *)clrs percentageArray:(NSArray *)percent gradientType:(GradientType)type;@end