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 android.util.Log;

import org.distorted.library.effect.Effect;

import org.distorted.library.effect.EffectType;

import org.distorted.library.effect.VertexEffect;

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 java.nio.ByteOrder;

import java.nio.FloatBuffer;

  private static DistortedProgram mMainProgram;

  private static int mMainTextureH;

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

  private static int[] mLinkedListSSBO = new int[1];

  private static int[] mAtomicCounter = new int[1];

  static

    {

    mLinkedListSSBO[0]= -1;

    mAtomicCounter[0] = -1;

    }

  private static int mBufferSize=(0x1<<23);  // 8 million entries

private static IntBuffer mIntBuf, mAtomicIntBuf;

  private static DistortedProgram mBlitDepthRenderProgram;

  private static int mBlitDepthRenderSizeH;

  private static DistortedProgram mNormalProgram;

  private static int mNormalMVPMatrixH;

  /// END PROGRAMS //////

  private boolean matrixCloned, vertexCloned, fragmentCloned, postprocessCloned;

  throws FragmentCompilationException,VertexCompilationException,VertexUniformsException,FragmentUniformsException,LinkingException

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

    int numV = VertexEffect.getNumEnabled();

    String mainFragHeader= Distorted.GLSL_VERSION + ("#define NUM_FRAGMENT " + ( numF>0 ? getMax(EffectType.FRAGMENT) : 0 ) + "\n");

    String enabledEffectV= VertexEffect.getGLSL();

    String enabledEffectF= FragmentEffect.getGLSL();

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

    //android.util.Log.e("Effects", "enabledF= "+enabledEffectF);

      {

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

                                          enabledEffectV, enabledEffectF, Distorted.GLSL, feedback);

      }

    catch(Exception e)

      {

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

      throw new RuntimeException(e.getMessage());

      }

    EffectQueueMatrix.getUniforms(mainProgramH);

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

    String blitFragHeader= (Distorted.GLSL_VERSION + "#define NUM_FRAGMENT 0\n");

      {

    catch(Exception e)

      {

      }

    int blitProgramH = mBlitProgram.getProgramHandle();

    mBlitDepthH    = GLES31.glGetUniformLocation( blitProgramH, "u_Depth");

    final InputStream blitDepthVertStream = resources.openRawResource(R.raw.blit_depth_vertex_shader);

    final InputStream blitDepthFragStream = resources.openRawResource(R.raw.blit_depth_fragment_shader);

    try

      {

      mBlitDepthProgram = new DistortedProgram(blitDepthVertStream,blitDepthFragStream,blitVertHeader,blitFragHeader, Distorted.GLSL);

      }

    catch(Exception e)

      {

      }

    int blitDepthProgramH   = mBlitDepthProgram.getProgramHandle();

    mIntBuffer = ByteBuffer.allocateDirect(4).order(ByteOrder.nativeOrder()).asIntBuffer();

    mIntBuffer.put(0,0);

    if( mLinkedListSSBO[0]<0 )

      {

      GLES31.glGenBuffers(1,mLinkedListSSBO,0);

    if( mAtomicCounter[0]<0 )

      {

      GLES31.glGenBuffers(1,mAtomicCounter,0);

      GLES31.glBufferData(GLES31.GL_ATOMIC_COUNTER_BUFFER, 4, mIntBuffer, GLES31.GL_DYNAMIC_DRAW);

      GLES31.glBindBuffer(GLES31.GL_ATOMIC_COUNTER_BUFFER, 0);

      }

    // BLIT DEPTH RENDER PROGRAM ///////////////////////////

    final InputStream blitDepthRenderVertStream = resources.openRawResource(R.raw.blit_depth_vertex_shader);

    final InputStream blitDepthRenderFragStream = resources.openRawResource(R.raw.blit_depth_render_fragment_shader);

    try

      {

      mBlitDepthRenderProgram = new DistortedProgram(blitDepthRenderVertStream,blitDepthRenderFragStream,blitVertHeader,blitFragHeader, Distorted.GLSL);

      }

    catch(Exception e)

      {

      }

    int blitDepthRenderProgramH   = mBlitDepthRenderProgram.getProgramHandle();

    mBlitDepthRenderSizeH         = GLES31.glGetUniformLocation( blitDepthRenderProgramH, "u_Size");

    final InputStream normalVertexStream   = resources.openRawResource(R.raw.normal_vertex_shader);

    final InputStream normalFragmentStream = resources.openRawResource(R.raw.normal_fragment_shader);

      {

    catch(Exception e)

      {

      }

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

      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;

      }

    }

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

  void newNode(DistortedNode node)

    {

    mM.newNode(node);

    mF.newNode(node);

    mV.newNode(node);

    mP.newNode(node);

    }

    GLES31.glBeginTransformFeedback( GLES31.GL_POINTS);

    GLES31.glBindBufferBase(GLES31.GL_TRANSFORM_FEEDBACK_BUFFER, 0, 0);

    GLES31.glBindBuffer(GLES31.GL_ARRAY_BUFFER, mesh.mAttTFO[0]);

    GLES31.glVertexAttribPointer(mNormalProgram.mAttribute[0], MeshObject.POS_DATA_SIZE, GLES31.GL_FLOAT, false, 0, 0);

    GLES31.glBindBuffer(GLES31.GL_ARRAY_BUFFER, 0);

    GLES31.glLineWidth(8.0f);

    GLES31.glDrawArrays(GLES31.GL_LINES, 0, 2*mesh.numVertices);

    mV.compute(currTime,halfW,halfH,halfZ);

    mF.compute(currTime,halfW,halfH);

    GLES31.glVertexAttribPointer(mMainProgram.mAttribute[0], MeshObject.POS_DATA_SIZE, GLES31.GL_FLOAT, false, MeshObject.VERTSIZE, MeshObject.OFFSET0);

    GLES31.glVertexAttribPointer(mMainProgram.mAttribute[1], MeshObject.NOR_DATA_SIZE, GLES31.GL_FLOAT, false, MeshObject.VERTSIZE, MeshObject.OFFSET1);

    GLES31.glVertexAttribPointer(mMainProgram.mAttribute[2], MeshObject.TEX_DATA_SIZE, GLES31.GL_FLOAT, false, MeshObject.VERTSIZE, MeshObject.OFFSET2);

    GLES31.glBindBuffer(GLES31.GL_ARRAY_BUFFER, 0);

    mF.send();

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

 * Only for use by the library itself.

 *

 * @y.exclude

 */

  public static void blitPriv(DistortedOutputSurface surface)

    GLES31.glUniform1i(mBlitTextureH, 0);

    GLES31.glUniform1f( mBlitDepthH , 1.0f-surface.mNear);

    GLES31.glVertexAttribPointer(mBlitProgram.mAttribute[0], 2, GLES31.GL_FLOAT, false, 0, mQuadPositions);

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

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

    mBlitDepthProgram.useProgram();

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

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

// render all the transparent pixels from the per-pixel linked lists. This is in the 'merge

// postprocessing buckets' stage.

  static void blitDepthRenderPriv(DistortedOutputSurface surface, float corrW, float corrH)

    {

    mBlitDepthRenderProgram.useProgram();

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

    // reset atomic counter to 0

    GLES31.glBindBuffer(GLES31.GL_ATOMIC_COUNTER_BUFFER, mAtomicCounter[0] );

    mAtomicBuf = (ByteBuffer)GLES31.glMapBufferRange( GLES31.GL_ATOMIC_COUNTER_BUFFER, 0, 4,

                                                      GLES31.GL_MAP_READ_BIT|GLES31.GL_MAP_WRITE_BIT);

    if( mAtomicBuf!=null )

      {

      mAtomicIntBuf = mAtomicBuf.order(ByteOrder.nativeOrder()).asIntBuffer();

      int counter = mAtomicIntBuf.get(0);

      mAtomicIntBuf.put(0, 0);

      //android.util.Log.e("counter", "now = "+counter+" w="+surface.mWidth+" h="+surface.mHeight

      //                             +" diff="+(counter-surface.mWidth*surface.mHeight));

      }

    else

      {

      android.util.Log.e("counter", "failed to map buffer");

      }

    GLES31.glUnmapBuffer(GLES31.GL_ATOMIC_COUNTER_BUFFER);

  private static void analyzeBuffer(int w, int h)

    {

    int errors = 0;

    mBuf = (ByteBuffer)GLES31.glMapBufferRange(GLES31.GL_SHADER_STORAGE_BUFFER, 0, mBufferSize*4, GLES31.GL_MAP_READ_BIT);

    mIntBuf = mBuf.order(ByteOrder.nativeOrder()).asIntBuffer();

      for(int row=0; row<h; row++)

        ptr = mIntBuf.get(index);

        if( ptr!=0 )

          {

          if( ptr>0 && ptr<mBufferSize )

            {

            ptr = mIntBuf.get(ptr);

              errors++;

            }

          else

            {

          }

        }

    GLES31.glBindBuffer(GLES31.GL_SHADER_STORAGE_BUFFER, 0);

    if( errors>0 ) android.util.Log.e("surface", "errors: "+errors);

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

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

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

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

    mV = null;

    mF = null;

    mLinkedListSSBO[0]= -1;

    mAtomicCounter[0] = -1;

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

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

/**

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

 */

    mF.deregisterForMessages(el);

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

/**

 * @return Number of effects aborted.

    return mM.abortAll(true) + mV.abortAll(true) + mF.abortAll(true);

    }

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

/**

    switch(type)

      case VERTEX     : return mV.abortAll(true);

      case FRAGMENT   : return mF.abortAll(true);

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

/**

 * @param id the Id of the Effect to be removed, as returned by getID().

 * @return Number of effects aborted.

 */

  public int abortById(long id)

    {

    long type = id&EffectType.MASK;

    if( type == EffectType.VERTEX.ordinal()      ) return mV.removeById(id);

    if( type == EffectType.FRAGMENT.ordinal()    ) return mF.removeById(id);

    return 0;

    }

/**

 * Aborts a single Effect.

 * 

      {

      case VERTEX     : return mV.removeEffect(effect);

      case FRAGMENT   : return mF.removeEffect(effect);

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

/**

      case VERTEX     : return mV.removeByName(name);

      case FRAGMENT   : return mF.removeByName(name);