在iOS上生成渐变图像纹理

import UIKit

func generateGradientImage(size: CGSize, colors: [CGColor], locations: [NSNumber]) -> UIImage? {
    UIGraphicsBeginImageContextWithOptions(size, false, 0)
    guard let context = UIGraphicsGetCurrentContext() else {
        return nil
    }
    
    let colorSpace = CGColorSpaceCreateDeviceRGB()
    let gradient = CGGradient(colorsSpace: colorSpace, colors: colors as CFArray, locations: locations)
    
    let startPoint = CGPoint.zero
    let endPoint = CGPoint(x: size.width, y: size.height)
    
    context.drawLinearGradient(gradient!, start: startPoint, end: endPoint, options: [])
    
    let image = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    
    return image
}

// 使用示例
let size = CGSize(width: 200, height: 200)
let colors = [UIColor.red.cgColor, UIColor.blue.cgColor]
let locations: [NSNumber] = [0, 1]

if let gradientImage = generateGradientImage(size: size, colors: colors, locations: locations) {
    // 使用生成的渐变图像纹理
    // 可以将其设置为UIImageView的image属性，或者在绘制UI元素时使用
} else {
    // 生成渐变图像纹理失败
}