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 static org.distorted.objectlib.main.TwistyObject.SQ2;

import android.content.res.Resources;

import org.distorted.library.type.Static3D;

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

public class TBPyraminxDiamond extends TablebasesAbstract

{

  private static final int[][] CORNER_QUATS =

      {

          {0,0,0,0},

          {0,0,0,0},

          {0,0,0,0},

          {0,0,0,0},

          {0,0,0,0},

          {0,0,0,0},

      };

  private static final int[][] CENTER_QUATS =

      {

          {0,0,0,0,0,0,0,0},

          {0,0,0,0,0,0,0,0},

          {0,0,0,0,0,0,0,0},

          {0,0,0,0,0,0,0,0},

          {0,0,0,0,0,0,0,0},

          {0,0,0,0,0,0,0,0},

          {0,0,0,0,0,0,0,0},

          {0,0,0,0,0,0,0,0},

      };

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

  public TBPyraminxDiamond()

    {

    super();

    }

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

  public TBPyraminxDiamond(Resources res)

    {

    super(res, 0);

    }

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

  int[][] getBasicAngles()

    {

    int[] tmp = {4,4,4};

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

    }

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

  Static3D[] getRotationAxis()

    {

    return new Static3D[]

         {

         new Static3D(SQ2/2, 0, SQ2/2),

         new Static3D(    0, 1,     0),

         new Static3D(SQ2/2, 0,-SQ2/2)

         };

    }

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

  float[][] getPosition()

    {

    float B0 = 1.5f*SQ2;

    float C0 = 1.5f;

    float A  = 1.03f;

    float B1 = 0.5f*SQ2*A;

    float C1 = A;

    return new float[][]

         {

             {  0, B0,  0 },

             {  0,-B0,  0 },

             { C0,  0, C0 },

             { C0,  0,-C0 },

             {-C0,  0, C0 },

             {-C0,  0,-C0 },

             {  0, B1, C1 },

             {  0, B1,-C1 },

             {  0,-B1, C1 },

             {  0,-B1,-C1 },

             { C1, B1,  0 },

             { C1,-B1,  0 },

             {-C1, B1,  0 },

             {-C1,-B1,  0 },

         };

    }

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

  float[][] getCuts()

    {

    float C = 0.5f*SQ2;

    float[] cut = { -C,C};

    return new float[][] { cut,cut,cut };

    }

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

  boolean[][] getRotatable()

    {

    boolean[] tmp = {true,false,true};

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

    }

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

// specifically for the tablebase

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

// 4^6 - corner twists; 8! - center permutation; divide by 2 because parity of permutation must

// agree with parity of total twists = (4^6)*8! / 2 = 82.575.360

  int getSize()

    {

    return 82575360;

    }

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

  int getMinScramble()

    {

    return 8;

    }

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

  private int findIndex(int[] table, int entry)

    {

    int len = table.length;

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

      if( table[i]==entry ) return i;

    return -1;

    }

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

  public int[] getQuats(int index)

    {

    int[] quats = new int[14];

    int[] perm  = new int[8];

    int[] twist = new int[6];

    int centers_perm_num = index/2048;

    int total_twist = index%2048;

    TablebaseHelpers.getPermutationFromNum(perm,8,centers_perm_num);

    boolean even = TablebaseHelpers.permutationIsEven(perm);

    for(int i=0; i<5; i++) { twist[i] = total_twist%4; total_twist /=4; }

    int total = twist[0]+twist[1]+twist[2]+twist[3]+twist[4]+(even?0:1);

    twist[5] = 2*(1-total_twist) + (total%2);

    for(int i=0; i<6; i++) quats[i  ] = CORNER_QUATS[i][twist[i]];

    for(int i=0; i<8; i++) quats[i+6] = CENTER_QUATS[i][perm[i]];

    return quats;

    }

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

  public int getIndex(int[] quats)

    {

    int[] twist = new int[6];

    int[] perm  = new int[8];

    for(int i=0; i<6; i++) twist[i] = findIndex(CORNER_QUATS[i],quats[i]);

    for(int i=0; i<8; i++) perm[i]  = findIndex(CENTER_QUATS[i],quats[i]);

    int centers_perm_num = TablebaseHelpers.computePermutationNum(perm);

    int total_twist = twist[0]+ 4*(twist[1]+ 4*(twist[2]+ 4*(twist[3]+ 4*(twist[4]+ 4*(twist[5]>1 ? 0:1)))));

    return total_twist + 2048*centers_perm_num;

    }

}