How do I export UIImage array as a movie?

后端 未结 10 1856
臣服心动
臣服心动 2020-11-22 02:11

I have a serious problem: I have an NSArray with several UIImage objects. What I now want to do, is create movie from those UIImages.

相关标签:
10条回答
  • 2020-11-22 02:35

    Here is the latest working code on iOS8 in Objective-C.

    We had to make a variety of tweaks to @Zoul's answer above to get it to work on the latest version of Xcode and iOS8. Here is our complete working code that takes an array of UIImages, makes them into a .mov file, saves it to a temp directory, then moves it to the camera roll. We assembled code from multiple different posts to get this working. We have highlighted the traps we had to solve to get the code working in our comments.

    (1) Create a collection of UIImages

    [self saveMovieToLibrary]
    
    
    - (IBAction)saveMovieToLibrary
    {
        // You just need the height and width of the video here
        // For us, our input and output video was 640 height x 480 width
        // which is what we get from the iOS front camera
        ATHSingleton *singleton = [ATHSingleton singletons];
        int height = singleton.screenHeight;
        int width = singleton.screenWidth;
    
        // You can save a .mov or a .mp4 file        
        //NSString *fileNameOut = @"temp.mp4";
        NSString *fileNameOut = @"temp.mov";
    
        // We chose to save in the tmp/ directory on the device initially
        NSString *directoryOut = @"tmp/";
        NSString *outFile = [NSString stringWithFormat:@"%@%@",directoryOut,fileNameOut];
        NSString *path = [NSHomeDirectory() stringByAppendingPathComponent:[NSString stringWithFormat:outFile]];
        NSURL *videoTempURL = [NSURL fileURLWithPath:[NSString stringWithFormat:@"%@%@", NSTemporaryDirectory(), fileNameOut]];
    
        // WARNING: AVAssetWriter does not overwrite files for us, so remove the destination file if it already exists
        NSFileManager *fileManager = [NSFileManager defaultManager];
        [fileManager removeItemAtPath:[videoTempURL path]  error:NULL];
    
    
        // Create your own array of UIImages        
        NSMutableArray *images = [NSMutableArray array];
        for (int i=0; i<singleton.numberOfScreenshots; i++)
        {
            // This was our routine that returned a UIImage. Just use your own.
            UIImage *image =[self uiimageFromCopyOfPixelBuffersUsingIndex:i];
            // We used a routine to write text onto every image 
            // so we could validate the images were actually being written when testing. This was it below. 
            image = [self writeToImage:image Text:[NSString stringWithFormat:@"%i",i ]];
            [images addObject:image];     
        }
    
    // If you just want to manually add a few images - here is code you can uncomment
    // NSString *path = [NSHomeDirectory() stringByAppendingPathComponent:[NSString stringWithFormat:@"Documents/movie.mp4"]];
    //    NSArray *images = [[NSArray alloc] initWithObjects:
    //                      [UIImage imageNamed:@"add_ar.png"],
    //                      [UIImage imageNamed:@"add_ja.png"],
    //                      [UIImage imageNamed:@"add_ru.png"],
    //                      [UIImage imageNamed:@"add_ru.png"],
    //                      [UIImage imageNamed:@"add_ar.png"],
    //                      [UIImage imageNamed:@"add_ja.png"],
    //                      [UIImage imageNamed:@"add_ru.png"],
    //                      [UIImage imageNamed:@"add_ar.png"],
    //                      [UIImage imageNamed:@"add_en.png"], nil];
    
    
    
        [self writeImageAsMovie:images toPath:path size:CGSizeMake(height, width)];
    }
    

    This is the main method that creates your AssetWriter and adds images to it for writing.

    (2) Wire up an AVAssetWriter

    -(void)writeImageAsMovie:(NSArray *)array toPath:(NSString*)path size:(CGSize)size
    {
    
        NSError *error = nil;
    
        // FIRST, start up an AVAssetWriter instance to write your video
        // Give it a destination path (for us: tmp/temp.mov)
        AVAssetWriter *videoWriter = [[AVAssetWriter alloc] initWithURL:[NSURL fileURLWithPath:path]
                                                               fileType:AVFileTypeQuickTimeMovie
                                                                  error:&error];
    
    
        NSParameterAssert(videoWriter);
    
        NSDictionary *videoSettings = [NSDictionary dictionaryWithObjectsAndKeys:
                                       AVVideoCodecH264, AVVideoCodecKey,
                                       [NSNumber numberWithInt:size.width], AVVideoWidthKey,
                                       [NSNumber numberWithInt:size.height], AVVideoHeightKey,
                                       nil];
    
        AVAssetWriterInput* writerInput = [AVAssetWriterInput assetWriterInputWithMediaType:AVMediaTypeVideo
                                                                             outputSettings:videoSettings];
    
        AVAssetWriterInputPixelBufferAdaptor *adaptor = [AVAssetWriterInputPixelBufferAdaptor assetWriterInputPixelBufferAdaptorWithAssetWriterInput:writerInput
                                                                                                                         sourcePixelBufferAttributes:nil];
        NSParameterAssert(writerInput);
        NSParameterAssert([videoWriter canAddInput:writerInput]);
        [videoWriter addInput:writerInput];
    

    (3) Start a writing Session (NOTE: the method is continuing from above)

        //Start a SESSION of writing. 
        // After you start a session, you will keep adding image frames 
        // until you are complete - then you will tell it you are done.
        [videoWriter startWriting];
        // This starts your video at time = 0
        [videoWriter startSessionAtSourceTime:kCMTimeZero];
    
        CVPixelBufferRef buffer = NULL;
    
        // This was just our utility class to get screen sizes etc.    
        ATHSingleton *singleton = [ATHSingleton singletons];
    
        int i = 0;
        while (1)
        {
            // Check if the writer is ready for more data, if not, just wait
            if(writerInput.readyForMoreMediaData){
    
                CMTime frameTime = CMTimeMake(150, 600);
                // CMTime = Value and Timescale.
                // Timescale = the number of tics per second you want
                // Value is the number of tics
                // For us - each frame we add will be 1/4th of a second
                // Apple recommend 600 tics per second for video because it is a 
                // multiple of the standard video rates 24, 30, 60 fps etc.
                CMTime lastTime=CMTimeMake(i*150, 600);
                CMTime presentTime=CMTimeAdd(lastTime, frameTime);
    
                if (i == 0) {presentTime = CMTimeMake(0, 600);} 
                // This ensures the first frame starts at 0.
    
    
                if (i >= [array count])
                {
                    buffer = NULL;
                }
                else
                {
                    // This command grabs the next UIImage and converts it to a CGImage
                    buffer = [self pixelBufferFromCGImage:[[array objectAtIndex:i] CGImage]];
                }
    
    
                if (buffer)
                {
                    // Give the CGImage to the AVAssetWriter to add to your video
                    [adaptor appendPixelBuffer:buffer withPresentationTime:presentTime];
                    i++;
                }
                else
                {
    

    (4) Finish the Session (Note: Method continues from above)

                    //Finish the session:
                    // This is important to be done exactly in this order
                    [writerInput markAsFinished];
                    // WARNING: finishWriting in the solution above is deprecated. 
                    // You now need to give a completion handler.
                    [videoWriter finishWritingWithCompletionHandler:^{
                        NSLog(@"Finished writing...checking completion status...");
                        if (videoWriter.status != AVAssetWriterStatusFailed && videoWriter.status == AVAssetWriterStatusCompleted)
                        {
                            NSLog(@"Video writing succeeded.");
    
                            // Move video to camera roll
                            // NOTE: You cannot write directly to the camera roll. 
                            // You must first write to an iOS directory then move it!
                            NSURL *videoTempURL = [NSURL fileURLWithPath:[NSString stringWithFormat:@"%@", path]];
                            [self saveToCameraRoll:videoTempURL];
    
                        } else
                        {
                            NSLog(@"Video writing failed: %@", videoWriter.error);
                        }
    
                    }]; // end videoWriter finishWriting Block
    
                    CVPixelBufferPoolRelease(adaptor.pixelBufferPool);
    
                    NSLog (@"Done");
                    break;
                }
            }
        }    
    }
    

    (5) Convert your UIImages to a CVPixelBufferRef
    This method will give you a CV pixel buffer reference which is needed by the AssetWriter. This is obtained from a CGImageRef which you get from your UIImage (above).

    - (CVPixelBufferRef) pixelBufferFromCGImage: (CGImageRef) image
    {
        // This again was just our utility class for the height & width of the
        // incoming video (640 height x 480 width)
        ATHSingleton *singleton = [ATHSingleton singletons];
        int height = singleton.screenHeight;
        int width = singleton.screenWidth;
    
        NSDictionary *options = [NSDictionary dictionaryWithObjectsAndKeys:
                                 [NSNumber numberWithBool:YES], kCVPixelBufferCGImageCompatibilityKey,
                                 [NSNumber numberWithBool:YES], kCVPixelBufferCGBitmapContextCompatibilityKey,
                                 nil];
        CVPixelBufferRef pxbuffer = NULL;
    
        CVReturn status = CVPixelBufferCreate(kCFAllocatorDefault, width,
                                              height, kCVPixelFormatType_32ARGB, (__bridge CFDictionaryRef) options,
                                              &pxbuffer);
    
        NSParameterAssert(status == kCVReturnSuccess && pxbuffer != NULL);
    
        CVPixelBufferLockBaseAddress(pxbuffer, 0);
        void *pxdata = CVPixelBufferGetBaseAddress(pxbuffer);
        NSParameterAssert(pxdata != NULL);
    
        CGColorSpaceRef rgbColorSpace = CGColorSpaceCreateDeviceRGB();
    
        CGContextRef context = CGBitmapContextCreate(pxdata, width,
                                                     height, 8, 4*width, rgbColorSpace,
                                                     kCGImageAlphaNoneSkipFirst);
        NSParameterAssert(context);
        CGContextConcatCTM(context, CGAffineTransformMakeRotation(0));
        CGContextDrawImage(context, CGRectMake(0, 0, CGImageGetWidth(image),
                                               CGImageGetHeight(image)), image);
        CGColorSpaceRelease(rgbColorSpace);
        CGContextRelease(context);
    
        CVPixelBufferUnlockBaseAddress(pxbuffer, 0);
    
        return pxbuffer;
    }
    

    (6) Move Your Video to the Camera Roll Because AVAssetWriter cannot write directly to the camera roll, this moves the video from "tmp/temp.mov" (or whatever filename you named it above) to the camera roll.

    - (void) saveToCameraRoll:(NSURL *)srcURL
    {
        NSLog(@"srcURL: %@", srcURL);
    
        ALAssetsLibrary *library = [[ALAssetsLibrary alloc] init];
        ALAssetsLibraryWriteVideoCompletionBlock videoWriteCompletionBlock =
        ^(NSURL *newURL, NSError *error) {
            if (error) {
                NSLog( @"Error writing image with metadata to Photo Library: %@", error );
            } else {
                NSLog( @"Wrote image with metadata to Photo Library %@", newURL.absoluteString);
            }
        };
    
        if ([library videoAtPathIsCompatibleWithSavedPhotosAlbum:srcURL])
        {
            [library writeVideoAtPathToSavedPhotosAlbum:srcURL
                                        completionBlock:videoWriteCompletionBlock];
        }
    }
    

    Zoul's answer above gives a nice outline of what you will be doing. We extensively commented this code so you can then see how it was done using working code.

    0 讨论(0)
  • 2020-11-22 02:38

    For those still doing the journey in 2020, and getting distortion in their movies because its not width 16px

    change

    CGContextRef context = CGBitmapContextCreate(pxdata,
                                                 width, height,
                                                 8, 4 * width,
                                                 rgbColorSpace,
                                                 kCGImageAlphaNoneSkipFirst);
    

    to

    CGContextRef context = CGBitmapContextCreate(pxdata,
                                                 width, height,
                                                 8, CVPixelBufferGetBytesPerRow(pxbuffer),
                                                 rgbColorSpace,
                                                 kCGImageAlphaNoneSkipFirst);
    

    Credit to @bluedays Output from AVAssetWriter (UIImages written to video) distorted

    0 讨论(0)
  • 2020-11-22 02:43

    NOTE: This is a Swift 2.1 solution (iOS8+, XCode 7.2).

    Last week I set out to write the iOS code to generate a video from images. I had a little bit of AVFoundation experience, but had never even heard of a CVPixelBuffer. I came across the answers on this page and also here. It took several days to dissect everything and put it all back together in Swift in a way that made sense to my brain. Below is what I came up with.

    NOTE: If you copy/paste all the code below into a single Swift file, it should compile. You'll just need to tweak loadImages() and the RenderSettings values.

    Part 1: Setting things up

    Here I group all the export-related settings into a single RenderSettings struct.

    import AVFoundation
    import UIKit
    import Photos
    
    struct RenderSettings {
    
        var width: CGFloat = 1280
        var height: CGFloat = 720
        var fps: Int32 = 2   // 2 frames per second
        var avCodecKey = AVVideoCodecH264
        var videoFilename = "render"
        var videoFilenameExt = "mp4"
    
        var size: CGSize {
            return CGSize(width: width, height: height)
        }
    
        var outputURL: NSURL {
            // Use the CachesDirectory so the rendered video file sticks around as long as we need it to.
            // Using the CachesDirectory ensures the file won't be included in a backup of the app.
            let fileManager = NSFileManager.defaultManager()
            if let tmpDirURL = try? fileManager.URLForDirectory(.CachesDirectory, inDomain: .UserDomainMask, appropriateForURL: nil, create: true) {
                return tmpDirURL.URLByAppendingPathComponent(videoFilename).URLByAppendingPathExtension(videoFilenameExt)
            }
            fatalError("URLForDirectory() failed")
        }
    }
    

    Part 2: The ImageAnimator

    The ImageAnimator class knows about your images and uses the VideoWriter class to perform the rendering. The idea is to keep the video content code separate from the low-level AVFoundation code. I also added saveToLibrary() here as a class function which gets called at the end of the chain to save the video to the Photo Library.

    class ImageAnimator {
    
        // Apple suggests a timescale of 600 because it's a multiple of standard video rates 24, 25, 30, 60 fps etc.
        static let kTimescale: Int32 = 600
    
        let settings: RenderSettings
        let videoWriter: VideoWriter
        var images: [UIImage]!
    
        var frameNum = 0
    
        class func saveToLibrary(videoURL: NSURL) {
            PHPhotoLibrary.requestAuthorization { status in
                guard status == .Authorized else { return }
    
                PHPhotoLibrary.sharedPhotoLibrary().performChanges({
                    PHAssetChangeRequest.creationRequestForAssetFromVideoAtFileURL(videoURL)
                    }) { success, error in
                        if !success {
                            print("Could not save video to photo library:", error)
                        }
                }
            }
        }
    
        class func removeFileAtURL(fileURL: NSURL) {
            do {
                try NSFileManager.defaultManager().removeItemAtPath(fileURL.path!)
            }
            catch _ as NSError {
                // Assume file doesn't exist.
            }
        }
    
        init(renderSettings: RenderSettings) {
            settings = renderSettings
            videoWriter = VideoWriter(renderSettings: settings)
            images = loadImages()
        }
    
        func render(completion: ()->Void) {
    
            // The VideoWriter will fail if a file exists at the URL, so clear it out first.
            ImageAnimator.removeFileAtURL(settings.outputURL)
    
            videoWriter.start()
            videoWriter.render(appendPixelBuffers) {
                ImageAnimator.saveToLibrary(self.settings.outputURL)
                completion()
            }
    
        }
    
        // Replace this logic with your own.
        func loadImages() -> [UIImage] {
            var images = [UIImage]()
            for index in 1...10 {
                let filename = "\(index).jpg"
                images.append(UIImage(named: filename)!)
            }
            return images
        }
    
        // This is the callback function for VideoWriter.render()
        func appendPixelBuffers(writer: VideoWriter) -> Bool {
    
            let frameDuration = CMTimeMake(Int64(ImageAnimator.kTimescale / settings.fps), ImageAnimator.kTimescale)
    
            while !images.isEmpty {
    
                if writer.isReadyForData == false {
                    // Inform writer we have more buffers to write.
                    return false
                }
    
                let image = images.removeFirst()
                let presentationTime = CMTimeMultiply(frameDuration, Int32(frameNum))
                let success = videoWriter.addImage(image, withPresentationTime: presentationTime)
                if success == false {
                    fatalError("addImage() failed")
                }
    
                frameNum++
            }
    
            // Inform writer all buffers have been written.
            return true
        }
    
    }
    

    Part 3: The VideoWriter

    The VideoWriter class does all AVFoundation heavy lifting. It's mostly a wrapper around AVAssetWriter and AVAssetWriterInput. It also contains fancy code written by not me that knows how to translate an image into a CVPixelBuffer.

    class VideoWriter {
    
        let renderSettings: RenderSettings
    
        var videoWriter: AVAssetWriter!
        var videoWriterInput: AVAssetWriterInput!
        var pixelBufferAdaptor: AVAssetWriterInputPixelBufferAdaptor!
    
        var isReadyForData: Bool {
            return videoWriterInput?.readyForMoreMediaData ?? false
        }
    
        class func pixelBufferFromImage(image: UIImage, pixelBufferPool: CVPixelBufferPool, size: CGSize) -> CVPixelBuffer {
    
            var pixelBufferOut: CVPixelBuffer?
    
            let status = CVPixelBufferPoolCreatePixelBuffer(kCFAllocatorDefault, pixelBufferPool, &pixelBufferOut)
            if status != kCVReturnSuccess {
                fatalError("CVPixelBufferPoolCreatePixelBuffer() failed")
            }
    
            let pixelBuffer = pixelBufferOut!
    
            CVPixelBufferLockBaseAddress(pixelBuffer, 0)
    
            let data = CVPixelBufferGetBaseAddress(pixelBuffer)
            let rgbColorSpace = CGColorSpaceCreateDeviceRGB()
            let context = CGBitmapContextCreate(data, Int(size.width), Int(size.height),
                8, CVPixelBufferGetBytesPerRow(pixelBuffer), rgbColorSpace, CGImageAlphaInfo.PremultipliedFirst.rawValue)
    
            CGContextClearRect(context, CGRectMake(0, 0, size.width, size.height))
    
            let horizontalRatio = size.width / image.size.width
            let verticalRatio = size.height / image.size.height
            //aspectRatio = max(horizontalRatio, verticalRatio) // ScaleAspectFill
            let aspectRatio = min(horizontalRatio, verticalRatio) // ScaleAspectFit
    
            let newSize = CGSize(width: image.size.width * aspectRatio, height: image.size.height * aspectRatio)
    
            let x = newSize.width < size.width ? (size.width - newSize.width) / 2 : 0
            let y = newSize.height < size.height ? (size.height - newSize.height) / 2 : 0
    
            CGContextDrawImage(context, CGRectMake(x, y, newSize.width, newSize.height), image.CGImage)
            CVPixelBufferUnlockBaseAddress(pixelBuffer, 0)
    
            return pixelBuffer
        }
    
        init(renderSettings: RenderSettings) {
            self.renderSettings = renderSettings
        }
    
        func start() {
    
            let avOutputSettings: [String: AnyObject] = [
                AVVideoCodecKey: renderSettings.avCodecKey,
                AVVideoWidthKey: NSNumber(float: Float(renderSettings.width)),
                AVVideoHeightKey: NSNumber(float: Float(renderSettings.height))
            ]
    
            func createPixelBufferAdaptor() {
                let sourcePixelBufferAttributesDictionary = [
                    kCVPixelBufferPixelFormatTypeKey as String: NSNumber(unsignedInt: kCVPixelFormatType_32ARGB),
                    kCVPixelBufferWidthKey as String: NSNumber(float: Float(renderSettings.width)),
                    kCVPixelBufferHeightKey as String: NSNumber(float: Float(renderSettings.height))
                ]
                pixelBufferAdaptor = AVAssetWriterInputPixelBufferAdaptor(assetWriterInput: videoWriterInput,
                    sourcePixelBufferAttributes: sourcePixelBufferAttributesDictionary)
            }
    
            func createAssetWriter(outputURL: NSURL) -> AVAssetWriter {
                guard let assetWriter = try? AVAssetWriter(URL: outputURL, fileType: AVFileTypeMPEG4) else {
                    fatalError("AVAssetWriter() failed")
                }
    
                guard assetWriter.canApplyOutputSettings(avOutputSettings, forMediaType: AVMediaTypeVideo) else {
                    fatalError("canApplyOutputSettings() failed")
                }
    
                return assetWriter
            }
    
            videoWriter = createAssetWriter(renderSettings.outputURL)
            videoWriterInput = AVAssetWriterInput(mediaType: AVMediaTypeVideo, outputSettings: avOutputSettings)
    
            if videoWriter.canAddInput(videoWriterInput) {
                videoWriter.addInput(videoWriterInput)
            }
            else {
                fatalError("canAddInput() returned false")
            }
    
            // The pixel buffer adaptor must be created before we start writing.
            createPixelBufferAdaptor()
    
            if videoWriter.startWriting() == false {
                fatalError("startWriting() failed")
            }
    
            videoWriter.startSessionAtSourceTime(kCMTimeZero)
    
            precondition(pixelBufferAdaptor.pixelBufferPool != nil, "nil pixelBufferPool")
        }
    
        func render(appendPixelBuffers: (VideoWriter)->Bool, completion: ()->Void) {
    
            precondition(videoWriter != nil, "Call start() to initialze the writer")
    
            let queue = dispatch_queue_create("mediaInputQueue", nil)
            videoWriterInput.requestMediaDataWhenReadyOnQueue(queue) {
                let isFinished = appendPixelBuffers(self)
                if isFinished {
                    self.videoWriterInput.markAsFinished()
                    self.videoWriter.finishWritingWithCompletionHandler() {
                        dispatch_async(dispatch_get_main_queue()) {
                            completion()
                        }
                    }
                }
                else {
                    // Fall through. The closure will be called again when the writer is ready.
                }
            }
        }
    
        func addImage(image: UIImage, withPresentationTime presentationTime: CMTime) -> Bool {
    
            precondition(pixelBufferAdaptor != nil, "Call start() to initialze the writer")
    
            let pixelBuffer = VideoWriter.pixelBufferFromImage(image, pixelBufferPool: pixelBufferAdaptor.pixelBufferPool!, size: renderSettings.size)
            return pixelBufferAdaptor.appendPixelBuffer(pixelBuffer, withPresentationTime: presentationTime)
        }
    
    }
    

    Part 4: Make it happen

    Once everything is in place, these are your 3 magic lines:

    let settings = RenderSettings()
    let imageAnimator = ImageAnimator(renderSettings: settings)
    imageAnimator.render() {
        print("yes")
    }
    
    0 讨论(0)
  • 2020-11-22 02:44

    Use AVAssetWriter to write images as movie. I already have answered here:- https://stackoverflow.com/a/19166876/1582217

    0 讨论(0)
提交回复
热议问题