Program.cpp revision ac670c0433d19397d4e36ced2110475b6f54fe26 
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#define LOG_TAG "OpenGLRenderer"
18
19#include "Program.h"
20
21namespace android {
22namespace uirenderer {
23
24///////////////////////////////////////////////////////////////////////////////
25// Shaders
26///////////////////////////////////////////////////////////////////////////////
27
28#define SHADER_SOURCE(name, source) const char* name = #source
29
30#include "shaders/drawColor.frag"
31
32#include "shaders/drawTexture.vert"
33#include "shaders/drawTexture.frag"
34
35#include "shaders/drawText.frag"
36
37#include "shaders/drawLinearGradient.vert"
38#include "shaders/drawLinearGradient.frag"
39
40///////////////////////////////////////////////////////////////////////////////
41// Base program
42///////////////////////////////////////////////////////////////////////////////
43
44Program::Program(const char* vertex, const char* fragment) {
45    vertexShader = buildShader(vertex, GL_VERTEX_SHADER);
46    fragmentShader = buildShader(fragment, GL_FRAGMENT_SHADER);
47
48    id = glCreateProgram();
49    glAttachShader(id, vertexShader);
50    glAttachShader(id, fragmentShader);
51    glLinkProgram(id);
52
53    GLint status;
54    glGetProgramiv(id, GL_LINK_STATUS, &status);
55    if (status != GL_TRUE) {
56        GLint infoLen = 0;
57        glGetProgramiv(id, GL_INFO_LOG_LENGTH, &infoLen);
58        if (infoLen > 1) {
59            char* log = (char*) malloc(sizeof(char) * infoLen);
60            glGetProgramInfoLog(id, infoLen, 0, log);
61            LOGE("Error while linking shaders: %s", log);
62            delete log;
63        }
64        glDeleteProgram(id);
65    }
66
67    mUse = false;
68}
69
70Program::~Program() {
71    glDeleteShader(vertexShader);
72    glDeleteShader(fragmentShader);
73    glDeleteProgram(id);
74}
75
76void Program::use() {
77    glUseProgram(id);
78    mUse = true;
79}
80
81void Program::remove() {
82    mUse = false;
83}
84
85int Program::addAttrib(const char* name) {
86    int slot = glGetAttribLocation(id, name);
87    attributes.add(name, slot);
88    return slot;
89}
90
91int Program::getAttrib(const char* name) {
92    return attributes.valueFor(name);
93}
94
95int Program::addUniform(const char* name) {
96    int slot = glGetUniformLocation(id, name);
97    uniforms.add(name, slot);
98    return slot;
99}
100
101int Program::getUniform(const char* name) {
102    return uniforms.valueFor(name);
103}
104
105GLuint Program::buildShader(const char* source, GLenum type) {
106    GLuint shader = glCreateShader(type);
107    glShaderSource(shader, 1, &source, 0);
108    glCompileShader(shader);
109
110    GLint status;
111    glGetShaderiv(shader, GL_COMPILE_STATUS, &status);
112    if (status != GL_TRUE) {
113        // Some drivers return wrong values for GL_INFO_LOG_LENGTH
114        // use a fixed size instead
115        GLchar log[512];
116        glGetShaderInfoLog(shader, sizeof(log), 0, &log[0]);
117        LOGE("Error while compiling shader: %s", log);
118        glDeleteShader(shader);
119    }
120
121    return shader;
122}
123
124///////////////////////////////////////////////////////////////////////////////
125// Draw color
126///////////////////////////////////////////////////////////////////////////////
127
128DrawColorProgram::DrawColorProgram():
129        Program(gDrawTextureVertexShader, gDrawColorFragmentShader) {
130    getAttribsAndUniforms();
131}
132
133DrawColorProgram::DrawColorProgram(const char* vertex, const char* fragment):
134        Program(vertex, fragment) {
135    getAttribsAndUniforms();
136}
137
138void DrawColorProgram::getAttribsAndUniforms() {
139    position = addAttrib("position");
140    texCoords = addAttrib("texCoords");
141    color = addUniform("color");
142    transform = addUniform("transform");
143}
144
145void DrawColorProgram::set(const mat4& projectionMatrix, const mat4& modelViewMatrix,
146        const mat4& transformMatrix) {
147    mat4 t(projectionMatrix);
148    t.multiply(transformMatrix);
149    t.multiply(modelViewMatrix);
150
151    glUniformMatrix4fv(transform, 1, GL_FALSE, &t.data[0]);
152}
153
154void DrawColorProgram::use() {
155    Program::use();
156    glEnableVertexAttribArray(position);
157    glEnableVertexAttribArray(texCoords);
158}
159
160void DrawColorProgram::remove() {
161    Program::remove();
162    glDisableVertexAttribArray(position);
163    glDisableVertexAttribArray(texCoords);
164}
165
166///////////////////////////////////////////////////////////////////////////////
167// Draw texture
168///////////////////////////////////////////////////////////////////////////////
169
170DrawTextureProgram::DrawTextureProgram():
171        DrawColorProgram(gDrawTextureVertexShader, gDrawTextureFragmentShader) {
172    sampler = addUniform("sampler");
173}
174
175DrawTextureProgram::DrawTextureProgram(const char* vertex, const char* fragment):
176        DrawColorProgram(vertex, fragment) {
177    sampler = addUniform("sampler");
178}
179
180void DrawTextureProgram::use() {
181    DrawColorProgram::use();
182    glUniform1i(sampler, 0);
183}
184
185void DrawTextureProgram::remove() {
186    DrawColorProgram::remove();
187}
188
189///////////////////////////////////////////////////////////////////////////////
190// Draw text
191///////////////////////////////////////////////////////////////////////////////
192
193DrawTextProgram::DrawTextProgram():
194        DrawTextureProgram(gDrawTextureVertexShader, gDrawTextFragmentShader) {
195}
196
197///////////////////////////////////////////////////////////////////////////////
198// Draw linear gradient
199///////////////////////////////////////////////////////////////////////////////
200
201DrawLinearGradientProgram::DrawLinearGradientProgram():
202        DrawColorProgram(gDrawLinearGradientVertexShader, gDrawLinearGradientFragmentShader) {
203    gradient = addUniform("gradient");
204    gradientLength = addUniform("gradientLength");
205    sampler = addUniform("sampler");
206    start = addUniform("start");
207    screenSpace = addUniform("screenSpace");
208}
209
210void DrawLinearGradientProgram::use() {
211    DrawColorProgram::use();
212    glActiveTexture(GL_TEXTURE0);
213    glUniform1i(sampler, 0);
214}
215
216void DrawLinearGradientProgram::remove() {
217    DrawColorProgram::remove();
218}
219
220}; // namespace uirenderer
221}; // namespace android
222