1/* 2 * Copyright (C) 2010 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17#include <stdlib.h> 18#include <stdio.h> 19#include <time.h> 20#include <sched.h> 21#include <sys/resource.h> 22 23#include <EGL/egl.h> 24#include <EGL/eglext.h> 25#include <GLES2/gl2.h> 26#include <GLES2/gl2ext.h> 27 28#include <utils/Timers.h> 29 30#include <WindowSurface.h> 31#include <ui/GraphicBuffer.h> 32#include <EGLUtils.h> 33 34using namespace android; 35 36static void printGLString(const char *name, GLenum s) { 37 // fprintf(stderr, "printGLString %s, %d\n", name, s); 38 const char *v = (const char *) glGetString(s); 39 // int error = glGetError(); 40 // fprintf(stderr, "glGetError() = %d, result of glGetString = %x\n", error, 41 // (unsigned int) v); 42 // if ((v < (const char*) 0) || (v > (const char*) 0x10000)) 43 // fprintf(stderr, "GL %s = %s\n", name, v); 44 // else 45 // fprintf(stderr, "GL %s = (null) 0x%08x\n", name, (unsigned int) v); 46 fprintf(stderr, "GL %s = %s\n", name, v); 47} 48 49static void checkEglError(const char* op, EGLBoolean returnVal = EGL_TRUE) { 50 if (returnVal != EGL_TRUE) { 51 fprintf(stderr, "%s() returned %d\n", op, returnVal); 52 } 53 54 for (EGLint error = eglGetError(); error != EGL_SUCCESS; error 55 = eglGetError()) { 56 fprintf(stderr, "after %s() eglError %s (0x%x)\n", op, EGLUtils::strerror(error), 57 error); 58 } 59} 60 61static void checkGlError(const char* op) { 62 for (GLint error = glGetError(); error; error 63 = glGetError()) { 64 fprintf(stderr, "after %s() glError (0x%x)\n", op, error); 65 } 66} 67 68static const char gVertexShader[] = "attribute vec4 vPosition;\n" 69 "varying vec2 yuvTexCoords;\n" 70 "void main() {\n" 71 " yuvTexCoords = vPosition.xy + vec2(0.5, 0.5);\n" 72 " gl_Position = vPosition;\n" 73 "}\n"; 74 75static const char gFragmentShader[] = "#extension GL_OES_EGL_image_external : require\n" 76 "precision mediump float;\n" 77 "uniform samplerExternalOES yuvTexSampler;\n" 78 "varying vec2 yuvTexCoords;\n" 79 "void main() {\n" 80 " gl_FragColor = texture2D(yuvTexSampler, yuvTexCoords);\n" 81 "}\n"; 82 83GLuint loadShader(GLenum shaderType, const char* pSource) { 84 GLuint shader = glCreateShader(shaderType); 85 if (shader) { 86 glShaderSource(shader, 1, &pSource, NULL); 87 glCompileShader(shader); 88 GLint compiled = 0; 89 glGetShaderiv(shader, GL_COMPILE_STATUS, &compiled); 90 if (!compiled) { 91 GLint infoLen = 0; 92 glGetShaderiv(shader, GL_INFO_LOG_LENGTH, &infoLen); 93 if (infoLen) { 94 char* buf = (char*) malloc(infoLen); 95 if (buf) { 96 glGetShaderInfoLog(shader, infoLen, NULL, buf); 97 fprintf(stderr, "Could not compile shader %d:\n%s\n", 98 shaderType, buf); 99 free(buf); 100 } 101 } else { 102 fprintf(stderr, "Guessing at GL_INFO_LOG_LENGTH size\n"); 103 char* buf = (char*) malloc(0x1000); 104 if (buf) { 105 glGetShaderInfoLog(shader, 0x1000, NULL, buf); 106 fprintf(stderr, "Could not compile shader %d:\n%s\n", 107 shaderType, buf); 108 free(buf); 109 } 110 } 111 glDeleteShader(shader); 112 shader = 0; 113 } 114 } 115 return shader; 116} 117 118GLuint createProgram(const char* pVertexSource, const char* pFragmentSource) { 119 GLuint vertexShader = loadShader(GL_VERTEX_SHADER, pVertexSource); 120 if (!vertexShader) { 121 return 0; 122 } 123 124 GLuint pixelShader = loadShader(GL_FRAGMENT_SHADER, pFragmentSource); 125 if (!pixelShader) { 126 return 0; 127 } 128 129 GLuint program = glCreateProgram(); 130 if (program) { 131 glAttachShader(program, vertexShader); 132 checkGlError("glAttachShader"); 133 glAttachShader(program, pixelShader); 134 checkGlError("glAttachShader"); 135 glLinkProgram(program); 136 GLint linkStatus = GL_FALSE; 137 glGetProgramiv(program, GL_LINK_STATUS, &linkStatus); 138 if (linkStatus != GL_TRUE) { 139 GLint bufLength = 0; 140 glGetProgramiv(program, GL_INFO_LOG_LENGTH, &bufLength); 141 if (bufLength) { 142 char* buf = (char*) malloc(bufLength); 143 if (buf) { 144 glGetProgramInfoLog(program, bufLength, NULL, buf); 145 fprintf(stderr, "Could not link program:\n%s\n", buf); 146 free(buf); 147 } 148 } 149 glDeleteProgram(program); 150 program = 0; 151 } 152 } 153 return program; 154} 155 156GLuint gProgram; 157GLint gvPositionHandle; 158GLint gYuvTexSamplerHandle; 159 160bool setupGraphics(int w, int h) { 161 gProgram = createProgram(gVertexShader, gFragmentShader); 162 if (!gProgram) { 163 return false; 164 } 165 gvPositionHandle = glGetAttribLocation(gProgram, "vPosition"); 166 checkGlError("glGetAttribLocation"); 167 fprintf(stderr, "glGetAttribLocation(\"vPosition\") = %d\n", 168 gvPositionHandle); 169 gYuvTexSamplerHandle = glGetUniformLocation(gProgram, "yuvTexSampler"); 170 checkGlError("glGetUniformLocation"); 171 fprintf(stderr, "glGetUniformLocation(\"yuvTexSampler\") = %d\n", 172 gYuvTexSamplerHandle); 173 174 glViewport(0, 0, w, h); 175 checkGlError("glViewport"); 176 return true; 177} 178 179int align(int x, int a) { 180 return (x + (a-1)) & (~(a-1)); 181} 182 183const int yuvTexWidth = 608; 184const int yuvTexHeight = 480; 185const int yuvTexUsage = GraphicBuffer::USAGE_HW_TEXTURE | 186 GraphicBuffer::USAGE_SW_WRITE_RARELY; 187const int yuvTexFormat = HAL_PIXEL_FORMAT_YV12; 188const int yuvTexOffsetY = 0; 189const bool yuvTexSameUV = false; 190static sp<GraphicBuffer> yuvTexBuffer; 191static GLuint yuvTex; 192 193bool setupYuvTexSurface(EGLDisplay dpy, EGLContext /*context*/) { 194 int blockWidth = yuvTexWidth > 16 ? yuvTexWidth / 16 : 1; 195 int blockHeight = yuvTexHeight > 16 ? yuvTexHeight / 16 : 1; 196 yuvTexBuffer = new GraphicBuffer(yuvTexWidth, yuvTexHeight, yuvTexFormat, 197 yuvTexUsage); 198 int yuvTexStrideY = yuvTexBuffer->getStride(); 199 int yuvTexOffsetV = yuvTexStrideY * yuvTexHeight; 200 int yuvTexStrideV = (yuvTexStrideY/2 + 0xf) & ~0xf; 201 int yuvTexOffsetU = yuvTexOffsetV + yuvTexStrideV * yuvTexHeight/2; 202 int yuvTexStrideU = yuvTexStrideV; 203 char* buf = NULL; 204 status_t err = yuvTexBuffer->lock(GRALLOC_USAGE_SW_WRITE_OFTEN, (void**)(&buf)); 205 if (err != 0) { 206 fprintf(stderr, "yuvTexBuffer->lock(...) failed: %d\n", err); 207 return false; 208 } 209 for (int x = 0; x < yuvTexWidth; x++) { 210 for (int y = 0; y < yuvTexHeight; y++) { 211 int parityX = (x / blockWidth) & 1; 212 int parityY = (y / blockHeight) & 1; 213 unsigned char intensity = (parityX ^ parityY) ? 63 : 191; 214 buf[yuvTexOffsetY + (y * yuvTexStrideY) + x] = intensity; 215 if (x < yuvTexWidth / 2 && y < yuvTexHeight / 2) { 216 buf[yuvTexOffsetU + (y * yuvTexStrideU) + x] = intensity; 217 if (yuvTexSameUV) { 218 buf[yuvTexOffsetV + (y * yuvTexStrideV) + x] = intensity; 219 } else if (x < yuvTexWidth / 4 && y < yuvTexHeight / 4) { 220 buf[yuvTexOffsetV + (y*2 * yuvTexStrideV) + x*2 + 0] = 221 buf[yuvTexOffsetV + (y*2 * yuvTexStrideV) + x*2 + 1] = 222 buf[yuvTexOffsetV + ((y*2+1) * yuvTexStrideV) + x*2 + 0] = 223 buf[yuvTexOffsetV + ((y*2+1) * yuvTexStrideV) + x*2 + 1] = intensity; 224 } 225 } 226 } 227 } 228 229 err = yuvTexBuffer->unlock(); 230 if (err != 0) { 231 fprintf(stderr, "yuvTexBuffer->unlock() failed: %d\n", err); 232 return false; 233 } 234 235 EGLClientBuffer clientBuffer = (EGLClientBuffer)yuvTexBuffer->getNativeBuffer(); 236 EGLImageKHR img = eglCreateImageKHR(dpy, EGL_NO_CONTEXT, EGL_NATIVE_BUFFER_ANDROID, 237 clientBuffer, 0); 238 checkEglError("eglCreateImageKHR"); 239 if (img == EGL_NO_IMAGE_KHR) { 240 return false; 241 } 242 243 glGenTextures(1, &yuvTex); 244 checkGlError("glGenTextures"); 245 glBindTexture(GL_TEXTURE_EXTERNAL_OES, yuvTex); 246 checkGlError("glBindTexture"); 247 glEGLImageTargetTexture2DOES(GL_TEXTURE_EXTERNAL_OES, (GLeglImageOES)img); 248 checkGlError("glEGLImageTargetTexture2DOES"); 249 250 return true; 251} 252 253const GLfloat gTriangleVertices[] = { 254 -0.5f, 0.5f, 255 -0.5f, -0.5f, 256 0.5f, -0.5f, 257 0.5f, 0.5f, 258}; 259 260void renderFrame() { 261 glClearColor(0.0f, 0.0f, 1.0f, 1.0f); 262 checkGlError("glClearColor"); 263 glClear( GL_DEPTH_BUFFER_BIT | GL_COLOR_BUFFER_BIT); 264 checkGlError("glClear"); 265 266 glUseProgram(gProgram); 267 checkGlError("glUseProgram"); 268 269 glVertexAttribPointer(gvPositionHandle, 2, GL_FLOAT, GL_FALSE, 0, gTriangleVertices); 270 checkGlError("glVertexAttribPointer"); 271 glEnableVertexAttribArray(gvPositionHandle); 272 checkGlError("glEnableVertexAttribArray"); 273 274 glUniform1i(gYuvTexSamplerHandle, 0); 275 checkGlError("glUniform1i"); 276 glBindTexture(GL_TEXTURE_EXTERNAL_OES, yuvTex); 277 checkGlError("glBindTexture"); 278 279 glDrawArrays(GL_TRIANGLE_FAN, 0, 4); 280 checkGlError("glDrawArrays"); 281} 282 283void printEGLConfiguration(EGLDisplay dpy, EGLConfig config) { 284 285#define X(VAL) {VAL, #VAL} 286 struct {EGLint attribute; const char* name;} names[] = { 287 X(EGL_BUFFER_SIZE), 288 X(EGL_ALPHA_SIZE), 289 X(EGL_BLUE_SIZE), 290 X(EGL_GREEN_SIZE), 291 X(EGL_RED_SIZE), 292 X(EGL_DEPTH_SIZE), 293 X(EGL_STENCIL_SIZE), 294 X(EGL_CONFIG_CAVEAT), 295 X(EGL_CONFIG_ID), 296 X(EGL_LEVEL), 297 X(EGL_MAX_PBUFFER_HEIGHT), 298 X(EGL_MAX_PBUFFER_PIXELS), 299 X(EGL_MAX_PBUFFER_WIDTH), 300 X(EGL_NATIVE_RENDERABLE), 301 X(EGL_NATIVE_VISUAL_ID), 302 X(EGL_NATIVE_VISUAL_TYPE), 303 X(EGL_SAMPLES), 304 X(EGL_SAMPLE_BUFFERS), 305 X(EGL_SURFACE_TYPE), 306 X(EGL_TRANSPARENT_TYPE), 307 X(EGL_TRANSPARENT_RED_VALUE), 308 X(EGL_TRANSPARENT_GREEN_VALUE), 309 X(EGL_TRANSPARENT_BLUE_VALUE), 310 X(EGL_BIND_TO_TEXTURE_RGB), 311 X(EGL_BIND_TO_TEXTURE_RGBA), 312 X(EGL_MIN_SWAP_INTERVAL), 313 X(EGL_MAX_SWAP_INTERVAL), 314 X(EGL_LUMINANCE_SIZE), 315 X(EGL_ALPHA_MASK_SIZE), 316 X(EGL_COLOR_BUFFER_TYPE), 317 X(EGL_RENDERABLE_TYPE), 318 X(EGL_CONFORMANT), 319 }; 320#undef X 321 322 for (size_t j = 0; j < sizeof(names) / sizeof(names[0]); j++) { 323 EGLint value = -1; 324 EGLint returnVal = eglGetConfigAttrib(dpy, config, names[j].attribute, &value); 325 EGLint error = eglGetError(); 326 if (returnVal && error == EGL_SUCCESS) { 327 printf(" %s: ", names[j].name); 328 printf("%d (0x%x)", value, value); 329 } 330 } 331 printf("\n"); 332} 333 334int main(int /*argc*/, char** /*argv*/) { 335 EGLBoolean returnValue; 336 EGLConfig myConfig = {0}; 337 338 EGLint context_attribs[] = { EGL_CONTEXT_CLIENT_VERSION, 2, EGL_NONE }; 339 EGLint s_configAttribs[] = { 340 EGL_SURFACE_TYPE, EGL_WINDOW_BIT, 341 EGL_RENDERABLE_TYPE, EGL_OPENGL_ES2_BIT, 342 EGL_NONE }; 343 EGLint majorVersion; 344 EGLint minorVersion; 345 EGLContext context; 346 EGLSurface surface; 347 EGLint w, h; 348 349 EGLDisplay dpy; 350 351 checkEglError("<init>"); 352 dpy = eglGetDisplay(EGL_DEFAULT_DISPLAY); 353 checkEglError("eglGetDisplay"); 354 if (dpy == EGL_NO_DISPLAY) { 355 printf("eglGetDisplay returned EGL_NO_DISPLAY.\n"); 356 return 0; 357 } 358 359 returnValue = eglInitialize(dpy, &majorVersion, &minorVersion); 360 checkEglError("eglInitialize", returnValue); 361 fprintf(stderr, "EGL version %d.%d\n", majorVersion, minorVersion); 362 if (returnValue != EGL_TRUE) { 363 printf("eglInitialize failed\n"); 364 return 0; 365 } 366 367 WindowSurface windowSurface; 368 EGLNativeWindowType window = windowSurface.getSurface(); 369 returnValue = EGLUtils::selectConfigForNativeWindow(dpy, s_configAttribs, window, &myConfig); 370 if (returnValue) { 371 printf("EGLUtils::selectConfigForNativeWindow() returned %d", returnValue); 372 return 1; 373 } 374 375 checkEglError("EGLUtils::selectConfigForNativeWindow"); 376 377 printf("Chose this configuration:\n"); 378 printEGLConfiguration(dpy, myConfig); 379 380 surface = eglCreateWindowSurface(dpy, myConfig, window, NULL); 381 checkEglError("eglCreateWindowSurface"); 382 if (surface == EGL_NO_SURFACE) { 383 printf("gelCreateWindowSurface failed.\n"); 384 return 1; 385 } 386 387 context = eglCreateContext(dpy, myConfig, EGL_NO_CONTEXT, context_attribs); 388 checkEglError("eglCreateContext"); 389 if (context == EGL_NO_CONTEXT) { 390 printf("eglCreateContext failed\n"); 391 return 1; 392 } 393 returnValue = eglMakeCurrent(dpy, surface, surface, context); 394 checkEglError("eglMakeCurrent", returnValue); 395 if (returnValue != EGL_TRUE) { 396 return 1; 397 } 398 eglQuerySurface(dpy, surface, EGL_WIDTH, &w); 399 checkEglError("eglQuerySurface"); 400 eglQuerySurface(dpy, surface, EGL_HEIGHT, &h); 401 checkEglError("eglQuerySurface"); 402 403 fprintf(stderr, "Window dimensions: %d x %d\n", w, h); 404 405 printGLString("Version", GL_VERSION); 406 printGLString("Vendor", GL_VENDOR); 407 printGLString("Renderer", GL_RENDERER); 408 printGLString("Extensions", GL_EXTENSIONS); 409 410 if(!setupYuvTexSurface(dpy, context)) { 411 fprintf(stderr, "Could not set up texture surface.\n"); 412 return 1; 413 } 414 415 if(!setupGraphics(w, h)) { 416 fprintf(stderr, "Could not set up graphics.\n"); 417 return 1; 418 } 419 420 for (;;) { 421 renderFrame(); 422 eglSwapBuffers(dpy, surface); 423 checkEglError("eglSwapBuffers"); 424 } 425 426 return 0; 427} 428