151 lines
3.8 KiB
Objective-C
151 lines
3.8 KiB
Objective-C
//
|
|
// FFMPEGFileProtocols.m
|
|
// FFMPEG
|
|
//
|
|
// Created by Christopher Snowhill on 10/4/13.
|
|
// Copyright 2013 __NoWork, Inc__. All rights reserved.
|
|
//
|
|
|
|
#include "Plugin.h"
|
|
|
|
#include <libavformat/avformat.h>
|
|
#include <libavformat/url.h> // INTERNAL
|
|
#include <libavutil/opt.h>
|
|
|
|
/* standard file protocol */
|
|
|
|
typedef struct FileContext {
|
|
const AVClass *class;
|
|
void *fd;
|
|
} FileContext;
|
|
|
|
static const AVOption file_options[] = {
|
|
{ NULL }
|
|
};
|
|
|
|
static const AVClass file_class = {
|
|
.class_name = "file",
|
|
.item_name = av_default_item_name,
|
|
.option = file_options,
|
|
.version = LIBAVUTIL_VERSION_INT,
|
|
};
|
|
|
|
static const AVClass http_class = {
|
|
.class_name = "http",
|
|
.item_name = av_default_item_name,
|
|
.option = file_options,
|
|
.version = LIBAVUTIL_VERSION_INT,
|
|
};
|
|
|
|
static const AVClass unpack_class = {
|
|
.class_name = "unpack",
|
|
.item_name = av_default_item_name,
|
|
.option = file_options,
|
|
.version = LIBAVUTIL_VERSION_INT,
|
|
};
|
|
|
|
static int file_read(URLContext *h, unsigned char *buf, int size)
|
|
{
|
|
FileContext *c = h->priv_data;
|
|
NSObject* _fd = (__bridge NSObject *)(c->fd);
|
|
id<CogSource> __unsafe_unretained fd = (id) _fd;
|
|
return [fd read:buf amount:size];
|
|
}
|
|
|
|
static int file_check(URLContext *h, int mask)
|
|
{
|
|
return mask & AVIO_FLAG_READ;
|
|
}
|
|
|
|
static int file_open(URLContext *h, const char *filename, int flags)
|
|
{
|
|
FileContext *c = h->priv_data;
|
|
id<CogSource> fd;
|
|
|
|
if (flags & AVIO_FLAG_WRITE) {
|
|
return -1;
|
|
}
|
|
|
|
NSString * urlString = [NSString stringWithUTF8String:filename];
|
|
NSURL * url = [NSURL URLWithString:[urlString stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding]];
|
|
|
|
id audioSourceClass = NSClassFromString(@"AudioSource");
|
|
fd = [audioSourceClass audioSourceForURL:url];
|
|
|
|
if (![fd open:url])
|
|
return -1;
|
|
|
|
c->fd = (void*)CFBridgingRetain(fd);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int http_open(URLContext *h, const char *filename, int flags)
|
|
{
|
|
int rval = file_open( h, filename, flags );
|
|
h->is_streamed = 1;
|
|
return rval;
|
|
}
|
|
|
|
/* XXX: use llseek */
|
|
static int64_t file_seek(URLContext *h, int64_t pos, int whence)
|
|
{
|
|
FileContext *c = h->priv_data;
|
|
NSObject* _fd = (__bridge NSObject *)(c->fd);
|
|
id<CogSource> __unsafe_unretained fd = (id) _fd;
|
|
return [fd seek:pos whence:whence] ? [fd tell] : -1;
|
|
}
|
|
|
|
static int64_t http_seek(URLContext *h, int64_t pos, int whence)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
static int file_close(URLContext *h)
|
|
{
|
|
FileContext *c = h->priv_data;
|
|
CFBridgingRelease(c->fd);
|
|
return 0;
|
|
}
|
|
|
|
static URLProtocol ff_file_protocol = {
|
|
.name = "file",
|
|
.url_open = file_open,
|
|
.url_read = file_read,
|
|
.url_seek = file_seek,
|
|
.url_close = file_close,
|
|
.url_check = file_check,
|
|
.priv_data_size = sizeof(FileContext),
|
|
.priv_data_class = &file_class,
|
|
};
|
|
|
|
static URLProtocol ff_http_protocol = {
|
|
.name = "http",
|
|
.url_open = http_open,
|
|
.url_read = file_read,
|
|
.url_seek = http_seek,
|
|
.url_close = file_close,
|
|
.url_check = file_check,
|
|
.priv_data_size = sizeof(FileContext),
|
|
.flags = URL_PROTOCOL_FLAG_NETWORK,
|
|
.priv_data_class = &http_class,
|
|
};
|
|
|
|
static URLProtocol ff_unpack_protocol = {
|
|
.name = "unpack",
|
|
.url_open = file_open,
|
|
.url_read = file_read,
|
|
.url_seek = file_seek,
|
|
.url_close = file_close,
|
|
.url_check = file_check,
|
|
.priv_data_size = sizeof(FileContext),
|
|
.priv_data_class = &unpack_class,
|
|
};
|
|
|
|
void registerCogProtocols()
|
|
{
|
|
ffurl_register_protocol(&ff_file_protocol);
|
|
ffurl_register_protocol(&ff_http_protocol);
|
|
ffurl_register_protocol(&ff_unpack_protocol);
|
|
}
|