Magic Cube

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

// Copyright 2022 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.bandaged;

import android.graphics.Bitmap;

import android.graphics.Canvas;

import android.graphics.Paint;

import org.distorted.library.effect.FragmentEffectBrightness;

import org.distorted.library.effect.MatrixEffectMove;

import org.distorted.library.effect.MatrixEffectQuaternion;

import org.distorted.library.effect.MatrixEffectScale;

import org.distorted.library.main.DistortedEffects;

import org.distorted.library.main.DistortedNode;

import org.distorted.library.main.DistortedTexture;

import org.distorted.library.main.QuatHelper;

import org.distorted.library.mesh.MeshBase;

import org.distorted.library.type.Static1D;

import org.distorted.library.type.Static3D;

import org.distorted.library.type.Static4D;

import org.distorted.objectlib.helpers.FactoryBandagedCubit;

import org.distorted.objectlib.helpers.FactoryCubit;

import static org.distorted.objectlib.main.TwistyObject.COLOR_BLUE;

import static org.distorted.objectlib.main.TwistyObject.COLOR_GREEN;

import static org.distorted.objectlib.main.TwistyObject.COLOR_ORANGE;

import static org.distorted.objectlib.main.TwistyObject.COLOR_RED;

import static org.distorted.objectlib.main.TwistyObject.COLOR_WHITE;

import static org.distorted.objectlib.main.TwistyObject.COLOR_YELLOW;

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

public class BandagedCubit

{

    private static final Static3D CENTER = new Static3D(0,0,0);

    private static final float[] mTmp = new float[4];

    private static final Static1D mAlpha = new Static1D(2.0f);

    private static Bitmap mBitmap;

    private static Static4D[] mTextureMaps;

    private final DistortedNode mNode;

    private final DistortedEffects mEffects;

    private final DistortedTexture mTexture;

    private final Static3D mMove;

    private final boolean mRoundCorners;

    private final int mX, mY, mZ;

    private float mUnscaledX, mUnscaledY, mUnscaledZ;

    private float[] mPosition;

    private boolean mIsAttached;

    private long mMarkedEffectID;

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

    private static void createBitmap()

      {

      final int[] FACE_COLORS = new int[]

         {

           COLOR_YELLOW, COLOR_WHITE,

           COLOR_BLUE  , COLOR_GREEN,

           COLOR_RED   , COLOR_ORANGE

         };

      final int NUM = FACE_COLORS.length;

      final int SIZE= 32;

      mTextureMaps = new Static4D[NUM];

      Paint paint = new Paint();

      paint.setStyle(Paint.Style.FILL);

      mBitmap = Bitmap.createBitmap( NUM*SIZE, SIZE, Bitmap.Config.ARGB_4444);

      Canvas canvas = new Canvas(mBitmap);

      for(int color=0; color<NUM; color++)

        {

        paint.setColor(FACE_COLORS[color]);

        canvas.drawRect(color*SIZE, 0, (color+1)*SIZE, SIZE, paint);

        mTextureMaps[color] = new Static4D( ((float)color)/NUM, 0.0f, 1.0f/NUM, 1.0f );

        }

      }

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

// (x,y,z) ==

//

// ( 1,0,0) --> 0

// (-1,0,0) --> 1

// (0, 1,0) --> 2

// (0,-1,0) --> 3

// (0,0, 1) --> 4

// (0,0,-1) --> 5

    private int computeMapsIndex(float x, float y, float z)

      {

      int ix = x>0 ? (int)(x+0.5f) : (int)(x-0.5f);

      int iy = y>0 ? (int)(y+0.5f) : (int)(y-0.5f);

      int iz = z>0 ? (int)(z+0.5f) : (int)(z-0.5f);

      if( ix== 1 ) return 0;

      if( ix==-1 ) return 1;

      if( iy== 1 ) return 2;

      if( iy==-1 ) return 3;

      if( iz== 1 ) return 4;

      if( iz==-1 ) return 5;

      return 0;

      }

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

    private void resetTextureMaps(MeshBase mesh)

      {

      FactoryCubit fact = FactoryCubit.getInstance();

      int numComponents = mesh.getNumTexComponents();

      Static4D[] maps = new Static4D[numComponents];

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

        {

        Static4D q = fact.getQuaternion(i);

        QuatHelper.rotateVectorByQuat(mTmp,0,0,1,0,q);

        int index = computeMapsIndex(mTmp[0], mTmp[1], mTmp[2]);

        maps[i] = mTextureMaps[index];

        }

      mesh.setTextureMap(maps,0);

      }

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

    private void computeMove(float[] position)

      {

      int numCenters = position.length/3;

      mUnscaledX=0.0f;

      mUnscaledY=0.0f;

      mUnscaledZ=0.0f;

      for(int center=0; center<numCenters; center++)

        {

        mUnscaledX += position[3*center  ];

        mUnscaledY += position[3*center+1];

        mUnscaledZ += position[3*center+2];

        }

      mUnscaledX /= numCenters;

      mUnscaledY /= numCenters;

      mUnscaledZ /= numCenters;

      }

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

// PUBLIC API

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

    public BandagedCubit(float[] position, int x, int y, int z, Static4D quat1,

                         Static4D quat2, Static3D scale, boolean roundCorners)

      {

      mX = x;

      mY = y;

      mZ = z;

      mRoundCorners = roundCorners;

      mPosition = position;

      mIsAttached = true;

      mMarkedEffectID = -1;

      computeMove(mPosition);

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

      FactoryBandagedCubit factory = FactoryBandagedCubit.getInstance();

      MeshBase mesh = factory.createMesh(mPosition,mX,mY,mZ,false,mRoundCorners);

      mTexture = new DistortedTexture();

      if( mBitmap==null ) createBitmap();

      mTexture.setTextureAlreadyInverted(mBitmap);

      resetTextureMaps(mesh);

      MatrixEffectScale scaleEffect = new MatrixEffectScale(scale);

      MatrixEffectQuaternion quat1Effect = new MatrixEffectQuaternion(quat1, CENTER);

      MatrixEffectQuaternion quat2Effect = new MatrixEffectQuaternion(quat2, CENTER);

      MatrixEffectMove moveEffect = new MatrixEffectMove(mMove);

      mEffects = new DistortedEffects();

      mEffects.apply(scaleEffect);

      mEffects.apply(moveEffect);

      mEffects.apply(quat2Effect);

      mEffects.apply(quat1Effect);

      mNode = new DistortedNode(mTexture,mEffects,mesh);

      }

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

    public void join(float[] position, float scale)

      {

      int len1 = mPosition.length;

      int len2 = position.length;

      float[] tmpPosition = new float[len1+len2];

      System.arraycopy(mPosition, 0, tmpPosition,    0, len1);

      System.arraycopy(position , 0, tmpPosition, len1, len2);

      computeMove(tmpPosition);

      mPosition = tmpPosition;

      FactoryBandagedCubit factory = FactoryBandagedCubit.getInstance();

      MeshBase mesh = factory.createMesh(mPosition,mX,mY,mZ,false,mRoundCorners);

      resetTextureMaps(mesh);

      mNode.setMesh(mesh);

      mMove.set( scale*mUnscaledX, scale*mUnscaledY, scale*mUnscaledZ);

      }

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

    public void reset(float scale)

      {

      float x = mPosition[0];

      float y = mPosition[1];

      float z = mPosition[2];

      mPosition = new float[3];

      mPosition[0] = x;

      mPosition[1] = y;

      mPosition[2] = z;

      computeMove(mPosition);

      FactoryBandagedCubit factory = FactoryBandagedCubit.getInstance();

      MeshBase mesh = factory.createMesh(mPosition,mX,mY,mZ,false,mRoundCorners);

      resetTextureMaps(mesh);

      mNode.setMesh(mesh);

      mMove.set( scale*mUnscaledX, scale*mUnscaledY, scale*mUnscaledZ);

      }

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

    public void recreateBitmap()

      {

      if( mBitmap==null ) createBitmap();

      mTexture.setTextureAlreadyInverted(mBitmap);

      }

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

    public void scaleMove(float scale)

      {

      mMove.set( scale*mUnscaledX, scale*mUnscaledY, scale*mUnscaledZ);

      }

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

    public void setMarked()

      {

      if( mMarkedEffectID<0 )

        {

        FragmentEffectBrightness effect = new FragmentEffectBrightness(mAlpha);

        mMarkedEffectID = effect.getID();

        mEffects.apply(effect);

        }

      }

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

    public void setUnmarked()

      {

      if( mMarkedEffectID>=0 )

        {

        mEffects.abortById(mMarkedEffectID);

        mMarkedEffectID = -1;

        }

      }

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

    public void detach()

      {

      mIsAttached = false;

      }

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

    public void attach()

      {

      mIsAttached = true;

      }

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

    public boolean isAttached()

      {

      return mIsAttached;

      }

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

    public float[] getPosition()

      {

      return mPosition;

      }

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

    public DistortedNode getNode()

      {

      return mNode;

      }

}