Определите цвет в UIImage и сохраните CGPoints

Question

У меня есть UIImage, и я хочу, чтобы все cgPoints определенного цвета были у этого изображения.Например, у меня есть это изображение

Теперь я хочу получить красные CGPoints из этого UIImage.Красный цвет может быть прямой горизонтальной / вертикальной линией.Это также может быть изогнутая / зигзагообразная линия.

Это то, что я пробовал, но я не могу обнаружить требуемые красные CGPoints.

цикл по размеру / пикселям изображения для определения цвета

var requiredPointsInImage = [CGPoint]()
    let testImage = UIImage.init(named: "imgToTest1")
    for heightIteration in 0..<Int(testImage!.size.height) {
        for widthIteration in 0..<Int(testImage!.size.width) {
            let colorOfPoints = testImage!.getPixelColor(pos: CGPoint(x:CGFloat(widthIteration), y:CGFloat(heightIteration)), withFrameSize: testImage!.size)
            if colorOfPoints == MYColor {
                print(colorOfPoints)
                requiredPointsInImage.append(CGPoint(x:CGFloat(widthIteration), y:CGFloat(heightIteration)))
            }

        }
    }

    let newImage = drawShapesOnImage(image: testImage!, points: requiredPointsInImage)

// Обнаружение цвета

extension UIImage {
func getPixelColor(pos: CGPoint, withFrameSize size: CGSize) -> UIColor {
    let x: CGFloat = (self.size.width) * pos.x / size.width
    let y: CGFloat = (self.size.height) * pos.y / size.height
    let pixelPoint: CGPoint = CGPoint(x: x, y: y)
    let pixelData = self.cgImage!.dataProvider!.data
    let data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)
    let pixelInfo: Int = ((Int(self.size.width) * Int(pixelPoint.y)) + Int(pixelPoint.x)) * 3 //4

    let r = CGFloat(data[pixelInfo]) / CGFloat(255.0)
    let g = CGFloat(data[pixelInfo+1]) / CGFloat(255.0)
    let b = CGFloat(data[pixelInfo+2]) / CGFloat(255.0)
//        let a = CGFloat(data[pixelInfo+3]) / CGFloat(255.0)

    return UIColor(red: r, green: g, blue: b, alpha: 1)
}
}

// Рисование на обнаруженных точках

func drawShapesOnImage(image: UIImage, points:[CGPoint]) -> UIImage {

    UIGraphicsBeginImageContext(image.size)
    image.draw(at: CGPoint.zero)
    let context = UIGraphicsGetCurrentContext()
    context!.setLineWidth(2.0)
    context!.setStrokeColor(UIColor.green.cgColor)
    let radius: CGFloat = 5.0
    for point in points  {
        let center = CGPoint(x: point.x, y: point.y);
        context!.addArc(center: CGPoint(x: center.x,y: center.y), radius: radius, startAngle: 0, endAngle: 360, clockwise: true)
        context!.strokePath()
    }
    let resultImage = UIGraphicsGetImageFromCurrentImageContext()
    UIGraphicsEndImageContext()
    return resultImage!
}

Это то, что у меня есть .....

Примечание: фон не всегда может быть белым, но, конечно, не красным.

Используя ту же функцию getPixelColor(), я могу получить точный цвет из этого изображения CGPoint ниже, но с постоянной y (скажем, y: 1)

Если вы считаете, что существует какой-либо другой лучший способ обнаружения всех этих точек, предложите.

Answer 1

Так что я нашел решение, На самом деле проблема была в том, что я игнорировал альфа-канал в расширении UIImage getPixelColor ().с помощью альфы решить мою проблему.

вот обновленный код!

extension UIImage {
func getPixelColor(pos: CGPoint, withFrameSize size: CGSize) -> UIColor {
let x: CGFloat = (self.size.width) * pos.x / size.width
let y: CGFloat = (self.size.height) * pos.y / size.height
let pixelPoint: CGPoint = CGPoint(x: x, y: y)
let pixelData = self.cgImage!.dataProvider!.data
let data: UnsafePointer<UInt8> = CFDataGetBytePtr(pixelData)
let pixelInfo: Int = ((Int(self.size.width) * Int(pixelPoint.y)) + Int(pixelPoint.x)) * 4

let r = CGFloat(data[pixelInfo]) / CGFloat(255.0)
let g = CGFloat(data[pixelInfo+1]) / CGFloat(255.0)
let b = CGFloat(data[pixelInfo+2]) / CGFloat(255.0)
let a = CGFloat(data[pixelInfo+3]) / CGFloat(255.0)

return UIColor(red: r, green: g, blue: b, alpha: a)
}
}

И результат: