react-native-vision-camera/example/ios/Frame Processor Plugins/Example Plugin (Swift)/ExamplePluginSwift.swift

//
//  ExamplePluginSwift.swift
//  VisionCamera
//
//  Created by Marc Rousavy on 30.04.21.
//  Copyright © 2021 mrousavy. All rights reserved.
//

import AVKit
import Vision

@objc
public class ExamplePluginSwift : FrameProcessorPlugin {
  
  override public func name() -> String! {
    return "example_plugin_swift"
  }
  
  public override func callback(_ frame: Frame!, withArguments arguments: [Any]!) -> Any! {
      guard let imageBuffer = CMSampleBufferGetImageBuffer(frame.buffer) else {
          return nil
      }
      NSLog("ExamplePlugin: \(CVPixelBufferGetWidth(imageBuffer)) x \(CVPixelBufferGetHeight(imageBuffer)) Image. Logging \(arguments.count) parameters:")

      arguments.forEach { arg in
          var string = "\(arg)"
          if let array = arg as? NSArray {
              string = (array as Array).description
          } else if let map = arg as? NSDictionary {
              string = (map as Dictionary).description
          }
          NSLog("ExamplePlugin:   -> \(string) (\(type(of: arg)))")
      }

      return [
          "example_str": "Test",
          "example_bool": true,
          "example_double": 5.3,
          "example_array": [
              "Hello",
              true,
              17.38,
          ],
      ]
  }
}