OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright 2015 The WebRTC project authors. All Rights Reserved. | |
3 * | |
4 * Use of this source code is governed by a BSD-style license | |
5 * that can be found in the LICENSE file in the root of the source | |
6 * tree. An additional intellectual property rights grant can be found | |
7 * in the file PATENTS. All contributing project authors may | |
8 * be found in the AUTHORS file in the root of the source tree. | |
9 */ | |
10 package org.webrtc; | |
11 | |
12 import android.graphics.SurfaceTexture; | |
13 import android.opengl.GLES20; | |
14 import android.test.ActivityTestCase; | |
15 import android.test.suitebuilder.annotation.MediumTest; | |
16 import android.test.suitebuilder.annotation.SmallTest; | |
17 | |
18 import java.nio.ByteBuffer; | |
19 import java.util.Random; | |
20 | |
21 public final class GlRectDrawerTest extends ActivityTestCase { | |
22 // Resolution of the test image. | |
23 private static final int WIDTH = 16; | |
24 private static final int HEIGHT = 16; | |
25 // Seed for random pixel creation. | |
26 private static final int SEED = 42; | |
27 // When comparing pixels, allow some slack for float arithmetic and integer ro
unding. | |
28 private static final float MAX_DIFF = 1.5f; | |
29 | |
30 private static float normalizedByte(byte b) { | |
31 return (b & 0xFF) / 255.0f; | |
32 } | |
33 | |
34 private static float saturatedConvert(float c) { | |
35 return 255.0f * Math.max(0, Math.min(c, 1)); | |
36 } | |
37 | |
38 // Assert RGB ByteBuffers are pixel perfect identical. | |
39 private static void assertEquals(int width, int height, ByteBuffer actual, Byt
eBuffer expected) { | |
40 actual.rewind(); | |
41 expected.rewind(); | |
42 assertEquals(actual.remaining(), width * height * 3); | |
43 assertEquals(expected.remaining(), width * height * 3); | |
44 for (int y = 0; y < height; ++y) { | |
45 for (int x = 0; x < width; ++x) { | |
46 final int actualR = actual.get() & 0xFF; | |
47 final int actualG = actual.get() & 0xFF; | |
48 final int actualB = actual.get() & 0xFF; | |
49 final int expectedR = expected.get() & 0xFF; | |
50 final int expectedG = expected.get() & 0xFF; | |
51 final int expectedB = expected.get() & 0xFF; | |
52 if (actualR != expectedR || actualG != expectedG || actualB != expectedB
) { | |
53 fail("ByteBuffers of size " + width + "x" + height + " not equal at po
sition " | |
54 + "(" + x + ", " + y + "). Expected color (R,G,B): " | |
55 + "(" + expectedR + ", " + expectedG + ", " + expectedB + ")" | |
56 + " but was: " | |
57 + "(" + actualR + ", " + actualG + ", " + actualB + ")."); | |
58 } | |
59 } | |
60 } | |
61 } | |
62 | |
63 // Convert RGBA ByteBuffer to RGB ByteBuffer. | |
64 private static ByteBuffer stripAlphaChannel(ByteBuffer rgbaBuffer) { | |
65 rgbaBuffer.rewind(); | |
66 assertEquals(rgbaBuffer.remaining() % 4, 0); | |
67 final int numberOfPixels = rgbaBuffer.remaining() / 4; | |
68 final ByteBuffer rgbBuffer = ByteBuffer.allocateDirect(numberOfPixels * 3); | |
69 while (rgbaBuffer.hasRemaining()) { | |
70 // Copy RGB. | |
71 for (int channel = 0; channel < 3; ++channel) { | |
72 rgbBuffer.put(rgbaBuffer.get()); | |
73 } | |
74 // Drop alpha. | |
75 rgbaBuffer.get(); | |
76 } | |
77 return rgbBuffer; | |
78 } | |
79 | |
80 @SmallTest | |
81 public void testRgbRendering() { | |
82 // Create EGL base with a pixel buffer as display output. | |
83 final EglBase eglBase = EglBase.create(null, EglBase.CONFIG_PIXEL_BUFFER); | |
84 eglBase.createPbufferSurface(WIDTH, HEIGHT); | |
85 eglBase.makeCurrent(); | |
86 | |
87 // Create RGB byte buffer plane with random content. | |
88 final ByteBuffer rgbPlane = ByteBuffer.allocateDirect(WIDTH * HEIGHT * 3); | |
89 final Random random = new Random(SEED); | |
90 random.nextBytes(rgbPlane.array()); | |
91 | |
92 // Upload the RGB byte buffer data as a texture. | |
93 final int rgbTexture = GlUtil.generateTexture(GLES20.GL_TEXTURE_2D); | |
94 GLES20.glActiveTexture(GLES20.GL_TEXTURE0); | |
95 GLES20.glBindTexture(GLES20.GL_TEXTURE_2D, rgbTexture); | |
96 GLES20.glTexImage2D(GLES20.GL_TEXTURE_2D, 0, GLES20.GL_RGB, WIDTH, HEIGHT, 0
, GLES20.GL_RGB, | |
97 GLES20.GL_UNSIGNED_BYTE, rgbPlane); | |
98 GlUtil.checkNoGLES2Error("glTexImage2D"); | |
99 | |
100 // Draw the RGB frame onto the pixel buffer. | |
101 final GlRectDrawer drawer = new GlRectDrawer(); | |
102 drawer.drawRgb(rgbTexture, RendererCommon.identityMatrix(), WIDTH, HEIGHT, 0
/* viewportX */, | |
103 0 /* viewportY */, WIDTH, HEIGHT); | |
104 | |
105 // Download the pixels in the pixel buffer as RGBA. Not all platforms suppor
t RGB, e.g. Nexus 9. | |
106 final ByteBuffer rgbaData = ByteBuffer.allocateDirect(WIDTH * HEIGHT * 4); | |
107 GLES20.glReadPixels(0, 0, WIDTH, HEIGHT, GLES20.GL_RGBA, GLES20.GL_UNSIGNED_
BYTE, rgbaData); | |
108 GlUtil.checkNoGLES2Error("glReadPixels"); | |
109 | |
110 // Assert rendered image is pixel perfect to source RGB. | |
111 assertEquals(WIDTH, HEIGHT, stripAlphaChannel(rgbaData), rgbPlane); | |
112 | |
113 drawer.release(); | |
114 GLES20.glDeleteTextures(1, new int[] {rgbTexture}, 0); | |
115 eglBase.release(); | |
116 } | |
117 | |
118 @SmallTest | |
119 public void testYuvRendering() { | |
120 // Create EGL base with a pixel buffer as display output. | |
121 EglBase eglBase = EglBase.create(null, EglBase.CONFIG_PIXEL_BUFFER); | |
122 eglBase.createPbufferSurface(WIDTH, HEIGHT); | |
123 eglBase.makeCurrent(); | |
124 | |
125 // Create YUV byte buffer planes with random content. | |
126 final ByteBuffer[] yuvPlanes = new ByteBuffer[3]; | |
127 final Random random = new Random(SEED); | |
128 for (int i = 0; i < 3; ++i) { | |
129 yuvPlanes[i] = ByteBuffer.allocateDirect(WIDTH * HEIGHT); | |
130 random.nextBytes(yuvPlanes[i].array()); | |
131 } | |
132 | |
133 // Generate 3 texture ids for Y/U/V. | |
134 final int yuvTextures[] = new int[3]; | |
135 for (int i = 0; i < 3; i++) { | |
136 yuvTextures[i] = GlUtil.generateTexture(GLES20.GL_TEXTURE_2D); | |
137 } | |
138 | |
139 // Upload the YUV byte buffer data as textures. | |
140 for (int i = 0; i < 3; ++i) { | |
141 GLES20.glActiveTexture(GLES20.GL_TEXTURE0 + i); | |
142 GLES20.glBindTexture(GLES20.GL_TEXTURE_2D, yuvTextures[i]); | |
143 GLES20.glTexImage2D(GLES20.GL_TEXTURE_2D, 0, GLES20.GL_LUMINANCE, WIDTH, H
EIGHT, 0, | |
144 GLES20.GL_LUMINANCE, GLES20.GL_UNSIGNED_BYTE, yuvPlanes[i]); | |
145 GlUtil.checkNoGLES2Error("glTexImage2D"); | |
146 } | |
147 | |
148 // Draw the YUV frame onto the pixel buffer. | |
149 final GlRectDrawer drawer = new GlRectDrawer(); | |
150 drawer.drawYuv(yuvTextures, RendererCommon.identityMatrix(), WIDTH, HEIGHT,
0 /* viewportX */, | |
151 0 /* viewportY */, WIDTH, HEIGHT); | |
152 | |
153 // Download the pixels in the pixel buffer as RGBA. Not all platforms suppor
t RGB, e.g. Nexus 9. | |
154 final ByteBuffer data = ByteBuffer.allocateDirect(WIDTH * HEIGHT * 4); | |
155 GLES20.glReadPixels(0, 0, WIDTH, HEIGHT, GLES20.GL_RGBA, GLES20.GL_UNSIGNED_
BYTE, data); | |
156 GlUtil.checkNoGLES2Error("glReadPixels"); | |
157 | |
158 // Compare the YUV data with the RGBA result. | |
159 for (int y = 0; y < HEIGHT; ++y) { | |
160 for (int x = 0; x < WIDTH; ++x) { | |
161 // YUV color space. Y in [0, 1], UV in [-0.5, 0.5]. The constants are ta
ken from the YUV | |
162 // fragment shader code in GlRectDrawer. | |
163 final float y_luma = normalizedByte(yuvPlanes[0].get()); | |
164 final float u_chroma = normalizedByte(yuvPlanes[1].get()) - 0.5f; | |
165 final float v_chroma = normalizedByte(yuvPlanes[2].get()) - 0.5f; | |
166 // Expected color in unrounded RGB [0.0f, 255.0f]. | |
167 final float expectedRed = saturatedConvert(y_luma + 1.403f * v_chroma); | |
168 final float expectedGreen = | |
169 saturatedConvert(y_luma - 0.344f * u_chroma - 0.714f * v_chroma); | |
170 final float expectedBlue = saturatedConvert(y_luma + 1.77f * u_chroma); | |
171 | |
172 // Actual color in RGB8888. | |
173 final int actualRed = data.get() & 0xFF; | |
174 final int actualGreen = data.get() & 0xFF; | |
175 final int actualBlue = data.get() & 0xFF; | |
176 final int actualAlpha = data.get() & 0xFF; | |
177 | |
178 // Assert rendered image is close to pixel perfect from source YUV. | |
179 assertTrue(Math.abs(actualRed - expectedRed) < MAX_DIFF); | |
180 assertTrue(Math.abs(actualGreen - expectedGreen) < MAX_DIFF); | |
181 assertTrue(Math.abs(actualBlue - expectedBlue) < MAX_DIFF); | |
182 assertEquals(actualAlpha, 255); | |
183 } | |
184 } | |
185 | |
186 drawer.release(); | |
187 GLES20.glDeleteTextures(3, yuvTextures, 0); | |
188 eglBase.release(); | |
189 } | |
190 | |
191 /** | |
192 * The purpose here is to test GlRectDrawer.oesDraw(). Unfortunately, there is
no easy way to | |
193 * create an OES texture, which is needed for input to oesDraw(). Most of the
test is concerned | |
194 * with creating OES textures in the following way: | |
195 * - Create SurfaceTexture with help from SurfaceTextureHelper. | |
196 * - Create an EglBase with the SurfaceTexture as EGLSurface. | |
197 * - Upload RGB texture with known content. | |
198 * - Draw the RGB texture onto the EglBase with the SurfaceTexture as target. | |
199 * - Wait for an OES texture to be produced. | |
200 * The actual oesDraw() test is this: | |
201 * - Create an EglBase with a pixel buffer as target. | |
202 * - Render the OES texture onto the pixel buffer. | |
203 * - Read back the pixel buffer and compare it with the known RGB data. | |
204 */ | |
205 @MediumTest | |
206 public void testOesRendering() throws InterruptedException { | |
207 /** | |
208 * Stub class to convert RGB ByteBuffers to OES textures by drawing onto a S
urfaceTexture. | |
209 */ | |
210 class StubOesTextureProducer { | |
211 private final EglBase eglBase; | |
212 private final GlRectDrawer drawer; | |
213 private final int rgbTexture; | |
214 | |
215 public StubOesTextureProducer( | |
216 EglBase.Context sharedContext, SurfaceTexture surfaceTexture, int widt
h, int height) { | |
217 eglBase = EglBase.create(sharedContext, EglBase.CONFIG_PLAIN); | |
218 surfaceTexture.setDefaultBufferSize(width, height); | |
219 eglBase.createSurface(surfaceTexture); | |
220 assertEquals(eglBase.surfaceWidth(), width); | |
221 assertEquals(eglBase.surfaceHeight(), height); | |
222 | |
223 drawer = new GlRectDrawer(); | |
224 | |
225 eglBase.makeCurrent(); | |
226 rgbTexture = GlUtil.generateTexture(GLES20.GL_TEXTURE_2D); | |
227 } | |
228 | |
229 public void draw(ByteBuffer rgbPlane) { | |
230 eglBase.makeCurrent(); | |
231 | |
232 // Upload RGB data to texture. | |
233 GLES20.glActiveTexture(GLES20.GL_TEXTURE0); | |
234 GLES20.glBindTexture(GLES20.GL_TEXTURE_2D, rgbTexture); | |
235 GLES20.glTexImage2D(GLES20.GL_TEXTURE_2D, 0, GLES20.GL_RGB, WIDTH, HEIGH
T, 0, GLES20.GL_RGB, | |
236 GLES20.GL_UNSIGNED_BYTE, rgbPlane); | |
237 // Draw the RGB data onto the SurfaceTexture. | |
238 drawer.drawRgb(rgbTexture, RendererCommon.identityMatrix(), WIDTH, HEIGH
T, | |
239 0 /* viewportX */, 0 /* viewportY */, WIDTH, HEIGHT); | |
240 eglBase.swapBuffers(); | |
241 } | |
242 | |
243 public void release() { | |
244 eglBase.makeCurrent(); | |
245 drawer.release(); | |
246 GLES20.glDeleteTextures(1, new int[] {rgbTexture}, 0); | |
247 eglBase.release(); | |
248 } | |
249 } | |
250 | |
251 // Create EGL base with a pixel buffer as display output. | |
252 final EglBase eglBase = EglBase.create(null, EglBase.CONFIG_PIXEL_BUFFER); | |
253 eglBase.createPbufferSurface(WIDTH, HEIGHT); | |
254 | |
255 // Create resources for generating OES textures. | |
256 final SurfaceTextureHelper surfaceTextureHelper = SurfaceTextureHelper.creat
e( | |
257 "SurfaceTextureHelper test" /* threadName */, eglBase.getEglBaseContext(
)); | |
258 final StubOesTextureProducer oesProducer = new StubOesTextureProducer( | |
259 eglBase.getEglBaseContext(), surfaceTextureHelper.getSurfaceTexture(), W
IDTH, HEIGHT); | |
260 final SurfaceTextureHelperTest.MockTextureListener listener = | |
261 new SurfaceTextureHelperTest.MockTextureListener(); | |
262 surfaceTextureHelper.startListening(listener); | |
263 | |
264 // Create RGB byte buffer plane with random content. | |
265 final ByteBuffer rgbPlane = ByteBuffer.allocateDirect(WIDTH * HEIGHT * 3); | |
266 final Random random = new Random(SEED); | |
267 random.nextBytes(rgbPlane.array()); | |
268 | |
269 // Draw the frame and block until an OES texture is delivered. | |
270 oesProducer.draw(rgbPlane); | |
271 listener.waitForNewFrame(); | |
272 | |
273 // Real test starts here. | |
274 // Draw the OES texture on the pixel buffer. | |
275 eglBase.makeCurrent(); | |
276 final GlRectDrawer drawer = new GlRectDrawer(); | |
277 drawer.drawOes(listener.oesTextureId, listener.transformMatrix, WIDTH, HEIGH
T, | |
278 0 /* viewportX */, 0 /* viewportY */, WIDTH, HEIGHT); | |
279 | |
280 // Download the pixels in the pixel buffer as RGBA. Not all platforms suppor
t RGB, e.g. Nexus 9. | |
281 final ByteBuffer rgbaData = ByteBuffer.allocateDirect(WIDTH * HEIGHT * 4); | |
282 GLES20.glReadPixels(0, 0, WIDTH, HEIGHT, GLES20.GL_RGBA, GLES20.GL_UNSIGNED_
BYTE, rgbaData); | |
283 GlUtil.checkNoGLES2Error("glReadPixels"); | |
284 | |
285 // Assert rendered image is pixel perfect to source RGB. | |
286 assertEquals(WIDTH, HEIGHT, stripAlphaChannel(rgbaData), rgbPlane); | |
287 | |
288 drawer.release(); | |
289 surfaceTextureHelper.returnTextureFrame(); | |
290 oesProducer.release(); | |
291 surfaceTextureHelper.dispose(); | |
292 eglBase.release(); | |
293 } | |
294 } | |
OLD | NEW |