TwistyPuzzleLib

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

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

import static org.distorted.objectlib.main.Movement12.COS54;

import static org.distorted.objectlib.main.Movement12.SIN54;

import java.io.InputStream;

import org.distorted.library.type.Static3D;

import org.distorted.library.type.Static4D;

import org.distorted.library.main.QuatHelper;

import org.distorted.objectlib.main.ObjectControl;

import org.distorted.objectlib.main.ObjectType;

import org.distorted.objectlib.helpers.ObjectShape;

import org.distorted.objectlib.helpers.ObjectSticker;

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

public class TwistyMegaminx extends TwistyMinx

{

  static final float MEGA_D = 0.04f;

  private int[] mQuatCenterIndices;

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

  public TwistyMegaminx(int[] numL, Static4D quat, Static3D move, float scale, InputStream stream)

    {

    super(numL, quat, move, scale, stream);

    }

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

  private void initializeCenterIndices()

    {

    mQuatCenterIndices = new int[] { 16, 18, 22,  1, 20, 13, 14, 15,  0, 12,  2,  3 };

    }

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

  private int numCubitsPerCorner(int numLayers)

    {

    return 3*((numLayers-1)/2)*((numLayers-3)/2) + 1;

    }

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

  private int numCubitsPerEdge(int numLayers)

    {

    return numLayers-2;

    }

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

  public int getNumStickerTypes(int[] numLayers)

    {

    return (numLayers[0]+3)/2;

    }

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

  public float[][] getCuts(int[] numLayers)

    {

    return genericGetCuts(numLayers[0],0.5f-MEGA_D);

    }

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

  private float[] computeCenter(int center, int numLayers)

    {

    if( mCenterCoords==null ) initializeCenterCoords();

    float[] coords = mCenterCoords[center];

    float A = 0.33f*numLayers;

    return new float[] { A*coords[0], A*coords[1], A*coords[2] };

    }

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

// Fill out mCurrCorner{X,Y,Z} by applying appropriate Quat to mBasicCorner{X,Y,Z}

// Appropriate one: QUATS[QUAT_INDICES[corner]].

  private void computeBasicCornerVectors(int corner)

    {

    if( mQuatCornerIndices==null ) initializeQuatIndices();

    if( mQuats==null ) initializeQuats();

    if( mCurrCornerV==null || mBasicCornerV==null ) initializeCornerV();

    Static4D quat = mQuats[mQuatCornerIndices[corner]];

    mCurrCornerV[0] = QuatHelper.rotateVectorByQuat(mBasicCornerV[0],quat);

    mCurrCornerV[1] = QuatHelper.rotateVectorByQuat(mBasicCornerV[1],quat);

    mCurrCornerV[2] = QuatHelper.rotateVectorByQuat(mBasicCornerV[2],quat);

    }

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

  private float[] computeCorner(int numCubitsPerCorner, int numLayers, int corner, int part)

    {

    if( mCorners==null ) initializeCorners();

    if( mCurrCornerV==null || mBasicCornerV==null ) initializeCornerV();

    float D = numLayers/3.0f;

    float[] corn = mCorners[corner];

    if( part==0 )

      {

      return new float[] { corn[0]*D, corn[1]*D, corn[2]*D };

      }

    else

      {

      float E = 2.0f*D*(0.5f-MEGA_D)/(0.5f*(numLayers-1));

      int N = (numCubitsPerCorner-1)/3;

      int block = (part-1) % N;

      int index = (part-1) / N;

      Static4D pri = mCurrCornerV[index];

      Static4D sec = mCurrCornerV[(index+2)%3];

      int layers= (numLayers-3)/2;

      int multP = (block % layers) + 1;

      int multS = (block / layers);

      return new float[] {

                          corn[0]*D + (pri.get0()*multP + sec.get0()*multS)*E,

                          corn[1]*D + (pri.get1()*multP + sec.get1()*multS)*E,

                          corn[2]*D + (pri.get2()*multP + sec.get2()*multS)*E

                         };

      }

    }

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

  private int computeEdgeType(int cubit, int numCubitsPerCorner, int numCubitsPerEdge)

    {

    int part = (cubit - NUM_CORNERS*numCubitsPerCorner) % numCubitsPerEdge;

    return (part+1)/2;

    }

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

  private float[] computeEdge(int numLayers, int edge, int part)

    {

    if( mCenterCoords==null ) initializeCenterCoords();

    if( mCorners==null ) initializeCorners();

    if( mEdgeMap==null ) initializeEdgeMap();

    float D = numLayers/3.0f;

    float[] c1 = mCorners[ mEdgeMap[edge][0] ];

    float[] c2 = mCorners[ mEdgeMap[edge][1] ];

    float x = D * (c1[0]+c2[0]) / 2;

    float y = D * (c1[1]+c2[1]) / 2;

    float z = D * (c1[2]+c2[2]) / 2;

    if( part==0 )

      {

      return new float[] { x, y, z };

      }

    else

      {

      int mult = (part+1)/2;

      int dir  = (part+1)%2;

      float[] center = mCenterCoords[ mEdgeMap[edge][dir+2] ];

      float vX = D*center[0] - x;

      float vY = D*center[1] - y;

      float vZ = D*center[2] - z;

      float A = 3*mult*D*(0.5f-MEGA_D)*COS18/((numLayers-1)*0.5f);

      A /= (float)Math.sqrt(vX*vX+vY*vY+vZ*vZ);

      return new float[] { x+A*vX, y+A*vY, z+A*vZ };

      }

    }

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

  public float[][] getCubitPositions(int[] numLayers)

    {

    int numL = numLayers[0];

    int numCubitsPerCorner = numCubitsPerCorner(numL);

    int numCubitsPerEdge   = numCubitsPerEdge(numL);

    int numCubits = NUM_CORNERS*numCubitsPerCorner + NUM_EDGES*numCubitsPerEdge + NUM_CENTERS;

    int index=0;

    final float[][] CENTERS = new float[numCubits][];

    for(int corner=0; corner<NUM_CORNERS; corner++)

      {

      computeBasicCornerVectors(corner);

      for(int part=0; part<numCubitsPerCorner; part++, index++)

        {

        CENTERS[index] = computeCorner(numCubitsPerCorner,numL,corner,part);

        }

      }

    for(int edge=0; edge<NUM_EDGES; edge++)

      {

      for(int part=0; part<numCubitsPerEdge; part++, index++)

        {

        CENTERS[index] = computeEdge(numL, edge, part );

        }

      }

    for(int center=0; center<NUM_CENTERS; center++, index++)

      {

      CENTERS[index] = computeCenter(center, numL);

      }

    return CENTERS;

    }

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

  private int getQuat(int cubit, int numCubitsPerCorner, int numCubitsPerEdge)

    {

    if( mQuatCornerIndices==null || mQuatEdgeIndices==null ) initializeQuatIndices();

    if( mQuatCenterIndices==null ) initializeCenterIndices();

    if( cubit < NUM_CORNERS*numCubitsPerCorner )

      {

      int corner = cubit/numCubitsPerCorner;

      return mQuatCornerIndices[corner];

      }

    if( cubit < NUM_CORNERS*numCubitsPerCorner + NUM_EDGES*numCubitsPerEdge )

      {

      int edge = (cubit-NUM_CORNERS*numCubitsPerCorner)/numCubitsPerEdge;

      return mQuatEdgeIndices[edge];

      }

    int center = cubit - NUM_CORNERS*numCubitsPerCorner - NUM_EDGES*numCubitsPerEdge;

    return mQuatCenterIndices[center];

    }

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

  public ObjectShape getObjectShape(int variant)

    {

    int[] numLayers = getNumLayers();

    int numVariants = getNumCubitVariants(numLayers);

    int numL        = numLayers[0];

    boolean small   = numL<=3;

    if( variant==0 )

      {

      float width = numL*(0.5f-MEGA_D)/(0.5f*(numL-1));

      float A = (2*SQ3/3)*SIN54;

      float B = 0.4f;

      double X = width*COS18*SIN_HALFD;

      double Y = width*SIN18;

      double Z = width*COS18*COS_HALFD;

      double[][] vertices = new double[][]

        {

            { 0.0, 0.0      , 0.0 },

            {   X,   Y      ,  -Z },

            { 0.0, 2*Y      ,-2*Z },

            {  -X,   Y      ,  -Z },

            { 0.0, 0.0-width, 0.0 },

            {   X,   Y-width,  -Z },

            { 0.0, 2*Y-width,-2*Z },

            {  -X,   Y-width,  -Z },

        };

      int[][] vertIndexes = new int[][]

        {

            {4,5,1,0},

            {7,4,0,3},

            {0,1,2,3},

            {4,5,6,7},

            {6,5,1,2},

            {7,6,2,3}

        };

      float[][] bands    = new float[][]

        {

         {0.04f,34,0.3f,0.2f, 3, 0, 0},

         {0.00f, 0,0.0f,0.0f, 2, 0, 0}

        };

      int[] bandIndices   = new int[] { 0,0,0,1,1,1};

      float[][] corners   = new float[][] { {0.04f,0.10f} };

      int[] cornerIndices = new int[] { 0,-1,-1,-1,-1,-1,-1,-1 };

      float[][] centers   = new float[][] { {0.0f, -(float)Math.sqrt(1-A*A)*B,-A*B} };

      int[] centerIndices = new int[] { 0,-1,-1,-1,-1,-1,-1,-1 };

      return new ObjectShape(vertices,vertIndexes,bands,bandIndices,corners,cornerIndices,centers,centerIndices,getNumCubitFaces(), null);

      }

    if( variant<numVariants-1 )

      {

      int type = variant-1;

      float height= numL*(0.5f-MEGA_D)*COS18/((numL-1)*0.5f);

      float width = numL*2*MEGA_D + 2*type*height*SIN18/COS18;

      double W = width/2;

      double X = height*SIN_HALFD;

      double Y = height*SIN18/COS18;

      double Z = height*COS_HALFD;

      double[][] vertices = new double[][]

        {

            { 0.0,   W   , 0.0 },

            {   X, W+Y   ,  -Z },

            { 0.0, W+2*Y ,-2*Z },

            {  -X, W+Y   ,  -Z },

            { 0.0,  -W   , 0.0 },

            {   X,-W-Y   ,  -Z },

            { 0.0,-W-2*Y ,-2*Z },

            {  -X,-W-Y   ,  -Z },

        };

      int[][] vertIndexes = new int[][]

        {

            {4,5,1,0},

            {7,4,0,3},

            {7,6,2,3},

            {6,5,1,2},

            {0,1,2,3},

            {4,5,6,7}

        };

      int N = small ? 5 : 3;

      float[][] bands = new float[][]

        {

         {0.04f,34,0.2f,0.2f,N,0,0},

         {0.00f,34,0.3f,0.2f,2,0,0}

        };

      int[] bandIndices   = new int[] { 0,0,1,1,1,1};

      float[][] corners   = new float[][] { {0.04f,0.10f} };

      int[] cornerIndices = new int[] { -1,-1,-1,-1, -1,-1,-1,-1 };

      float[][] centers   = new float[][] { {0.0f, 0.0f, (float)(-2*Z)} };

      int[] centerIndices = new int[] { -1,-1,-1,-1, -1,-1,-1,-1 };

      return new ObjectShape(vertices,vertIndexes,bands,bandIndices,corners,cornerIndices,centers,centerIndices,getNumCubitFaces(), null);

      }

    else

      {

      float width = 2*numL*(MEGA_D+(0.5f-MEGA_D)*SIN18);

      final double V = 0.83;   // ??

      final double ANGLE = V*Math.PI;

      final double cosA  = Math.cos(ANGLE);

      final double sinA  = Math.sin(ANGLE);

      float R  = 0.5f*width/COS54;

      float X1 = R*COS54;

      float Y1 = R*SIN54;

      float X2 = R*COS18;

      float Y2 = R*SIN18;

      double[][] vertices = new double[][]

        {

          {-X1,+Y1*sinA, Y1*cosA},

          {-X2,-Y2*sinA,-Y2*cosA},

          {0.0f,-R*sinA, -R*cosA},

          {+X2,-Y2*sinA,-Y2*cosA},

          {+X1,+Y1*sinA, Y1*cosA}

        };

      int[][] vertIndexes = new int[][]

        {

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

          {0,1,2,3,4}

        };

      int N = small ? 4 : 3;

      float[][] bands = new float[][]

        {

         {0.04f,45, R/3,0.2f,N,0,0},

         {0.00f,45, R/3,0.2f,2,0,0}

        };

      int[] bandIndices   = new int[] { 0,1 };

      float[][] corners   = new float[][] { {0.04f,0.10f} };

      int[] cornerIndices = new int[] { -1,-1,-1,-1, -1 };

      float[][] centers   = new float[][] { {0.0f, 0.0f, 0.0f} };

      int[] centerIndices = new int[] { -1,-1,-1,-1, -1 };

      return new ObjectShape(vertices,vertIndexes,bands,bandIndices,corners,cornerIndices,centers,centerIndices,getNumCubitFaces(), null);

      }

    }

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

  public Static4D getQuat(int cubit, int[] numLayers)

    {

    if( mQuats==null ) initializeQuats();

    int numL = numLayers[0];

    int numCubitsPerCorner = numCubitsPerCorner(numL);

    int numCubitsPerEdge   = numCubitsPerEdge(numL);

    return mQuats[getQuat(cubit,numCubitsPerCorner,numCubitsPerEdge)];

    }

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

  public int getNumCubitVariants(int[] numLayers)

    {

    switch(numLayers[0])

      {

      case 3: return 3;

      case 5: return 4;

      }

    return 3;

    }

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

  public int getCubitVariant(int cubit, int[] numLayers)

    {

    int numL = numLayers[0];

    int numCubitsPerCorner = numCubitsPerCorner(numL);

    if( cubit<NUM_CORNERS*numCubitsPerCorner ) return 0;

    int numCubitsPerEdge = numCubitsPerEdge(numL);

    if( cubit<NUM_CORNERS*numCubitsPerCorner + NUM_EDGES*numCubitsPerEdge )

      {

      int type = computeEdgeType(cubit,numCubitsPerCorner,numCubitsPerEdge);

      return type+1;

      }

    return getNumCubitVariants(numLayers)-1;

    }

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

  int getCornerColor(int cubit, int cubitface, int numLayers, int numCubitsPerCorner)

    {

    if( mCornerFaceMap==null ) initializeCornerFaceMap();

    if( cubitface<0 || cubitface>2 ) return -1;

    int part  = cubit % numCubitsPerCorner;

    int corner= cubit / numCubitsPerCorner;

    if( part==0 )

      {

      return mCornerFaceMap[corner][cubitface];

      }

    else

      {

      int N = (numCubitsPerCorner-1)/3;

      int block = (part-1) % N;

      int index = (part-1) / N;

      if( block< (numLayers-3)/2 )

        {

        switch(index)

          {

          case 0: return cubitface==1 ? -1 : mCornerFaceMap[corner][cubitface];

          case 1: return cubitface==0 ? -1 : mCornerFaceMap[corner][cubitface];

          case 2: return cubitface==2 ? -1 : mCornerFaceMap[corner][cubitface];

          }

        }

      else

        {

        switch(index)

          {

          case 0: return cubitface==0 ? mCornerFaceMap[corner][cubitface] : -1;

          case 1: return cubitface==2 ? mCornerFaceMap[corner][cubitface] : -1;

          case 2: return cubitface==1 ? mCornerFaceMap[corner][cubitface] : -1;

          }

        }

      }

    return NUM_TEXTURES;

    }

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

  int getEdgeColor(int edge, int cubitface, int numCubitsPerEdge)

    {

    if( cubitface<0 || cubitface>1 ) return -1;

    int part    = edge % numCubitsPerEdge;

    int variant = edge / numCubitsPerEdge;

    if( mEdgeMap==null ) initializeEdgeMap();

    return (part==0 || cubitface==((part+1)%2)) ? mEdgeMap[variant][cubitface+2] : -1;

    }

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

  int getCenterColor(int center, int cubitface)

    {

    return cubitface>0 ? -1 : center;

    }

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

  public int getCubitFaceColor(int cubit, int face, int[] numLayers)

    {

    int numL = numLayers[0];

    int numCubitsPerCorner = numCubitsPerCorner(numL);

    int numCubitsPerEdge   = numCubitsPerEdge(numL);

    if( cubit < NUM_CORNERS*numCubitsPerCorner )

      {

      return getCornerColor(cubit,face,numL,numCubitsPerCorner);

      }

    else if( cubit<NUM_CORNERS*numCubitsPerCorner + NUM_EDGES*numCubitsPerEdge )

      {

      int edge = cubit - NUM_CORNERS*numCubitsPerCorner;

      return getEdgeColor(edge,face,numCubitsPerEdge);

      }

    else

      {

      int center = cubit-NUM_CORNERS*numCubitsPerCorner-NUM_EDGES*numCubitsPerEdge;

      return getCenterColor( center, face );

      }

    }

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

  public int getVariantFaceColor(int variant, int face, int[] numLayers)

    {

    if( variant==0 ) return 0;

    else

      {

      if( variant==getNumCubitVariants(numLayers)-1 ) return 1;

      else return variant+1;

      }

    }

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

  public ObjectSticker retSticker(int sticker)

    {

    if( mStickers==null )

      {

      float[][] STICKERS;

      int[] numLayers = getNumLayers();

      int numL = numLayers[0];

      if( numL==3 )

        {

        STICKERS = new float[][]

          {

            { -0.36327127f, -0.5f, 0.36327127f, -0.26393202f, 0.36327127f, 0.5f, -0.36327127f, 0.26393202f },

            { -0.29389262f, 0.4045085f, -0.47552824f, -0.1545085f, 0.0f, -0.5f, 0.47552824f, -0.1545085f, 0.29389262f, 0.4045085f },

            { -0.5f, -0.0914315f, 0.5f, -0.4163512f, 0.5f, 0.4163512f, -0.5f, 0.0914315f }

          };

        }

      else

        {

        STICKERS = new float[][]

          {

            { -0.36327127f, -0.5f, 0.36327127f, -0.26393202f, 0.36327127f, 0.5f, -0.36327127f, 0.26393202f },

            { -0.29389262f, 0.4045085f, -0.47552824f, -0.1545085f, 0.0f, -0.5f, 0.47552824f, -0.1545085f, 0.29389262f, 0.4045085f },

            { -0.49233657f, -0.18006028f, 0.49233657f, -0.5f, 0.49233657f, 0.5f, -0.49233657f, 0.18006028f },

            { -0.3002273f, -0.30490047f, 0.3002273f, -0.5f, 0.3002273f, 0.5f, -0.3002273f, 0.30490047f }

          };

        }

      mStickers = new ObjectSticker[STICKERS.length];

      final float R0 = 0.08f;

      final float R1 = 0.10f;

      final float R2 = 0.12f;

      final float R3 = 0.08f;

      final float[][] radii = { {R0,R0,R0,R0},{R1,R1,R1,R1,R1},{R2,R2,R2,R2},{R3,R3,R3,R3} };

      float[] strokes = { 0.10f,0.065f,0.12f,0.08f };

      if( ObjectControl.isInIconMode() )

        {

        float mult = numL==3 ? 1.5f : 2.2f;

        strokes[0]*=mult;

        strokes[1]*=mult;

        strokes[2]*=mult;

        strokes[3]*=mult;

        strokes[4]*=mult;

        }

      for(int s=0; s<STICKERS.length; s++)

        {

        mStickers[s] = new ObjectSticker(STICKERS[s],null,radii[s],strokes[s]);

        }

      }

    return mStickers[sticker];

    }

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

  public ObjectType intGetObjectType(int[] numLayers)

    {

    switch(numLayers[0])

      {

      case 3: return ObjectType.MEGA_3;

      case 5: return ObjectType.MEGA_5;

      }

    return ObjectType.MEGA_3;

    }

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

  public String getObjectName()

    {

    switch(getNumLayers()[0])

      {

      case 3: return "Megaminx";

      case 5: return "Gigaminx";

      }

    return "Megaminx";

    }

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

  public String getInventor()

    {

    switch(getNumLayers()[0])

      {

      case 3: return "Ferenc Szlivka";

      case 5: return "Tyler Fox";

      }

    return "Ferenc Szlivka";

    }

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

  public int getYearOfInvention()

    {

    switch(getNumLayers()[0])

      {

      case 3: return 1982;

      case 5: return 2006;

      }

    return 2006;

    }

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

  public int getComplexity()

    {

    switch(getNumLayers()[0])

      {

      case 3: return 7;

      case 5: return 9;

      }

    return 9;

    }

}