Project

General

Profile

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

library / src / main / java / org / distorted / library / program / DistortedProgram.java @ 7602a827

1 432442f9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
2
// Copyright 2016 Leszek Koltunski                                                               //
3
//                                                                                               //
4 7602a827 Leszek Koltunski
// This file is part of DistortedLibrary.                                                               //
5 432442f9 Leszek Koltunski
//                                                                                               //
6 7602a827 Leszek Koltunski
// DistortedLibrary is free software: you can redistribute it and/or modify                             //
7 432442f9 Leszek Koltunski
// 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 7602a827 Leszek Koltunski
// DistortedLibrary is distributed in the hope that it will be useful,                                  //
12 432442f9 Leszek Koltunski
// 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 7602a827 Leszek Koltunski
// along with DistortedLibrary.  If not, see <http://www.gnu.org/licenses/>.                            //
18 432442f9 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
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 a13dde77 Leszek Koltunski
  public DistortedProgram(final InputStream vert, final InputStream frag, final String vertHeader, final String fragHeader,
352
                          int glslVersion, final String[] feedback )
353
  throws FragmentCompilationException,VertexCompilationException,LinkingException
354 432442f9 Leszek Koltunski
    {
355 041b6dee Leszek Koltunski
    init(glslVersion);
356 432442f9 Leszek Koltunski
357 a13dde77 Leszek Koltunski
    final String vertShader = readTextFileFromRawResource(vert, true );
358
    final String fragShader = readTextFileFromRawResource(frag, false);
359 432442f9 Leszek Koltunski
360 a13dde77 Leszek Koltunski
    final int vertShaderHandle = compileShader(GLES31.GL_VERTEX_SHADER  , vertHeader + vertShader);
361
    final int fragShaderHandle = compileShader(GLES31.GL_FRAGMENT_SHADER, fragHeader + fragShader);
362 432442f9 Leszek Koltunski
363 a13dde77 Leszek Koltunski
    mProgramHandle = createAndLinkProgram(vertShaderHandle, fragShaderHandle, mAttributeName, glslVersion>= 300 ? feedback:null );
364 7cd24173 leszek
365
    mAttribute = new int[mNumAttributes];
366
367
    for(int i=0; i<mNumAttributes; i++)
368
      {
369 e6519ac8 Leszek Koltunski
      mAttribute[i] = GLES31.glGetAttribLocation( mProgramHandle, mAttributeName[i]);
370 7cd24173 leszek
      }
371 041b6dee Leszek Koltunski
372
    if( mNumUniforms>0 )
373
      {
374
      mUniform = new int[mNumUniforms];
375
      mUniformName = mUniList.split(" ");
376
377
      for(int i=0; i<mNumUniforms; i++)
378
        {
379 e6519ac8 Leszek Koltunski
        mUniform[i] = GLES31.glGetUniformLocation( mProgramHandle, mUniformName[i]);
380 041b6dee Leszek Koltunski
        }
381
      }
382
    else mUniform = null;
383
    }
384
385
///////////////////////////////////////////////////////////////////////////////////////////////////
386 faa3ff56 Leszek Koltunski
/**
387
 * Only for use by the library itself.
388
 *
389
 * @y.exclude
390
 */
391 a13dde77 Leszek Koltunski
  public DistortedProgram(final InputStream vert, final InputStream frag, final String vertHeader, final String fragHeader,
392
                          final String enabledVert, final String enabledFrag, int glslVersion, final String[] feedback )
393
  throws FragmentCompilationException,VertexCompilationException,LinkingException
394 041b6dee Leszek Koltunski
    {
395 faa3ff56 Leszek Koltunski
    init(glslVersion);
396 041b6dee Leszek Koltunski
397 a13dde77 Leszek Koltunski
    String vertShader = readTextFileFromRawResource( vert, true );
398
    String fragShader = readTextFileFromRawResource( frag, false);
399 041b6dee Leszek Koltunski
400 a13dde77 Leszek Koltunski
    if( enabledVert!=null ) vertShader = insertEnabledEffects(vertShader,enabledVert);
401
    if( enabledFrag!=null ) fragShader = insertEnabledEffects(fragShader,enabledFrag);
402 041b6dee Leszek Koltunski
403 a13dde77 Leszek Koltunski
    final int vertShaderHandle = compileShader(GLES31.GL_VERTEX_SHADER  , vertHeader + vertShader);
404
    final int fragShaderHandle = compileShader(GLES31.GL_FRAGMENT_SHADER, fragHeader + fragShader);
405 faa3ff56 Leszek Koltunski
406 a13dde77 Leszek Koltunski
    mProgramHandle = createAndLinkProgram(vertShaderHandle, fragShaderHandle, mAttributeName, glslVersion>= 300 ? feedback:null );
407 041b6dee Leszek Koltunski
408
    mAttribute = new int[mNumAttributes];
409
410
    for(int i=0; i<mNumAttributes; i++)
411
      {
412 e6519ac8 Leszek Koltunski
      mAttribute[i] = GLES31.glGetAttribLocation( mProgramHandle, mAttributeName[i]);
413 041b6dee Leszek Koltunski
      }
414
415
    if( mNumUniforms>0 )
416
      {
417
      mUniform = new int[mNumUniforms];
418
      mUniformName = mUniList.split(" ");
419
420
      for(int i=0; i<mNumUniforms; i++)
421
        {
422 e6519ac8 Leszek Koltunski
        mUniform[i] = GLES31.glGetUniformLocation( mProgramHandle, mUniformName[i]);
423 041b6dee Leszek Koltunski
        }
424
      }
425
    else mUniform = null;
426 7cd24173 leszek
    }
427
428
///////////////////////////////////////////////////////////////////////////////////////////////////
429 faa3ff56 Leszek Koltunski
/**
430
 * Only for use by the library itself.
431
 *
432
 * @y.exclude
433
 */
434
  public DistortedProgram(final InputStream vertex, final InputStream fragment, final String vertexHeader, final String fragmentHeader, int glslVersion )
435 7cd24173 leszek
  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException
436
    {
437 faa3ff56 Leszek Koltunski
    this(vertex,fragment,vertexHeader,fragmentHeader,glslVersion,null);
438
    }
439 7cd24173 leszek
440 faa3ff56 Leszek Koltunski
///////////////////////////////////////////////////////////////////////////////////////////////////
441
// PUBLIC API
442
///////////////////////////////////////////////////////////////////////////////////////////////////
443
/**
444
 * Create a new Shader Program from two source strings.
445
 * <p>
446
 * Needs to be called from a thread holding the OpenGL context.
447
 *
448
 * @param vertex   Vertex shader code.
449
 * @param fragment Fragment shader code.
450
 * @throws FragmentCompilationException
451
 * @throws VertexCompilationException
452
 * @throws VertexUniformsException
453
 * @throws FragmentUniformsException
454
 * @throws LinkingException
455
 */
456
  public DistortedProgram(final String vertex, final String fragment)
457
  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException
458
    {
459
    init(300);
460 7cd24173 leszek
461 faa3ff56 Leszek Koltunski
    doAttributes(vertex  , true );
462
    doAttributes(fragment, false);
463 7cd24173 leszek
464 e6519ac8 Leszek Koltunski
    final int vertexShaderHandle   = compileShader(GLES31.GL_VERTEX_SHADER  , vertex  );
465
    final int fragmentShaderHandle = compileShader(GLES31.GL_FRAGMENT_SHADER, fragment);
466 7cd24173 leszek
467 faa3ff56 Leszek Koltunski
    mProgramHandle = createAndLinkProgram(vertexShaderHandle, fragmentShaderHandle, mAttributeName, null );
468 432442f9 Leszek Koltunski
469 02de77c9 Leszek Koltunski
    mAttribute = new int[mNumAttributes];
470 432442f9 Leszek Koltunski
471 02de77c9 Leszek Koltunski
    for(int i=0; i<mNumAttributes; i++)
472
      {
473 e6519ac8 Leszek Koltunski
      mAttribute[i] = GLES31.glGetAttribLocation( mProgramHandle, mAttributeName[i]);
474 02de77c9 Leszek Koltunski
      }
475 041b6dee Leszek Koltunski
476
    if( mNumUniforms>0 )
477
      {
478
      mUniform = new int[mNumUniforms];
479
      mUniformName = mUniList.split(" ");
480
481
      for(int i=0; i<mNumUniforms; i++)
482
        {
483 e6519ac8 Leszek Koltunski
        mUniform[i] = GLES31.glGetUniformLocation( mProgramHandle, mUniformName[i]);
484 041b6dee Leszek Koltunski
        }
485
      }
486
    else mUniform = null;
487 432442f9 Leszek Koltunski
    }
488
489
///////////////////////////////////////////////////////////////////////////////////////////////////
490 3d804c91 Leszek Koltunski
/**
491
 * Return the handle of the created program so that we can later, say, call glUseProgram.
492
 */
493 432442f9 Leszek Koltunski
  public int getProgramHandle()
494
    {
495
    return mProgramHandle;
496
    }
497
498
///////////////////////////////////////////////////////////////////////////////////////////////////
499 3d804c91 Leszek Koltunski
/**
500 02de77c9 Leszek Koltunski
 * Use the program and enable all vertex attribute arrays.
501
 *
502 3d804c91 Leszek Koltunski
 * Needs to be called from a thread holding the OpenGL context.
503
 */
504 02de77c9 Leszek Koltunski
  public void useProgram()
505 432442f9 Leszek Koltunski
    {
506 e6519ac8 Leszek Koltunski
    GLES31.glUseProgram(mProgramHandle);
507 02de77c9 Leszek Koltunski
508 432442f9 Leszek Koltunski
    for(int i=0; i<mNumAttributes; i++)
509 e6519ac8 Leszek Koltunski
      GLES31.glEnableVertexAttribArray(mAttribute[i]);
510 432442f9 Leszek Koltunski
    }
511
  }
512 02de77c9 Leszek Koltunski