FreeJ/xcode/CVFFmpegLayer.mm

//
//  CVFFmpegLayer.mm
//  freej
//
//  Created by xant on 3/27/10.
//  Copyright 2010 dyne.org. All rights reserved.
//

#import "CVFFmpegLayer.h"
#import "CVFFmpegLayerController.h"
#include "CVLayer.h"
#include <libavutil/pixfmt.h>

extern "C"
{
#include "ffdec.h"
}

CVFFmpegLayer::CVFFmpegLayer(CVLayerController *controller) : CVLayer(controller)
{
    ff = NULL;
    pixelBuffer = NULL;
    currentFrame = NULL;
    repeat = NO;
}

CVFFmpegLayer::~CVFFmpegLayer()
{
}

bool CVFFmpegLayer::open(const char *movie)
{
    if (ff)
        close_and_free_ff(&ff);
    strncpy(filename, movie, sizeof(filename)-1);
    return true;
}

void *CVFFmpegLayer::feed()
{
    if (!ff) 
    {
        if (currentFrame) {
            free(currentFrame);
            currentFrame = NULL;
        }
        if (strlen(filename)) { // XXX - (argh!, find a better way)
            if (open_movie(&ff, filename) == 0) {
                init_moviebuffer(ff, geo.w, geo.h, PIX_FMT_ARGB);
                decode_frame(ff);
            } else {
                close_and_free_ff(&ff);
                if (!repeat)
                    memset(filename, 0, sizeof(filename)); // XXX
            }
        }
    } else {
        uint8_t *ffbuffer = get_bufptr(ff);
        if (!ffbuffer)
            return NULL;
        // TODO - handle geometry changes and ensure using the same size of ffmpeg buffer
        if (!currentFrame)
            currentFrame = malloc(geo.bytesize);
        memcpy(currentFrame, ffbuffer, geo.bytesize);
        if (pixelBuffer)
            CVPixelBufferRelease(pixelBuffer);
        CVReturn err = CVPixelBufferCreateWithBytes (
                                                     NULL,
                                                     geo.w,
                                                     geo.h,
                                                     k32ARGBPixelFormat,
                                                     currentFrame,
                                                     geo.w*4,
                                                     NULL,
                                                     NULL,
                                                     NULL,
                                                     &pixelBuffer
                                                    ); 
        if (err == kCVReturnSuccess) {
            [input feedFrame:pixelBuffer];
            if (!decode_frame(ff)) {
                close_and_free_ff(&ff);
                //buffer = NULL;
                // XXX - find a cleaner way instead of blindly resetting the filename
                // TODO - allow looping on a stream by reopening it
                if (!repeat) {
                    memset(filename, 0, sizeof(filename));
                    if (input) {
                        [(CVFFmpegLayerController *)input deactivate];
                        [(CVFFmpegLayerController *)input clearPreview];
                    }
                }
            }
        } else {
            // TODO - Error messages
        }
    }
    return currentFrame;
}

bool CVFFmpegLayer::isDecoding()
{
    return (ff && (get_pt_status(ff)&3) == 1)
           ? true
           : false;
}

int CVFFmpegLayer::scaledWidth()
{
    return get_scaled_width(ff);
}

int CVFFmpegLayer::scaledHeight()
{
    return get_scaled_height(ff);
}

bool CVFFmpegLayer::hasFF()
{
    return ff ? true : false;
}

void
CVFFmpegLayer::close()
{
    lock();
    close_and_free_ff(&ff);
    //buffer = NULL;
    memset(filename, 0, sizeof(filename));
    if (input) {
        [(CVFFmpegLayerController *)input deactivate];
        [(CVFFmpegLayerController *)input clearPreview];
    }
    unlock();
}