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;

import android.content.res.Resources;

import android.opengl.GLES20;

import org.distorted.library.message.EffectMessage;

import org.distorted.library.program.DistortedProgram;

import org.distorted.library.program.FragmentCompilationException;

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.Data1D;

import org.distorted.library.type.Data2D;

import org.distorted.library.type.Dynamic1D;

import org.distorted.library.type.Dynamic2D;

import org.distorted.library.type.Static1D;

import org.distorted.library.type.Static2D;

import java.io.InputStream;

import java.nio.ByteBuffer;

import java.nio.ByteOrder;

import java.nio.FloatBuffer;

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

class EffectQueuePostprocess extends EffectQueue

  {

  private static final int BYTES_PER_FLOAT   = 4;

  private static final int POSITION_DATA_SIZE= 2; // Post Program: size of the position data in elements

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

  private static final int NUM_UNIFORMS = 3;

  private static final int NUM_CACHE    = 0;

  private static final int INDEX = EffectTypes.POSTPROCESS.ordinal();

  private static int mNumEffectsH;

  private static int mTypeH;

  private static int mUniformsH;

  private static int mObjDH;

  private static int mMVPMatrixH;

  private static final FloatBuffer mQuadPositions, mQuadTexture;

  private static DistortedProgram mProgram;

  static

    {

    int dataLength = 4;

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

    float[] textureData = {  0.0f,  0.0f,   0.0f, 1.0f,  1.0f, 0.0f,  1.0f, 1.0f };

    mQuadPositions = ByteBuffer.allocateDirect(POSITION_DATA_SIZE*dataLength*BYTES_PER_FLOAT).order(ByteOrder.nativeOrder()).asFloatBuffer();

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

    mQuadTexture   = ByteBuffer.allocateDirect(TEX_DATA_SIZE     *dataLength*BYTES_PER_FLOAT).order(ByteOrder.nativeOrder()).asFloatBuffer();

    mQuadTexture.put(textureData).position(0);

    }

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

  EffectQueuePostprocess(long id)

    { 

    super(id,NUM_UNIFORMS,NUM_CACHE,INDEX );

    }

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

  static void createProgram(Resources resources)

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

    {

    final InputStream postVertexStream   = resources.openRawResource(R.raw.post_vertex_shader);

    final InputStream postFragmentStream = resources.openRawResource(R.raw.post_fragment_shader);

    String postFragmentHeader= ("#version 100\n#define NUM_POSTPROCESS "  + DistortedEffects.getMaxPostprocess()+"\n");

    for(EffectNames name: EffectNames.values() )

      {

      if( name.getType()== EffectTypes.POSTPROCESS)

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

      }

    mProgram = new DistortedProgram(postVertexStream,postFragmentStream, "", postFragmentHeader);

    int postProgramH = mProgram.getProgramHandle();

    getUniforms(postProgramH);

    }

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

  static void getUniforms(int mProgramH)

    {

    mNumEffectsH= GLES20.glGetUniformLocation( mProgramH, "pNumEffects");

    mTypeH      = GLES20.glGetUniformLocation( mProgramH, "pType");

    mUniformsH  = GLES20.glGetUniformLocation( mProgramH, "pUniforms");

    mObjDH      = GLES20.glGetUniformLocation( mProgramH, "u_objD");

    mMVPMatrixH = GLES20.glGetUniformLocation(mProgramH, "u_MVPMatrix");

    }

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

  synchronized void compute(long currTime) 

    {

    if( currTime==mTime ) return;

    if( mTime==0 ) mTime = currTime;

    long step = (currTime-mTime);

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

      {

      if( mInter[0][i]!=null && mInter[0][i].interpolateMain(mUniforms ,NUM_UNIFORMS*i, mCurrentDuration[i], step) )

        {

        for(int j=0; j<mNumListeners; j++)

          EffectMessageSender.newMessage( mListeners.elementAt(j),

                                          EffectMessage.EFFECT_FINISHED,

                                         (mID[i]<<EffectTypes.LENGTH)+EffectTypes.POSTPROCESS.type,

                                          mName[i],

                                          mObjectID);

        if( EffectNames.isUnity(mName[i], mUniforms, NUM_UNIFORMS*i) )

          {

          remove(i);

          i--;

          continue;

          }

        else mInter[0][i] = null;

        }

      if( mInter[1][i]!=null )

        {

        mInter[1][i].interpolateMain(mUniforms, NUM_UNIFORMS*i+1, mCurrentDuration[i], step);

        }

      mCurrentDuration[i] += step;

      }

    mTime = currTime;  

    }  

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

  protected void moveEffect(int index)

    {

    mUniforms[NUM_UNIFORMS*index  ] = mUniforms[NUM_UNIFORMS*(index+1)  ];

    mUniforms[NUM_UNIFORMS*index+1] = mUniforms[NUM_UNIFORMS*(index+1)+1];

    mUniforms[NUM_UNIFORMS*index+2] = mUniforms[NUM_UNIFORMS*(index+1)+2];

    }

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

  synchronized static void sendZero(float w, float h, float[] mvp)

    {

    GLES20.glUniform1i( mNumEffectsH, 0);

    GLES20.glUniform2f( mObjDH , w, h);

    GLES20.glUniformMatrix4fv(mMVPMatrixH, 1, false, mvp , 0);

    }

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

// w,h - width and height of hte input texture. MVP - Model-View-Projection matrix to apply to the

// texture; df - output FBO.

  synchronized void render(float w, float h, float[] mvp, DistortedFramebuffer df)

    {

    mProgram.useProgram();

    df.setAsOutput();

    GLES20.glUniform1i( mNumEffectsH, mNumEffects);

    GLES20.glUniform2f( mObjDH , w, h );

    GLES20.glUniformMatrix4fv(mMVPMatrixH, 1, false, mvp , 0);

    if( mNumEffects>0 )

      {

      GLES20.glUniform1iv( mTypeH    ,  mNumEffects, mName    ,0);

      GLES20.glUniform4fv( mUniformsH,2*mNumEffects, mUniforms,0);

      }

    GLES20.glVertexAttribPointer(mProgram.mAttribute[0], POSITION_DATA_SIZE, GLES20.GL_FLOAT, false, 0, mQuadPositions);

    GLES20.glVertexAttribPointer(mProgram.mAttribute[1], TEX_DATA_SIZE     , GLES20.GL_FLOAT, false, 0, mQuadTexture);

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

    }

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

// blur

  synchronized long add(EffectNames eln, Data1D degree, Data2D center)

    {

    if( mMax[INDEX]>mNumEffects )

      {

      if( degree instanceof Dynamic1D)

        {

        mInter[0][mNumEffects] = (Dynamic1D)degree;

        }

      else if( degree instanceof Static1D)

        {

        mInter[0][mNumEffects] = null;

        mUniforms[NUM_UNIFORMS*mNumEffects] = ((Static1D)degree).getX();

        }

      else return -1;

      if( center instanceof Dynamic2D)

        {

        mInter[1][mNumEffects] = (Dynamic2D)center;

        }

      else if( center instanceof Static2D)

        {

        mInter[1][mNumEffects] = null;

        mUniforms[NUM_UNIFORMS*mNumEffects+1] = ((Static2D)center).getX();

        mUniforms[NUM_UNIFORMS*mNumEffects+2] = ((Static2D)center).getY();

        }

      else return -1;

      return addBase(eln);

      }

    return -1;

    }

  }