像素陣列到UIImage Swift

Question

我一直想弄清楚如何在Swift中將rgb像素數據數組轉換爲UIImage。

我讓每個像素的RGB數據在一個簡單的結構：

public struct PixelData { 
    var a: Int 
    var r: Int 
    var g: Int 
    var b: Int 
} 

我做了我的方式，下面的函數，但由此產生的圖像是不正確：

func imageFromARGB32Bitmap(pixels:[PixelData], width: Int, height: Int)-> UIImage { 
    let rgbColorSpace = CGColorSpaceCreateDeviceRGB() 
    let bitmapInfo:CGBitmapInfo = CGBitmapInfo(CGImageAlphaInfo.PremultipliedFirst.rawValue) 
    let bitsPerComponent:Int = 8 
    let bitsPerPixel:Int = 32 

    assert(pixels.count == Int(width * height)) 

    var data = pixels // Copy to mutable [] 
    let providerRef = CGDataProviderCreateWithCFData(
     NSData(bytes: &data, length: data.count * sizeof(PixelData)) 
    ) 

    let cgim = CGImageCreate(
     width, 
     height, 
     bitsPerComponent, 
     bitsPerPixel, 
     width * Int(sizeof(PixelData)), 
     rgbColorSpace, 
     bitmapInfo, 
     providerRef, 
     nil, 
     true, 
     kCGRenderingIntentDefault 
    ) 
    return UIImage(CGImage: cgim)! 
} 

有關如何正確地將rgb數組轉換爲UIImage的技巧或指導？

Answer 1

您唯一的問題是您的PixelData結構中的數據類型需要爲UInt8。我在操場創建了一個測試圖像有以下：

public struct PixelData { 
    var a: UInt8 
    var r: UInt8 
    var g: UInt8 
    var b: UInt8 
} 

var pixels = [PixelData]() 

let red = PixelData(a: 255, r: 255, g: 0, b: 0) 
let green = PixelData(a: 255, r: 0, g: 255, b: 0) 
let blue = PixelData(a: 255, r: 0, g: 0, b: 255) 

for _ in 1...300 { 
    pixels.append(red) 
} 
for _ in 1...300 { 
    pixels.append(green) 
} 
for _ in 1...300 { 
    pixels.append(blue) 
} 

let image = imageFromARGB32Bitmap(pixels, width: 30, height: 30) 

---

更新雨燕2.2：

我更新imageFromARGB32Bitmap與雨燕2.2的工作：

func imageFromARGB32Bitmap(pixels:[PixelData], width: Int, height: Int)-> UIImage { 
    let rgbColorSpace = CGColorSpaceCreateDeviceRGB() 
    let bitmapInfo:CGBitmapInfo = CGBitmapInfo(rawValue: CGImageAlphaInfo.PremultipliedFirst.rawValue) 
    let bitsPerComponent = 8 
    let bitsPerPixel = 32 

    assert(pixels.count == width * height) 

    var data = pixels // Copy to mutable [] 
    let providerRef = CGDataProviderCreateWithCFData(
     NSData(bytes: &data, length: data.count * sizeof(PixelData)) 
    ) 

    let cgim = CGImageCreate(
     width, 
     height, 
     bitsPerComponent, 
     bitsPerPixel, 
     width * Int(sizeof(PixelData)), 
     rgbColorSpace, 
     bitmapInfo, 
     providerRef, 
     nil, 
     true, 
     .RenderingIntentDefault 
    ) 
    return UIImage(CGImage: cgim!) 
} 

Answer 2

更新for Swift 3

struct PixelData { 
    var a: UInt8 = 0 
    var r: UInt8 = 0 
    var g: UInt8 = 0 
    var b: UInt8 = 0 
} 

func imageFromBitmap(pixels: [PixelData], width: Int, height: Int) -> UIImage? { 
    assert(width > 0) 

    assert(height > 0) 

    let pixelDataSize = MemoryLayout<PixelData>.size 
    assert(pixelDataSize == 4) 

    assert(pixels.count == Int(width * height)) 

    let data: Data = pixels.withUnsafeBufferPointer { 
     return Data(buffer: $0) 
    } 

    let cfdata = NSData(data: data) as CFData 
    let provider: CGDataProvider! = CGDataProvider(data: cfdata) 
    if provider == nil { 
     print("CGDataProvider is not supposed to be nil") 
     return nil 
    } 
    let cgimage: CGImage! = CGImage(
     width: width, 
     height: height, 
     bitsPerComponent: 8, 
     bitsPerPixel: 32, 
     bytesPerRow: width * pixelDataSize, 
     space: CGColorSpaceCreateDeviceRGB(), 
     bitmapInfo: CGBitmapInfo(rawValue: CGImageAlphaInfo.premultipliedFirst.rawValue), 
     provider: provider, 
     decode: nil, 
     shouldInterpolate: true, 
     intent: .defaultIntent 
    ) 
    if cgimage == nil { 
     print("CGImage is not supposed to be nil") 
     return nil 
    } 
    return UIImage(cgImage: cgimage) 
}