#import <UIKit/UIKit.h>

typedef NS_ENUM(NSInteger, GradientType) {
 GradientFromTopToBottom = 1,   //從上到下
 GradientFromLeftToRight,    //從做到右
 GradientFromLeftTopToRightBottom,  //從上到下
 GradientFromLeftBottomToRightTop  //從上到下
};

@interface UIImage (Gradient)

/**
 * 根據(jù)給定的顏色，生成漸變色的圖片
 * @param imageSize  要生成的圖片的大小
 * @param colorArr   漸變顏色的數(shù)組
 * @param percents   漸變顏色的占比數(shù)組
 * @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)

/**
 * 根據(jù)給定的顏色，設置按鈕的顏色
 * @param btnSize 這里要求手動設置下生成圖片的大小，防止coder使用第三方layout,沒有設置大小
 * @param clrs  漸變顏色的數(shù)組
 * @param percent 漸變顏色的占比數(shù)組
 * @param type  漸變色的類型
 */
- (UIButton *)gradientButtonWithSize:(CGSize)btnSize colorArray:(NSArray *)clrs percentageArray:(NSArray *)percent gradientType:(GradientType)type;

@end