Swift 掩蔽图像

我有一个UIImage，我计划为其创建一个掩码，这样当用户继续注册过程时，他们会看到每个数字从非活动到活动再到完成（复选标记）。我还创建了一个“错误”标志，如果有错误，我想放置它。这有点像pageviewcontroller，但我不想使用pageviewcontroller。它们是5个圆圈，在激活或不激活时会改变颜色，完成时会变为复选标记，错误时会变为“X”。我应该使用CGImageMaskCreate还是CALayer？这就是我到目前为止所做的：

func progressImage（名称：String）->UIImage{

let stepOne: UIImage = UIImage(named: name)!
print("UIImage 1: \(stepOne.size.width) H\(stepOne.size.height)")

let maskImage: CGImage = stepOne.CGImage!

}

func maskImage（图像：UIImage，掩码：（UIImage））->UIImage{

let stepOne: UIImage = UIImage(named: name)!
print("UIImage 1: \(stepOne.size.width) H\(stepOne.size.height)")

let maskImage: CGImage = stepOne.CGImage!

let imageReference = image.CGImage
let maskReference = mask.CGImage

let imageMask = CGImageMaskCreate(CGImageGetWidth(maskReference),
CGImageGetHeight(maskReference),
CGImageGetBitsPerComponent(maskReference),
CGImageGetBitsPerPixel(maskReference),
CGImageGetBytesPerRow(maskReference),
CGImageGetDataProvider(maskReference), nil, true)

let maskedReference = CGImageCreateWithMask(imageReference, imageMask)

let maskedImage = UIImage(CGImage:maskedReference!)

return maskedImage

}

重写func viewDidLoad（）{ super.viewDidLoad（）