[feat]: image operations

Author: DanielCech

Sources/ScriptToolkit/FileExtension.swift

@@ -51,7 +51,7 @@ public extension File {
             throw OperationError.renameFailed(self)
         }
     }
-    
+
     ////////////////////////////////////////////////////////////////////////////////
     // MARK: - Modification date
 
@@ -151,7 +151,7 @@ public extension File {
             try FileManager.default.removeItem(atPath: newName)
         }
 
-        run(ScriptToolkit.soxPath, path, newName, "tempo", "-s", String(percent))
+        runAndDebug(ScriptToolkit.soxPath, path, newName, "tempo", "-s", String(percent))
         return try File(path: newName)
     }
 
@@ -161,7 +161,7 @@ public extension File {
             try FileManager.default.removeItem(atPath: newName)
         }
 
-        run(ScriptToolkit.ffmpegPath, "-i", path, "-sample_rate", "44100", newName.deletingPathExtension + ".wav")
+        runAndDebug(ScriptToolkit.ffmpegPath, "-i", path, "-sample_rate", "44100", newName.deletingPathExtension + ".wav")
         return try File(path: newName)
     }
 
@@ -171,7 +171,7 @@ public extension File {
             try FileManager.default.removeItem(atPath: newName)
         }
 
-        run(ScriptToolkit.soxPath, path, "-r", "44100", newName)
+        runAndDebug(ScriptToolkit.soxPath, path, "-r", "44100", "--channels", "2", newName)
         return try File(path: newName)
     }
 
@@ -181,7 +181,7 @@ public extension File {
             try FileManager.default.removeItem(atPath: newName)
         }
 
-        run(ScriptToolkit.ffmpegPath, "-i", path, "-sample_rate", "44100", newName.deletingPathExtension + ".m4a")
+        runAndDebug(ScriptToolkit.ffmpegPath, "-i", path, "-sample_rate", "44100", newName.deletingPathExtension + ".m4a")
         return try File(path: newName)
     }
 
@@ -253,6 +253,7 @@ public extension File {
         try silencedFile75.delete()
         try silencedFile90.delete()
         try silencedFile100.delete()
+        try normWavFile.delete()
     }
 
     ////////////////////////////////////////////////////////////////////////////////
@@ -276,7 +277,7 @@ public extension File {
             if !overwrite { return try File(path: newName) }
             try FileManager.default.removeItem(atPath: newName)
         }
-        
+
         let left = Int(insets.left)
         let top = Int(insets.top)
         let bottom = Int(insets.bottom)

Sources/ScriptToolkit/NSColorExtension.swift

@@ -0,0 +1,47 @@
+//
+//  NSColorExtension.swift
+//  ScriptToolkit
+//
+//  Created by Dan Cech on 05.03.2019.
+//
+
+import Foundation
+import AppKit
+
+public extension NSColor {
+    convenience init(hexString: String) {
+        let hexString = hexString.trimmingCharacters(in: .whitespacesAndNewlines)
+        let scanner = Scanner(string: hexString)
+
+        if (hexString.hasPrefix("#")) {
+            scanner.scanLocation = 1
+        }
+
+        var color: UInt32 = 0
+        scanner.scanHexInt32(&color)
+
+        let mask = 0x000000FF
+        let r = Int(color >> 16) & mask
+        let g = Int(color >> 8) & mask
+        let b = Int(color) & mask
+
+        let red   = CGFloat(r) / 255.0
+        let green = CGFloat(g) / 255.0
+        let blue  = CGFloat(b) / 255.0
+
+        self.init(red:red, green:green, blue:blue, alpha:1)
+    }
+
+    func toHexString() -> String {
+        var r:CGFloat = 0
+        var g:CGFloat = 0
+        var b:CGFloat = 0
+        var a:CGFloat = 0
+
+        getRed(&r, green: &g, blue: &b, alpha: &a)
+
+        let rgb:Int = (Int)(r*255)<<16 | (Int)(g*255)<<8 | (Int)(b*255)<<0
+
+        return String(format:"#%06x", rgb)
+    }
+}

Sources/ScriptToolkit/NSImageExtension.swift

@@ -0,0 +1,135 @@
+//
+//  NSImageExtension.swift
+//  ScriptToolkit
+//
+//  Created by Dan Cech on 05.03.2019.
+//
+
+import Foundation
+import AppKit
+
+public extension NSImage {
+
+    /// Returns the height of the current image.
+    var height: CGFloat {
+        return self.size.height
+    }
+
+    /// Returns the width of the current image.
+    var width: CGFloat {
+        return self.size.width
+    }
+
+    /// Returns a png representation of the current image.
+    var PNGRepresentation: Data? {
+        if let tiff = self.tiffRepresentation, let tiffData = NSBitmapImageRep(data: tiff) {
+            return tiffData.representation(using: .png, properties: [:])
+        }
+
+        return nil
+    }
+
+    ///  Copies the current image and resizes it to the given size.
+    ///
+    ///  - parameter size: The size of the new image.
+    ///
+    ///  - returns: The resized copy of the given image.
+    func copy(size: NSSize) throws -> NSImage {
+        // Create a new rect with given width and height
+        let frame = NSMakeRect(0, 0, size.width, size.height)
+
+        // Get the best representation for the given size.
+        guard let rep = self.bestRepresentation(for: frame, context: nil, hints: nil) else {
+            throw ScriptError.generalError(message: "Unable to resize image")
+        }
+
+        // Create an empty image with the given size.
+        let img = NSImage(size: size)
+
+        // Set the drawing context and make sure to remove the focus before returning.
+        img.lockFocus()
+        defer { img.unlockFocus() }
+
+        // Draw the new image
+        if rep.draw(in: frame) {
+            return img
+        }
+
+        // Return nil in case something went wrong.
+        throw ScriptError.generalError(message: "Unable to resize image")
+    }
+
+    ///  Saves the PNG representation of the current image to the HD.
+    ///
+    /// - parameter url: The location url to which to write the png file.
+    func savePNGRepresentationToURL(url: URL) throws {
+        if let png = self.PNGRepresentation {
+            try png.write(to: url, options: .atomicWrite)
+        }
+    }
+
+    func combine(withImage image: NSImage) throws -> NSImage {
+        guard
+            let firstImageData = image.tiffRepresentation,
+            let firstImage = CIImage(data: firstImageData),
+            let secondImageData = tiffRepresentation,
+            let secondImage = CIImage(data: secondImageData)
+            else {
+                throw ScriptError.generalError(message: "Image processing error")
+        }
+
+        let filter = CIFilter(name: "CISourceOverCompositing")!
+        filter.setDefaults()
+
+        filter.setValue(firstImage, forKey: "inputImage")
+        filter.setValue(secondImage, forKey: "inputBackgroundImage")
+
+        let resultImage = filter.outputImage
+
+        let rep = NSCIImageRep(ciImage: resultImage!)
+        let finalResult = NSImage(size: rep.size)
+        finalResult.addRepresentation(rep)
+
+        return finalResult
+    }
+
+    func image(withText text: String, attributes: [NSAttributedString.Key: Any]) -> NSImage {
+        let image = self
+        let text = text as NSString
+        let options: NSString.DrawingOptions = [.usesLineFragmentOrigin, .usesFontLeading]
+
+        let textSize = text.boundingRect(with: image.size, options: options, attributes: attributes).size
+
+        let x = (image.size.width - textSize.width) / 2
+        let y = (image.size.height - textSize.height) / 2
+        let point = NSMakePoint(x, y * 0.2)
+
+        image.lockFocus()
+        text.draw(at: point, withAttributes: attributes)
+        image.unlockFocus()
+
+        return image
+    }
+
+    func annotate(text: String, size: CGFloat, fill: NSColor, stroke: NSColor) -> NSImage {
+
+        // Solid color text
+        let fillText = image(
+            withText: text,
+            attributes: [
+                .foregroundColor: fill,
+                .font: NSFont(name: "Impact", size: size)!
+            ])
+
+        // Add strokes
+        return fillText.image(
+            withText: text,
+            attributes: [
+                .foregroundColor: fill,
+                .strokeColor: stroke,
+                .strokeWidth: 1,
+                .font: NSFont(name: "Impact", size: size)!
+            ])
+    }
+}
+