feature: Frame Processors (iOS) (#2)

* Clean up Frame Processor

* Create FrameProcessorHolder

* Create FrameProcessorDelegate in ObjC++

* Move frame processor to FrameProcessorDelegate

* Decorate runtime, check for null

* Update FrameProcessorDelegate.mm

* Cleanup FrameProcessorBindings.mm

* Fix RuntimeDecorator.h import

* Update FrameProcessorDelegate.mm

* "React" -> "React Helper" to avoid confusion

* Rename folders again

* Fix podspec flattening a lot of headers, causing REA nameclash

* Fix header imports to avoid REA naming collision

* Lazily initialize jsi::Runtime on DispatchQueue

* Install frame processor bindings from Swift

* First try to call jsi::Function (frame processor) 👀

* Call viewForReactTag on RCT main thread

* Fix bridge accessing

* Add more logs

* Update CameraViewManager.swift

* Add more TODOs

* Re-indent .cpp files

* Fix RCTTurboModule import podspec

* Remove unnecessary include check for swift umbrella header

* Merge branch 'main' into frame-processors

* Docs: use static width for images (283)

* Create validate-cpp.yml

* Update a lot of packages to latest

* Set SWIFT_VERSION to 5.2 in podspec

* Create clean.sh

* Delete unused C++ files

* podspec: Remove CLANG_CXX_LANGUAGE_STANDARD and OTHER_CFLAGS

* Update pod lockfiles

* Regenerate lockfiles

* Remove IOSLogger

* Use NSLog

* Create FrameProcessorManager (inherits from REA RuntimeManager)

* Create reanimated::RuntimeManager shared_ptr

* Re-integrate pods

* Add react-native-reanimated >=2 peerDependency

* Add metro-config

* blacklist -> exclusionList

* Try to call worklet

* Fix jsi::Value* initializer

* Call ShareableValue::adapt (makeShareable) with React/JS Runtime

* Add null-checks

* Lift runtime manager creation out of delegate, into bindings

* Remove debug statement

* Make RuntimeManager unique_ptr

* Set _FRAME_PROCESSOR

* Extract convertJSIFunctionToFrameProcessorCallback

* Print frame

* Merge branch 'main' into frame-processors

* Reformat Swift code

* Install reanimated from npm again

* Re-integrate Pods

* Dependabot: Also scan example/ and docs/

* Update validate-cpp.yml

* Create FrameProcessorUtils

* Create Frame.h

* Abstract HostObject creation away

* Fix types

* Fix frame processor call

* Add todo

* Update lockfiles

* Add C++ contributing instructions

* Update CONTRIBUTING.md

* Add android/src/main/cpp to cpplint

* Update cpplint.sh

* Fix a few cpplint errors

* Fix globals

* Fix a few more cpplint errors

* Update App.tsx

* Update AndroidLogger.cpp

* Format

* Fix cpplint script (check-cpp)

* Try to simplify frame processor

* y

* Update FrameProcessorUtils.mm

* Update FrameProcessorBindings.mm

* Update CameraView.swift

* Update CameraViewManager.m

* Restructure everything

* fix

* Fix `@objc` export (make public)

* Refactor installFrameProcessorBindings into FrameProcessorRuntimeManager

* Add swift RCTBridge.runOnJS helper

* Fix run(onJS)

* Add pragma once

* Add `&self` to lambda

* Update FrameProcessorRuntimeManager.mm

* reorder imports

* Fix imports

* forward declare

* Rename extension

* Destroy buffer after execution

* Add FrameProcessorPluginRegistry base

* Merge branch 'main' into frame-processors

* Add frameProcessor to types

* Update Camera.tsx

* Fix rebase merge

* Remove movieOutput

* Use `useFrameProcessor`

* Fix bad merge

* Add additional ESLint rules

* Update lockfiles

* Update CameraViewManager.m

* Add support for V8 runtime

* Add frame processor plugins API

* Print plugin invoke

* Fix React Utils in podspec

* Fix runOnJS swift name

* Remove invalid redecl of `captureSession`

* Use REA 2.1.0 which includes all my big PRs 🎉

* Update validate-cpp.yml

* Update Podfile.lock

* Remove Flipper

* Fix dereferencing

* Capture `self` by value. Fucking hell, what a dumb mistake.

* Override a few HostObject functions

* Expose isReady, width, height, bytesPerRow and planesCount

* use hook again

* Expose property names

* FrameProcessor -> Frame

* Update CameraView+RecordVideo.swift

* Add Swift support for Frame Processors Plugins

* Add macros for plugin installation

* Add ObjC frame processor plugin

* Correctly install frame processor plugins

* Don't require custom name for macro

* Check if plugin already exists

* Implement QR Code Frame Processor Plugin in Swift

* Adjust ObjC style frame processor macro

* optimize

* Add `frameProcessorFrameDropRate`

* Fix types

* Only log once

* Log if it executes slowly

* Implement `frameProcessorFps`

* Implement manual encoded video recordings

* Use recommended video settings

* Add fileType types

* Ignore if input is not ready for media data

* Add completion handler

* Add audio buffer sampling

* Init only for video frame

* use AVAssetWriterInputPixelBufferAdaptor

* Remove AVAssetWriterInputPixelBufferAdaptor

* Rotate VideoWriter

* Always assume portrait orientation

* Update RecordingSession.swift

* Use a separate Queue for Audio

* Format Swift

* Update CameraView+RecordVideo.swift

* Use `videoQueue` instead of `cameraQueue`

* Move example plugins to example app

* Fix hardcoded name in plugin macro

* QRFrame... -> QRCodeFrame...

* Update FrameProcessorPlugin.h

* Add example frame processors to JS base

* Update QRCodeFrameProcessorPluginSwift.m

* Add docs to create FP Plugins

* Update FRAME_PROCESSORS_CREATE.mdx

* Update FRAME_PROCESSORS_CREATE.mdx

* Use `AVAssetWriterInputPixelBufferAdaptor` for efficient pixel buffer recycling

* Add customizable `pixelFormat`

* Use native format if available

* Update project.pbxproj

* Set video width and height as source-pixel-buffer attributes

* Catch

* Update App.tsx

* Don't explicitly set video dimensions, let CVPixelBufferPool handle it

* Add a few logs

* Cleanup

* Update CameraView+RecordVideo.swift

* Eagerly initialize asset writer to fix stutter at first frame

* Use `cameraQueue` DispatchQueue to not block CaptureDataOutputDelegate

* Fix duration calculation

* cleanup

* Cleanup

* Swiftformat

* Return available video codecs

* Only show frame drop notification for video output

* Remove photo and video codec functionality

It was too much complexity and probably never used anyways.

* Revert all android related changes for now

* Cleanup

* Remove unused header

* Update AVAssetWriter.Status+descriptor.swift

* Only call Frame Processor for Video Frames

* Fix `if`

* Add support for Frame Processor plugin parameters/arguments

* Fix arg support

* Move to JSIUtils.mm

* Update JSIUtils.h

* Update FRAME_PROCESSORS_CREATE.mdx

* Update FRAME_PROCESSORS_CREATE.mdx

* Upgrade packages for docs/

* fix docs

* Rename

* highlight lines

* docs

* community plugins

* Update FRAME_PROCESSOR_CREATE_FINAL.mdx

* Update FRAME_PROCESSOR_PLUGIN_LIST.mdx

* Update FRAME_PROCESSOR_PLUGIN_LIST.mdx

* Update dependencies (1/2)

* Update dependencies (2/2)

* Update Gemfile.lock

* add FP docs

* Update README.md

* Make `lastFrameProcessor` private

* add `frameProcessor` docs

* fix docs

* adjust docs

* Update DEVICES.mdx

* fix

* s

* Add logs demo

* add metro restart note

* Update FRAME_PROCESSOR_CREATE_PLUGIN_IOS.mdx

* Mirror video device

* Update AVCaptureVideoDataOutput+mirror.swift

* Create .swift-version

* Enable whole module optimization

* Fix recording mirrored video

* Swift format

* Clean dictionary on `markInvalid`

* Fix cleanup

* Add docs for disabling frame processors

* Update project.pbxproj

* Revert "Update project.pbxproj"

This reverts commit e67861e51b88b4888a6940e2d20388f3044211d0.

* Log frame drop reason

* Format

* add more samples

* Add clang-format

* also check .mm

* Revert "also check .mm"

This reverts commit 8b9d5e2c29866b05909530d104f6633d6c49eadd.

* Revert "Add clang-format"

This reverts commit 7643ac808e0fc34567ea1f814e73d84955381636.

* Use `kCVPixelFormatType_420YpCbCr8BiPlanarVideoRange` as default

* Read matching video attributes from videoSettings

* Add TODO

* Swiftformat

* Conditionally disable frame processors

* Assert if trying to use frame processors when disabled

* Add frame-processors demo gif

* Allow disabling frame processors via `VISION_CAMERA_DISABLE_FRAME_PROCESSORS`

* Update FrameProcessorRuntimeManager.mm

* Update FRAME_PROCESSORS.mdx

* Update project.pbxproj

* Update FRAME_PROCESSORS_CREATE_OVERVIEW.mdx

ios/RecordingSession.swift

@@ -0,0 +1,135 @@
+//
+//  RecordingSession.swift
+//  VisionCamera
+//
+//  Created by Marc Rousavy on 01.05.21.
+//  Copyright © 2021 Facebook. All rights reserved.
+//
+
+import AVFoundation
+import Foundation
+
+// MARK: - BufferType
+
+enum BufferType {
+  case audio
+  case video
+}
+
+// MARK: - RecordingSession
+
+class RecordingSession {
+  private let assetWriter: AVAssetWriter
+  private let audioWriter: AVAssetWriterInput
+  private let videoWriter: AVAssetWriterInput
+  private let bufferAdaptor: AVAssetWriterInputPixelBufferAdaptor
+  private let completionHandler: (AVAssetWriter.Status, Error?) -> Void
+
+  private let initialTimestamp: CMTime
+  private var latestTimestamp: CMTime?
+  private var hasWrittenFirstVideoFrame = false
+
+  var url: URL {
+    return assetWriter.outputURL
+  }
+
+  var duration: Double {
+    guard let latestTimestamp = latestTimestamp else {
+      return 0.0
+    }
+    return (latestTimestamp - initialTimestamp).seconds
+  }
+
+  init(url: URL,
+       fileType: AVFileType,
+       videoSettings: [String: Any],
+       audioSettings: [String: Any],
+       isVideoMirrored: Bool,
+       completion: @escaping (AVAssetWriter.Status, Error?) -> Void) throws {
+    do {
+      assetWriter = try AVAssetWriter(outputURL: url, fileType: fileType)
+      audioWriter = AVAssetWriterInput(mediaType: .audio, outputSettings: audioSettings)
+      videoWriter = AVAssetWriterInput(mediaType: .video, outputSettings: videoSettings)
+      completionHandler = completion
+    } catch let error as NSError {
+      throw CameraError.capture(.createRecorderError(message: error.description))
+    }
+
+    audioWriter.expectsMediaDataInRealTime = true
+    videoWriter.expectsMediaDataInRealTime = true
+    if isVideoMirrored {
+      videoWriter.transform = CGAffineTransform(rotationAngle: -(.pi / 2))
+    } else {
+      videoWriter.transform = CGAffineTransform(rotationAngle: .pi / 2)
+    }
+
+    bufferAdaptor = AVAssetWriterInputPixelBufferAdaptor(assetWriterInput: videoWriter, withVideoSettings: videoSettings)
+
+    assetWriter.add(videoWriter)
+    assetWriter.add(audioWriter)
+
+    assetWriter.startWriting()
+    initialTimestamp = CMTime(seconds: CACurrentMediaTime(), preferredTimescale: 1_000_000_000)
+    assetWriter.startSession(atSourceTime: initialTimestamp)
+    ReactLogger.log(level: .info, message: "Initialized Video and Audio AssetWriter.")
+  }
+
+  deinit {
+    if assetWriter.status == .writing {
+      ReactLogger.log(level: .info, message: "Cancelling AssetWriter...")
+      assetWriter.cancelWriting()
+    }
+  }
+
+  func appendBuffer(_ buffer: CMSampleBuffer, type bufferType: BufferType) {
+    if !CMSampleBufferDataIsReady(buffer) {
+      return
+    }
+
+    let timestamp = CMSampleBufferGetPresentationTimeStamp(buffer)
+    latestTimestamp = timestamp
+
+    switch bufferType {
+    case .video:
+      if !videoWriter.isReadyForMoreMediaData {
+        ReactLogger.log(level: .warning, message: "The Video AVAssetWriterInput was not ready for more data! Is your frame rate too high?")
+        return
+      }
+      guard let imageBuffer = CMSampleBufferGetImageBuffer(buffer) else {
+        ReactLogger.log(level: .error, message: "Failed to get the CVImageBuffer!")
+        return
+      }
+      bufferAdaptor.append(imageBuffer, withPresentationTime: timestamp)
+      if !hasWrittenFirstVideoFrame {
+        hasWrittenFirstVideoFrame = true
+        ReactLogger.log(level: .warning, message: "VideoWriter: First frame arrived \((timestamp - initialTimestamp).seconds) seconds late.")
+      }
+    case .audio:
+      if !audioWriter.isReadyForMoreMediaData {
+        return
+      }
+      if !hasWrittenFirstVideoFrame {
+        // first video frame has not been written yet, so skip this audio frame.
+        return
+      }
+      audioWriter.append(buffer)
+    }
+
+    if assetWriter.status == .failed {
+      // TODO: Should I call the completion handler or is this instance still valid?
+      ReactLogger.log(level: .error, message: "AssetWriter failed to write buffer! Error: \(assetWriter.error?.localizedDescription ?? "none")")
+    }
+  }
+
+  func finish() {
+    ReactLogger.log(level: .info, message: "Finishing Recording with AssetWriter status \"\(assetWriter.status.descriptor)\"...")
+    if assetWriter.status == .writing {
+      videoWriter.markAsFinished()
+      assetWriter.finishWriting {
+        self.completionHandler(self.assetWriter.status, self.assetWriter.error)
+      }
+    } else {
+      completionHandler(assetWriter.status, assetWriter.error)
+    }
+  }
+}