TwistyPuzzleLib

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

// Copyright 2021 Leszek Koltunski                                                               //

//                                                                                               //

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

//                                                                                               //

// Magic Cube 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.                                                           //

//                                                                                               //

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

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

package org.distorted.objectlib.automator;

import java.io.IOException;

import java.io.InputStream;

import android.app.Activity;

import android.graphics.Bitmap;

import android.graphics.BitmapFactory;

import org.distorted.library.effect.MatrixEffectQuaternion;

import org.distorted.library.effect.MatrixEffectScale;

import org.distorted.library.main.DistortedEffects;

import org.distorted.library.main.DistortedFramebuffer;

import org.distorted.library.main.DistortedNode;

import org.distorted.library.main.DistortedTexture;

import org.distorted.library.mesh.MeshQuad;

import org.distorted.library.message.EffectListener;

import org.distorted.library.type.Dynamic;

import org.distorted.library.type.Dynamic3D;

import org.distorted.library.type.Dynamic4D;

import org.distorted.library.type.Static3D;

import org.distorted.library.type.Static4D;

import org.distorted.library.main.QuatHelper;

import org.distorted.objectlib.R;

import org.distorted.objectlib.main.TwistyObject;

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

class ObjectAutomatorRotate implements EffectListener

  {

  private static final int NUM_SCREEN = 0;

  private static final int NUM_OBJECT = 1;

  private static Static4D INIT_QUAT;

  private final ObjectAutomator mControl;

  private DistortedEffects[] mScreenEffects, mObjectEffects;

  private DistortedNode[] mScreenNodes, mObjectNodes;

  private long mScreenEffectID, mObjectEffectID, mCubeEffectID;

  private Static4D mObjRotQuat;

  private Dynamic3D mDynamic3;

  private Dynamic4D mDynamic4;

  private MatrixEffectScale mScale;

  private MatrixEffectQuaternion mQuaternion;

  private MeshQuad mScreenQuad, mObjectQuad;

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

   private Bitmap openBitmap(Activity act, int resource)

     {

     try( InputStream is = act.getResources().openRawResource(resource) )

       {

       return BitmapFactory.decodeStream(is);

       }

     catch( IOException e )

       {

       // ignore

       }

     return null;

     }

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

  private void computeInitQuat()

    {

    double alphaZ = 0;//-Math.PI* 0.1250f;

    double alphaY = 0;//-Math.PI* 0.0625f;

    alphaY /= 2;

    alphaZ /= 2;

    float sinZ = (float)Math.sin(alphaZ);

    float cosZ = (float)Math.cos(alphaZ);

    float sinY = (float)Math.sin(alphaY);

    float cosY = (float)Math.cos(alphaY);

    Static4D qZ = new Static4D(0,0,sinZ,cosZ);

    Static4D qY = new Static4D(0,sinY,0,cosY);

    INIT_QUAT = QuatHelper.quatMultiply(qY,qZ);

    }

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

// Take 3D vector 'ax', rotate it by quaternion 'objQuat' to get vector V1.

// Take 3D vector (1,0,0) and rotate it by INIT_QUAT to get vector V2.

// Return a quaternion Q such that if we rotate V1 by Q, we get V2.

  private Static4D computeQuat(Static3D ax, Static4D objQuat, float x, float y, float z)

    {

    Static4D ax4D = new Static4D( ax.get0(), ax.get1(), ax.get2(), 0);

    Static4D axRo = QuatHelper.rotateVectorByQuat(ax4D,objQuat);

    return QuatHelper.retRotationQuat(axRo.get0(),axRo.get1(),axRo.get2(),x,y,z);

    }

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

  private void computeRotQuat()

    {

    TwistyObject object = mControl.getObject();

    Static3D[] axis = object.getRotationAxis();

    int chosen=-1,numAxis = axis.length;

    float cos,maxCos = -1.0f;

    Static4D quat;

    Static4D objCurrQuat = mControl.getQuat();

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

    Static4D rotAxisX = QuatHelper.rotateVectorByQuat(axisX,INIT_QUAT);

    float axX = rotAxisX.get0();

    float axY = rotAxisX.get1();

    float axZ = rotAxisX.get2();

    for (int a=0; a<numAxis; a++)

      {

      quat = computeQuat(axis[a],objCurrQuat,axX,axY,axZ);

      cos = quat.get3();

android.util.Log.e("D", "axis="+a+" "+quat.get0()+" "+quat.get1()+" "+quat.get2()+" "+quat.get3() );

      if (cos > maxCos)

        {

        maxCos = cos;

        chosen = a;

        if( mObjRotQuat==null ) mObjRotQuat = new Static4D(quat);

        else mObjRotQuat.set(quat);

        }

      }

    android.util.Log.e("D", "axis chosen: "+chosen);

android.util.Log.e("D", mObjRotQuat.get0()+" "+mObjRotQuat.get1()+" "+mObjRotQuat.get2()+" "+mObjRotQuat.get3() );

    }

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

  private void setScreenEffectsStage1()

    {

    mDynamic3.resetToBeginning();

    mScale.notifyWhenFinished(mControl);

    }

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

// the whole cube rotates so that its axis is alignled according to the Euler angles defined in

// computeInitQuat()

  private void setObjectEffectsStage1()

    {

    TwistyObject obj = mControl.getObject();

    obj.apply(mQuaternion,0);

    mDynamic4.resetToBeginning();

    mQuaternion.notifyWhenFinished(this);

Static4D d = mDynamic4.getPoint(1);

android.util.Log.e("D", "set: "+d.get0()+" "+d.get1()+" "+d.get2()+" "+d.get3() );

    }

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

  private void setObjectEffectsStage2()

    {

    }

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

  private void createScreenEffects()

    {

    mScreenEffects   = new DistortedEffects[NUM_SCREEN];

    mScreenEffects[0]= new DistortedEffects();

    DistortedFramebuffer frame = mControl.getFramebuffer();

    int wid = frame.getWidth();

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

    Static3D scaleEnd  = new Static3D(wid,wid,wid);

    mDynamic3 = new Dynamic3D(10000,0.5f);

    mDynamic3.add(scaleStart);

    mDynamic3.add(scaleEnd  );

    mDynamic3.add(scaleStart);

    mDynamic3.setMode(Dynamic.MODE_PATH);

    mScale = new MatrixEffectScale(mDynamic3);

    mScreenEffectID = mScale.getID();

    mScreenEffects[0].apply(mScale);

    }

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

  private void createObjectEffects()

    {

    mObjectEffects   = new DistortedEffects[NUM_OBJECT];

    mObjectEffects[0]= new DistortedEffects();

    mDynamic4 = new Dynamic4D(5000,0.5f);

    mDynamic4.add(new Static4D(0,0,0,1));

    mDynamic4.add(mObjRotQuat);

    mDynamic4.add(mObjRotQuat);

    mDynamic4.add(mObjRotQuat);

    mDynamic4.setMode(Dynamic.MODE_PATH);

android.util.Log.e("D", "create: "+mObjRotQuat.get0()+" "+mObjRotQuat.get1()+" "+mObjRotQuat.get2()+" "+mObjRotQuat.get3() );

TwistyObject obj = mControl.getObject();

Static3D ax = obj.getRotationAxis()[0];

Static4D axis = new Static4D(ax.get0(), ax.get1(), ax.get2(), 0);

Static4D v1 = QuatHelper.rotateVectorByQuat( new Static4D(1,0,0,0), INIT_QUAT);

Static4D v2 = QuatHelper.rotateVectorByQuat( axis                 , mObjRotQuat);

android.util.Log.e("D", "v1: "+v1.get0()+" "+v1.get1()+" "+v1.get2()+" "+v1.get3());

android.util.Log.e("D", "v2: "+v2.get0()+" "+v2.get1()+" "+v2.get2()+" "+v2.get3());

android.util.Log.e("D", "ax: "+ax.get0()+" "+ax.get1()+" "+ax.get2());

    Static3D center = new Static3D(0,0,0);

    mQuaternion = new MatrixEffectQuaternion(mDynamic4, center);

    mCubeEffectID = mQuaternion.getID();

    DistortedFramebuffer frame = mControl.getFramebuffer();

    int wid = frame.getWidth();

    Static3D scaleFactor = new Static3D(wid,wid*0.1f,10);

    MatrixEffectScale scale = new MatrixEffectScale(scaleFactor);

    MatrixEffectQuaternion quat = new MatrixEffectQuaternion(INIT_QUAT,center);

    mObjectEffects[0].apply(scale);

    mObjectEffects[0].apply(quat);

    }

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

  private void createScreenNodes()

    {

    if( mScreenNodes==null )

      {

      mScreenNodes= new DistortedNode[NUM_SCREEN];

      mScreenQuad = new MeshQuad();

      }

    DistortedTexture texture = new DistortedTexture();

    texture.setColorARGB(0xff00ff00);

    mScreenNodes[0] = new DistortedNode(texture, mScreenEffects[0], mScreenQuad);

    }

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

  private void createObjectNodes()

    {

    if( mObjectNodes==null )

      {

      mObjectNodes= new DistortedNode[NUM_OBJECT];

      mObjectQuad = new MeshQuad();

      }

    Activity act = mControl.getActivity();

    if( act!=null )

      {

      Bitmap bmpArrow = openBitmap(act, R.drawable.ui_axis_arrow);

      DistortedTexture textureArrow = new DistortedTexture();

      if( bmpArrow!=null ) textureArrow.setTexture(bmpArrow);

      mObjectNodes[0] = new DistortedNode(textureArrow, mObjectEffects[0], mObjectQuad);

      }

    else

      {

      android.util.Log.e("D", "Activity NULL!!");

      }

    }

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

  long getEffectID()

    {

    return mObjectEffectID;

    }

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

  DistortedNode[] getScreenNodes()

    {

    if( NUM_SCREEN>0 )

      {

      if( mScreenEffects==null ) createScreenEffects();

      createScreenNodes();

      setScreenEffectsStage1();

      }

    return mScreenNodes;

    }

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

  DistortedNode[] returnScreenNodes()

    {

    return mScreenNodes;

    }

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

   DistortedNode[] getObjectNodes()

    {

    if( NUM_OBJECT>0 )

      {

      if( INIT_QUAT==null ) computeInitQuat();

      computeRotQuat();

      if( mObjectEffects==null ) createObjectEffects();

      createObjectNodes();

      setObjectEffectsStage1();

      }

    return mObjectNodes;

    }

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

   DistortedNode[] returnObjectNodes()

    {

    return mObjectNodes;

    }

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

  ObjectAutomatorRotate(ObjectAutomator control)

    {

    mControl = control;

    }

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

  @Override

  public void effectFinished(long effectID)

    {

    if( effectID==mCubeEffectID)

      {

      setObjectEffectsStage2();

      TwistyObject obj = mControl.getObject();

      obj.remove(mCubeEffectID);

      mObjectEffectID = -1;

      mControl.effectFinished(mObjectEffectID);

      }

    }

  }