kdan_autotest_mac/KdanAutoTest/KdanAuto/Class/Tools/ImageProcess/ImageProcess.swift

//
//  ImageProcess.swift
//  KdanAuto
//
//  Created by 朱东勇 on 2023/2/7.
//

import Foundation
import AppKit

class ImageProcess : NSObject {
    
    //  Image compare
    class func compareJPEG(_ resultPath:String, checkPath:String, processCover:Bool) -> Double {
        if !FileManager.default.fileExists(atPath: resultPath) || !FileManager.default.fileExists(atPath: checkPath) {
            return -1
        }
        
        let rImage = NSImage.init(contentsOfFile: resultPath) ?? nil
        let cImage = NSImage.init(contentsOfFile: checkPath) ?? nil
        
        if nil == rImage || nil == cImage {
            return -1
        }
        
        let resultImage = rImage as! NSImage
        let checkImage = cImage as! NSImage
        
        let resultImageRep = NSBitmapImageRep.init(cgImage: resultImage.cgImage(forProposedRect: nil, context: nil, hints: nil)!)
        let checkImageRep = NSBitmapImageRep.init(cgImage: checkImage.cgImage(forProposedRect: nil, context: nil, hints: nil)!)
        
        let rWidth = resultImageRep.pixelsWide
        let rHeight = resultImageRep.pixelsHigh
        let rBitPerPixel = resultImageRep.bitsPerPixel / 8
        let rBytePerRow = resultImageRep.bytesPerRow
        
        let cWidth = checkImageRep.pixelsWide
        let cHeight = checkImageRep.pixelsHigh
        let cBitPerPixel = checkImageRep.bitsPerPixel / 8
        let cBytePerRow = checkImageRep.bytesPerRow
        
        let maxWidth = min(rWidth, cWidth) - 1
        let maxHeight = min(rHeight, cHeight) - 1
        
        // check background color
        // 挑选图片 对角斜线 上的相素进行识别
        var markInfo = NSMutableDictionary.init()
        for w in 0...maxWidth {
            let x = Int(w)
            for mark in 0...1 {
                let color = checkImageRep.colorAt(x: min(x, cWidth-1), y: min(mark == 0 ? x : (maxWidth - x), (cHeight-1))) as! NSColor
                
                if (markInfo[color] != nil) {
                    let count = (markInfo[color] as? NSNumber)!.intValue ?? 0
                    
                    markInfo[color] = NSNumber.init(value: count + 1)
                }else {
                    markInfo[color] = NSNumber.init(value: 1)
                }
            }
        }
        
        var maxCount = Int(0);
        var bgColor : NSColor? = nil
        for color in markInfo.allKeys {
            let count = (markInfo[color] as? NSNumber)!.intValue ?? 0
            
            if count > maxCount {
                maxCount = count
                bgColor = color as! NSColor
            }
        }
        
        if nil != bgColor {
            NSLog(String("识别到背景色\(bgColor)"))
        }
        
        let data = NSMutableData.init(length: cWidth * cHeight * 4)
        // Compare
        let compareDifValue = DataModel.shared.comparativeDifference()
        var equalCount = 0 as Double
        var bgCount = 0 as Double
        for w in 0...maxWidth {
            let x = Int(w)
            
            for h in 0...maxHeight {
                let y = Int(h)
                
                let cColor = checkImageRep.colorAt(x: x, y: y) as! NSColor
                let rColor = resultImageRep.colorAt(x: x, y: y) as! NSColor
                
                let cr = Int(cColor.redComponent*255)
                let cg = Int(cColor.greenComponent*255)
                let cb = Int(cColor.blueComponent*255)
                let ca = Int(cColor.blueComponent*255)
                
                let rr = Int(rColor.redComponent*255)
                let rg = Int(rColor.greenComponent*255)
                let rb = Int(rColor.blueComponent*255)
                let ra = Int(rColor.blueComponent*255)
                
//                if (cColor.isEqual(to: rColor)) {
                if (abs(cr - rr) <= compareDifValue &&
                    abs(cg - rg) <= compareDifValue &&
                    abs(cb - rb) <= compareDifValue &&
                    abs(ca - ra) <= compareDifValue) {
                    equalCount = equalCount + 1
                    
                    if bgColor != nil &&
                        cColor.isEqual(to: bgColor) {
                        bgCount += 1
                    }
                }else if (processCover && nil != data){
                    let addr = cWidth * 4 * y + x * 4
                    
                    var r = uint8(255)
                    data!.replaceBytes(in: NSRange.init(location: addr+0, length: 1), withBytes: &r)
                    var g = uint8(255 * rColor.greenComponent/2)
                    data!.replaceBytes(in: NSRange.init(location: addr+1, length: 1), withBytes: &g)
                    var b = uint8(255 * rColor.blueComponent/2)
                    data!.replaceBytes(in: NSRange.init(location: addr+2, length: 1), withBytes: &b)
                    var a = uint8(255 * rColor.alphaComponent * 0.7)
                    data!.replaceBytes(in: NSRange.init(location: addr+3, length: 1), withBytes: &a)
                }
            }
        }
        
        let outDegree = Double(max(equalCount-bgCount, 1)/(max(Double(cWidth) * Double(cHeight)-bgCount, 1)) * 100.0)
        if (abs(outDegree - 100) > 0 && processCover && nil != data) {
            DispatchQueue.global().async {
                let cfData = CFDataCreate(kCFAllocatorDefault, data?.bytes, data!.length)
                let dataProvider = CGDataProvider.init(data: cfData!)
                
                let colorSpace = CGColorSpaceCreateDeviceRGB()
                let cgImage = CGImage.init(width: cWidth,
                                           height: cHeight,
                                           bitsPerComponent: 8,
                                           bitsPerPixel: 8 * 4,
                                           bytesPerRow: cWidth * 4,
                                           space: colorSpace,
                                           bitmapInfo: CGBitmapInfo(rawValue: CGImageAlphaInfo.premultipliedLast.rawValue | CGBitmapInfo.byteOrderDefault.rawValue) ?? CGBitmapInfo.byteOrderDefault,
                                           provider: dataProvider!,
                                           decode: nil,
                                           shouldInterpolate: true,
                                           intent: CGColorRenderingIntent.defaultIntent)
                
                if nil != cgImage {
                    let coverPath = NSString(format: "%@_cover.png", NSString(string: resultPath).deletingPathExtension) as! String
                    let rep = NSBitmapImageRep.init(cgImage: cgImage!)
                    
                    if let saveData = rep.representation(using: .png, properties: [:]) {
                        let url = URL.init(fileURLWithPath: coverPath, isDirectory: false)
                        try? saveData.write(to: url)
                    }
                }
            }
        }
        
        NSLog(String("过滤点数目\(bgCount)"))
        
        return outDegree
    }
    
    //Genera Image
    class func processImage(_ resultPath:String, checkPath:String) -> NSImage? {
        if !FileManager.default.fileExists(atPath: resultPath) || !FileManager.default.fileExists(atPath: checkPath) {
            return nil
        }
        
        let rImage = NSImage.init(contentsOfFile: resultPath) ?? nil
        let cImage = NSImage.init(contentsOfFile: checkPath) ?? nil
        
        if nil == rImage || nil == cImage {
            return nil
        }
        
        let resultImage = rImage as! NSImage
        let checkImage = cImage as! NSImage
        
        let resultImageRep = NSBitmapImageRep.init(cgImage: resultImage.cgImage(forProposedRect: nil, context: nil, hints: nil)!)
        let checkImageRep = NSBitmapImageRep.init(cgImage: checkImage.cgImage(forProposedRect: nil, context: nil, hints: nil)!)
        
        let rWidth = resultImageRep.pixelsWide
        let rHeight = resultImageRep.pixelsHigh
        let rBitPerPixel = resultImageRep.bitsPerPixel / 8
        let rBytePerRow = resultImageRep.bytesPerRow
        
        let cWidth = checkImageRep.pixelsWide
        let cHeight = checkImageRep.pixelsHigh
        let cBitPerPixel = checkImageRep.bitsPerPixel / 8
        let cBytePerRow = checkImageRep.bytesPerRow
        let data = NSMutableData.init(length: cWidth * cHeight * 4)
        if nil == data {
            return nil
        }
        
        let maxWidth = min(rWidth, cWidth) - 1
        let maxHeight = min(rHeight, cHeight) - 1
        
        // check background color
        // 挑选图片 对角斜线 上的相素进行识别
        var markInfo = NSMutableDictionary.init()
        for w in 0...maxWidth {
            let x = Int(w)
            for mark in 0...1 {
                let color = checkImageRep.colorAt(x: min(x, cWidth-1), y: min(mark == 0 ? x : (maxWidth - x), (cHeight-1))) as! NSColor
                
                if (markInfo[color] != nil) {
                    let count = (markInfo[color] as? NSNumber)!.intValue ?? 0
                    
                    markInfo[color] = NSNumber.init(value: count + 1)
                }else {
                    markInfo[color] = NSNumber.init(value: 1)
                }
            }
        }
        
        var maxCount = Int(0);
        var bgColor : NSColor? = nil
        for color in markInfo.allKeys {
            let count = (markInfo[color] as? NSNumber)!.intValue ?? 0
            
            if count > maxCount {
                maxCount = count
                bgColor = color as! NSColor
            }
        }
        
        if nil != bgColor {
            NSLog(String("识别到背景色\(bgColor)"))
        }
        
        // Compare
        let compareDifValue = DataModel.shared.comparativeDifference()
        for w in 0...maxWidth {
            let x = Int(w)
            
            for h in 0...maxHeight {
                let y = Int(h)
                
                let cColor = checkImageRep.colorAt(x: x, y: y) as! NSColor
                let rColor = resultImageRep.colorAt(x: x, y: y) as! NSColor
                
                let cr = Int(cColor.redComponent*255)
                let cg = Int(cColor.greenComponent*255)
                let cb = Int(cColor.blueComponent*255)
                let ca = Int(cColor.blueComponent*255)
                
                let rr = Int(rColor.redComponent*255)
                let rg = Int(rColor.greenComponent*255)
                let rb = Int(rColor.blueComponent*255)
                let ra = Int(rColor.blueComponent*255)
                
                //                if (cColor.isEqual(to: rColor)) {
                if (abs(cr - rr) <= compareDifValue &&
                    abs(cg - rg) <= compareDifValue &&
                    abs(cb - rb) <= compareDifValue &&
                    abs(ca - ra) <= compareDifValue) {
                    if bgColor != nil &&
                        cColor.isEqual(to: bgColor) {
                    }
                }else {
//                    NSLog("(\(cr),\(cg),\(cb),\(cr))=(\(rr),\(rg),\(rb),\(rr))")
                    let addr = cWidth * 4 * y + x * 4
                    
                    var r = uint8(255)
                    data!.replaceBytes(in: NSRange.init(location: addr+0, length: 1), withBytes: &r)
                    var g = uint8(255 * rColor.greenComponent/2)
                    data!.replaceBytes(in: NSRange.init(location: addr+1, length: 1), withBytes: &g)
                    var b = uint8(255 * rColor.blueComponent/2)
                    data!.replaceBytes(in: NSRange.init(location: addr+2, length: 1), withBytes: &b)
                    var a = uint8(255 * rColor.alphaComponent * 0.7)
                    data!.replaceBytes(in: NSRange.init(location: addr+3, length: 1), withBytes: &a)
                }
            }
        }
        
        let cfData = CFDataCreate(kCFAllocatorDefault, data?.bytes, data!.length)
        let dataProvider = CGDataProvider.init(data: cfData!)
        
        let colorSpace = CGColorSpaceCreateDeviceRGB()
        let cgImage = CGImage.init(width: cWidth,
                                   height: cHeight,
                                   bitsPerComponent: 8,
                                   bitsPerPixel: 8 * 4,
                                   bytesPerRow: cWidth * 4,
                                   space: colorSpace,
                                   bitmapInfo: CGBitmapInfo(rawValue: CGImageAlphaInfo.premultipliedLast.rawValue | CGBitmapInfo.byteOrderDefault.rawValue) ?? CGBitmapInfo.byteOrderDefault,
                                   provider: dataProvider!,
                                   decode: nil,
                                   shouldInterpolate: true,
                                   intent: CGColorRenderingIntent.defaultIntent)
        
        if nil != cgImage {
            return NSImage.init(cgImage: cgImage!, size: NSSize.init(width: cWidth, height: cHeight))
        }
        
        return nil
    }
}