Magic Cube

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

// Copyright 2020 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.objects;

import android.content.res.Resources;

import android.graphics.Canvas;

import android.graphics.Paint;

import org.distorted.library.effect.MatrixEffectQuaternion;

import org.distorted.library.main.DistortedEffects;

import org.distorted.library.main.DistortedTexture;

import org.distorted.library.mesh.MeshBase;

import org.distorted.library.mesh.MeshSquare;

import org.distorted.library.type.Static3D;

import org.distorted.library.type.Static4D;

import org.distorted.main.R;

import org.distorted.main.RubikSurfaceView;

import java.util.Random;

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

public class TwistyDiamond extends TwistyObject

{

  private static final int FACES_PER_CUBIT =8;

  // the four rotation axis of a Diamond. Must be normalized.

  static final Static3D[] ROT_AXIS = new Static3D[]

         {

           new Static3D(+SQ6/3,+SQ3/3,     0),

           new Static3D(-SQ6/3,+SQ3/3,     0),

           new Static3D(     0,-SQ3/3,-SQ6/3),

           new Static3D(     0,-SQ3/3,+SQ6/3)

         };

  private static final int[] FACE_COLORS = new int[]

         {

           COLOR_GREY  , COLOR_VIOLET,

           COLOR_YELLOW, COLOR_WHITE ,

           COLOR_BLUE  , COLOR_RED   ,

           COLOR_ORANGE, COLOR_GREEN

         };

  // All legal rotation quats of a Diamond: unit + three 180 deg turns + 8 generators

  private static final Static4D[] QUATS = new Static4D[]

         {

           new Static4D(  0.0f,  0.0f,   0.0f,  1.0f ),

           new Static4D(  0.0f,  1.0f,   0.0f,  0.0f ),

           new Static4D(+SQ2/2,  0.0f, -SQ2/2,  0.0f ),

           new Static4D(-SQ2/2,  0.0f, -SQ2/2,  0.0f ),

           new Static4D(+SQ2/2,  0.5f,   0.0f,  0.5f ),

           new Static4D(-SQ2/2,  0.5f,   0.0f,  0.5f ),

           new Static4D(  0.0f,  0.5f, +SQ2/2,  0.5f ),

           new Static4D(  0.0f,  0.5f, -SQ2/2,  0.5f ),

           new Static4D(+SQ2/2,  0.5f,   0.0f, -0.5f ),

           new Static4D(-SQ2/2,  0.5f,   0.0f, -0.5f ),

           new Static4D(  0.0f,  0.5f, +SQ2/2, -0.5f ),

           new Static4D(  0.0f,  0.5f, -SQ2/2, -0.5f )

         };

  private static final float DIST = 0.50f;

  // Colors of the faces of cubits. Each cubit has 8 faces

  private static final int[][] mOctaFaceMap = new int[][]

         {

           { 6,1,8,8, 2,5,8,8 },

           { 8,1,3,8, 8,5,7,8 },

           { 8,8,3,4, 8,8,7,0 },

           { 6,8,8,4, 2,8,8,0 },

           { 6,1,3,4, 8,8,8,8 },

           { 8,8,8,8, 2,5,7,0 },

         };

  private static final int[] mTetraFaceMap = new int[] { 1, 3, 4, 6, 5, 7, 0, 2 };

  private static MeshBase mOctaMesh, mTetraMesh;

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

  TwistyDiamond(int size, Static4D quat, DistortedTexture texture,

                MeshSquare mesh, DistortedEffects effects, int[][] moves, Resources res, int scrWidth)

    {

    super(size, size, quat, texture, mesh, effects, moves, ObjectList.DIAM, res, scrWidth);

    }

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

  float getScreenRatio()

    {

    return 0.65f;

    }

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

  Static4D[] getQuats()

    {

    return QUATS;

    }

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

  int getNumFaces()

    {

    return FACE_COLORS.length;

    }

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

  boolean shouldResetTextureMaps()

    {

    return false;

    }

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

  int getNumStickerTypes(int numLayers)

    {

    return 1;

    }

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

  float[] getCuts(int numLayers)

    {

    if( numLayers<2 )

      {

      return null;

      }

    else

      {

      float[] cuts = new float[numLayers-1];

      float dist = SQ6*0.666f;

      float cut  = 0.5f*dist*(2-numLayers);

      for(int i=0; i<numLayers-1; i++)

        {

        cuts[i] = cut;

        cut += dist;

        }

      return cuts;

      }

    }

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

  int getNumCubitFaces()

    {

    return FACES_PER_CUBIT;

    }

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

  private int getNumOctahedrons(int layers)

    {

    return layers==1 ? 1 : 4*(layers-1)*(layers-1) + 2;

    }

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

  private int getNumTetrahedrons(int layers)

    {

    return 4*layers*(layers-1);

    }

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

  private int createOctahedrons(float[][] centers, int index, int layers, float height)

    {

    float x = DIST*(layers-1);

    float z = DIST*(layers+1);

    for(int i=0; i<layers; i++, index++)

      {

      z -= 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    for(int i=0; i<layers-1; i++, index++)

      {

      x -= 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    for(int i=0; i<layers-1; i++, index++)

      {

      z += 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    for(int i=0; i<layers-2; i++, index++)

      {

      x += 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    return index;

    }

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

  private int createTetrahedrons(float[][] centers, int index, int layers, float height)

    {

    float x = DIST*(layers-1);

    float z = DIST*layers;

    for(int i=0; i<layers-1; i++, index++)

      {

      z -= 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    x += DIST;

    z -= DIST;

    for(int i=0; i<layers-1; i++, index++)

      {

      x -= 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    x -= DIST;

    z -= DIST;

    for(int i=0; i<layers-1; i++, index++)

      {

      z += 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    x -= DIST;

    z += DIST;

    for(int i=0; i<layers-1; i++, index++)

      {

      x += 2*DIST;

      centers[index][0] = x;

      centers[index][1] = height;

      centers[index][2] = z;

      }

    return index;

    }

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

  float[][] getCubitPositions(int layers)

    {

    int numO = getNumOctahedrons(layers);

    int numT = getNumTetrahedrons(layers);

    int index = 0;

    float height = 0.0f;

    float[][] CENTERS = new float[numO+numT][3];

    index = createOctahedrons(CENTERS,index,layers,height);

    for(int i=layers-1; i>0; i--)

      {

      height += SQ2*DIST;

      index = createOctahedrons(CENTERS,index,i,+height);

      index = createOctahedrons(CENTERS,index,i,-height);

      }

    height = DIST*SQ2/2;

    for(int i=layers; i>1; i--)

      {

      index = createTetrahedrons(CENTERS,index,i,+height);

      index = createTetrahedrons(CENTERS,index,i,-height);

      height += SQ2*DIST;

      }

    return CENTERS;

    }

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

// TODO:

  private int retFaceTetraBelongsTo(int tetra, int numLayers)

    {

    switch(tetra)

      {

      case 0 : return 0;

      case 1 : return 1;

      case 2 : return 2;

      case 3 : return 3;

      case 4 : return 4;

      case 5 : return 5;

      case 6 : return 6;

      case 7 : return 7;

      default: return 8;

      }

    }

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

  private Static4D getQuat(int cubit, int numLayers, int numO)

    {

    if( cubit<numO ) return QUATS[0];

    switch( retFaceTetraBelongsTo(cubit-numO, numLayers) )

      {

      case 0: return new Static4D(0,-SQ2/2,0,SQ2/2);    //  90 along Y

      case 1: return QUATS[1];                          // 180 along Y

      case 2: return new Static4D(0,+SQ2/2,0,SQ2/2);    //  90 along Y

      case 3: return QUATS[0];                          // unit quat

      case 4: return new Static4D(SQ2/2, 0,SQ2/2,0);    //

      case 5: return new Static4D(     1,0,0,    0);    // 180 along X

      case 6: return new Static4D(-SQ2/2,0,SQ2/2,0);    //

      case 7: return new Static4D(0,     0,1,    0);    // 180 along Z

      }

    return null;

    }

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

  MeshBase createCubitMesh(int cubit, int numLayers)

    {

    MeshBase mesh;

    int numO = getNumOctahedrons(numLayers);

    if( cubit<numO )

      {

      if( mOctaMesh==null ) mOctaMesh = FactoryCubit.getInstance().createOctaMesh();

      mesh = mOctaMesh.copy(true);

      }

    else

      {

      if( mTetraMesh==null ) mTetraMesh = FactoryCubit.getInstance().createTetraMesh();

      mesh = mTetraMesh.copy(true);

      }

    Static4D sQ = getQuat(cubit,numLayers,numO);

    MatrixEffectQuaternion quat = new MatrixEffectQuaternion( sQ, new Static3D(0,0,0) );

    mesh.apply(quat,0xffffffff,0);

    return mesh;

    }

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

// TODO

  int getFaceColor(int cubit, int cubitface, int numLayers)

    {

    int numO = getNumOctahedrons(numLayers);

    if( cubit<numO )

      {

      return mOctaFaceMap[cubit][cubitface];

      }

    else

      {

      return cubitface>0 ? 8 : mTetraFaceMap[retFaceTetraBelongsTo(cubit-numO, numLayers)];

      }

    }

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

  void createFaceTexture(Canvas canvas, Paint paint, int face, int left, int top)

    {

    float E = 0.75f;

    float F = 0.50f;

    float R = 0.06f;

    float S = 0.07f;

    float[] vertices = { -F,-E/3, +F,-E/3, 0.0f,2*E/3};

    FactorySticker factory = FactorySticker.getInstance();

    factory.drawRoundedPolygon(canvas, paint, left, top, vertices, S, FACE_COLORS[face], R);

    }

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

  float returnMultiplier()

    {

    return 1.5f;

    }

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

  float[] getRowChances(int numLayers)

    {

    float[] chances = new float[numLayers];

    for(int i=0; i<numLayers; i++) chances[i] = ((float)(i+1))/numLayers;

    return chances;

    }

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

// PUBLIC API

  public Static3D[] getRotationAxis()

    {

    return ROT_AXIS;

    }

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

  public int getBasicAngle()

    {

    return 3;

    }

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

  public void randomizeNewScramble(int[][] scramble, Random rnd, int num)

    {

    if( num==0 )

      {

      scramble[num][0] = rnd.nextInt(ROTATION_AXIS.length);

      }

    else

      {

      int newVector = rnd.nextInt(ROTATION_AXIS.length-1);

      scramble[num][0] = (newVector>=scramble[num-1][0] ? newVector+1 : newVector);

      }

    float rowFloat = rnd.nextFloat();

    for(int row=0; row<mRowChances.length; row++)

      {

      if( rowFloat<=mRowChances[row] )

        {

        scramble[num][1] = row;

        break;

        }

      }

    switch( rnd.nextInt(2) )

      {

      case 0: scramble[num][2] = -1; break;

      case 1: scramble[num][2] =  1; break;

      }

    }

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

  int mulQuat(int q1, int q2)

    {

    Static4D result = RubikSurfaceView.quatMultiply(QUATS[q1],QUATS[q2]);

    float rX = result.get0();

    float rY = result.get1();

    float rZ = result.get2();

    float rW = result.get3();

    final float MAX_ERROR = 0.1f;

    float dX,dY,dZ,dW;

    for(int i=0; i<QUATS.length; i++)

      {

      dX = QUATS[i].get0() - rX;

      dY = QUATS[i].get1() - rY;

      dZ = QUATS[i].get2() - rZ;

      dW = QUATS[i].get3() - rW;

      if( dX<MAX_ERROR && dX>-MAX_ERROR &&

          dY<MAX_ERROR && dY>-MAX_ERROR &&

          dZ<MAX_ERROR && dZ>-MAX_ERROR &&

          dW<MAX_ERROR && dW>-MAX_ERROR  ) return i;

      dX = QUATS[i].get0() + rX;

      dY = QUATS[i].get1() + rY;

      dZ = QUATS[i].get2() + rZ;

      dW = QUATS[i].get3() + rW;

      if( dX<MAX_ERROR && dX>-MAX_ERROR &&

          dY<MAX_ERROR && dY>-MAX_ERROR &&

          dZ<MAX_ERROR && dZ>-MAX_ERROR &&

          dW<MAX_ERROR && dW>-MAX_ERROR  ) return i;

      }

    return -1;

    }

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

// The Diamond is solved if and only if:

//

// 1) all 6 octahedrons are rotated with the same quat

// 2) the 8 tetrahedrons are rotated with the quat and, optionally, the can also be rotated

//    by multitudes of 120 degrees along the face they are the center of.

//

// so:

// 1) cubits 6,12: can also be QUAT 6,10

// 2) cubits 7,13: can also be QUAT 4,8

// 3) cubits 8,10: can also be QUAT 7,11

// 4) cubits 9,11: can also be QUAT 5,9

// TODO

  public boolean isSolved()

    {

    int q = CUBITS[0].mQuatIndex;

    if ( CUBITS[ 1].mQuatIndex == q &&

         CUBITS[ 2].mQuatIndex == q &&

         CUBITS[ 3].mQuatIndex == q &&

         CUBITS[ 4].mQuatIndex == q &&

         CUBITS[ 5].mQuatIndex == q  )

      {

      int q1 = mulQuat(q,5);

      int q2 = mulQuat(q,9);

      if( CUBITS[ 9].mQuatIndex != q && CUBITS[ 9].mQuatIndex != q1 && CUBITS[ 9].mQuatIndex != q2 ) return false;

      if( CUBITS[11].mQuatIndex != q && CUBITS[11].mQuatIndex != q1 && CUBITS[11].mQuatIndex != q2 ) return false;

      q1 = mulQuat(q,4);

      q2 = mulQuat(q,8);

      if( CUBITS[ 7].mQuatIndex != q && CUBITS[ 7].mQuatIndex != q1 && CUBITS[ 7].mQuatIndex != q2 ) return false;

      if( CUBITS[13].mQuatIndex != q && CUBITS[13].mQuatIndex != q1 && CUBITS[13].mQuatIndex != q2 ) return false;

      q1 = mulQuat(q,6);

      q2 = mulQuat(q,10);

      if( CUBITS[ 6].mQuatIndex != q && CUBITS[ 6].mQuatIndex != q1 && CUBITS[ 6].mQuatIndex != q2 ) return false;

      if( CUBITS[12].mQuatIndex != q && CUBITS[12].mQuatIndex != q1 && CUBITS[12].mQuatIndex != q2 ) return false;

      q1 = mulQuat(q,7);

      q2 = mulQuat(q,11);

      if( CUBITS[ 8].mQuatIndex != q && CUBITS[ 8].mQuatIndex != q1 && CUBITS[ 8].mQuatIndex != q2 ) return false;

      if( CUBITS[10].mQuatIndex != q && CUBITS[10].mQuatIndex != q1 && CUBITS[10].mQuatIndex != q2 ) return false;

      return true;

      }

    return false;

    }

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

// only needed for solvers - there are no Diamond solvers ATM

  public String retObjectString()

    {

    return "";

    }

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

  public int getObjectName(int numLayers)

    {

    return R.string.diam2;

    }

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

  public int getInventor(int numLayers)

    {

    return R.string.diam2_inventor;

    }

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

  public int getComplexity(int numLayers)

    {

    return 5;

    }

}