TwistyPuzzleLib

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

// Copyright 2023 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.objectlib.tablebases;

import android.content.res.Resources;

import org.distorted.library.main.QuatHelper;

import org.distorted.library.type.Static3D;

import org.distorted.library.type.Static4D;

import org.distorted.objectlib.R;

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

public class TablebasesCube2 extends TablebasesMITM

{

  private static final int[][] P =

      {

            { 2,-1,-1},

            { 2,-1, 1},

            { 2, 1,-1},

            { 2, 1, 1},

            {-2,-1,-1},

            {-2,-1, 1},

            {-2, 1,-1},

            {-2, 1, 1},

            {-1, 2,-1},

            { 1, 2,-1},

            {-1, 2, 1},

            { 1, 2, 1},

            {-1,-2,-1},

            { 1,-2,-1},

            {-1,-2, 1},

            { 1,-2, 1},

            {-1,-1, 2},

            { 1,-1, 2},

            {-1, 1, 2},

            { 1, 1, 2},

            {-1,-1,-2},

            { 1,-1,-2},

            {-1, 1,-2},

            { 1, 1,-2},

      };

  private static final int[][] PT =

      {

          {4,0},{5,0},{6,0},{7,0},

          {0,0},{1,0},{2,0},{3,0},

          {2,2},{6,1},{3,1},{7,2},

          {0,1},{4,2},{1,2},{5,1},

          {1,1},{5,2},{3,2},{7,1},

          {0,2},{4,1},{2,1},{6,2}

      };

  private int[][][] mQuatsMap;

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

  public TablebasesCube2()

    {

    super();

    }

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

  public TablebasesCube2(Resources res)

    {

    super(res, new int[] {R.raw.cube_2_pruning5,R.raw.cube_2_pruning11} );

    }

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

  int[][] getBasicAngles()

    {

    int[] tmp = {4,4};

    return new int[][] { tmp,tmp,tmp };

    }

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

  Static3D[] getRotationAxis()

    {

    return new Static3D[]

         {

           new Static3D(1,0,0),

           new Static3D(0,1,0),

           new Static3D(0,0,1)

         };

    }

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

  float[][] getPosition()

    {

    return new float[][]

      {

        { -0.5f, -0.5f, -0.5f },

        { -0.5f, -0.5f,  0.5f },

        { -0.5f,  0.5f, -0.5f },

        { -0.5f,  0.5f,  0.5f },

        {  0.5f, -0.5f, -0.5f },

        {  0.5f, -0.5f,  0.5f },

        {  0.5f,  0.5f, -0.5f },

        {  0.5f,  0.5f,  0.5f },

      };

    }

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

  float[][] getCuts()

    {

    return new float[][] { {0.0f}, {0.0f}, {0.0f} };

    }

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

  boolean[][] getRotatable()

    {

    return new boolean[][] { {false,true},{false,true},{true,false} };

    }

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

// specifically for the tablebase

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

// 7!*3^6 --> https://www.jaapsch.net/puzzles/cube2.htm

  int getSize()

    {

    return 3674160;

    }

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

  int getMinScramble()

    {

    return 9;

    }

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

  int[] getPruningLevels()

    {

    return new int[] {5,11};

    }

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

  private int[] getPermTwist(float[] point)

    {

    float ERR = 0.01f;

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

      {

      float dx = point[0]-P[i][0];

      float dy = point[1]-P[i][1];

      float dz = point[2]-P[i][2];

      if( dx*dx + dy*dy + dz*dz < ERR ) return PT[i];

      }

    return null;

    }

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

  private void fillOutMap(float[] buffer, int[][] map, float[] point, Static4D quat, int quatIndex)

    {

    QuatHelper.rotateVectorByQuat(buffer,point[0],point[1],point[2],1,quat);

    int[] pt = getPermTwist(buffer);

    map[pt[0]][pt[1]] = quatIndex;

    }

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

  private void computeMap()

    {

    final float[][] POINTS =

        {

            {-2,-1,-1},

            {-2,-1, 1},

            {-2, 1,-1},

            {-2, 1, 1},

            { 2,-1,-1},

            { 2,-1, 1},

            { 2, 1,-1},

            { 2, 1, 1},

        };

    mQuatsMap = new int[8][8][3];

    Static4D[] quats = getQuats();

    float[] buffer = new float[4];

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

      for(int q=0; q<24; q++)

        fillOutMap(buffer,mQuatsMap[c],POINTS[c],quats[q],q);

    }

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

  public static int[] shrinkPerm(int[] perm8)

    {

    int[] perm7 = new int[7];

    int tmp = perm8[0];

    perm7[0] = tmp>1 ? tmp-1 : tmp;

    tmp = perm8[2];

    perm7[1] = tmp>1 ? tmp-1 : tmp;

    tmp = perm8[3];

    perm7[2] = tmp>1 ? tmp-1 : tmp;

    tmp = perm8[4];

    perm7[3] = tmp>1 ? tmp-1 : tmp;

    tmp = perm8[5];

    perm7[4] = tmp>1 ? tmp-1 : tmp;

    tmp = perm8[6];

    perm7[5] = tmp>1 ? tmp-1 : tmp;

    tmp = perm8[7];

    perm7[6] = tmp>1 ? tmp-1 : tmp;

    return perm7;

    }

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

  private static int[] growPerm(int[] perm7)

    {

    int[] perm8 = new int[8];

    int tmp = perm7[0];

    perm8[0] = tmp>=1 ? tmp+1 : tmp;

    perm8[1] = 1;

    tmp = perm7[1];

    perm8[2] = tmp>=1 ? tmp+1 : tmp;

    tmp = perm7[2];

    perm8[3] = tmp>=1 ? tmp+1 : tmp;

    tmp = perm7[3];

    perm8[4] = tmp>=1 ? tmp+1 : tmp;

    tmp = perm7[4];

    perm8[5] = tmp>=1 ? tmp+1 : tmp;

    tmp = perm7[5];

    perm8[6] = tmp>=1 ? tmp+1 : tmp;

    tmp = perm7[6];

    perm8[7] = tmp>=1 ? tmp+1 : tmp;

    return perm8;

    }

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

  private int[] getPerm(int permNum)

    {

    int[] perm7 = new int[7];

    TablebaseHelpers.getPermutationFromNum(perm7,7,permNum);

    return growPerm(perm7);

    }

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

  private int[] getTwist(int twistNum)

    {

    int[] twist = new int[8];

    twist[0] = twistNum%3;

    twist[1] = 0;

    twistNum /=3;

    twist[2] = twistNum%3;

    twistNum /=3;

    twist[3] = twistNum%3;

    twistNum /=3;

    twist[4] = twistNum%3;

    twistNum /=3;

    twist[5] = twistNum%3;

    twistNum /=3;

    twist[6] = twistNum%3;

    int total = (twist[0]+twist[2]+twist[3]+twist[4]+twist[5]+twist[6])%3;

    if( total==0 ) twist[7] = 0;

    else twist[7] = 3-total;

    return twist;

    }

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

  public int[] getQuats(int index)

    {

    if( mQuatsMap==null ) computeMap();

    int twistNum = index%729;

    int permuNum = index/729;

    int[] twist = getTwist(twistNum);

    int[] perm  = getPerm(permuNum);

    int[] quats = new int[8];

    for(int i=0; i<8; i++) quats[i] = mQuatsMap[i][perm[i]][twist[i]];

    return quats;

    }

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

  private void extractTwistAndPerm(int quat, int[][] map, int[] twist, int[] perm, int index)

    {

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

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

        if( map[i][j]==quat )

          {

          twist[index] = j;

          perm[index] = i;

          break;

          }

    }

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

  public int getIndex(int[] quats)

    {

    if( mQuatsMap==null ) computeMap();

    int[] twist = new int[8];

    int[] perm8 = new int[8];

    for(int i=0; i<8; i++) extractTwistAndPerm(quats[i],mQuatsMap[i],twist,perm8,i);

    int[] perm7 = shrinkPerm(perm8);

    int twistNum = twist[0] + 3*(twist[2] + 3*(twist[3] + 3*(twist[4] + 3*(twist[5] + 3*twist[6]))));

    int permNum  = TablebaseHelpers.computePermutationNum(perm7);

    return twistNum + 729*permNum;

    }

}