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

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

import org.distorted.library.program.FragmentUniformsException;

import org.distorted.library.program.LinkingException;

import org.distorted.library.program.VertexCompilationException;

import org.distorted.library.program.VertexUniformsException;

import org.distorted.library.type.Data4D;

import java.nio.ByteOrder;

import java.nio.FloatBuffer;

  private static final int NORMAL_DATA_SIZE  = 3; // Main Program: size of the normal data in elements

  private static final int TEX_DATA_SIZE     = 2; // Main Program: size of the texture coordinate data in elements.

  private static DistortedProgram mMainProgram;

  private static int mMainTextureH;

  static

    {

    int len = EffectNames.size();

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

      {

      mEffectEnabled[i] = false;

      }

    }

  private static DistortedProgram mBlitProgram;

  private static int mBlitTextureH;

  static

    {

    float[] positionData= { -0.5f, -0.5f,  -0.5f, 0.5f,  0.5f,-0.5f,  0.5f, 0.5f };

    mQuadPositions = ByteBuffer.allocateDirect(32).order(ByteOrder.nativeOrder()).asFloatBuffer();

    mQuadPositions.put(positionData).position(0);

    }

  private static DistortedProgram mDebugProgram;

  private static int mDebugObjDH;

  private static int mDebugMVPMatrixH;

  private static float[] mTmpMatrix = new float[16];

  private EffectQueueFragment    mF;

  private EffectQueueVertex      mV;

  private EffectQueuePostprocess mP;

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

    final InputStream mainFragStream = resources.openRawResource(R.raw.main_fragment_shader);

    String mainVertHeader= ("#version 100\n");

    String mainFragHeader= ("#version 100\n");

    EffectTypes type;

    boolean foundF = false;

    boolean foundV = false;

        {

        name = EffectNames.getName(i);

        type = EffectNames.getType(i);

        if( type == EffectTypes.VERTEX )

          {

          mainVertHeader += ("#define "+name.name()+" "+name.ordinal()+"\n");

          foundV = true;

          }

        else if( type == EffectTypes.FRAGMENT )

          {

          mainFragHeader += ("#define "+name.name()+" "+name.ordinal()+"\n");

          foundF = true;

          }

        }

    mainFragHeader += ("#define NUM_FRAGMENT " + ( foundF ? getMaxFragment() : 0 ) + "\n");

    //android.util.Log.e("Effects", "fragHeader= "+mainFragHeader);

    EffectQueueVertex.getUniforms(mainProgramH);

    EffectQueueMatrix.getUniforms(mainProgramH);

    // BLIT PROGRAM ////////////////////////////////////

    final InputStream blitVertStream = resources.openRawResource(R.raw.blit_vertex_shader);

    final InputStream blitFragStream = resources.openRawResource(R.raw.blit_fragment_shader);

    String blitVertHeader= ("#version 100\n#define NUM_VERTEX 0\n"  );

    String blitFragHeader= ("#version 100\n#define NUM_FRAGMENT 0\n");

    mBlitProgram = new DistortedProgram(blitVertStream,blitFragStream,blitVertHeader,blitFragHeader);

    int blitProgramH = mBlitProgram.getProgramHandle();

    mBlitTextureH  = GLES30.glGetUniformLocation( blitProgramH, "u_Texture");

    // DEBUG ONLY //////////////////////////////////////

    final InputStream debugVertexStream   = resources.openRawResource(R.raw.test_vertex_shader);

    final InputStream debugFragmentStream = resources.openRawResource(R.raw.test_fragment_shader);

    mDebugProgram = new DistortedProgram(debugVertexStream,debugFragmentStream, "#version 100\n", "#version 100\n");

    int debugProgramH = mDebugProgram.getProgramHandle();

    mDebugMVPMatrixH = GLES30.glGetUniformLocation( debugProgramH, "u_MVPMatrix");

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

      matrixCloned = true;

      }

    else

      {

      }

      {

      mV = d.mV;

      vertexCloned = true;

      }

    else

      {

      }

      {

      mF = d.mF;

      fragmentCloned = true;

      {

      }

    if( (flags & Distorted.CLONE_POSTPROCESS) != 0 )

      {

      mP = d.mP;

      postprocessCloned = true;

      }

    else

      {

      mP = new EffectQueuePostprocess(mID);

      postprocessCloned = false;

      }

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

    int len  = vertices.length/3;

    int minx = Integer.MAX_VALUE;

    int maxx = Integer.MIN_VALUE;

    int miny = Integer.MAX_VALUE;

    int maxy = Integer.MIN_VALUE;

    int wx,wy;

    float x,y,z, X,Y,W, ndcX,ndcY;

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

      {

      x = 2*halfX*vertices[3*i  ];

      y = 2*halfY*vertices[3*i+1];

      z = 2*halfZ*vertices[3*i+2];

      X = mvp[ 0]*x + mvp[ 4]*y + mvp[ 8]*z + mvp[12];

      Y = mvp[ 1]*x + mvp[ 5]*y + mvp[ 9]*z + mvp[13];

      W = mvp[ 3]*x + mvp[ 7]*y + mvp[11]*z + mvp[15];

      ndcX = X/W;

      ndcY = Y/W;

      wy = (int)(surface.mHeight*(ndcY+1)/2);

      if( wx>maxx ) maxx = wx;

      if( wy<miny ) miny = wy;

      if( wy>maxy ) maxy = wy;

    mDebugProgram.useProgram();

    Matrix.translateM  ( mTmpMatrix, 0, halfX,-halfY, 0);

    GLES30.glUniformMatrix4fv(mDebugMVPMatrixH, 1, false, mMVPMatrix , 0);

    GLES30.glDrawArrays(GLES30.GL_TRIANGLE_STRIP, 0, 4);

  DistortedFramebuffer getPostprocessingBuffer()

    {

    return mP.mNumEffects==0 ? null : mP.mMainBuffer;

    }

    mM.compute(currTime);

    mV.compute(currTime);

    mF.compute(currTime);

    GLES30.glUniform1i(mMainTextureH, 0);

    surface.setAsOutput();

    mM.send(surface,halfW,halfH,halfZ);

    mV.send(halfW,halfH,halfZ);

    mF.send(halfW,halfH);

    GLES30.glVertexAttribPointer(mMainProgram.mAttribute[0], POSITION_DATA_SIZE, GLES30.GL_FLOAT, false, 0, mesh.mMeshPositions);

    GLES30.glVertexAttribPointer(mMainProgram.mAttribute[1], NORMAL_DATA_SIZE  , GLES30.GL_FLOAT, false, 0, mesh.mMeshNormals);

    GLES30.glVertexAttribPointer(mMainProgram.mAttribute[2], TEX_DATA_SIZE     , GLES30.GL_FLOAT, false, 0, mesh.mMeshTexture);

    GLES30.glDrawArrays(GLES30.GL_TRIANGLE_STRIP, 0, mesh.dataLength);

    /// DEBUG ONLY //////

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

    GLES30.glDrawArrays(GLES30.GL_TRIANGLE_STRIP, 0, 4);

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

    if( !vertexCloned     ) mV.abortAll(false);

    if( !fragmentCloned   ) mF.abortAll(false);

    if( !postprocessCloned) mP.abortAll(false);

    mV = null;

    mF = null;

    mNextID = 0;

    int len = EffectNames.size();

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

      {

      mEffectEnabled[i] = false;

      }

// PUBLIC API

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

 * Whatever we do not clone gets created just like in the default constructor.

 *

 * @param dc    Source object to create our object from

 * @param flags A bitmask of values specifying what to copy.

/**

    releasePriv();

    }

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

/**

 *

 * @return ID of the object.

/**

 * Adds the calling class to the list of Listeners that get notified each time some event happens 

 * @param el A class implementing the EffectListener interface that wants to get notifications.

 */

    mF.registerForMessages(el);

    mM.registerForMessages(el);

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

/**

 * Removes the calling class from the list of Listeners.

 * 

 * @param el A class implementing the EffectListener interface that no longer wants to get notifications.

 */

    mF.deregisterForMessages(el);

    mM.deregisterForMessages(el);

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

/**

 * @return Number of effects aborted.

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

/**

 * @return Number of effects aborted.

    {

    switch(type)

      case VERTEX     : return mV.abortAll(true);

      case FRAGMENT   : return mF.abortAll(true);

      case POSTPROCESS: return mP.abortAll(true);

      default         : return 0;

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

/**

 * Aborts a single Effect.

 * 

 * @param id ID of the Effect we want to abort.

    {

    int type = (int)(id&EffectTypes.MASK);

    if( type==EffectTypes.VERTEX.type      ) return mV.removeByID(id>>EffectTypes.LENGTH);

    if( type==EffectTypes.FRAGMENT.type    ) return mF.removeByID(id>>EffectTypes.LENGTH);

    if( type==EffectTypes.POSTPROCESS.type ) return mP.removeByID(id>>EffectTypes.LENGTH);

    }

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

/**

    {

    switch(name.getType())

      case VERTEX     : return mV.removeByType(name);

      case FRAGMENT   : return mF.removeByType(name);

      case POSTPROCESS: return mP.removeByType(name);

      default         : return 0;

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

/**

 * Print some info about a given Effect to Android's standard out. Used for debugging only.

 * 

 * @param id Effect ID we want to print info about

 * @return <code>true</code> if a single Effect of type effectType has been found.

 */

    {

    int type = (int)(id&EffectTypes.MASK);

    if( type==EffectTypes.VERTEX.type      )  return mV.printByID(id>>EffectTypes.LENGTH);

    if( type==EffectTypes.FRAGMENT.type    )  return mF.printByID(id>>EffectTypes.LENGTH);

    if( type==EffectTypes.POSTPROCESS.type )  return mP.printByID(id>>EffectTypes.LENGTH);

    }

/**

 * Enables a given Effect.

 * <p>

 * By default, all effects are disabled. One has to explicitly enable each effect one intends to use.

 * This needs to be called BEFORE shaders get compiled, i.e. before the call to Distorted.onCreate().

 * The point: by enabling only the effects we need, we can optimize the shaders.

 *

 * @param name Name of the Effect to enable.

 */

  public static void enableEffect(EffectNames name)

    {

    mEffectEnabled[name.ordinal()] = true;

    }

/**

 * Returns the maximum number of Matrix effects.

 *

 * @return The maximum number of Matrix effects

 */

  public static int getMaxMatrix()

    {

    return EffectQueue.getMax(EffectTypes.MATRIX.ordinal());

    }

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

/**

 * Returns the maximum number of Vertex effects.

 *

 * @return The maximum number of Vertex effects

 */

  public static int getMaxVertex()

    {

    return EffectQueue.getMax(EffectTypes.VERTEX.ordinal());

    }

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

/**

 * Returns the maximum number of Fragment effects.

 *

 * @return The maximum number of Fragment effects

 */

  public static int getMaxFragment()

    {

    return EffectQueue.getMax(EffectTypes.FRAGMENT.ordinal());

    }

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

/**

 * Returns the maximum number of Postprocess effects.

 *

 * @return The maximum number of Postprocess effects

 */

  public static int getMaxPostprocess()

    {

    return EffectQueue.getMax(EffectTypes.POSTPROCESS.ordinal());

    }

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

/**

 * Sets the maximum number of Matrix effects that can be stored in a single EffectQueue at one time.

 * This can fail if:

 * <ul>

 * <li>the value of 'max' is outside permitted range (0 &le; max &le; Byte.MAX_VALUE)

 * <li>We try to increase the value of 'max' when it is too late to do so already. It needs to be called

 *     before the Vertex Shader gets compiled, i.e. before the call to {@link Distorted#onCreate}. After this

 *     time only decreasing the value of 'max' is permitted.

 * <li>Furthermore, this needs to be called before any instances of the DistortedEffects class get created.

 * </ul>

 *

 * @param max new maximum number of simultaneous Matrix Effects. Has to be a non-negative number not greater

 *            than Byte.MAX_VALUE

 * @return <code>true</code> if operation was successful, <code>false</code> otherwise.

 */

  public static boolean setMaxMatrix(int max)

    {

    return EffectQueue.setMax(EffectTypes.MATRIX.ordinal(),max);

    }

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

/**

 * Sets the maximum number of Vertex effects that can be stored in a single EffectQueue at one time.

 * This can fail if:

 * <ul>

 * <li>the value of 'max' is outside permitted range (0 &le; max &le; Byte.MAX_VALUE)

 * <li>We try to increase the value of 'max' when it is too late to do so already. It needs to be called

 *     before the Vertex Shader gets compiled, i.e. before the call to {@link Distorted#onCreate}. After this

 *     time only decreasing the value of 'max' is permitted.

* <li>Furthermore, this needs to be called before any instances of the DistortedEffects class get created.

 * </ul>

 *

 * @param max new maximum number of simultaneous Vertex Effects. Has to be a non-negative number not greater

 *            than Byte.MAX_VALUE

 * @return <code>true</code> if operation was successful, <code>false</code> otherwise.

 */

  public static boolean setMaxVertex(int max)

    {

    return EffectQueue.setMax(EffectTypes.VERTEX.ordinal(),max);

    }

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

/**

 * Sets the maximum number of Fragment effects that can be stored in a single EffectQueue at one time.

 * This can fail if:

 * <ul>

 * <li>the value of 'max' is outside permitted range (0 &le; max &le; Byte.MAX_VALUE)

 * <li>We try to increase the value of 'max' when it is too late to do so already. It needs to be called

 *     before the Fragment Shader gets compiled, i.e. before the call to {@link Distorted#onCreate}. After this

 *     time only decreasing the value of 'max' is permitted.

 * <li>Furthermore, this needs to be called before any instances of the DistortedEffects class get created.

 * </ul>

 *

 * @param max new maximum number of simultaneous Fragment Effects. Has to be a non-negative number not greater

 *            than Byte.MAX_VALUE

 * @return <code>true</code> if operation was successful, <code>false</code> otherwise.

 */

  public static boolean setMaxFragment(int max)

    {

    return EffectQueue.setMax(EffectTypes.FRAGMENT.ordinal(),max);

    }

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

/**

 * Sets the maximum number of Postprocess effects that can be stored in a single EffectQueue at one time.

 * This can fail if:

 * <ul>

 * <li>the value of 'max' is outside permitted range (0 &le; max &le; Byte.MAX_VALUE)

 * <li>We try to increase the value of 'max' when it is too late to do so already. It needs to be called

 *     before the Fragment Shader gets compiled, i.e. before the call to {@link Distorted#onCreate}. After this

 *     time only decreasing the value of 'max' is permitted.

 * <li>Furthermore, this needs to be called before any instances of the DistortedEffects class get created.

 * </ul>

 *

 * @param max new maximum number of simultaneous Postprocess Effects. Has to be a non-negative number not greater

 *            than Byte.MAX_VALUE

 * @return <code>true</code> if operation was successful, <code>false</code> otherwise.

 */

  public static boolean setMaxPostprocess(int max)

    {

    return EffectQueue.setMax(EffectTypes.POSTPROCESS.ordinal(),max);

    }

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

// Individual effect functions.

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

// Matrix-based effects

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

/**

 * @return       ID of the effect added, or -1 if we failed to add one.

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

/**

 * @return      ID of the effect added, or -1 if we failed to add one.

/**

 * @param scale The factor to scale all 3 dimensions with.

 * @return      ID of the effect added, or -1 if we failed to add one.

 */