Swift Core Image Perspective Correction POC

skewbacca_poc.swift

import Cocoa

let context = CIContext()

let lidimg = NSImage(named: "lid")?.cgImage(forProposedRect: nil, context: nil, hints: nil)

let originalCIImage = CIImage(cgImage: lidimg!)


func perspFilter(_ input: CIImage) -> CIImage?
{
    let persFilter = CIFilter(name:"CIPerspectiveCorrection")

    persFilter!.setValue(input, forKey: "inputImage")

    persFilter!.setValue(CIVector(x: 1000, y: 2000), forKey: "inputTopLeft")
    persFilter!.setValue(CIVector(x: 3654, y: 2000), forKey: "inputTopRight")
    persFilter!.setValue(CIVector(x: 700, y: 300), forKey: "inputBottomLeft")
    persFilter!.setValue(CIVector(x: 4000, y: 250), forKey: "inputBottomRight")

    return persFilter!.outputImage
}


let sepiaCIImage = perspFilter(originalCIImage)

var rep: NSCIImageRep = NSCIImageRep(ciImage: sepiaCIImage!)
var nsImage: NSImage = NSImage(size: rep.size)
nsImage.addRepresentation(rep)