| Index: webrtc/sdk/android/src/java/org/webrtc/TextureBufferImpl.java
|
| diff --git a/webrtc/sdk/android/src/java/org/webrtc/TextureBufferImpl.java b/webrtc/sdk/android/src/java/org/webrtc/TextureBufferImpl.java
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..ebcb22ff9c5c75692be5cd171fa112f8aebbf0da
|
| --- /dev/null
|
| +++ b/webrtc/sdk/android/src/java/org/webrtc/TextureBufferImpl.java
|
| @@ -0,0 +1,131 @@
|
| +/*
|
| + * Copyright 2017 The WebRTC project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +package org.webrtc;
|
| +
|
| +import android.graphics.Matrix;
|
| +import java.nio.ByteBuffer;
|
| +
|
| +/**
|
| + * Android texture buffer backed by a SurfaceTextureHelper's texture. The buffer calls
|
| + * |releaseCallback| when it is released.
|
| + */
|
| +class TextureBufferImpl implements VideoFrame.TextureBuffer {
|
| + private final int width;
|
| + private final int height;
|
| + private final Type type;
|
| + private final int id;
|
| + private final Matrix transformMatrix;
|
| + private final SurfaceTextureHelper surfaceTextureHelper;
|
| + private final Runnable releaseCallback;
|
| + private int refCount;
|
| +
|
| + public TextureBufferImpl(int width, int height, Type type, int id, Matrix transformMatrix,
|
| + SurfaceTextureHelper surfaceTextureHelper, Runnable releaseCallback) {
|
| + this.width = width;
|
| + this.height = height;
|
| + this.type = type;
|
| + this.id = id;
|
| + this.transformMatrix = transformMatrix;
|
| + this.surfaceTextureHelper = surfaceTextureHelper;
|
| + this.releaseCallback = releaseCallback;
|
| + this.refCount = 1; // Creator implicitly holds a reference.
|
| + }
|
| +
|
| + @Override
|
| + public VideoFrame.TextureBuffer.Type getType() {
|
| + return type;
|
| + }
|
| +
|
| + @Override
|
| + public int getTextureId() {
|
| + return id;
|
| + }
|
| +
|
| + @Override
|
| + public Matrix getTransformMatrix() {
|
| + return transformMatrix;
|
| + }
|
| +
|
| + @Override
|
| + public int getWidth() {
|
| + return width;
|
| + }
|
| +
|
| + @Override
|
| + public int getHeight() {
|
| + return height;
|
| + }
|
| +
|
| + @Override
|
| + public VideoFrame.I420Buffer toI420() {
|
| + // SurfaceTextureHelper requires a stride that is divisible by 8. Round width up.
|
| + // See SurfaceTextureHelper for details on the size and format.
|
| + int stride = ((width + 7) / 8) * 8;
|
| + int uvHeight = (height + 1) / 2;
|
| + // Due to the layout used by SurfaceTextureHelper, vPos + stride * uvHeight would overrun the
|
| + // buffer. Add one row at the bottom to compensate for this. There will never be data in the
|
| + // extra row, but now other code does not have to deal with v stride * v height exceeding the
|
| + // buffer's capacity.
|
| + int size = stride * (height + uvHeight + 1);
|
| + ByteBuffer buffer = ByteBuffer.allocateDirect(size);
|
| + surfaceTextureHelper.textureToYUV(buffer, width, height, stride, id,
|
| + RendererCommon.convertMatrixFromAndroidGraphicsMatrix(transformMatrix));
|
| +
|
| + int yPos = 0;
|
| + int uPos = yPos + stride * height;
|
| + // Rows of U and V alternate in the buffer, so V data starts after the first row of U.
|
| + int vPos = uPos + stride / 2;
|
| +
|
| + buffer.position(yPos);
|
| + buffer.limit(yPos + stride * height);
|
| + ByteBuffer dataY = buffer.slice();
|
| +
|
| + buffer.position(uPos);
|
| + buffer.limit(uPos + stride * uvHeight);
|
| + ByteBuffer dataU = buffer.slice();
|
| +
|
| + buffer.position(vPos);
|
| + buffer.limit(vPos + stride * uvHeight);
|
| + ByteBuffer dataV = buffer.slice();
|
| +
|
| + // SurfaceTextureHelper uses the same stride for Y, U, and V data.
|
| + return new I420BufferImpl(width, height, dataY, stride, dataU, stride, dataV, stride, null);
|
| + }
|
| +
|
| + @Override
|
| + public void retain() {
|
| + ++refCount;
|
| + }
|
| +
|
| + @Override
|
| + public void release() {
|
| + if (--refCount == 0) {
|
| + releaseCallback.run();
|
| + }
|
| + }
|
| +
|
| + @Override
|
| + public VideoFrame.Buffer cropAndScale(
|
| + int cropX, int cropY, int cropWidth, int cropHeight, int scaleWidth, int scaleHeight) {
|
| + retain();
|
| + Matrix newMatrix = new Matrix(transformMatrix);
|
| + newMatrix.postScale(cropWidth / (float) width, cropHeight / (float) height);
|
| + newMatrix.postTranslate(cropX / (float) width, cropY / (float) height);
|
| +
|
| + return new TextureBufferImpl(
|
| + scaleWidth, scaleHeight, type, id, newMatrix, surfaceTextureHelper, new Runnable() {
|
| + @Override
|
| + public void run() {
|
| + release();
|
| + }
|
| + });
|
| + }
|
| +}
|
|
|