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 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, mUniformStr, mUniList;

  private int mAttributeLen, mUniformLen;

  private int mProgramHandle;

  private int mNumAttributes;

  private int mNumUniforms;

  private String[] mAttributeName;

  private String[] mUniformName;

  public final int[] mAttribute;

  public final int[] mUniform;

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

  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 void init(int glslVersion)

    {

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

    mAttributeLen  = mAttributeStr.length();

    mNumAttributes = 0;

    mUniformStr    = "uniform ";

    mUniformLen    = mUniformStr.length();

    mNumUniforms   = 0;

    mUniList       = "";

    }

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

  private String parseOutUniform(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>=mUniformLen+1 )

      {

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

      int subLen = semicolon-whiteSpace;

      if( subline.startsWith(mUniformStr))

        {

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

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

          {

          currChar=subline.charAt(nameBegin);

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

            {

            mNumUniforms++;

            String uniform = subline.substring(nameBegin+1,subLen);

            int brace = uniform.indexOf("[");

            return brace>=0 ? uniform.substring(0,brace) : uniform;

            }

          }

        }

      }

    return null;

    }

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

  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 void doAttributes(final String shader, boolean doAttributes)

    {

    String attribute, attrList="", uniform;

    String[] lines = shader.split("\n");

    int length = lines.length;

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

      {

      if( doAttributes )

        {

        attribute = parseOutAttribute(lines[i]);

        if( attribute!=null )

          {

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

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

          attrList += attribute;

          }

        }

      uniform = parseOutUniform(lines[i]);

      if( uniform!=null )

        {

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

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

        mUniList += uniform;

        }

      }

    if( doAttributes ) mAttributeName = attrList.split(" ");

    }

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

  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 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

    {

    init(glslVersion);

    final String vertexShader   = readTextFileFromRawResource(vertex  , true );

    final String fragmentShader = readTextFileFromRawResource(fragment, false);

    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]);

      }

    if( mNumUniforms>0 )

      {

      mUniform = new int[mNumUniforms];

      mUniformName = mUniList.split(" ");

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

        {

        mUniform[i] = GLES30.glGetUniformLocation( mProgramHandle, mUniformName[i]);

        }

      }

    else mUniform = null;

    }

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

  public DistortedProgram(final String vertex, final String fragment)

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

    {

    init(300);

    doAttributes(vertex  , true );

    doAttributes(fragment, false);

    final int vertexShaderHandle   = compileShader(GLES30.GL_VERTEX_SHADER  , vertex  );

    final int fragmentShaderHandle = compileShader(GLES30.GL_FRAGMENT_SHADER, fragment);

    mProgramHandle = createAndLinkProgram(vertexShaderHandle, fragmentShaderHandle, mAttributeName, null );

    mAttribute = new int[mNumAttributes];

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

      {

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

      }

    if( mNumUniforms>0 )

      {

      mUniform = new int[mNumUniforms];

      mUniformName = mUniList.split(" ");

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

        {

        mUniform[i] = GLES30.glGetUniformLocation( mProgramHandle, mUniformName[i]);

        }

      }

    else mUniform = 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,

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

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

    {

    init(glslVersion);

    String vertexShader   = readTextFileFromRawResource(vertex  , true );

    String fragmentShader = readTextFileFromRawResource(fragment, false);

    vertexShader   = insertEnabledEffects(vertexShader  ,enabledVertex  );

    fragmentShader = insertEnabledEffects(fragmentShader,enabledFragment);

    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]);

      }

    if( mNumUniforms>0 )

      {

      mUniform = new int[mNumUniforms];

      mUniformName = mUniList.split(" ");

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

        {

        mUniform[i] = GLES30.glGetUniformLocation( mProgramHandle, mUniformName[i]);

        }

      }

    else mUniform = null;

    }

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

// 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]);

    }

  }