使用Mediacodec进行解码并使用OpenCV

时间:2013-12-03 13:59:08

标签: java android opencv mediacodec

我正在开发使用OpenCV的Android应用程序。

我有mp4视频文件,我需要从中读取300帧1920x1080并对它们进行一些图像处理操作。

经过大量搜索后,我发现最后只有this个例子。

我的问题是我需要做一件简单的事情,我只是想读取帧并将它们保存在设备内存中或者只是将它们转换为OpenCV Matrix。

这是我的尝试(最后解释):

public void run() {

        extractor = new MediaExtractor();
        extractor.setDataSource(SAMPLE);

        for (int i = 0; i < extractor.getTrackCount(); i++) {
            MediaFormat format = extractor.getTrackFormat(i);
            String mime = format.getString(MediaFormat.KEY_MIME);
            if (mime.startsWith("video/")) {
                extractor.selectTrack(i);
                decoder = MediaCodec.createDecoderByType(mime);
                decoder.configure(format, surface, null, 0);

                break;
            }
        }

        if (decoder == null) {
            Log.e("DecodeActivity", "Can't find video info!");
            return;
        }

        decoder.start();

        ByteBuffer[] inputBuffers = decoder.getInputBuffers();
        ByteBuffer[] outputBuffers = decoder.getOutputBuffers();

        BufferInfo info = new BufferInfo();

        boolean isEOS = false;
        long startMs = System.currentTimeMillis();

        while (!Thread.interrupted()) {
            if (!isEOS) {
                int inIndex = decoder.dequeueInputBuffer(10000);
                if (inIndex >= 0) {
                    ByteBuffer buffer = inputBuffers[inIndex];
                    int sampleSize = extractor.readSampleData(buffer, 0);
                    if (sampleSize < 0) {
                        Log.d("DecodeActivity", "InputBuffer BUFFER_FLAG_END_OF_STREAM");
                        decoder.queueInputBuffer(inIndex, 0, 0, 0, MediaCodec.BUFFER_FLAG_END_OF_STREAM);
                        isEOS = true;
                    } else {
                        decoder.queueInputBuffer(inIndex, 0, sampleSize, extractor.getSampleTime(), 0);
                        extractor.advance();
                    }

                }
            }
            int outIndex = decoder.dequeueOutputBuffer(info, 10000);
            switch (outIndex) {
            case MediaCodec.INFO_OUTPUT_BUFFERS_CHANGED:
                Log.d("DecodeActivity", "INFO_OUTPUT_BUFFERS_CHANGED");
                outputBuffers = decoder.getOutputBuffers();
                break;
            case MediaCodec.INFO_OUTPUT_FORMAT_CHANGED:
                Log.d("DecodeActivity", "New format " + decoder.getOutputFormat());
                break;
            case MediaCodec.INFO_TRY_AGAIN_LATER:
                Log.d("DecodeActivity", "dequeueOutputBuffer timed out!");
                break;
            default:
                ByteBuffer buffer = outputBuffers[outIndex];

                Log.v("DecodeActivity", "We can't use this buffer but render it due to the API limit, " + buffer);
                byte[] b = new byte[buffer.remaining()];

                // We use a very simple clock to keep the video FPS, or the video
                // playback will be too fast
                while (info.presentationTimeUs / 1000 > System.currentTimeMillis() - startMs) {
                    try {
                        sleep(10);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                        break;
                    }
                }
                decoder.releaseOutputBuffer(outIndex, true);
                break;
            }

            // All decoded frames have been rendered, we can stop playing now
            if ((info.flags & MediaCodec.BUFFER_FLAG_END_OF_STREAM) != 0) {
                Log.d("DecodeActivity", "OutputBuffer BUFFER_FLAG_END_OF_STREAM");
                break;
            }
        }

        decoder.stop();
        decoder.release();
        extractor.release();

    }

在这个例子中,我读取了帧并将它们显示为Surface。

为了将其保存为位图/矩阵或将其保存在设备中,我需要更改哪些内容?

由于

1 个答案:

答案 0 :(得分:1)

我看到了代码的两条路径:

  1. 对于Android 4.3及更高版本,您可以使用Grafika中的示例作为fadden建议。 MediaCodec解码器应使用表面for configure方法,并将releaseOutputBuffer与布尔渲染设置为“true”,以便渲染发生在曲面上。然后,您可以使用着色器执行操作,同时渲染到将由MediaCodec编码器用于将其编码为视频的表面。此解决方案在大多数具有exceptions的设备上都很快,但它是新的,bugs仍出现在设备硬件中。
  2. 如果你可以接受一个较慢的解决方案,OpenCV有一个nice port/integration on Android,带有代码样本和所有,甚至使用ffmpeg编码回h264。对于解码,您仍然可以使用MediaCodec(如果将结果渲染到曲面,则需要使用glReadPixels,这很慢,以便将数据恢复到cpu)。这也适用于以前版本的Android。
  3. 无论哪种方式,你需要时间,耐心和精力,因为它不会没有一些斗争。