| Index: client/deps/glbench/src/yuvtest.cc
|
| diff --git a/client/deps/glbench/src/yuvtest.cc b/client/deps/glbench/src/yuvtest.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..fb1b9107751585a460dc29549edfca828c965248
|
| --- /dev/null
|
| +++ b/client/deps/glbench/src/yuvtest.cc
|
| @@ -0,0 +1,140 @@
|
| +// Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include <stdio.h>
|
| +#include <sys/mman.h>
|
| +
|
| +#include "base/logging.h"
|
| +
|
| +#include "main.h"
|
| +#include "testbase.h"
|
| +#include "utils.h"
|
| +#include "yuv2rgb.h"
|
| +
|
| +namespace glbench {
|
| +
|
| +class YuvToRgbTest : public DrawArraysTestFunc {
|
| + public:
|
| + YuvToRgbTest(int type, const char* name) : type_(type), name_(name) {}
|
| + virtual ~YuvToRgbTest() {}
|
| + virtual bool Run();
|
| +
|
| + private:
|
| + int type_;
|
| + const char* name_;
|
| + DISALLOW_COPY_AND_ASSIGN(YuvToRgbTest);
|
| +};
|
| +
|
| +
|
| +GLuint YuvToRgbShaderProgram(int type, GLuint vertex_buffer,
|
| + int width, int height) {
|
| + const char *vertex = type == 1 ? YUV2RGB_VERTEX_1 : YUV2RGB_VERTEX_2;
|
| + const char *fragment = type == 1 ? YUV2RGB_FRAGMENT_1 : YUV2RGB_FRAGMENT_2;
|
| + size_t size_vertex = 0;
|
| + size_t size_fragment = 0;
|
| + char *yuv_to_rgb_vertex = static_cast<char *>(
|
| + MmapFile(vertex, &size_vertex));
|
| + char *yuv_to_rgb_fragment = static_cast<char *>(
|
| + MmapFile(fragment, &size_fragment));
|
| + GLuint program = 0;
|
| +
|
| + if (!yuv_to_rgb_fragment || !yuv_to_rgb_vertex)
|
| + goto done;
|
| +
|
| + {
|
| + program = InitShaderProgram(yuv_to_rgb_vertex, yuv_to_rgb_fragment);
|
| +
|
| + int imageWidthUniform = glGetUniformLocation(program, "imageWidth");
|
| + int imageHeightUniform = glGetUniformLocation(program, "imageHeight");
|
| + int textureSampler = glGetUniformLocation(program, "textureSampler");
|
| + int evenLinesSampler = glGetUniformLocation(program, "paritySampler");
|
| +
|
| + glUniform1f(imageWidthUniform, width);
|
| + glUniform1f(imageHeightUniform, height);
|
| + glUniform1i(textureSampler, 0);
|
| + glUniform1i(evenLinesSampler, 1);
|
| +
|
| + int attribute_index = glGetAttribLocation(program, "c");
|
| + glBindBuffer(GL_ARRAY_BUFFER, vertex_buffer);
|
| + glVertexAttribPointer(attribute_index, 2, GL_FLOAT, GL_FALSE, 0, NULL);
|
| + glEnableVertexAttribArray(attribute_index);
|
| + return program;
|
| + }
|
| +
|
| +
|
| +done:
|
| + munmap(yuv_to_rgb_fragment, size_fragment);
|
| + munmap(yuv_to_rgb_fragment, size_vertex);
|
| + return program;
|
| +}
|
| +
|
| +
|
| +bool YuvToRgbTest::Run() {
|
| + size_t size = 0;
|
| + GLuint texture[2];
|
| + GLuint program = 0;
|
| + GLuint vertex_buffer = 0;
|
| + GLfloat vertices[8] = {
|
| + 0.f, 0.f,
|
| + 1.f, 0.f,
|
| + 0.f, 1.f,
|
| + 1.f, 1.f,
|
| + };
|
| + char evenodd[2] = {0, 255};
|
| + const int pixel_height = YUV2RGB_HEIGHT * 2 / 3;
|
| +
|
| + char *pixels = static_cast<char *>(MmapFile(YUV2RGB_NAME, &size));
|
| + if (!pixels) {
|
| + printf("# Could not open image file: %s\n", YUV2RGB_NAME);
|
| + goto done;
|
| + }
|
| + if (size != YUV2RGB_SIZE) {
|
| + printf("# Image file of wrong size, got %d, expected %d\n",
|
| + static_cast<int>(size), YUV2RGB_SIZE);
|
| + goto done;
|
| + }
|
| +
|
| + glGenTextures(2, texture);
|
| + glActiveTexture(GL_TEXTURE0);
|
| + glBindTexture(GL_TEXTURE_2D, texture[0]);
|
| + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
|
| + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
|
| + glTexImage2D(GL_TEXTURE_2D, 0, GL_LUMINANCE, YUV2RGB_WIDTH, YUV2RGB_HEIGHT,
|
| + 0, GL_LUMINANCE, GL_UNSIGNED_BYTE, pixels);
|
| +
|
| + glActiveTexture(GL_TEXTURE1);
|
| + glBindTexture(GL_TEXTURE_2D, texture[1]);
|
| + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
|
| + glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
|
| + glTexImage2D(GL_TEXTURE_2D, 0, GL_LUMINANCE, 2, 1,
|
| + 0, GL_LUMINANCE, GL_UNSIGNED_BYTE, evenodd);
|
| +
|
| + glViewport(-YUV2RGB_WIDTH, -pixel_height, YUV2RGB_WIDTH*2, pixel_height * 2);
|
| + vertex_buffer = SetupVBO(GL_ARRAY_BUFFER, sizeof(vertices), vertices);
|
| +
|
| + program = YuvToRgbShaderProgram(type_, vertex_buffer,
|
| + YUV2RGB_WIDTH, pixel_height);
|
| +
|
| + if (program) {
|
| + FillRateTestNormalSubWindow(name_, std::min(YUV2RGB_WIDTH, g_width),
|
| + std::min(pixel_height, g_height));
|
| + } else {
|
| + printf("# Could not set up YUV shader.\n");
|
| + }
|
| +
|
| +done:
|
| + glDeleteProgram(program);
|
| + glDeleteTextures(2, texture);
|
| + glDeleteBuffers(1, &vertex_buffer);
|
| + munmap(pixels, size);
|
| +
|
| + return true;
|
| +}
|
| +
|
| +
|
| +TestBase* GetYuvToRgbTest(int type, const char* name) {
|
| + return new YuvToRgbTest(type, name);
|
| +}
|
| +
|
| +} // namespace glbench
|
|
|