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

import org.distorted.library.effect.Effect;

import org.distorted.library.effect.EffectName;

import org.distorted.library.effect.EffectType;

import org.distorted.library.main.DistortedLibrary;

import org.distorted.library.main.InternalMaster;

import org.distorted.library.main.InternalStackFrameList;

import java.util.ArrayList;

import java.util.HashMap;

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

/**

 * Not part of public API, do not document

 *

 * @y.exclude

 */

public abstract class EffectQueue implements InternalMaster.Slave

  {

  public static final int MAIN_VARIANTS = 4; // Number of Main program variants (ATM 4: MAIN, MAIN OIT, PREPROCESS, FULL)

  private static final int CREATE = 0;

  private static final int ATTACH = 1;

  private static final int DETACH = 2;

  private static final int DETALL = 3;

  long mTime;

  int mNumEffects;              // 'ToBe' will be more than mNumEffects if doWork() hasn't

  private int mNumEffectsToBe;  // added them yet (or less if it hasn't removed some yet)

  private int mNumFloatUniforms, mNumIntUniforms;

  Effect[] mEffects;

  float[] mFloatUniforms;

  int[] mIntUniforms;

  private long mID;

  private int mIndex;

  private boolean mCreated;

  private static class Job

    {

    int type;

    int num1, num2;

    boolean notify;

    Effect effect;

    Job(int t, int m1, int m2, boolean n, Effect e)

      {

      type  = t;

      num1  = m1;

      num2  = m2;

      notify= n;

      effect= e;

      }

    }

  private ArrayList<Job> mJobs = new ArrayList<>();

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

  EffectQueue(int numFloatUniforms, int numIntUniforms, int index)

    {

    mCreated          = false;

    mTime             = 0;

    mID               = 0;

    mNumEffects       = 0;

    mNumEffectsToBe   = 0;

    mIndex            = index;

    mNumFloatUniforms = numFloatUniforms;

    mNumIntUniforms   = numIntUniforms;

    mJobs.add(new Job(CREATE,numFloatUniforms,numIntUniforms, false,null)); // create the stuff that depends on max number

    InternalMaster.newSlave(this);                                          // of uniforms later, on first render.

    }

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

// this is not thread safe! The 'source' might change while we're copying it.

  EffectQueue(EffectQueue source)

    {

    if( !source.mCreated )

      {

      mCreated          = false;

      mTime             = 0;

      mID               = 0;

      mNumEffects       = 0;

      mNumEffectsToBe   = 0;

      mIndex            = source.mIndex;

      mNumFloatUniforms = source.mNumFloatUniforms;

      mNumIntUniforms   = source.mNumIntUniforms;

      int numJobs = source.mJobs.size();

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

        {

        Job job = source.mJobs.get(i);

        mJobs.add(job);

        }

      InternalMaster.newSlave(this);

      }

    else

      {

      mCreated          = true;

      mTime             = source.mTime;

      mID               = source.mID;

      mNumEffects       = source.mNumEffects;

      mNumEffectsToBe   = source.mNumEffectsToBe;

      mIndex            = source.mIndex;

      mNumFloatUniforms = source.mNumFloatUniforms;

      mNumIntUniforms   = source.mNumIntUniforms;

      int max = InternalStackFrameList.getMax(mIndex);

      if( max>0 )

        {

        mEffects       = new Effect[max];

        mFloatUniforms = new float[max*source.mNumFloatUniforms];

        mIntUniforms   = new int[max*source.mNumIntUniforms];

        if( mNumEffects>=0 )

          {

          System.arraycopy(source.mEffects, 0, mEffects, 0, mNumEffects);

          }

        if( mNumIntUniforms*mNumEffects>=0 )

          {

          System.arraycopy(source.mIntUniforms, 0, mIntUniforms, 0, mNumIntUniforms*mNumEffects);

          }

        }

      }

    }

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

  public static void allocateQueues(EffectQueue[] queues, EffectQueue[] from, int flags)

    {

    queues[0] = (flags & DistortedLibrary.CLONE_MATRIX     ) != 0 ? from[0] : new EffectQueueMatrix();

    queues[1] = (flags & DistortedLibrary.CLONE_VERTEX     ) != 0 ? from[1] : new EffectQueueVertex();

    queues[2] = (flags & DistortedLibrary.CLONE_FRAGMENT   ) != 0 ? from[2] : new EffectQueueFragment();

    queues[3] = (flags & DistortedLibrary.CLONE_POSTPROCESS) != 0 ? from[3] : new EffectQueuePostprocess();

    }

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

  public static void compute(EffectQueue[] queues, long currTime)

    {

    ((EffectQueueMatrix     )queues[0]).compute(currTime);

    ((EffectQueueVertex     )queues[1]).compute(currTime);

    ((EffectQueueFragment   )queues[2]).compute(currTime);

    ((EffectQueuePostprocess)queues[3]).compute(currTime);

    }

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

  public static void send(EffectQueue[] queues, float distance, float mipmap,

                          float[] projection, float inflate, int variant )

    {

    ((EffectQueueMatrix  )queues[0]).send(distance, mipmap, projection, variant);

    ((EffectQueueVertex  )queues[1]).send(inflate, variant);

    ((EffectQueueFragment)queues[2]).send(variant);

    }

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

// variant: 0 --> MAIN  1 --> OIT  2 --> prePOST  3 --> FULL

  public static void getUniforms(int programH, int variant)

    {

    EffectQueueFragment.uniforms(programH,variant);

    EffectQueueVertex  .uniforms(programH,variant);

    EffectQueueMatrix  .uniforms(programH,variant);

    }

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

// Every effect queue has an ID, which should be the same iff two queues hold the same effects.

// (this is a speedup: then both queues can be applied once, which seriously speeds up stuff -

// especially important in case of postprocessing)

  private void regenerateID()

    {

    if( mNumEffects>0 )

      {

      HashMap<ArrayList<Long>,Long> map = InternalStackFrameList.getMap();

      ArrayList<Long> list = new ArrayList<>();

      for (int i=0; i<mNumEffects; i++) list.add(mEffects[i].getID());

      Long id = map.get(list);

      if( id!=null )

        {

        mID = id;

        }

      else

        {

        mID = InternalStackFrameList.getNextQueueID();

        map.put(list,mID);

        }

      }

    else

      {

      mID = 0;

      }

    }

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

  public long getID()

    {

    return mID;

    }

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

  public static boolean setMax(int index, int m)

    {

    return InternalStackFrameList.setMax(index, Math.max(m,0));

    }

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

  public static int getMax(int index)

    {

    return InternalStackFrameList.getMax(index);

    }

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

// this assumes 0<=effect

  private void removeNow(int pos)

    {

    if( mNumEffects>pos )

      {

      mNumEffects--;

      mEffects[pos].remQueue(this);

      System.arraycopy(mEffects, pos+1, mEffects, pos, mNumEffects-pos);

      System.arraycopy(mIntUniforms, mNumIntUniforms*(pos+1), mIntUniforms, mNumIntUniforms*pos, mNumIntUniforms*(mNumEffects-pos) );

      mEffects[mNumEffects] = null;

      }

    }

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

  private void addNow(int pos, Effect effect)

    {

    mEffects[pos]                     = effect;

    mIntUniforms[mNumIntUniforms*pos] = effect.getName().ordinal();

    effect.addQueue(this);

    if( mIndex==EffectType.VERTEX.ordinal() )

      {

      effect.writeAssociations(mIntUniforms, mNumIntUniforms*pos+1, mNumIntUniforms*pos+3);

      }

    }

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

  public synchronized int removeByName(EffectName name)

    {

    int ret = 0;

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

      {

      if( mEffects[i].getName() == name )

        {

        mJobs.add(new Job(DETACH,0,0,true,mEffects[i]));

        ret++;

        }

      }

    if( ret>0 )

      {

      InternalMaster.newSlave(this);

      mNumEffectsToBe-=ret;

      }

    return ret;

    }

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

  public synchronized int removeById(long id)

    {

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

      {

      if( mEffects[i].getID() == id )

        {

        mJobs.add(new Job(DETACH,0,0,true,mEffects[i]));

        InternalMaster.newSlave(this);

        mNumEffectsToBe--;

        return 1;

        }

      }

    return 0;

    }

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

  public synchronized int removeEffect(Effect effect)

    {

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

      {

      if( mEffects[i]==effect )

        {

        mJobs.add(new Job(DETACH,0,0,true,mEffects[i]));

        InternalMaster.newSlave(this);

        mNumEffectsToBe--;

        return 1;

        }

      }

    return 0;

    }

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

// we do want to notify Listeners if they called 'abortAll' themselves but don't want to notify

// them if it is the library itself which is releasing resources.

  public synchronized int removeAll(boolean notify)

    {

    mJobs.add(new Job(DETALL,0,0,notify,null));

    InternalMaster.newSlave(this);

    mNumEffectsToBe = 0;

    return mNumEffects;

    }

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

  public boolean add(Effect effect)

    {

    if( InternalStackFrameList.getMax(mIndex)>mNumEffectsToBe || !mCreated )

      {

      mJobs.add(new Job(ATTACH,-1,0,false,effect));

      InternalMaster.newSlave(this);

      mNumEffectsToBe++;

      return true;

      }

    return false;

    }

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

  public boolean add(Effect effect, int position)

    {

    if( InternalStackFrameList.getMax(mIndex)>mNumEffectsToBe || !mCreated )

      {

      mJobs.add(new Job(ATTACH,position,0,false,effect));

      InternalMaster.newSlave(this);

      mNumEffectsToBe++;

      return true;

      }

    return false;

    }

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

  public int getNumEffects()

    {

    return mNumEffects;

    }

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

  public int getNumEffectsToBe()

    {

    return mNumEffectsToBe;

    }

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

  public Effect getEffect(int position)

    {

    if( position>=0 && position< mNumEffects )

      {

      return mEffects[position];

      }

    else

      {

      android.util.Log.e("queue", "getEffect: out of range "+position);

      return null;

      }

    }

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

  public void setAssociation(long effectID)

    {

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

      {

      if (mEffects[j].getID() == effectID)

        {

        mEffects[j].writeAssociations(mIntUniforms, mNumIntUniforms*j+1, mNumIntUniforms*j+3);

        }

      }

    }

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

  public void doWork()

    {

    boolean changed = false;

    int num = mJobs.size();

    Job job;

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

      {

      job = mJobs.remove(0);

      switch(job.type)

        {

        case CREATE: int max = InternalStackFrameList.getMax(mIndex);

                     if( max>0 )

                       {

                       mEffects       = new Effect[max];

                       mFloatUniforms = new float[max*job.num1];

                       mIntUniforms   = new int[max*job.num2];

                       }

                     mCreated = true;

                     break;

        case ATTACH: if( InternalStackFrameList.getMax(mIndex)>mNumEffects ) // it is possible that we have first

                       {                                                     // added effects and then lowered mMax

                       int position = job.num1;

                       if( position<0 )

                         {

                         addNow(mNumEffects,job.effect);

                         mNumEffects++;

                         changed = true;

                         }

                       else if( position<=mNumEffects )

                         {

                         System.arraycopy(mEffects    , position, mEffects    , position+1, mNumEffects-position);

                         System.arraycopy(mIntUniforms, mNumIntUniforms*position, mIntUniforms, mNumIntUniforms*(position+1), mNumIntUniforms*(mNumEffects-position) );

                         addNow(position,job.effect);

                         mNumEffects++;

                         changed = true;

                         }

                       }

                     else

                       {

                       android.util.Log.e("queue", "failed to add effect "+job.effect.getName());

                       }

                     break;

        case DETACH: for(int j=0; j<mNumEffects; j++)

                       {

                       if (mEffects[j] == job.effect)

                         {

                         removeNow(j);

                         changed = true;

                         break;

                         }

                       }

                     break;

        case DETALL: for(int j=0; j<mNumEffects; j++ )

                       {

                       changed = true;

                       mEffects[j] = null;

                       }

                     mNumEffects= 0;

                     break;

        }

      }

    if( changed && mIndex==EffectType.POSTPROCESS.ordinal() ) regenerateID();

    }

  }