Distorted Android

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

// Copyright 2016 Leszek Koltunski                                                               //

//                                                                                               //

// This file is part of DistortedLibrary.                                                               //

//                                                                                               //

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

//                                                                                               //

// DistortedLibrary 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 DistortedLibrary.  If not, see <http://www.gnu.org/licenses/>.                            //

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

package org.distorted.library.effectqueue;

import android.content.res.Resources;

import android.opengl.GLES31;

import android.util.Log;

import org.distorted.library.R;

import org.distorted.library.effect.EffectType;

import org.distorted.library.effect.PostprocessEffect;

import org.distorted.library.effect.VertexEffect;

import org.distorted.library.main.DistortedLibrary;

import org.distorted.library.main.DistortedFramebuffer;

import org.distorted.library.main.DistortedNode;

import org.distorted.library.main.InternalOutputSurface;

import org.distorted.library.main.InternalRenderState;

import org.distorted.library.main.InternalSurface;

import org.distorted.library.mesh.MeshBase;

import org.distorted.library.message.EffectMessage;

import org.distorted.library.message.EffectMessageSender;

import org.distorted.library.program.DistortedProgram;

import java.io.InputStream;

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

/**

 * Not part of public API, do not document

 *

 * @y.exclude

 */

public class EffectQueuePostprocess extends EffectQueue

  {

  private static final int NUM_UNIFORMS = PostprocessEffect.NUM_UNIFORMS;

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

  private int mHalo;

  private float mR, mG, mB, mA;

  private static DistortedProgram mPreProgram;

  private static int mPreColorH;

  private static int mPreTextureH;

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

  EffectQueuePostprocess(long id)

    { 

    super(id,NUM_UNIFORMS,INDEX );

    }

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

  void compute(long currTime)

    {

    if( currTime==mTime ) return;

    if( mTime==0 ) mTime = currTime;

    long step = (currTime-mTime);

    mR = mG = mB = mA = 0.0f;

    mHalo = 0;

    int halo;

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

      {

      mCurrentDuration[i] += step;

      // first zero out the 'alpha' because BLUR effect will not overwrite this (it is a 1D effect)

      // and if previously there was a GLOW effect here then mA would be non-zero and we don't want

      // that (see preprocess())

      mUniforms[NUM_UNIFORMS*i+4]=0.0f;

      if( mEffects[i].compute(mUniforms, NUM_UNIFORMS*i, mCurrentDuration[i], step) )

        {

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

          EffectMessageSender.newMessage( mListeners.get(j), EffectMessage.EFFECT_FINISHED, mEffects[i].getID(), mDistortedEffectsID);

        }

      halo = (int)mUniforms[NUM_UNIFORMS*i];

      if( halo>mHalo ) mHalo = halo;

      }

    // TODO  (now only really works in case of 1 effect!)

    if( mNumEffects>0 )

      {

      mR = mUniforms[1];

      mG = mUniforms[2];

      mB = mUniforms[3];

      mA = mUniforms[4];

      }

    mTime = currTime;

    }

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

  public static void createPrograms(Resources resources)

    {

    final InputStream mainVertStream = resources.openRawResource(R.raw.main_vertex_shader);

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

    int numV = VertexEffect.getNumEnabled();

    String mainVertHeader= DistortedLibrary.GLSL_VERSION + ("#define NUM_VERTEX "   + ( numV>0 ? DistortedLibrary.getMax(EffectType.VERTEX  ) : 0 ) + "\n");

    String mainFragHeader= DistortedLibrary.GLSL_VERSION + "\n";

    String enabledEffectV= VertexEffect.getGLSL();

    try

      {

      mPreProgram = new DistortedProgram(mainVertStream, mainFragStream, mainVertHeader, mainFragHeader,

                                         enabledEffectV, null, DistortedLibrary.GLSL, null);

      }

    catch(Exception e)

      {

      Log.e("POSTPROCESS", e.getClass().getSimpleName()+" trying to compile PRE program: "+e.getMessage());

      throw new RuntimeException(e.getMessage());

      }

    int preProgramH = mPreProgram.getProgramHandle();

    EffectQueueVertex.getUniforms( preProgramH,2 );

    EffectQueueMatrix.getUniforms( preProgramH,2 );

    mPreColorH  = GLES31.glGetUniformLocation( preProgramH, "u_Color"  );

    mPreTextureH= GLES31.glGetUniformLocation( preProgramH, "u_Texture");

    }

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

// TODO  (now only really works in case of 1 effect!)

  public int getQuality()

    {

    return mNumEffects>0 ? ((PostprocessEffect)mEffects[0]).getQuality() : 0;

    }

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

// TODO  (now only really works in case of 1 effect!)

  public boolean getRender()

    {

    return mNumEffects>0 ? ((PostprocessEffect)mEffects[0]).getRender() : false;

    }

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

  public int preprocess(InternalOutputSurface buffer, DistortedNode node, float distance, float mipmap, float[] projection)

    {

    buffer.setAsOutput();

    InternalSurface input = node.getInternalSurface();

    if( input.setAsInput() )

      {

      MeshBase mesh = node.getMesh();

      float halfW = input.getWidth() / 2.0f;

      float halfH = input.getHeight()/ 2.0f;

      float halfZ = halfW*mesh.getZFactor();

      int width = buffer.getWidth();

      int height = buffer.getHeight();

      InternalRenderState.setUpStencilMark(mA!=0.0f);

      InternalRenderState.disableBlending();

      GLES31.glViewport(0, 0, width, height );

      mPreProgram.useProgram();

      mesh.bindVertexAttribs(mPreProgram);

      EffectQueue[] queues = node.getEffects().getQueues();

      EffectQueueMatrix matrix = (EffectQueueMatrix)queues[0];

      EffectQueueVertex vertex = (EffectQueueVertex)queues[1];

      float inflate=0.0f;

      matrix.send(width, height, distance, mipmap, projection, halfW, halfH, halfZ, 2);

      if( mHalo!=0.0f )

        {

        inflate = matrix.magnify(projection, width, height, mipmap, halfW, halfH, halfZ, mHalo);

        }

      vertex.send(inflate,2);

      if( mA!=0.0f )

        {

        GLES31.glUniform4f(mPreColorH, mR, mG, mB, mA);

        GLES31.glUniform1i(mPreTextureH, 0);

        }

      GLES31.glDrawArrays(GLES31.GL_TRIANGLE_STRIP, 0, mesh.getNumVertices() );

      InternalRenderState.restoreBlending();

      InternalRenderState.unsetUpStencilMark();

      return 1;

      }

    return 0;

    }

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

  public int postprocess(DistortedFramebuffer buffer)

    {

    int numRenders = 0;

    GLES31.glDisable(GLES31.GL_BLEND);

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

      {

      numRenders += ((PostprocessEffect)mEffects[i]).apply(mUniforms,NUM_UNIFORMS*i, buffer);

      }

    GLES31.glEnable(GLES31.GL_BLEND);

    return numRenders;

    }

  }