Chromium Code Reviews| 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..ba90f7fffb4f65cbbe7bb8747d8d13ca50bf110b |
| --- /dev/null |
| +++ b/webrtc/sdk/android/src/java/org/webrtc/TextureBufferImpl.java |
| @@ -0,0 +1,120 @@ |
| +/* |
| + * 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 VideoFrame.TextureBuffer.Type.OES; |
|
magjed_webrtc
2017/07/27 14:02:29
Is it possible to return just Type.OES?
sakal
2017/07/28 08:01:36
This should 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; |
| + |
| + // SurfaceTextureHelper uses the same stride for Y, U, and V data. |
| + return new I420BufferImpl( |
| + buffer, width, height, yPos, stride, uPos, stride, vPos, 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(); |
| + } |
| + }); |
| + } |
| +} |