Only composit if necessary

bgayman · Feb 2, 2020 · 522e9cd · 522e9cd
1 parent 4ca7cb8
commit 522e9cd
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Sources/FlipBook/FlipBook.swift b/Sources/FlipBook/FlipBook.swift
@@ -191,8 +191,14 @@ public final class FlipBook: NSObject {
                         self.onCompletion?(.failure(error))
                         self.onCompletion = nil
                     } else {
+                        let composition: ((CALayer) ->  Void)?
+                        if self.compositionAnimation != nil  {
+                            composition = { [weak self] layer in  self?.compositionAnimation?(layer) }
+                        } else {
+                            composition = nil
+                        }
                         self.writer.endLiveCapture(assetType: self.assetType,
-                                                   compositionAnimation: { [weak self] layer in  self?.compositionAnimation?(layer) },
+                                                   compositionAnimation: composition,
                                                    progress: { [weak self] prog in DispatchQueue.main.async { self?.onProgress?(prog) }
                             }, completion: { [weak self] result in
                                 guard let self = self else {