Magic Cube

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

// Copyright 2022 Leszek Koltunski                                                               //

//                                                                                               //

// This file is part of Magic Cube.                                                              //

//                                                                                               //

// Magic Cube is proprietary software licensed under an EULA which you should have received      //

// along with the code. If not, check https://distorted.org/magic/License-Magic-Cube.html        //

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

package org.distorted.bandaged;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.nio.ByteBuffer;

import java.nio.ByteOrder;

import javax.microedition.khronos.egl.EGLConfig;

import javax.microedition.khronos.opengles.GL10;

import android.app.Activity;

import android.opengl.GLES31;

import android.opengl.GLSurfaceView;

import android.widget.Toast;

import org.distorted.dialogs.RubikDialogBandagedSave;

import org.distorted.library.effect.EffectType;

import org.distorted.library.effect.FragmentEffectBrightness;

import org.distorted.library.effect.PostprocessEffectBorder;

import org.distorted.library.main.DistortedEffects;

import org.distorted.library.main.DistortedFramebuffer;

import org.distorted.library.main.DistortedLibrary;

import org.distorted.library.main.DistortedNode;

import org.distorted.library.main.DistortedScreen;

import org.distorted.library.main.InternalOutputSurface;

import org.distorted.library.mesh.MeshBase;

import org.distorted.library.type.Static1D;

import org.distorted.library.type.Static3D;

import org.distorted.library.type.Static4D;

import org.distorted.objectlib.json.JsonWriter;

import org.distorted.objectlib.main.InitData;

import org.distorted.objectlib.shape.ShapeHexahedron;

import org.distorted.objectlib.main.TwistyObject;

import org.distorted.objectlib.objects.TwistyBandagedCuboid;

import org.json.JSONException;

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

public class BandagedCreatorRenderer implements GLSurfaceView.Renderer, DistortedLibrary.ExceptionListener

{

   public static final float BRIGHTNESS = 0.333f;

   private static final int RESET_DURATION = 1000;

   private static final float MAX_SIZE_CHANGE = 1.70f;

   private static final float MIN_SIZE_CHANGE = 0.50f;

   private static final float INIT_RATIO = 0.5f;

   private final BandagedCreatorView mView;

   private final DistortedScreen mScreen;

   private final Static3D mScale;

   private final Static4D mQuatT, mQuatA;

   private final int[] mObjSize;

   private BandagedCubit[] mCubits;

   private boolean mInitialPhase;

   private long mStartTime;

   private float mQuatX, mQuatY, mQuatZ, mQuatW;

   private int mX, mY, mZ, mNumCubits;

   private boolean mResetQuats, mSetQuatT, mResettingObject, mConnectingCubits, mCreatingCubits, mRescaling;

   private int mIndex1, mIndex2;

   private int mSaveIcon;

   private DistortedFramebuffer mFramebuffer;

   private String mPath;

   private boolean mCubitsCreated;

   private int mWidth, mHeight;

   private float mScaleValue, mObjectScreenRatio;

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

   BandagedCreatorRenderer(BandagedCreatorView v)

     {

     mQuatT = new Static4D(0,0,0,1);

     mQuatA = new Static4D(-0.25189602f,0.3546389f,0.009657208f,0.90038127f);

     mView = v;

     mResetQuats       = false;

     mSetQuatT         = false;

     mResettingObject  = false;

     mConnectingCubits = false;

     mCubitsCreated    = false;

     mCreatingCubits   = false;

     mRescaling        = false;

     mObjectScreenRatio= INIT_RATIO;

     mSaveIcon = -1;

     mObjSize  = new int[3];

     mScreen = new DistortedScreen();

     mScreen.glClearColor(BRIGHTNESS, BRIGHTNESS, BRIGHTNESS, 1.0f);

     mScale = new Static3D(1,1,1);

     }

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

   private boolean isAdjacent(float[] pos1, float[] pos2)

     {

     int len1 = pos1.length/3;

     int len2 = pos2.length/3;

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

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

         {

         float d0 = pos1[3*i  ] - pos2[3*j  ];

         float d1 = pos1[3*i+1] - pos2[3*j+1];

         float d2 = pos1[3*i+2] - pos2[3*j+2];

         if( d0*d0 + d1*d1 + d2*d2 == 1 ) return true;

         }

     return false;

     }

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

   private int computeNumCubits(int x, int y, int z)

     {

     return ( x<=1 || y<=1 || z<=1 ) ? x*y*z : x*y*z-(x-2)*(y-2)*(z-2);

     }

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

   private void createCubits()

     {

     mCubits = new BandagedCubit[mNumCubits];

     int c=0;

     float begX = 0.5f*(1-mX);

     float begY = 0.5f*(1-mY);

     float begZ = 0.5f*(1-mZ);

     for(int x=0; x<mX; x++)

       for(int y=0; y<mY; y++)

          for(int z=0; z<mZ; z++)

            if( x==0 || x==mX-1 || y==0 || y==mY-1 || z==0 || z==mZ-1 )

              {

              float[] pos = new float[] { begX+x,begY+y,begZ+z };

              mCubits[c] = new BandagedCubit(pos,mX,mY,mZ,mQuatT,mQuatA,mScale,false);

              c++;

              }

     mView.setCubits(mCubits,mX,mY,mZ);

     }

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

   private void resetObject()

     {

     mView.resetCubits();

     for(int c=0; c<mNumCubits; c++)

       {

       if( !mCubits[c].isAttached() )

         {

         mCubits[c].attach();

         mScreen.attach(mCubits[c].getNode());

         }

       if( mCubits[c].getPosition().length>3 )

         {

         mCubits[c].reset(mScaleValue);

         }

       mCubits[c].setUnmarked();

       }

     }

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

   @Override

   public void onDrawFrame(GL10 glUnused)

     {

     long time = System.currentTimeMillis();

     mScreen.render(time);

     if( mSetQuatT )

       {

       mSetQuatT = false;

       mQuatT.set(mQuatX,mQuatY,mQuatZ,mQuatW);

       }

     if( mResetQuats )

       {

       mResetQuats = false;

       float qx = mQuatT.get0();

       float qy = mQuatT.get1();

       float qz = mQuatT.get2();

       float qw = mQuatT.get3();

       float rx = mQuatA.get0();

       float ry = mQuatA.get1();

       float rz = mQuatA.get2();

       float rw = mQuatA.get3();

       float tx = rw*qx - rz*qy + ry*qz + rx*qw;

       float ty = rw*qy + rz*qx + ry*qw - rx*qz;

       float tz = rw*qz + rz*qw - ry*qx + rx*qy;

       float tw = rw*qw - rz*qz - ry*qy - rx*qx;

       mQuatT.set(0f, 0f, 0f, 1f);

       mQuatA.set(tx, ty, tz, tw);

       }

     if( mResettingObject )

       {

       boolean done = continueResetting(time);

       if( done ) mResettingObject = false;

       }

     if( mSaveIcon>=0 )

       {

       renderIcon(time); // for some reason we need to call render() twice here, otherwise the

       mSaveIcon++;      // icon turns out black. Probably some problem with binding the texture.

       }

     if( mSaveIcon>=2 )

       {

       saveIcon();

       mSaveIcon = -1;

       }

     if( mConnectingCubits )

       {

       mConnectingCubits = false;

       tryConnectingCubits(mIndex1,mIndex2);

       }

     if( mCreatingCubits )

       {

       mCreatingCubits = false;

       rescaleObject();

       if( mCubitsCreated )

         {

         createCubits();

         mScreen.detachAll();

         mView.resetCubits();

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

           {

           mCubits[i].scaleMove(mScaleValue);

           DistortedNode node = mCubits[i].getNode();

           mScreen.attach(node);

           }

         }

       }

     if( mRescaling )

       {

       mRescaling = false;

       rescaleObject();

       for(int i=0; i<mNumCubits; i++) mCubits[i].scaleMove(mScaleValue);

       BandagedCreatorTouchControl control = mView.getTouchControl();

       control.setObjectRatio(mObjectScreenRatio);

       }

     }

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

   @Override

   public void onSurfaceChanged(GL10 glUnused, int width, int height)

      {

      if( width!=mWidth || height!=mHeight )

        {

        mWidth = width;

        mHeight= height;

        rescaleObject();

        mScreen.detachAll();

        int touched = mView.getTouched();

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

          if( mCubits[i].isAttached() )

            {

            mCubits[i].scaleMove(mScaleValue);

            if( touched==i ) mCubits[i].setMarked();

            else             mCubits[i].setUnmarked();

            DistortedNode node = mCubits[i].getNode();

            mScreen.attach(node);

            }

        mView.setScreenSize(width,height);

        mScreen.resize(width,height);

        }

      }

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

   @Override

   public void onSurfaceCreated(GL10 glUnused, EGLConfig config)

      {

      DistortedLibrary.setMax(EffectType.VERTEX,0);

      MeshBase.setMaxEffComponents(120);

      FragmentEffectBrightness.enable();

      DistortedLibrary.onSurfaceCreated(mView.getContext(),this,1);

      DistortedLibrary.setCull(true);

      if( mCubits==null )

        {

        createCubits();

        }

      else

        {

        for(int i=0; i<mNumCubits; i++) mCubits[i].recreateBitmap();

        }

      mCubitsCreated = true;

      mWidth = 0;

      mHeight= 0;

      }

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

   public void distortedException(Exception ex)

     {

     android.util.Log.e("CREATOR", "unexpected exception: "+ex.getMessage() );

     }

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

   public DistortedScreen getScreen()

     {

     return mScreen;

     }

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

   void setConnecting(int index1, int index2)

     {

     mIndex1 = index1;

     mIndex2 = index2;

     mConnectingCubits = true;

     }

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

   private void tryConnectingCubits(int index1, int index2)

     {

     if( index1!=index2 )

       {

       float[] pos1 = mCubits[index1].getPosition();

       float[] pos2 = mCubits[index2].getPosition();

       if( isAdjacent(pos1,pos2) )

         {

         mCubits[index2].join(pos1,mScaleValue);

         mCubits[index1].detach();

         mScreen.detach(mCubits[index1].getNode());

         }

       }

     }

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

   public Static4D getQuatAccu()

     {

     return mQuatA;

     }

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

   public void setQuatTemp(float x, float y, float z, float w)

     {

     mSetQuatT = false;

     mQuatX = x;

     mQuatY = y;

     mQuatZ = z;

     mQuatW = w;

     mSetQuatT = true;

     }

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

   public void resetQuats()

     {

     mResetQuats = true;

     }

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

   public boolean isBusy()

     {

     return mResettingObject;

     }

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

   public void saveObject()

     {

     int numAttached=0;

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

       if( mCubits[i].isAttached() ) numAttached++;

     float[][] pos = new float[numAttached][];

     int attached=0;

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

       if( mCubits[i].isAttached() )

         {

         pos[attached++] = mCubits[i].getPosition();

         }

     InitData data = new InitData(mObjSize,pos);

     TwistyObject obj = new TwistyBandagedCuboid( TwistyObject.MESH_NICE, TwistyObject.MODE_NORM,

                                                  new Static4D(0,0,0,1), new Static3D(0,0,0), 1.0f, data, null );

     String name = obj.getShortName();

     BandagedCreatorActivity act = (BandagedCreatorActivity) mView.getContext();

     if( act.objectDoesntExist(name) && createObjectJson(obj,act) )

       {

       setupIconCreation(act,data);

       act.addObject(obj.getShortName());

       }

     }

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

   private boolean createObjectJson(TwistyObject object, Activity act)

     {

     final String name = object.getShortName()+"_object.json";

     File file = new File(act.getFilesDir(), name);

     String filename = file.getAbsolutePath();

     try

       {

       JsonWriter writer = JsonWriter.getInstance();

       String json = writer.createObjectString(object,24,0);

       writer.write(filename,json);

       return true;

       }

     catch(JSONException ex)

       {

       act.runOnUiThread(new Runnable()

         {

         public void run()

           {

           String message = "JSON Exception saving to \n\n"+filename+"\n\n failed:\n\n"+ex.getMessage();

           Toast.makeText(act,message,Toast.LENGTH_LONG).show();

           }

         });

       return false;

       }

     catch(FileNotFoundException ex)

       {

       act.runOnUiThread(new Runnable()

         {

         public void run()

           {

           String message = "FileNotFound exception saving to \n\n"+filename+"\n\n failed:\n\n"+ex.getMessage();

           Toast.makeText(act,message,Toast.LENGTH_LONG).show();

           }

         });

       return false;

       }

     catch(IOException ex)

       {

       act.runOnUiThread(new Runnable()

         {

         public void run()

           {

           String message = "IO exception saving to \n\n"+filename+"\n\n failed:\n\n"+ex.getMessage();

           Toast.makeText(act,message,Toast.LENGTH_LONG).show();

           }

         });

       return false;

       }

     }

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

   private int computeProjectionAngle()

     {

     float quot1 = mObjSize[2]/ (float)mObjSize[0];

     float quot2 = mObjSize[2]/ (float)mObjSize[1];

     float quot3 = mObjSize[0]/ (float)mObjSize[2];

     float quot4 = mObjSize[0]/ (float)mObjSize[1];

     float quot5 = Math.max(quot1,quot2);

     float quot6 = Math.max(quot3,quot4);

     float quot7 = Math.max(quot5,quot6);

          if( quot7<=1.0f ) return 120;

     else if( quot7<=1.5f ) return 90;

     else if( quot7<=2.0f ) return 60;

     else                   return 30;

     }

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

   private void setupIconCreation(Activity act, InitData data)

     {

     final float R=1.0f;

     final int FBO_WIDTH  = (int)(R*720);

     final int FBO_HEIGHT = (int)(R*1280);

     final float OBJECT_SIZE = R*0.35f;

     TwistyObject obj = new TwistyBandagedCuboid( TwistyObject.MESH_NICE, TwistyObject.MODE_ICON,

                                                  ShapeHexahedron.DEFAULT_ROT, new Static3D(0,0,0), OBJECT_SIZE, data, null );

     DistortedEffects effects = obj.getObjectEffects();

     DistortedNode node = obj.getNode();

     if( mFramebuffer==null )

       {

       mFramebuffer = new DistortedFramebuffer(FBO_WIDTH,FBO_HEIGHT,1, InternalOutputSurface.DEPTH_NO_STENCIL);

       mFramebuffer.glClearColor(BRIGHTNESS, BRIGHTNESS, BRIGHTNESS, 1.0f);

       }

     mFramebuffer.setProjection( computeProjectionAngle() ,0.1f);

     mFramebuffer.detachAll();

     mFramebuffer.attach(node);

     Static1D halo = new Static1D(5);

     Static4D color = new Static4D(0,0,0,1);

     PostprocessEffectBorder border = new PostprocessEffectBorder(halo,color);

     border.setHaloDepth(false);

     effects.apply(border);

     final String name = obj.getShortName()+".png";

     File file = new File(act.getFilesDir(), name);

     String filename = file.getAbsolutePath();

     mSaveIcon = 0;

     mPath = filename;

     }

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

   private void renderIcon(long time)

     {

     mFramebuffer.render(time);

     }

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

   private void saveIcon()

     {

     int fW = mFramebuffer.getWidth();

     int fH = mFramebuffer.getHeight();

     ByteBuffer buf = ByteBuffer.allocateDirect(fW*fH*4);

     buf.order(ByteOrder.LITTLE_ENDIAN);

     mFramebuffer.setAsReadFramebuffer(0);

     GLES31.glReadBuffer(GLES31.GL_COLOR_ATTACHMENT0);

     GLES31.glReadPixels( 0, 0, fW, fH, GLES31.GL_RGBA, GLES31.GL_UNSIGNED_BYTE, buf);

     BandagedCreatorWorkerThread.newBuffer(buf,fW,fH,6,mPath);

     GLES31.glBindFramebuffer(GLES31.GL_READ_FRAMEBUFFER, 0);

     mSaveIcon = -1;

     }

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

   void mulObjectRatio(float ratio)

     {

     mObjectScreenRatio *= ratio;

     if( mObjectScreenRatio>MAX_SIZE_CHANGE*INIT_RATIO) mObjectScreenRatio = MAX_SIZE_CHANGE*INIT_RATIO;

     if( mObjectScreenRatio<MIN_SIZE_CHANGE*INIT_RATIO) mObjectScreenRatio = MIN_SIZE_CHANGE*INIT_RATIO;

     mRescaling = true;

     }

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

   float getObjectRatio()

     {

     return mObjectScreenRatio;

     }

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

   private void rescaleObject()

     {

     final int size = mX>mY ? Math.max(mX, mZ) : Math.max(mY, mZ);

     final float Q = mObjectScreenRatio/size;

     mScaleValue = mWidth<mHeight ? Q*mWidth : Q*mHeight;

     mScale.set( mScaleValue,mScaleValue,mScaleValue );

     }

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

   public void changeObject(int x, int y, int z)

     {

     if( mX!=x || mY!=y || mZ!=z )

       {

       mX = x;

       mY = y;

       mZ = z;

       mObjSize[0] = mX;

       mObjSize[1] = mY;

       mObjSize[2] = mZ;

       mNumCubits = computeNumCubits(mX,mY,mZ);

       mCreatingCubits = true;

       }

     }

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

   public void displaySavingDialog()

     {

     BandagedCreatorActivity act = (BandagedCreatorActivity)mView.getContext();

     RubikDialogBandagedSave saveDiag = new RubikDialogBandagedSave();

     saveDiag.show(act.getSupportFragmentManager(), null);

     }

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

   public void setupReset()

     {

     mResettingObject = true;

     mInitialPhase    = true;

     mStartTime       = System.currentTimeMillis();

     }

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

   public boolean continueResetting(long time)

     {

     long diff = time-mStartTime;

     float quotient = ((float)diff)/RESET_DURATION;

     if( mInitialPhase && quotient>0.5f )

       {

       mInitialPhase=false;

       resetObject();

       }

     double angle = 2*Math.PI*quotient*quotient*(3-2*quotient);

     float sinA = (float)Math.sin(angle);

     float cosA = (float)Math.cos(angle);

     mQuatT.set(0, -sinA, 0, cosA);

     return quotient>1.0f;

     }

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

  public void touchCubit(int index)

    {

    if( index>=0 && index<mNumCubits && mCubits[index]!=null ) mCubits[index].setMarked();

    }

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

  public void untouchCubit(int index)

    {

    if( index>=0 && index<mNumCubits && mCubits[index]!=null ) mCubits[index].setUnmarked();

    }

}