[iOS][Swift]이미지로부터 픽셀 데이터 얻어 새로운 이미지 만들기

import UIKit

let image = [#Image(imageLiteral: "2.png")#] // 이미지를 끌어놓아 사용합니다.
let pixelData = CGDataProviderCopyData(CGImageGetDataProvider(image.CGImage))	// 이미지에서를 Data 형태로 만듭니다.
let data = CFDataGetBytePtr(pixelData)	// 주소로 접근할 수 있도록 선언합니다.

// 특정 위치에 색상 값을 뽑아내는 함수입니다.
func getRGBA(pData: UnsafePointer<UInt8>, _ pixel: Int) -> UIColor {
	let red = pData[pixel]
	let green = pData[(pixel + 1)];
	let blue = pData[pixel + 2];
	let alpha = pData[pixel + 3];
	return UIColor(red: CGFloat(red)/255.0, green: CGFloat(green)/255.0, blue: CGFloat(blue)/255.0, alpha: CGFloat(alpha)/255.0)
}

// 색상 값을 찾습니다.
func findColor(pData: UnsafePointer<UInt8>) -> UIColor {
	var resultColor = UIColor.blackColor()	// 색상이 없는 경우 대비해서 검은 색으로 설정
	let width = Int(image.size.width)	// 이미지의 width 값을 얻습니다.
	for i in 0...Int(image.size.height) where resultColor == UIColor.blackColor() {
		// (0,0)에서 색상값을 찾기 위해 대각선으로 이동하며, 각 pixel은 4byte이므로 4를 곱해서 좌표를 넘겨줍니다.
		let color = getRGBA(data, (width * i + i) * 4)
		if UIColor(red: 0, green: 0, blue: 0, alpha: 0) != color {
			resultColor = color
		}
	}
	return resultColor
}

// 뽑아낸 색상값을 배경색으로 하는 이미지를 만드는 함수입니다.
func renderImage(image: UIImage, andBackgroundColor backgroundColor: UIColor) -> UIImage {
	UIGraphicsBeginImageContext(image.size)
	let context = UIGraphicsGetCurrentContext()
	let rect = CGRectMake(0, 0, image.size.width, image.size.height)	// 원본 이미지 크기와 동일하게 합니다.
	backgroundColor.setFill()	// 배경색으로 칠합니다.
	CGContextFillRect(context, rect)
	image.drawInRect(rect)	// 이미지를 그립니다.

	let maskedImage = UIGraphicsGetImageFromCurrentImageContext()	// 배경색이 추가된 이미지를 얻습니다.
	UIGraphicsEndImageContext()

	return maskedImage
}

renderImage(image, andBackgroundColor: findColor(data))