Objective-C to Swift Conversion Issue on CGBitmapInfo

Question

У меня есть этот код Objective-C, который извлекает непрозрачный фон фильтра. Я пытаюсь преобразовать его в последнюю версию Swift и иметь ошибки повсюду.

-(UIImage*)removeColorFromImage:(UIImage*)sourceImage grayLevel:(int)grayLevel 
{ 
    int width = sourceImage.size.width * sourceImage.scale; 
    int height = sourceImage.size.height * sourceImage.scale; 
    CGFloat scale = sourceImage.scale; 

    CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB(); 
    CGContextRef context = CGBitmapContextCreate(NULL, width, height, 8, width * 4, colorSpace, kCGBitmapAlphaInfoMask & kCGImageAlphaPremultipliedFirst); 
    CGColorSpaceRelease(colorSpace); 

    CGContextDrawImage(context, CGRectMake(0, 0, width, height), sourceImage.CGImage); 

    unsigned int *colorData = CGBitmapContextGetData(context); 

    for (int i = 0; i < width * height; i++) 
    { 
     unsigned int color = *colorData; 

     short a = color & 0xFF; 
     short r = (color >> 8) & 0xFF; 
     short g = (color >> 16) & 0xFF; 
     short b = (color >> 24) & 0xFF; 

     if ((r == grayLevel) && (g == grayLevel) && (b == grayLevel)) 
     { 
      a = r = g = b = 0; 
      *colorData = (unsigned int)(r << 8) + ((unsigned int)(g) << 16) + ((unsigned int)(b) << 24) + ((unsigned int)(a)); 
     } 

     colorData++; 
    } 

    CGImageRef output = CGBitmapContextCreateImage(context); 
    UIImage* retImage = [UIImage imageWithCGImage:output scale:scale orientation:UIImageOrientationUp]; 

    CGImageRelease(output); 
    CGContextRelease(context); 

    return retImage; 
} 

Когда я конвертирую строку за строкой. Я получил до этого, но возникли проблемы преобразования строки:

var context = CGBitmapContextCreate(nil, width, height, 8, width * 4, colorSpace, CGBitmapInfo(CGImageAlphaInfo.PremultipliedFirst.rawValue)) 

EDIT ** Это как ошибки выглядят:

Answer 1

Я считаю, что следующее является правильным портом исходного кода :

func removeColorFromImage(sourceImage: UIImage, grayLevel: UInt32) -> UIImage? { 
    let scale = sourceImage.scale 
    let width = Int(sourceImage.size.width * scale) 
    let height = Int(sourceImage.size.height * scale) 
    let colorSpace = CGColorSpaceCreateDeviceRGB() 
    let bitmapInfo = CGBitmapInfo(rawValue: CGBitmapInfo.AlphaInfoMask.rawValue & CGImageAlphaInfo.PremultipliedFirst.rawValue) 
    let context = CGBitmapContextCreate(nil, width, height, 8, width * 4, colorSpace, bitmapInfo.rawValue) 
    CGContextDrawImage(context, CGRectMake(0, 0, CGFloat(width), CGFloat(height)), sourceImage.CGImage) 

    let uncastedData = CGBitmapContextGetData(context) 
    let colorData = UnsafeMutablePointer<UInt32>(uncastedData) 
    for i in 0..<width * height { 
     let color = colorData[i] 
     var a = color & 0xFF 
     var r = (color >> 8) & 0xFF 
     var g = (color >> 16) & 0xFF 
     var b = (color >> 24) & 0xFF 
     if ((r == grayLevel) && (g == grayLevel) && (b == grayLevel)) { 
      (a, r, g, b) = (0, 0, 0, 0) 
      colorData[i] = (r << 8) + (g << 16) + (b << 24) + a 
     } 
     colorData.advancedBy(1) 
    } 

    if let output = CGBitmapContextCreateImage(context) { 
     return UIImage(CGImage: output, scale: scale, orientation: UIImageOrientation.Up) 
    } 

    return nil 
} 

Ниже приведен пример (оригинал -> выход):

removeColorFromImage(original, grayLevel: 175) 

объясняющие изменения

С Swift 2 (я верю), вы должны использовать RawValue для последнего параметра CGBitmapContextCreate:

let context = CGBitmapContextCreate(nil, width, height, 8, width * 4, colorSpace, bitmapInfo.rawValue) 

Кроме того, в вашем порту, не использовали исходные значения из кода Objective-C:

// your new code was just using CGImageAlphaInfo.PremultipliedFirst.rawValue 
// to match, it should be: 
let bitmapInfo = CGBitmapInfo(rawValue: CGBitmapInfo.AlphaInfoMask.rawValue & CGImageAlphaInfo.PremultipliedFirst.rawValue) 

Смещение битов, которое вы делали в цикле стиля C, было недействительным. Сейчас на UInt32. Я избавилась от предупреждения контура C-стиль для вас, как хорошо :)

Наконец, вам нужно использовать для инициализации доработанный UIImage:

UIImage(CGImage: output, scale: scale, orientation: UIImageOrientation.Up) 

Функция класса вы пытаетесь использовать не действительный.