1 /*
2  * Copyright 2013 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 #ifndef SCREENRECORD_PROGRAM_H
18 #define SCREENRECORD_PROGRAM_H
19 
20 #include <utils/Errors.h>
21 
22 #include <EGL/egl.h>
23 #include <GLES2/gl2.h>
24 
25 namespace android {
26 
27 /*
28  * Utility class for GLES rendering.
29  *
30  * Not thread-safe.
31  */
32 class Program {
33 public:
34     enum ProgramType { PROGRAM_UNKNOWN=0, PROGRAM_EXTERNAL_TEXTURE,
35             PROGRAM_TEXTURE_2D };
36 
Program()37     Program() :
38         mProgramType(PROGRAM_UNKNOWN),
39         mProgram(0),
40         maPositionLoc(0),
41         maTextureCoordLoc(0),
42         muMVPMatrixLoc(0),
43         muGLCMatrixLoc(0),
44         muTextureLoc(0)
45         {}
~Program()46     ~Program() { release(); }
47 
48     // Initialize the program for use with the specified texture type.
49     status_t setup(ProgramType type);
50 
51     // Release the program and associated resources.
52     void release();
53 
54     // Blit the specified texture to { x, y, x+w, y+h }.  Inverts the
55     // content if "invert" is set.
56     status_t blit(GLuint texName, const float* texMatrix,
57             int32_t x, int32_t y, int32_t w, int32_t h,
58             bool invert = false) const;
59 
60     // Draw a number of triangles.
61     status_t drawTriangles(GLuint texName, const float* texMatrix,
62             const float* vertices, const float* texes, size_t count) const;
63 
64     static const float kIdentity[];
65 
66 private:
67     Program(const Program&);
68     Program& operator=(const Program&);
69 
70     // Common code for draw functions.
71     status_t beforeDraw(GLuint texName, const float* texMatrix,
72             const float* vertices, const float* texes, bool invert) const;
73     status_t afterDraw() const;
74 
75     // GLES 2 shader utilities.
76     status_t createProgram(GLuint* outPgm, const char* vertexShader,
77             const char* fragmentShader);
78     static status_t compileShader(GLenum shaderType, const char* src,
79             GLuint* outShader);
80     static status_t linkShaderProgram(GLuint vs, GLuint fs, GLuint* outPgm);
81 
82     ProgramType mProgramType;
83     GLuint mProgram;
84 
85     GLint maPositionLoc;
86     GLint maTextureCoordLoc;
87     GLint muMVPMatrixLoc;
88     GLint muGLCMatrixLoc;
89     GLint muTextureLoc;
90 };
91 
92 }; // namespace android
93 
94 #endif /*SCREENRECORD_PROGRAM_H*/
95