Project

General

Profile

Download (17.1 KB) Statistics
| Branch: | Revision:

library / src / main / java / org / distorted / library / program / DistortedProgram.java @ 040cd18c

1 432442f9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
2
// Copyright 2016 Leszek Koltunski                                                               //
3
//                                                                                               //
4
// This file is part of Distorted.                                                               //
5
//                                                                                               //
6
// Distorted is free software: you can redistribute it and/or modify                             //
7
// it under the terms of the GNU General Public License as published by                          //
8
// the Free Software Foundation, either version 2 of the License, or                             //
9
// (at your option) any later version.                                                           //
10
//                                                                                               //
11
// Distorted is distributed in the hope that it will be useful,                                  //
12
// but WITHOUT ANY WARRANTY; without even the implied warranty of                                //
13
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                                 //
14
// GNU General Public License for more details.                                                  //
15
//                                                                                               //
16
// You should have received a copy of the GNU General Public License                             //
17
// along with Distorted.  If not, see <http://www.gnu.org/licenses/>.                            //
18
///////////////////////////////////////////////////////////////////////////////////////////////////
19
20
package org.distorted.library.program;
21
22 e6519ac8 Leszek Koltunski
import android.opengl.GLES31;
23 432442f9 Leszek Koltunski
24
import java.io.BufferedReader;
25
import java.io.IOException;
26
import java.io.InputStream;
27
import java.io.InputStreamReader;
28
29
///////////////////////////////////////////////////////////////////////////////////////////////////
30
/**
31
 * An object which encapsulates a vertex/fragment shader combo, aka Shader Program.
32
 */
33
public class DistortedProgram
34
  {
35 041b6dee Leszek Koltunski
  private String mAttributeStr, mUniformStr, mUniList;
36
  private int mAttributeLen, mUniformLen;
37 94f6d472 Leszek Koltunski
  private int mProgramHandle;
38 432442f9 Leszek Koltunski
  private int mNumAttributes;
39 041b6dee Leszek Koltunski
  private int mNumUniforms;
40 432442f9 Leszek Koltunski
  private String[] mAttributeName;
41 041b6dee Leszek Koltunski
  private String[] mUniformName;
42 432442f9 Leszek Koltunski
43 faa3ff56 Leszek Koltunski
/**
44 f81ebc3f Leszek Koltunski
 * List of Attributes (OpenGL ES 3.0: 'in' variables), in the same order as declared in the shader source.
45 faa3ff56 Leszek Koltunski
 */
46 02de77c9 Leszek Koltunski
  public final int[] mAttribute;
47 faa3ff56 Leszek Koltunski
/**
48 f81ebc3f Leszek Koltunski
 * List of Uniforms, in the same order as declared in the shader source.
49 faa3ff56 Leszek Koltunski
 */
50 041b6dee Leszek Koltunski
  public final int[] mUniform;
51 02de77c9 Leszek Koltunski
52 432442f9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
53
54 4782b4e6 Leszek Koltunski
  private int createAndLinkProgram(final int vertexShaderHandle, final int fragmentShaderHandle, final String[] attributes, final String[] feedbackVaryings)
55 432442f9 Leszek Koltunski
  throws LinkingException
56
    {
57 e6519ac8 Leszek Koltunski
    int programHandle = GLES31.glCreateProgram();
58 432442f9 Leszek Koltunski
59
    if (programHandle != 0)
60
      {
61 e6519ac8 Leszek Koltunski
      GLES31.glAttachShader(programHandle, vertexShaderHandle);
62
      GLES31.glAttachShader(programHandle, fragmentShaderHandle);
63 432442f9 Leszek Koltunski
64 4782b4e6 Leszek Koltunski
      if( feedbackVaryings!=null )
65
        {
66 e6519ac8 Leszek Koltunski
        GLES31.glTransformFeedbackVaryings(programHandle, feedbackVaryings, GLES31.GL_INTERLEAVED_ATTRIBS);
67 4782b4e6 Leszek Koltunski
        }
68
69 432442f9 Leszek Koltunski
      if (attributes != null)
70
        {
71
        final int size = attributes.length;
72
73
        for (int i = 0; i < size; i++)
74
          {
75 e6519ac8 Leszek Koltunski
          GLES31.glBindAttribLocation(programHandle, i, attributes[i]);
76 432442f9 Leszek Koltunski
          }
77
        }
78
79 e6519ac8 Leszek Koltunski
      GLES31.glLinkProgram(programHandle);
80 432442f9 Leszek Koltunski
81
      final int[] linkStatus = new int[1];
82 e6519ac8 Leszek Koltunski
      GLES31.glGetProgramiv(programHandle, GLES31.GL_LINK_STATUS, linkStatus, 0);
83 432442f9 Leszek Koltunski
84 e6519ac8 Leszek Koltunski
      if (linkStatus[0] != GLES31.GL_TRUE )
85 432442f9 Leszek Koltunski
        {
86 e6519ac8 Leszek Koltunski
        String error = GLES31.glGetProgramInfoLog(programHandle);
87
        GLES31.glDeleteProgram(programHandle);
88 432442f9 Leszek Koltunski
        throw new LinkingException(error);
89
        }
90
91
      final int[] numberOfUniforms = new int[1];
92 e6519ac8 Leszek Koltunski
      GLES31.glGetProgramiv(programHandle, GLES31.GL_ACTIVE_UNIFORMS, numberOfUniforms, 0);
93 432442f9 Leszek Koltunski
94
      //android.util.Log.d("program", "number of active uniforms="+numberOfUniforms[0]);
95
      }
96
97
    return programHandle;
98
    }
99
100 041b6dee Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
101
102
  private void init(int glslVersion)
103
    {
104
    mAttributeStr  = (glslVersion == 100 ? "attribute " : "in ");
105
    mAttributeLen  = mAttributeStr.length();
106
    mNumAttributes = 0;
107
    mUniformStr    = "uniform ";
108
    mUniformLen    = mUniformStr.length();
109
    mNumUniforms   = 0;
110
    mUniList       = "";
111
    }
112
113
///////////////////////////////////////////////////////////////////////////////////////////////////
114
115
  private String parseOutUniform(final String line)
116
    {
117
    int len = line.length();
118
    int whiteSpace, semicolon, nameBegin;
119
    char currChar;
120
121
    for(whiteSpace=0; whiteSpace<len; whiteSpace++)
122
      {
123
      currChar = line.charAt(whiteSpace);
124
      if( currChar!=' ' && currChar!='\t') break;
125
      }
126
127
    for(semicolon=whiteSpace; semicolon<len; semicolon++)
128
      {
129
      currChar = line.charAt(semicolon);
130
      if( currChar==';') break;
131
      }
132
133
    if( semicolon<len && semicolon-whiteSpace>=mUniformLen+1 )
134
      {
135
      String subline = line.substring(whiteSpace,semicolon);
136
      int subLen = semicolon-whiteSpace;
137
138
      if( subline.startsWith(mUniformStr))
139
        {
140
        //android.util.Log.e("program", "GOOD LINE: " +subline+" subLen="+subLen);
141
142
        for(nameBegin=subLen-1; nameBegin>mUniformLen-2; nameBegin--)
143
          {
144
          currChar=subline.charAt(nameBegin);
145
146
          if( currChar==' ' || currChar=='\t' )
147
            {
148
            mNumUniforms++;
149
            String uniform = subline.substring(nameBegin+1,subLen);
150
            int brace = uniform.indexOf("[");
151
152
            return brace>=0 ? uniform.substring(0,brace) : uniform;
153
            }
154
          }
155
        }
156
      }
157
158
    return null;
159
    }
160
161 432442f9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
162
163 3d804c91 Leszek Koltunski
  private String parseOutAttribute(final String line)
164 432442f9 Leszek Koltunski
    {
165
    int len = line.length();
166
    int whiteSpace, semicolon, nameBegin;
167
    char currChar;
168
169
    for(whiteSpace=0; whiteSpace<len; whiteSpace++)
170
      {
171
      currChar = line.charAt(whiteSpace);
172
      if( currChar!=' ' && currChar!='\t') break;
173
      }
174
175
    for(semicolon=whiteSpace; semicolon<len; semicolon++)
176
      {
177
      currChar = line.charAt(semicolon);
178
      if( currChar==';') break;
179
      }
180
181 94f6d472 Leszek Koltunski
    if( semicolon<len && semicolon-whiteSpace>=mAttributeLen+1 )
182 432442f9 Leszek Koltunski
      {
183
      String subline = line.substring(whiteSpace,semicolon);
184
      int subLen = semicolon-whiteSpace;
185
186 94f6d472 Leszek Koltunski
      if( subline.startsWith(mAttributeStr))
187 432442f9 Leszek Koltunski
        {
188
        //android.util.Log.e("program", "GOOD LINE: " +subline+" subLen="+subLen);
189
190 94f6d472 Leszek Koltunski
        for(nameBegin=subLen-1; nameBegin>mAttributeLen-2; nameBegin--)
191 432442f9 Leszek Koltunski
          {
192
          currChar=subline.charAt(nameBegin);
193
194
          if( currChar==' ' || currChar=='\t' )
195
            {
196
            mNumAttributes++;
197 3d804c91 Leszek Koltunski
            return subline.substring(nameBegin+1,subLen);
198 432442f9 Leszek Koltunski
            }
199
          }
200
        }
201
      }
202 3d804c91 Leszek Koltunski
203
    return null;
204 432442f9 Leszek Koltunski
    }
205
206 041b6dee Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
207
208
  private void doAttributes(final String shader, boolean doAttributes)
209
    {
210
    String attribute, attrList="", uniform;
211
    String[] lines = shader.split("\n");
212
    int length = lines.length;
213
214
    for(int i=0; i<length; i++)
215
      {
216
      if( doAttributes )
217
        {
218
        attribute = parseOutAttribute(lines[i]);
219
220
        if( attribute!=null )
221
          {
222
          //android.util.Log.d("program", "new attribute: "+attribute);
223
          if( attrList.length()>0 ) attrList+=" ";
224
          attrList += attribute;
225
          }
226
        }
227
228
      uniform = parseOutUniform(lines[i]);
229
230
      if( uniform!=null )
231
        {
232 aa2f0486 Leszek Koltunski
        //android.util.Log.d("program", "new uniform: "+uniform);
233 041b6dee Leszek Koltunski
        if( mUniList.length()>0 ) mUniList+=" ";
234
        mUniList += uniform;
235
        }
236
      }
237
238
    if( doAttributes ) mAttributeName = attrList.split(" ");
239
    }
240
241 432442f9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
242
243
  private String readTextFileFromRawResource(final InputStream inputStream, boolean doAttributes)
244
    {
245
    final InputStreamReader inputStreamReader = new InputStreamReader(inputStream);
246
    final BufferedReader bufferedReader = new BufferedReader(inputStreamReader);
247
248 3d804c91 Leszek Koltunski
    String nextLine, attribute, attrList="";
249 432442f9 Leszek Koltunski
    final StringBuilder body = new StringBuilder();
250
251
    try
252
      {
253
      while ((nextLine = bufferedReader.readLine()) != null)
254
        {
255
        body.append(nextLine);
256
        body.append('\n');
257
258 3d804c91 Leszek Koltunski
        if( doAttributes )
259
          {
260
          attribute = parseOutAttribute(nextLine);
261
262
          if( attribute!=null )
263
            {
264
            //android.util.Log.d("program", "new attribute: "+attribute);
265
            if( attrList.length()>0 ) attrList+=" ";
266
            attrList += attribute;
267
            }
268
          }
269 432442f9 Leszek Koltunski
        }
270
      }
271
    catch (IOException e)
272
      {
273
      return null;
274
      }
275
276 041b6dee Leszek Koltunski
    if( doAttributes ) mAttributeName = attrList.split(" ");
277 432442f9 Leszek Koltunski
278
    return body.toString();
279
    }
280
281
///////////////////////////////////////////////////////////////////////////////////////////////////
282
283
  private static int compileShader(final int shaderType, final String shaderSource)
284
  throws FragmentCompilationException,VertexCompilationException
285
    {
286 e6519ac8 Leszek Koltunski
    int shaderHandle = GLES31.glCreateShader(shaderType);
287 432442f9 Leszek Koltunski
288
    if (shaderHandle != 0)
289
      {
290 e6519ac8 Leszek Koltunski
      GLES31.glShaderSource(shaderHandle, shaderSource);
291
      GLES31.glCompileShader(shaderHandle);
292 432442f9 Leszek Koltunski
      final int[] compileStatus = new int[1];
293 e6519ac8 Leszek Koltunski
      GLES31.glGetShaderiv(shaderHandle, GLES31.GL_COMPILE_STATUS, compileStatus, 0);
294 432442f9 Leszek Koltunski
295 341151fc Leszek Koltunski
      if (compileStatus[0] != GLES31.GL_TRUE)
296 432442f9 Leszek Koltunski
        {
297 341151fc Leszek Koltunski
        String error = GLES31.glGetShaderInfoLog(shaderHandle);
298 432442f9 Leszek Koltunski
299 341151fc Leszek Koltunski
        //android.util.Log.e("Program", "error compiling :" + error);
300 432442f9 Leszek Koltunski
301 341151fc Leszek Koltunski
        GLES31.glDeleteShader(shaderHandle);
302 81a0b906 leszek
303 341151fc Leszek Koltunski
        switch (shaderType)
304
          {
305
          case GLES31.GL_VERTEX_SHADER:   throw new VertexCompilationException(error);
306
          case GLES31.GL_FRAGMENT_SHADER: throw new FragmentCompilationException(error);
307
          default:                        throw new RuntimeException(error);
308
          }
309 432442f9 Leszek Koltunski
        }
310
      }
311
312
    return shaderHandle;
313
    }
314
315 7cd24173 leszek
///////////////////////////////////////////////////////////////////////////////////////////////////
316
317
  private static String insertEnabledEffects(String code, final String effects)
318
    {
319
    final String marker = "// ENABLED EFFECTS WILL BE INSERTED HERE";
320
    int length = marker.length();
321
322
    int place = code.indexOf(marker);
323
324
    if( place>=0 )
325
      {
326
      String begin = code.substring(0,place-1);
327
      String end   = code.substring(place+length);
328
/*
329
int len = begin.length();
330
331
android.util.Log.d("Program", begin.substring(len-40));
332
android.util.Log.d("Program", effects);
333
android.util.Log.d("Program", end.substring(0,40));
334
*/
335
      return begin + effects + end;
336
      }
337
    else
338
      {
339
      android.util.Log.e("Program", "Error: marker string not found in SHADER!");
340
      }
341
342
    return null;
343
    }
344
345 432442f9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
346 faa3ff56 Leszek Koltunski
/**
347
 * Only for use by the library itself.
348
 *
349
 * @y.exclude
350
 */
351 4782b4e6 Leszek Koltunski
  public DistortedProgram(final InputStream vertex, final InputStream fragment, final String vertexHeader, final String fragmentHeader, int glslVersion, final String[] feedback )
352 432442f9 Leszek Koltunski
  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException
353
    {
354 041b6dee Leszek Koltunski
    init(glslVersion);
355 432442f9 Leszek Koltunski
356
    final String vertexShader   = readTextFileFromRawResource(vertex  , true );
357
    final String fragmentShader = readTextFileFromRawResource(fragment, false);
358
359 e6519ac8 Leszek Koltunski
    final int vertexShaderHandle   = compileShader(GLES31.GL_VERTEX_SHADER  , vertexHeader   + vertexShader  );
360
    final int fragmentShaderHandle = compileShader(GLES31.GL_FRAGMENT_SHADER, fragmentHeader + fragmentShader);
361 432442f9 Leszek Koltunski
362 420836fc leszek
    mProgramHandle = createAndLinkProgram(vertexShaderHandle, fragmentShaderHandle, mAttributeName, glslVersion>= 300 ? feedback:null );
363 7cd24173 leszek
364
    mAttribute = new int[mNumAttributes];
365
366
    for(int i=0; i<mNumAttributes; i++)
367
      {
368 e6519ac8 Leszek Koltunski
      mAttribute[i] = GLES31.glGetAttribLocation( mProgramHandle, mAttributeName[i]);
369 7cd24173 leszek
      }
370 041b6dee Leszek Koltunski
371
    if( mNumUniforms>0 )
372
      {
373
      mUniform = new int[mNumUniforms];
374
      mUniformName = mUniList.split(" ");
375
376
      for(int i=0; i<mNumUniforms; i++)
377
        {
378 e6519ac8 Leszek Koltunski
        mUniform[i] = GLES31.glGetUniformLocation( mProgramHandle, mUniformName[i]);
379 041b6dee Leszek Koltunski
        }
380
      }
381
    else mUniform = null;
382
    }
383
384
///////////////////////////////////////////////////////////////////////////////////////////////////
385 faa3ff56 Leszek Koltunski
/**
386
 * Only for use by the library itself.
387
 *
388
 * @y.exclude
389
 */
390
  public DistortedProgram(final InputStream vertex, final InputStream fragment, final String vertexHeader, final String fragmentHeader,
391
                          final String enabledVertex, final String enabledFragment, int glslVersion, final String[] feedback )
392 041b6dee Leszek Koltunski
  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException
393
    {
394 faa3ff56 Leszek Koltunski
    init(glslVersion);
395 041b6dee Leszek Koltunski
396 faa3ff56 Leszek Koltunski
    String vertexShader   = readTextFileFromRawResource(vertex  , true );
397
    String fragmentShader = readTextFileFromRawResource(fragment, false);
398 041b6dee Leszek Koltunski
399 faa3ff56 Leszek Koltunski
    vertexShader   = insertEnabledEffects(vertexShader  ,enabledVertex  );
400
    fragmentShader = insertEnabledEffects(fragmentShader,enabledFragment);
401 041b6dee Leszek Koltunski
402 e6519ac8 Leszek Koltunski
    final int vertexShaderHandle   = compileShader(GLES31.GL_VERTEX_SHADER  , vertexHeader   + vertexShader  );
403
    final int fragmentShaderHandle = compileShader(GLES31.GL_FRAGMENT_SHADER, fragmentHeader + fragmentShader);
404 faa3ff56 Leszek Koltunski
405
    mProgramHandle = createAndLinkProgram(vertexShaderHandle, fragmentShaderHandle, mAttributeName, glslVersion>= 300 ? feedback:null );
406 041b6dee Leszek Koltunski
407
    mAttribute = new int[mNumAttributes];
408
409
    for(int i=0; i<mNumAttributes; i++)
410
      {
411 e6519ac8 Leszek Koltunski
      mAttribute[i] = GLES31.glGetAttribLocation( mProgramHandle, mAttributeName[i]);
412 041b6dee Leszek Koltunski
      }
413
414
    if( mNumUniforms>0 )
415
      {
416
      mUniform = new int[mNumUniforms];
417
      mUniformName = mUniList.split(" ");
418
419
      for(int i=0; i<mNumUniforms; i++)
420
        {
421 e6519ac8 Leszek Koltunski
        mUniform[i] = GLES31.glGetUniformLocation( mProgramHandle, mUniformName[i]);
422 041b6dee Leszek Koltunski
        }
423
      }
424
    else mUniform = null;
425 7cd24173 leszek
    }
426
427
///////////////////////////////////////////////////////////////////////////////////////////////////
428 faa3ff56 Leszek Koltunski
/**
429
 * Only for use by the library itself.
430
 *
431
 * @y.exclude
432
 */
433
  public DistortedProgram(final InputStream vertex, final InputStream fragment, final String vertexHeader, final String fragmentHeader, int glslVersion )
434 7cd24173 leszek
  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException
435
    {
436 faa3ff56 Leszek Koltunski
    this(vertex,fragment,vertexHeader,fragmentHeader,glslVersion,null);
437
    }
438 7cd24173 leszek
439 faa3ff56 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
440
// PUBLIC API
441
///////////////////////////////////////////////////////////////////////////////////////////////////
442
/**
443
 * Create a new Shader Program from two source strings.
444
 * <p>
445
 * Needs to be called from a thread holding the OpenGL context.
446
 *
447
 * @param vertex   Vertex shader code.
448
 * @param fragment Fragment shader code.
449
 * @throws FragmentCompilationException
450
 * @throws VertexCompilationException
451
 * @throws VertexUniformsException
452
 * @throws FragmentUniformsException
453
 * @throws LinkingException
454
 */
455
  public DistortedProgram(final String vertex, final String fragment)
456
  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException
457
    {
458
    init(300);
459 7cd24173 leszek
460 faa3ff56 Leszek Koltunski
    doAttributes(vertex  , true );
461
    doAttributes(fragment, false);
462 7cd24173 leszek
463 e6519ac8 Leszek Koltunski
    final int vertexShaderHandle   = compileShader(GLES31.GL_VERTEX_SHADER  , vertex  );
464
    final int fragmentShaderHandle = compileShader(GLES31.GL_FRAGMENT_SHADER, fragment);
465 7cd24173 leszek
466 faa3ff56 Leszek Koltunski
    mProgramHandle = createAndLinkProgram(vertexShaderHandle, fragmentShaderHandle, mAttributeName, null );
467 432442f9 Leszek Koltunski
468 02de77c9 Leszek Koltunski
    mAttribute = new int[mNumAttributes];
469 432442f9 Leszek Koltunski
470 02de77c9 Leszek Koltunski
    for(int i=0; i<mNumAttributes; i++)
471
      {
472 e6519ac8 Leszek Koltunski
      mAttribute[i] = GLES31.glGetAttribLocation( mProgramHandle, mAttributeName[i]);
473 02de77c9 Leszek Koltunski
      }
474 041b6dee Leszek Koltunski
475
    if( mNumUniforms>0 )
476
      {
477
      mUniform = new int[mNumUniforms];
478
      mUniformName = mUniList.split(" ");
479
480
      for(int i=0; i<mNumUniforms; i++)
481
        {
482 e6519ac8 Leszek Koltunski
        mUniform[i] = GLES31.glGetUniformLocation( mProgramHandle, mUniformName[i]);
483 041b6dee Leszek Koltunski
        }
484
      }
485
    else mUniform = null;
486 432442f9 Leszek Koltunski
    }
487
488
///////////////////////////////////////////////////////////////////////////////////////////////////
489 3d804c91 Leszek Koltunski
/**
490
 * Return the handle of the created program so that we can later, say, call glUseProgram.
491
 */
492 432442f9 Leszek Koltunski
  public int getProgramHandle()
493
    {
494
    return mProgramHandle;
495
    }
496
497
///////////////////////////////////////////////////////////////////////////////////////////////////
498 3d804c91 Leszek Koltunski
/**
499 02de77c9 Leszek Koltunski
 * Use the program and enable all vertex attribute arrays.
500
 *
501 3d804c91 Leszek Koltunski
 * Needs to be called from a thread holding the OpenGL context.
502
 */
503 02de77c9 Leszek Koltunski
  public void useProgram()
504 432442f9 Leszek Koltunski
    {
505 e6519ac8 Leszek Koltunski
    GLES31.glUseProgram(mProgramHandle);
506 02de77c9 Leszek Koltunski
507 432442f9 Leszek Koltunski
    for(int i=0; i<mNumAttributes; i++)
508 e6519ac8 Leszek Koltunski
      GLES31.glEnableVertexAttribArray(mAttribute[i]);
509 432442f9 Leszek Koltunski
    }
510
  }
511 02de77c9 Leszek Koltunski