iOS5 AVFoundation图像到视频

时间:2011-09-10 11:35:14

标签: iphone uiimage avfoundation ios5

我正在尝试从单张图片创建一个视频,并将其保存到我的照片库中,我一直在谷歌搜索多年 - 并且找不到解决方案。

我有这段代码:

    @autoreleasepool {
    NSString *path = [NSHomeDirectory() stringByAppendingPathComponent:[NSString stringWithFormat:@"Documents/movie2.mp4"]];

    UIImage *img = [UIImage imageWithData:[[self imageDataArrya]objectAtIndex:0]imageData];
    [self writeImageAsMovie:img toPath:path size:CGSizeMake(640, 960) duration:10];

    UISaveVideoAtPathToSavedPhotosAlbum (path,self, @selector(video:didFinishSavingWithError: contextInfo:), nil);
}

我在后台线程中调用上面提到的方法。这是'writeImageAsMovie'的代码:

- (void)writeImageAsMovie:(UIImage*)image toPath:(NSString*)path size:(CGSize)size duration:(int)duration {
NSError *error = nil;
AVAssetWriter *videoWriter = [[AVAssetWriter alloc] initWithURL:
                              [NSURL fileURLWithPath:path] fileType:AVFileTypeQuickTimeMovie
                                                          error:&error];

NSDictionary *videoSettings = [NSDictionary dictionaryWithObjectsAndKeys:
                               AVVideoCodecH264, AVVideoCodecKey,
                               [NSNumber numberWithInt:size.width], AVVideoWidthKey,
                               [NSNumber numberWithInt:size.height], AVVideoHeightKey,
                               nil];
[self setInput:[AVAssetWriterInput assetWriterInputWithMediaType:AVMediaTypeVideo 
                                                  outputSettings:videoSettings]];

AVAssetWriterInputPixelBufferAdaptor *adaptor = [AVAssetWriterInputPixelBufferAdaptor
                                                 assetWriterInputPixelBufferAdaptorWithAssetWriterInput:input
                                                 sourcePixelBufferAttributes:nil];

[videoWriter addInput:input];

[videoWriter startWriting];
[videoWriter startSessionAtSourceTime:kCMTimeZero];

CVPixelBufferRef buffer = [self pixelBufferFromCGImage:image.CGImage];
[adaptor appendPixelBuffer:buffer withPresentationTime:kCMTimeZero];
[adaptor appendPixelBuffer:buffer withPresentationTime:CMTimeMake(duration-1, 2)];

[input markAsFinished];
[videoWriter endSessionAtSourceTime:CMTimeMake(duration, 2)];
[videoWriter finishWriting];

}

将Image转换为CVPixelBufferRef的实用程序方法:

- (CVPixelBufferRef) pixelBufferFromCGImage: (CGImageRef) image {
NSDictionary *options = [NSDictionary dictionaryWithObjectsAndKeys:
                         [NSNumber numberWithBool:YES], kCVPixelBufferCGImageCompatibilityKey,
                         [NSNumber numberWithBool:YES], kCVPixelBufferCGBitmapContextCompatibilityKey,
                         nil];
CVPixelBufferRef pxbuffer = NULL;

CVReturn status = CVPixelBufferCreate(kCFAllocatorDefault, 
                                      self.view.frame.size.width,
                                      self.view.frame.size.height, 
                                      kCVPixelFormatType_32ARGB, 
                                      (__bridge CFDictionaryRef) options, 
                                      &pxbuffer);

CVPixelBufferLockBaseAddress(pxbuffer, 0);
void *pxdata = CVPixelBufferGetBaseAddress(pxbuffer);

CGColorSpaceRef rgbColorSpace = CGColorSpaceCreateDeviceRGB();
CGContextRef context = CGBitmapContextCreate(pxdata, self.view.frame.size.width,
                                             self.view.frame.size.height, 8, 4*self.view.frame.size.width, rgbColorSpace, 
                                             kCGImageAlphaNoneSkipFirst);
CGContextConcatCTM(context, CGAffineTransformMakeRotation(0));
CGContextDrawImage(context, CGRectMake(0, 0, CGImageGetWidth(image), 
                                       CGImageGetHeight(image)), image);
CGColorSpaceRelease(rgbColorSpace);
CGContextRelease(context);

CVPixelBufferUnlockBaseAddress(pxbuffer, 0);

return pxbuffer;
}

现在,如果我尝试从模拟器运行代码,它会给我一个错误,说明数据已损坏。

如果我在我的设备上运行它,它会将2秒的视频保存到我的照片库,但它只有绿色,我的图像不在那里。

任何帮助将不胜感激:)

1 个答案:

答案 0 :(得分:2)

我完成了这项工作 - 抱歉,我今天没有看到你的答复。 这是我用过的:

创建临时文件

 NSString *path = [NSHomeDirectory() stringByAppendingPathComponent:[NSString stringWithFormat:@"Documents/flipimator-tempfile.mp4"]];

//overwrites it if it already exists.
if([fileManager fileExistsAtPath:path]) 
    [fileManager removeItemAtPath:path error:NULL];

调用导出图像方法将图像保存到临时文件:

[self exportImages:frames 
         asVideoToPath:path 
         withFrameSize:imageSize 
       framesPerSecond:fps];

将临时文件保存到相册:

UISaveVideoAtPathToSavedPhotosAlbum (path,self, @selector(video:didFinishSavingWithError: contextInfo:), nil);

- (void)video:(NSString *) videoPath didFinishSavingWithError: (NSError *) error contextInfo: (void *) contextInfo {
    NSLog(@"Finished saving video with error: %@", error);
    UIAlertView *alert = [[UIAlertView alloc]initWithTitle:@"Done"
                                                   message:@"Movie succesfully exported." 
                                          delegate:nil 
                                 cancelButtonTitle:@"OK" 
                                 otherButtonTitles:nil, nil];
    [alert show];
}

exportImages方法的代码:

        - (void)exportImages:(NSArray *)imageArray 
           asVideoToPath:(NSString *)path 
           withFrameSize:(CGSize)imageSize
         framesPerSecond:(NSUInteger)fps {
        NSLog(@"Start building video from defined frames.");

        NSError *error = nil;

        AVAssetWriter *videoWriter = [[AVAssetWriter alloc] initWithURL:
                                      [NSURL fileURLWithPath:path] fileType:AVFileTypeQuickTimeMovie
                                                                  error:&error];    
        NSParameterAssert(videoWriter);

        NSDictionary *videoSettings = [NSDictionary dictionaryWithObjectsAndKeys:
                                       AVVideoCodecH264, AVVideoCodecKey,
                                       [NSNumber numberWithInt:imageSize.width], AVVideoWidthKey,
                                       [NSNumber numberWithInt:imageSize.height], AVVideoHeightKey,
                                       nil];

        AVAssetWriterInput* videoWriterInput = [AVAssetWriterInput
                                                assetWriterInputWithMediaType:AVMediaTypeVideo
                                                outputSettings:videoSettings];


        AVAssetWriterInputPixelBufferAdaptor *adaptor = [AVAssetWriterInputPixelBufferAdaptor
                                                         assetWriterInputPixelBufferAdaptorWithAssetWriterInput:videoWriterInput
                                                         sourcePixelBufferAttributes:nil];

        NSParameterAssert(videoWriterInput);
        NSParameterAssert([videoWriter canAddInput:videoWriterInput]);
        videoWriterInput.expectsMediaDataInRealTime = YES;
        [videoWriter addInput:videoWriterInput];

        //Start a session:
        [videoWriter startWriting];
        [videoWriter startSessionAtSourceTime:kCMTimeZero];

        CVPixelBufferRef buffer = NULL;

        //convert uiimage to CGImage.
        int frameCount = 0;

        for(UIImage * img in imageArray) {
            buffer = [self pixelBufferFromCGImage:[img CGImage] andSize:imageSize];

            BOOL append_ok = NO;
            int j = 0;
            while (!append_ok && j < 30) {
                if (adaptor.assetWriterInput.readyForMoreMediaData)  {
                    //print out status::
                    NSString *border = @"**************************************************";
                    NSLog(@"\n%@\nProcessing video frame (%d,%d).\n%@",border,frameCount,[imageArray count],border);

                    CMTime frameTime = CMTimeMake(frameCount,(int32_t) fps);
                    append_ok = [adaptor appendPixelBuffer:buffer withPresentationTime:frameTime];
                    if(!append_ok){
                        NSError *error = videoWriter.error;
                        if(error!=nil) {
                            NSLog(@"Unresolved error %@,%@.", error, [error userInfo]);
                        }
                    }

                } 
                else {
                    printf("adaptor not ready %d, %d\n", frameCount, j);
                    [NSThread sleepForTimeInterval:0.1];
                }
                j++;
            }
            if (!append_ok) {
                printf("error appending image %d times %d\n, with error.", frameCount, j);
            }
            frameCount++;
        }

        //Finish the session:
        [videoWriterInput markAsFinished];  
        [videoWriter finishWriting];
        NSLog(@"Write Ended");

    }

方法的参数

  • imageArray:UIImage的NSArray。
  • path:处理时写入的临时路径(上面定义的temp)。
  • imageSize:视频的大小(以像素为单位)(宽度和高度)。
  • fps:视频中每秒应显示多少张图片。

希望它有所帮助! 抱歉格式化 - 我仍然是StackOverflow.com的新手。

这是我使用代码的地方:http://www.youtube.com/watch?v=DDckJyF2bnA