如何添加一个连续的正弦波掩模到一个图像iOS？

import UIKit

func addSineWaveMask(to image: UIImage, amplitude: CGFloat, frequency: CGFloat, phase: CGFloat) -> UIImage? {
    guard let cgImage = image.cgImage else {
        return nil
    }
    
    let width = cgImage.width
    let height = cgImage.height
    let colorSpace = CGColorSpaceCreateDeviceRGB()
    let bytesPerPixel = 4
    let bytesPerRow = bytesPerPixel * width
    let bitsPerComponent = 8
    let bitmapInfo = CGImageAlphaInfo.premultipliedLast.rawValue | CGBitmapInfo.byteOrder32Big.rawValue
    
    guard let context = CGContext(data: nil, width: width, height: height, bitsPerComponent: bitsPerComponent, bytesPerRow: bytesPerRow, space: colorSpace, bitmapInfo: bitmapInfo) else {
        return nil
    }
    
    context.draw(cgImage, in: CGRect(x: 0, y: 0, width: width, height: height))
    
    let buffer = context.data?.assumingMemoryBound(to: UInt8.self)
    
    for y in 0..<height {
        for x in 0..<width {
            let pixelIndex = y * width + x
            let pixelOffset = pixelIndex * bytesPerPixel
            
            let red = CGFloat(buffer?[pixelOffset + 0] ?? 0) / 255.0
            let green = CGFloat(buffer?[pixelOffset + 1] ?? 0) / 255.0
            let blue = CGFloat(buffer?[pixelOffset + 2] ?? 0) / 255.0
            
            let amplitude = amplitude * sin(2 * CGFloat.pi * frequency * CGFloat(x) / CGFloat(width) + phase)
            
            let newRed = red * amplitude
            let newGreen = green * amplitude
            let newBlue = blue * amplitude
            
            buffer?[pixelOffset + 0] = UInt8(newRed * 255)
            buffer?[pixelOffset + 1] = UInt8(newGreen * 255)
            buffer?[pixelOffset + 2] = UInt8(newBlue * 255)
        }
    }
    
    guard let newCGImage = context.makeImage() else {
        return nil
    }
    
    let newImage = UIImage(cgImage: newCGImage)
    return newImage
}

// 示例用法
let originalImage = UIImage(named: "original_image.jpg")
let maskedImage = addSineWaveMask(to: originalImage, amplitude: 0.5, frequency: 0.1, phase: 0.0)

// 将maskedImage显示在界面上或保存到文件中