Swift -如何绘制一条线，其中每个点都可以有单独的颜色

import UIKit

// 创建一个继承自UIView的自定义视图
class LineView: UIView {
    override func draw(_ rect: CGRect) {
        // 获取绘图上下文
        guard let context = UIGraphicsGetCurrentContext() else { return }
        
        // 设置线条宽度
        context.setLineWidth(2.0)
        
        // 定义线条的起始点和终止点
        let startPoint = CGPoint(x: 50, y: 50)
        let endPoint = CGPoint(x: 250, y: 50)
        
        // 定义每个点的颜色
        let colors: [UIColor] = [.red, .green, .blue, .yellow, .orange]
        
        // 绘制线条
        context.move(to: startPoint)
        for i in 0..<5 {
            // 设置当前点的颜色
            context.setStrokeColor(colors[i].cgColor)
            
            // 绘制到下一个点
            let nextPoint = CGPoint(x: startPoint.x + CGFloat(i) * 50, y: startPoint.y)
            context.addLine(to: nextPoint)
            
            // 更新起始点
            context.move(to: nextPoint)
        }
        
        // 渲染线条
        context.strokePath()
    }
}

// 创建一个视图控制器，并在其中添加自定义视图
class ViewController: UIViewController {
    override func viewDidLoad() {
        super.viewDidLoad()
        
        // 创建自定义视图
        let lineView = LineView(frame: CGRect(x: 0, y: 0, width: 300, height: 100))
        lineView.center = view.center
        
        // 将自定义视图添加到视图控制器的视图中
        view.addSubview(lineView)
    }
}

// 运行视图控制器
let viewController = ViewController()
viewController.view.frame = CGRect(x: 0, y: 0, width: 375, height: 667) // 设置视图控制器的尺寸
viewController.view.backgroundColor = .white // 设置视图控制器的背景颜色

// 在iOS应用程序中，通常使用UIWindow来承载视图控制器
let window = UIWindow(frame: UIScreen.main.bounds)
window.rootViewController = viewController
window.makeKeyAndVisible()