Distorted Android

///////////////////////////////////////////////////////////////////////////////////////////////////

// Copyright 2016 Leszek Koltunski                                                               //

//                                                                                               //

// This file is part of Distorted.                                                               //

//                                                                                               //

// Distorted is free software: you can redistribute it and/or modify                             //

// it under the terms of the GNU General Public License as published by                          //

// the Free Software Foundation, either version 2 of the License, or                             //

// (at your option) any later version.                                                           //

//                                                                                               //

// Distorted is distributed in the hope that it will be useful,                                  //

// but WITHOUT ANY WARRANTY; without even the implied warranty of                                //

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                                 //

// GNU General Public License for more details.                                                  //

//                                                                                               //

// You should have received a copy of the GNU General Public License                             //

// along with Distorted.  If not, see <http://www.gnu.org/licenses/>.                            //

///////////////////////////////////////////////////////////////////////////////////////////////////

package org.distorted.library.program;

import android.opengl.GLES30;

import android.os.Build;

import org.distorted.library.effect.EffectType;

import org.distorted.library.main.DistortedEffects;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

///////////////////////////////////////////////////////////////////////////////////////////////////

/**

 * An object which encapsulates a vertex/fragment shader combo, aka Shader Program.

 */

public class DistortedProgram

  {

  private String mAttributeStr;

  private int mAttributeLen;

  private int mProgramHandle;

  private int mNumAttributes;

  private String[] mAttributeName;

  public final int[] mAttribute;

///////////////////////////////////////////////////////////////////////////////////////////////////

  private int createAndLinkProgram(final int vertexShaderHandle, final int fragmentShaderHandle, final String[] attributes, final String[] feedbackVaryings)

  throws LinkingException

    {

    int programHandle = GLES30.glCreateProgram();

    if (programHandle != 0)

      {

      GLES30.glAttachShader(programHandle, vertexShaderHandle);

      GLES30.glAttachShader(programHandle, fragmentShaderHandle);

      if( feedbackVaryings!=null )

        {

        GLES30.glTransformFeedbackVaryings(programHandle, feedbackVaryings, GLES30.GL_INTERLEAVED_ATTRIBS);

        }

      if (attributes != null)

        {

        final int size = attributes.length;

        for (int i = 0; i < size; i++)

          {

          GLES30.glBindAttribLocation(programHandle, i, attributes[i]);

          }

        }

      GLES30.glLinkProgram(programHandle);

      final int[] linkStatus = new int[1];

      GLES30.glGetProgramiv(programHandle, GLES30.GL_LINK_STATUS, linkStatus, 0);

      if (linkStatus[0] != GLES30.GL_TRUE )

        {

        String error = GLES30.glGetProgramInfoLog(programHandle);

        GLES30.glDeleteProgram(programHandle);

        throw new LinkingException(error);

        }

      final int[] numberOfUniforms = new int[1];

      GLES30.glGetProgramiv(programHandle, GLES30.GL_ACTIVE_UNIFORMS, numberOfUniforms, 0);

      //android.util.Log.d("program", "number of active uniforms="+numberOfUniforms[0]);

      }

    return programHandle;

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

  private String parseOutAttribute(final String line)

    {

    int len = line.length();

    int whiteSpace, semicolon, nameBegin;

    char currChar;

    for(whiteSpace=0; whiteSpace<len; whiteSpace++)

      {

      currChar = line.charAt(whiteSpace);

      if( currChar!=' ' && currChar!='\t') break;

      }

    for(semicolon=whiteSpace; semicolon<len; semicolon++)

      {

      currChar = line.charAt(semicolon);

      if( currChar==';') break;

      }

    if( semicolon<len && semicolon-whiteSpace>=mAttributeLen+1 )

      {

      String subline = line.substring(whiteSpace,semicolon);

      int subLen = semicolon-whiteSpace;

      if( subline.startsWith(mAttributeStr))

        {

        //android.util.Log.e("program", "GOOD LINE: " +subline+" subLen="+subLen);

        for(nameBegin=subLen-1; nameBegin>mAttributeLen-2; nameBegin--)

          {

          currChar=subline.charAt(nameBegin);

          if( currChar==' ' || currChar=='\t' )

            {

            mNumAttributes++;

            return subline.substring(nameBegin+1,subLen);

            }

          }

        }

      }

    return null;

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

  private String readTextFileFromRawResource(final InputStream inputStream, boolean doAttributes)

    {

    final InputStreamReader inputStreamReader = new InputStreamReader(inputStream);

    final BufferedReader bufferedReader = new BufferedReader(inputStreamReader);

    String nextLine, attribute, attrList="";

    final StringBuilder body = new StringBuilder();

    try

      {

      while ((nextLine = bufferedReader.readLine()) != null)

        {

        body.append(nextLine);

        body.append('\n');

        if( doAttributes )

          {

          attribute = parseOutAttribute(nextLine);

          if( attribute!=null )

            {

            //android.util.Log.d("program", "new attribute: "+attribute);

            if( attrList.length()>0 ) attrList+=" ";

            attrList += attribute;

            }

          }

        }

      }

    catch (IOException e)

      {

      return null;

      }

    if( doAttributes )

      {

      mAttributeName = attrList.split(" ");

      }

    return body.toString();

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

  private static void sanitizeMaxValues()

  throws VertexUniformsException,FragmentUniformsException

    {

    int maxV,maxF;

    int[] param = new int[1];

    GLES30.glGetIntegerv(GLES30.GL_MAX_VERTEX_UNIFORM_VECTORS, param, 0);

    maxV = param[0];

    GLES30.glGetIntegerv(GLES30.GL_MAX_FRAGMENT_UNIFORM_VECTORS, param, 0);

    maxF = param[0];

    //Log.d("program", "Max vectors in vertex shader: "+maxV);

    //Log.d("program", "Max vectors in fragment shader: "+maxF);

    if( !Build.FINGERPRINT.contains("generic") )

      {

      int realMaxV = (maxV-11)/4;   // adjust this in case of changes to the shaders...

      int realMaxF = (maxF- 2)/4;   //

      if( DistortedEffects.getMax(EffectType.VERTEX)   > realMaxV )

        {

        throw new VertexUniformsException("Too many effects in the vertex shader, max is "+realMaxV, realMaxV);

        }

      if( DistortedEffects.getMax(EffectType.FRAGMENT) > realMaxF )

        {

        throw new FragmentUniformsException("Too many effects in the fragment shader, max is "+realMaxF, realMaxF);

        }

      }

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

  private static int compileShader(final int shaderType, final String shaderSource)

  throws FragmentCompilationException,VertexCompilationException

    {

    int shaderHandle = GLES30.glCreateShader(shaderType);

    if (shaderHandle != 0)

      {

      GLES30.glShaderSource(shaderHandle, shaderSource);

      GLES30.glCompileShader(shaderHandle);

      final int[] compileStatus = new int[1];

      GLES30.glGetShaderiv(shaderHandle, GLES30.GL_COMPILE_STATUS, compileStatus, 0);

      if (compileStatus[0] != GLES30.GL_TRUE )

        {

        GLES30.glDeleteShader(shaderHandle);

        shaderHandle = 0;

        }

      }

    if (shaderHandle == 0)

      {

      String error = GLES30.glGetShaderInfoLog(shaderHandle);

      //android.util.Log.e("Program", "error compiling :"+error);

      switch(shaderType)

        {

        case GLES30.GL_VERTEX_SHADER  : throw new VertexCompilationException(error);

        case GLES30.GL_FRAGMENT_SHADER: throw new FragmentCompilationException(error);

        default                       : throw new RuntimeException(error);

        }

      }

    return shaderHandle;

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

  private static String insertEnabledEffects(String code, final String effects)

    {

    final String marker = "// ENABLED EFFECTS WILL BE INSERTED HERE";

    int length = marker.length();

    int place = code.indexOf(marker);

    if( place>=0 )

      {

      String begin = code.substring(0,place-1);

      String end   = code.substring(place+length);

/*

int len = begin.length();

android.util.Log.d("Program", begin.substring(len-40));

android.util.Log.d("Program", effects);

android.util.Log.d("Program", end.substring(0,40));

*/

      return begin + effects + end;

      }

    else

      {

      android.util.Log.e("Program", "Error: marker string not found in SHADER!");

      }

    return null;

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

// feedback: List of 'out' variables (OpenGL ES >= 3.0 only!) that will be transferred back to CPU

// using Transform Feedback.

  public DistortedProgram(final InputStream vertex, final InputStream fragment, final String vertexHeader, final String fragmentHeader, int glslVersion, final String[] feedback )

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

    {

    mAttributeStr = (glslVersion == 100 ? "attribute " : "in ");

    mAttributeLen = mAttributeStr.length();

    mNumAttributes = 0;

    final String vertexShader   = readTextFileFromRawResource(vertex  , true );

    final String fragmentShader = readTextFileFromRawResource(fragment, false);

    sanitizeMaxValues();

    final int vertexShaderHandle   = compileShader(GLES30.GL_VERTEX_SHADER  , vertexHeader   + vertexShader  );

    final int fragmentShaderHandle = compileShader(GLES30.GL_FRAGMENT_SHADER, fragmentHeader + fragmentShader);

    mProgramHandle = createAndLinkProgram(vertexShaderHandle, fragmentShaderHandle, mAttributeName, glslVersion>= 300 ? feedback:null );

    mAttribute = new int[mNumAttributes];

    for(int i=0; i<mNumAttributes; i++)

      {

      mAttribute[i] = GLES30.glGetAttribLocation( mProgramHandle, mAttributeName[i]);

      }

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

// feedback: List of 'out' variables (OpenGL ES >= 3.0 only!) that will be transferred back to CPU

// using Transform Feedback.

  public DistortedProgram(final InputStream vertex, final InputStream fragment, final String vertexHeader, final String fragmentHeader,

                          final String enabledVertex, final String enabledFragment, int glslVersion, final String[] feedback )

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

    {

    mAttributeStr = (glslVersion == 100 ? "attribute " : "in ");

    mAttributeLen = mAttributeStr.length();

    mNumAttributes = 0;

    String vertexShader   = readTextFileFromRawResource(vertex  , true );

    String fragmentShader = readTextFileFromRawResource(fragment, false);

    vertexShader   = insertEnabledEffects(vertexShader  ,enabledVertex  );

    fragmentShader = insertEnabledEffects(fragmentShader,enabledFragment);

    sanitizeMaxValues();

    final int vertexShaderHandle   = compileShader(GLES30.GL_VERTEX_SHADER  , vertexHeader   + vertexShader  );

    final int fragmentShaderHandle = compileShader(GLES30.GL_FRAGMENT_SHADER, fragmentHeader + fragmentShader);

    mProgramHandle = createAndLinkProgram(vertexShaderHandle, fragmentShaderHandle, mAttributeName, glslVersion>= 300 ? feedback:null );

    mAttribute = new int[mNumAttributes];

    for(int i=0; i<mNumAttributes; i++)

      {

      mAttribute[i] = GLES30.glGetAttribLocation( mProgramHandle, mAttributeName[i]);

      }

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

// PUBLIC API

///////////////////////////////////////////////////////////////////////////////////////////////////

  /**

   * Create a new Shader Program from source stored in resource files.

   * <p>

   * Needs to be called from a thread holding the OpenGL context.

   *

   * @param vertex   InputStream containing the opened Resource file from where to read vertex shader code.

   * @param fragment InputStream containing the opened Resource file from where to read fragment shader code.

   * @throws FragmentCompilationException

   * @throws VertexCompilationException

   * @throws VertexUniformsException

   * @throws FragmentUniformsException

   * @throws LinkingException

   */

  public DistortedProgram(final InputStream vertex, final InputStream fragment, final String vertexHeader, final String fragmentHeader, int glslVersion )

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

    {

    this(vertex,fragment,vertexHeader,fragmentHeader,glslVersion,null);

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

/**

 * Return the handle of the created program so that we can later, say, call glUseProgram.

 */

  public int getProgramHandle()

    {

    return mProgramHandle;

    }

///////////////////////////////////////////////////////////////////////////////////////////////////

/**

 * Use the program and enable all vertex attribute arrays.

 *

 * Needs to be called from a thread holding the OpenGL context.

 */

  public void useProgram()

    {

    GLES30.glUseProgram(mProgramHandle);

    for(int i=0; i<mNumAttributes; i++)

      GLES30.glEnableVertexAttribArray(mAttribute[i]);

    }

  }