web-dev-qa-db-ja.com

libavformatを使用してメモリ内にあるファイルを読み取る

私は現在、サーバーから送信された小さなビデオファイルを読み込もうとしています

Libavformatを使用してファイルを読み取るには、

_av_open_input_file(&avFormatContext, "C:\\path\\to\\video.avi", 0, 0, 0);
_

問題は、この場合、ファイルがディスク上ではなくメモリ内にあることです。

今のところ私がしているのは、ファイルをダウンロードし、一時名を使用してディスクに書き込み、一時ファイル名を使用して_av_open_input_file_を呼び出すことです。これは、あまりクリーンなソリューションではありません。

実際、私が欲しいのはav_open_custom(&avFormatContext, &myReadFunction, &mySeekFunction);のような関数ですが、ドキュメントには何も見つかりませんでした。ファイルの名前は、ライブラリが使用している形式を判別するのに役立つものではないため、技術的には可能だと思います。

それで、このような関数、またはav_open_input_fileの代替はありますか？

c++cfilelibavcodeclibavformat
28
Tomaka17

この問題に何時間も取り組んできたのに、このサイトに質問を投稿した直後に、いつも自分で解決策を見つけるのはおかしいです。

実際、avFormatContext->pbを呼び出す前にav_open_inputを初期化し、偽のファイル名を渡す必要があります。これはドキュメントには書かれていませんが、ライブラリのソースコードに直接解説されています。

Istreamからロードする場合のサンプルコード（テストされていないため、同じ問題を抱えている人がアイデアを得ることができます）

static int readFunction(void* opaque, uint8_t* buf, int buf_size) {
    auto& me = *reinterpret_cast<std::istream*>(opaque);
    me.read(reinterpret_cast<char*>(buf), buf_size);
    return me.gcount();
}

std::ifstream stream("file.avi", std::ios::binary);

const std::shared_ptr<unsigned char> buffer(reinterpret_cast<unsigned char*>(av_malloc(8192)), &av_free);
const std::shared_ptr<AVIOContext> avioContext(avio_alloc_context(buffer.get(), 8192, 0, reinterpret_cast<void*>(static_cast<std::istream*>(&stream)), &readFunction, nullptr, nullptr), &av_free);

const auto avFormat = std::shared_ptr<AVFormatContext>(avformat_alloc_context(), &avformat_free_context);
auto avFormatPtr = avFormat.get();
avFormat->pb = avioContext.get();
avformat_open_input(&avFormatPtr, "dummyFilename", nullptr, nullptr);
36
Tomaka17

これは素晴らしい情報であり、私をかなり助けてくれましたが、人々が知っておくべきいくつかの問題があります。 libavformatは、avio_alloc_contextに指定したバッファーを混乱させる可能性があります。これは、本当に厄介なダブルフリーエラーまたはメモリリークにつながります。私が問題を探し始めたとき、私はそれを完全に釘付けにした https://lists.ffmpeg.org/pipermail/libav-user/2012-December/003257.html を見つけました。

この作業をクリーンアップするときの私の回避策は、先に進んで電話することです。

    av_free(avioContext->buffer)

次に、必要に応じて、独自のバッファポインタ（avio_alloc_context呼び出しに割り当てたもの）をNULLに設定します。

11
keefer

Tomaka17の優れた回答は、std :: istreamではなくQtQIODeviceを使用して類似の問題を解決するための良いスタートを切りました。 Tomaka17のソリューションの側面と、関連するエクスペリエンスの側面をブレンドする必要があることがわかりました http://cdry.wordpress.com/2009/09/09/using-custom-io-callbacks-with-ffmpeg/

私のカスタム読み取り関数は次のようになります。

int readFunction(void* opaque, uint8_t* buf, int buf_size)
{
    QIODevice* stream = (QIODevice*)opaque;
    int numBytes = stream->read((char*)buf, buf_size);
    return numBytes;
}

...しかし、カスタムのSeek関数も作成する必要がありました。

int64_t seekFunction(void* opaque, int64_t offset, int whence)
{
    if (whence == AVSEEK_SIZE)
        return -1; // I don't know "size of my handle in bytes"
    QIODevice* stream = (QIODevice*)opaque;
    if (stream->isSequential())
        return -1; // cannot seek a sequential stream
    if (! stream->seek(offset) )
        return -1;
    return stream->pos();
}

...そして私はそれをこのように結びました：

...
const int ioBufferSize = 32768;
unsigned char * ioBuffer = (unsigned char *)av_malloc(ioBufferSize + FF_INPUT_BUFFER_PADDING_SIZE); // can get av_free()ed by libav
AVIOContext * avioContext = avio_alloc_context(ioBuffer, ioBufferSize, 0, (void*)(&fileStream), &readFunction, NULL, &seekFunction);
AVFormatContext * container = avformat_alloc_context();
container->pb = avioContext;
avformat_open_input(&container, "dummyFileName", NULL, NULL);
...

私はまだメモリ管理の問題を解決していないことに注意してください。

7
Christopher Bruns