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 {
        autoreleasepool {
            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!
            let checkImage = cImage!
            
            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 = NSColor.clear
            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 bg_r = Int(bgColor.redComponent*255);
            let bg_g = Int(bgColor.redComponent*255);
            let bg_b = Int(bgColor.redComponent*255);
            let bg_a = Int(bgColor.redComponent*255);
            
            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 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? {
        autoreleasepool {
            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
        }
    }
}