1
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
2
|
// Copyright 2023 Leszek Koltunski //
|
3
|
// //
|
4
|
// This file is part of Magic Cube. //
|
5
|
// //
|
6
|
// Magic Cube is proprietary software licensed under an EULA which you should have received //
|
7
|
// along with the code. If not, check https://distorted.org/magic/License-Magic-Cube.html //
|
8
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
9
|
|
10
|
package org.distorted.objectlib.tablebases;
|
11
|
|
12
|
import android.content.res.Resources;
|
13
|
|
14
|
import org.distorted.library.main.QuatHelper;
|
15
|
import org.distorted.library.type.Static3D;
|
16
|
import org.distorted.library.type.Static4D;
|
17
|
import org.distorted.objectlib.helpers.QuatGroupGenerator;
|
18
|
|
19
|
import java.io.ByteArrayOutputStream;
|
20
|
import java.io.IOException;
|
21
|
import java.io.InputStream;
|
22
|
import java.util.ArrayList;
|
23
|
import java.util.Random;
|
24
|
|
25
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
26
|
|
27
|
public abstract class TablebasesAbstract
|
28
|
{
|
29
|
private final Static3D[] mAxis;
|
30
|
private final int mSize, mMinScramble;
|
31
|
private final int[][] mAngles;
|
32
|
private final int[] mNumLayers;
|
33
|
private final int mNumQuats;
|
34
|
private final Static4D[] mQuats;
|
35
|
private final float[][] mCuts;
|
36
|
private final int[] mNumCuts;
|
37
|
|
38
|
private int[][] mQuatMult;
|
39
|
private boolean mInitialized;
|
40
|
|
41
|
Tablebase mTablebase;
|
42
|
final int mScalingFactor;
|
43
|
final int mNumAxis;
|
44
|
final float[][] mPosition;
|
45
|
final int[][] mRotRow;
|
46
|
final int mNumCubits;
|
47
|
final boolean[][] mRotatable;
|
48
|
|
49
|
private static final float[] mTmp = new float[4];
|
50
|
|
51
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
52
|
|
53
|
abstract int[][] getBasicAngles();
|
54
|
abstract Static3D[] getRotationAxis();
|
55
|
abstract float[][] getPosition();
|
56
|
abstract float[][] getCuts();
|
57
|
|
58
|
abstract boolean[][] getRotatable();
|
59
|
|
60
|
abstract int getMinScramble();
|
61
|
abstract int getSize();
|
62
|
abstract int[] getQuats(int index);
|
63
|
abstract int getIndex(int[] quats);
|
64
|
|
65
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
66
|
|
67
|
public TablebasesAbstract()
|
68
|
{
|
69
|
mSize = getSize();
|
70
|
mMinScramble = getMinScramble();
|
71
|
mAngles = getBasicAngles();
|
72
|
mAxis = getRotationAxis();
|
73
|
mNumAxis = mAxis.length;
|
74
|
mNumLayers = new int[mNumAxis];
|
75
|
for(int i=0; i<mNumAxis; i++) mNumLayers[i] = mAngles[i].length;
|
76
|
mQuats = QuatGroupGenerator.computeGroup(mAxis,mAngles);
|
77
|
mNumQuats = mQuats.length;
|
78
|
mPosition = getPosition();
|
79
|
mNumCubits = mPosition.length;
|
80
|
mRotatable = getRotatable();
|
81
|
mCuts = getCuts();
|
82
|
mNumCuts = new int[mNumAxis];
|
83
|
|
84
|
int scaling = 0;
|
85
|
|
86
|
for(int i=0; i<mNumAxis; i++)
|
87
|
{
|
88
|
mNumCuts[i] = (mCuts==null || mCuts[i]==null ? 0 : mCuts[i].length);
|
89
|
int numLayers = mNumLayers[i];
|
90
|
for(int j=0; j<numLayers; j++) scaling+=(mAngles[i][j]-1);
|
91
|
}
|
92
|
|
93
|
mScalingFactor = scaling;
|
94
|
mRotRow = new int[mNumCubits][mNumAxis];
|
95
|
mInitialized = false;
|
96
|
}
|
97
|
|
98
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
99
|
|
100
|
public TablebasesAbstract(Resources res, int resource)
|
101
|
{
|
102
|
this();
|
103
|
|
104
|
mInitialized = true;
|
105
|
InputStream stream = res.openRawResource(resource);
|
106
|
ByteArrayOutputStream buffer = new ByteArrayOutputStream();
|
107
|
|
108
|
int nRead;
|
109
|
byte[] tmp = new byte[16384];
|
110
|
|
111
|
try
|
112
|
{
|
113
|
while ((nRead = stream.read(tmp, 0, tmp.length)) != -1)
|
114
|
{
|
115
|
buffer.write(tmp, 0, nRead);
|
116
|
}
|
117
|
stream.close();
|
118
|
byte[] data = buffer.toByteArray();
|
119
|
buffer.close();
|
120
|
mTablebase = new Tablebase(data);
|
121
|
}
|
122
|
catch(IOException ex)
|
123
|
{
|
124
|
mInitialized = false;
|
125
|
}
|
126
|
}
|
127
|
|
128
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
129
|
|
130
|
int computeRow(float[] pos, int quat, int axisIndex)
|
131
|
{
|
132
|
int ret=0;
|
133
|
int len = pos.length/3;
|
134
|
Static3D axis = mAxis[axisIndex];
|
135
|
float axisX = axis.get0();
|
136
|
float axisY = axis.get1();
|
137
|
float axisZ = axis.get2();
|
138
|
float casted, xoff=0, yoff=0, zoff=0;
|
139
|
Static4D q = mQuats[quat];
|
140
|
|
141
|
for(int i=0; i<len; i++)
|
142
|
{
|
143
|
QuatHelper.rotateVectorByQuat(mTmp,pos[3*i],pos[3*i+1],pos[3*i+2],1.0f,q);
|
144
|
casted = (mTmp[0]+xoff)*axisX + (mTmp[1]+yoff)*axisY + (mTmp[2]+zoff)*axisZ;
|
145
|
ret |= computeSingleRow(axisIndex,casted);
|
146
|
}
|
147
|
|
148
|
return ret;
|
149
|
}
|
150
|
|
151
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
152
|
|
153
|
private int computeSingleRow(int axisIndex,float casted)
|
154
|
{
|
155
|
int num = mNumCuts[axisIndex];
|
156
|
|
157
|
for(int i=0; i<num; i++)
|
158
|
{
|
159
|
if( casted<mCuts[axisIndex][i] ) return (1<<i);
|
160
|
}
|
161
|
|
162
|
return (1<<num);
|
163
|
}
|
164
|
|
165
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
166
|
// remember about the double cover or unit quaternions!
|
167
|
|
168
|
private int mulQuat(int q1, int q2)
|
169
|
{
|
170
|
Static4D result = QuatHelper.quatMultiply(mQuats[q1],mQuats[q2]);
|
171
|
|
172
|
float rX = result.get0();
|
173
|
float rY = result.get1();
|
174
|
float rZ = result.get2();
|
175
|
float rW = result.get3();
|
176
|
|
177
|
final float MAX_ERROR = 0.1f;
|
178
|
float dX,dY,dZ,dW;
|
179
|
|
180
|
for(int i=0; i<mNumQuats; i++)
|
181
|
{
|
182
|
dX = mQuats[i].get0() - rX;
|
183
|
dY = mQuats[i].get1() - rY;
|
184
|
dZ = mQuats[i].get2() - rZ;
|
185
|
dW = mQuats[i].get3() - rW;
|
186
|
|
187
|
if( dX<MAX_ERROR && dX>-MAX_ERROR &&
|
188
|
dY<MAX_ERROR && dY>-MAX_ERROR &&
|
189
|
dZ<MAX_ERROR && dZ>-MAX_ERROR &&
|
190
|
dW<MAX_ERROR && dW>-MAX_ERROR ) return i;
|
191
|
|
192
|
dX = mQuats[i].get0() + rX;
|
193
|
dY = mQuats[i].get1() + rY;
|
194
|
dZ = mQuats[i].get2() + rZ;
|
195
|
dW = mQuats[i].get3() + rW;
|
196
|
|
197
|
if( dX<MAX_ERROR && dX>-MAX_ERROR &&
|
198
|
dY<MAX_ERROR && dY>-MAX_ERROR &&
|
199
|
dZ<MAX_ERROR && dZ>-MAX_ERROR &&
|
200
|
dW<MAX_ERROR && dW>-MAX_ERROR ) return i;
|
201
|
}
|
202
|
|
203
|
return -1;
|
204
|
}
|
205
|
|
206
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
207
|
|
208
|
int getMultQuat(int index1, int index2)
|
209
|
{
|
210
|
if( mQuatMult==null )
|
211
|
{
|
212
|
mQuatMult = new int[mNumQuats][mNumQuats];
|
213
|
|
214
|
for(int i=0; i<mNumQuats; i++)
|
215
|
for(int j=0; j<mNumQuats; j++) mQuatMult[i][j] = -1;
|
216
|
}
|
217
|
|
218
|
if( index1<mNumQuats && index2<mNumQuats )
|
219
|
{
|
220
|
if( mQuatMult[index1][index2]==-1 ) mQuatMult[index1][index2] = mulQuat(index1,index2);
|
221
|
return mQuatMult[index1][index2];
|
222
|
}
|
223
|
|
224
|
return -2;
|
225
|
}
|
226
|
|
227
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
228
|
// assumption: all layers have the same basicAngles!
|
229
|
|
230
|
private int insertAllChildren(int index, byte level)
|
231
|
{
|
232
|
int ret = 0;
|
233
|
int[] quats = getQuats(index);
|
234
|
int numQuats = quats.length;
|
235
|
int[] tmpQuats = new int[numQuats];
|
236
|
byte newLevel = (byte)(level+1);
|
237
|
int quatBasis = 0;
|
238
|
|
239
|
for(int ax=0; ax<mNumAxis; ax++)
|
240
|
for(int cubit=0; cubit<mNumCubits; cubit++)
|
241
|
mRotRow[cubit][ax] = computeRow(mPosition[cubit],quats[cubit],ax);
|
242
|
|
243
|
for(int ax=0; ax<mNumAxis; ax++)
|
244
|
{
|
245
|
for(int layer=0; layer<mNumLayers[ax]; layer++)
|
246
|
{
|
247
|
if( !mRotatable[ax][layer] ) continue;
|
248
|
int bitLayer = (1<<layer);
|
249
|
int maxAngle = mAngles[ax][layer];
|
250
|
|
251
|
for(int angle=1; angle<maxAngle; angle++ )
|
252
|
{
|
253
|
System.arraycopy(quats, 0, tmpQuats, 0, numQuats);
|
254
|
int quat = quatBasis + angle;
|
255
|
|
256
|
for(int cubit=0; cubit<mNumCubits; cubit++)
|
257
|
if( mRotRow[cubit][ax]==bitLayer )
|
258
|
{
|
259
|
int currQuat = tmpQuats[cubit];
|
260
|
int newQuat = getMultQuat(quat,currQuat);
|
261
|
tmpQuats[cubit] = newQuat;
|
262
|
}
|
263
|
|
264
|
int childIndex = getIndex(tmpQuats);
|
265
|
if( mTablebase.insertUnpacked(childIndex,newLevel) ) ret++;
|
266
|
}
|
267
|
}
|
268
|
|
269
|
quatBasis += (mAngles[ax][0]-1);
|
270
|
}
|
271
|
|
272
|
return ret;
|
273
|
}
|
274
|
|
275
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
276
|
|
277
|
public void createTablebase(int maxLevel)
|
278
|
{
|
279
|
mTablebase = new Tablebase(mSize);
|
280
|
mTablebase.insertUnpacked(0,(byte)0);
|
281
|
|
282
|
int numInserted, totalInserted=1;
|
283
|
byte insertingLevel = 0;
|
284
|
|
285
|
android.util.Log.e("D", "creating tablebase of size "+mSize);
|
286
|
|
287
|
do
|
288
|
{
|
289
|
numInserted = 0;
|
290
|
|
291
|
for(int i=0; i<mSize; i++)
|
292
|
{
|
293
|
byte level = mTablebase.retrieveUnpacked(i);
|
294
|
if( level==insertingLevel ) numInserted += insertAllChildren(i,level);
|
295
|
}
|
296
|
|
297
|
insertingLevel++;
|
298
|
totalInserted += numInserted;
|
299
|
android.util.Log.e("D", "inserted "+numInserted+" positions at level "+insertingLevel);
|
300
|
}
|
301
|
while( numInserted>0 && insertingLevel!=maxLevel );
|
302
|
|
303
|
android.util.Log.e("D", "total Inserted: "+totalInserted);
|
304
|
}
|
305
|
|
306
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
307
|
|
308
|
public void pack()
|
309
|
{
|
310
|
android.util.Log.e("D", "packing...");
|
311
|
mTablebase.pack();
|
312
|
android.util.Log.e("D", "all done");
|
313
|
mInitialized = true;
|
314
|
}
|
315
|
|
316
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
317
|
|
318
|
public byte[][] getPacked()
|
319
|
{
|
320
|
if( !mInitialized ) createTablebase(-1);
|
321
|
byte[] data = mTablebase.getPacked();
|
322
|
|
323
|
return new byte[][] { data };
|
324
|
}
|
325
|
|
326
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
327
|
|
328
|
void convertMoves(int[][] moves)
|
329
|
{
|
330
|
for(int[] move : moves )
|
331
|
{
|
332
|
int axis = move[0];
|
333
|
int layer= move[1];
|
334
|
int angle= move[2];
|
335
|
|
336
|
int maxAngle = mAngles[axis][layer];
|
337
|
angle = maxAngle-angle;
|
338
|
if( angle> 0.5f*maxAngle ) angle -= maxAngle;
|
339
|
|
340
|
move[1] = (1<<layer);
|
341
|
move[2] = angle;
|
342
|
}
|
343
|
}
|
344
|
|
345
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
346
|
|
347
|
private void addMove(ArrayList<int[]> moves, int axis, int layer, int angle)
|
348
|
{
|
349
|
int maxAngle = mAngles[axis][layer];
|
350
|
angle = maxAngle-angle;
|
351
|
if( angle> 0.5f*maxAngle ) angle -= maxAngle;
|
352
|
|
353
|
int[] move = new int[] { axis, (1<<layer), angle };
|
354
|
moves.add(move);
|
355
|
}
|
356
|
|
357
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
358
|
|
359
|
private int[][] convertMoves(ArrayList<int[]> moves)
|
360
|
{
|
361
|
int len = moves.size();
|
362
|
int[][] ret = new int[len][];
|
363
|
for(int i=0; i<len; i++) ret[i] = moves.get(i);
|
364
|
return ret;
|
365
|
}
|
366
|
|
367
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
368
|
|
369
|
void getNextAxisLayerAngleQuat(int[] data)
|
370
|
{
|
371
|
int axis = data[0];
|
372
|
int layer= data[1];
|
373
|
int angle= data[2];
|
374
|
|
375
|
if( angle< mAngles[axis][layer]-1 ) data[2]++;
|
376
|
else
|
377
|
{
|
378
|
data[2] = 1;
|
379
|
|
380
|
if( layer< mNumLayers[axis]-1 ) data[1]++;
|
381
|
else
|
382
|
{
|
383
|
data[1] = 0;
|
384
|
data[0] = (axis<mNumAxis-1) ? axis+1 : 0;
|
385
|
}
|
386
|
}
|
387
|
|
388
|
data[3] = data[2];
|
389
|
for(int i=0; i<data[0]; i++) data[3] += (mAngles[i][0]-1);
|
390
|
}
|
391
|
|
392
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
393
|
|
394
|
int[][] extraInfo(int[][] moves, int[] extra)
|
395
|
{
|
396
|
return moves;
|
397
|
}
|
398
|
|
399
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
400
|
|
401
|
public int[][] solution(int index, int[] extra)
|
402
|
{
|
403
|
if( !mInitialized ) return null;
|
404
|
|
405
|
int[] data = new int[4];
|
406
|
byte level = mTablebase.retrievePacked(index);
|
407
|
ArrayList<int[]> moves = new ArrayList<>();
|
408
|
int[] quats = getQuats(index);
|
409
|
int numQuats = quats.length;
|
410
|
int[] tmpQuats = new int[numQuats];
|
411
|
|
412
|
while(index!=0)
|
413
|
{
|
414
|
boolean found = false;
|
415
|
|
416
|
data[0]=0;
|
417
|
data[1]=0;
|
418
|
data[2]=1;
|
419
|
data[3]=1;
|
420
|
|
421
|
for(int ax=0; ax<mNumAxis; ax++)
|
422
|
for(int cubit=0; cubit<mNumCubits; cubit++)
|
423
|
mRotRow[cubit][ax] = computeRow(mPosition[cubit],quats[cubit],ax);
|
424
|
|
425
|
for(int s=0; s<mScalingFactor && !found; s++)
|
426
|
{
|
427
|
int ax = data[0];
|
428
|
int layer = data[1];
|
429
|
int angle = data[2];
|
430
|
int quat = data[3];
|
431
|
|
432
|
if( mRotatable[ax][layer] )
|
433
|
{
|
434
|
int bitLayer = (1<<layer);
|
435
|
System.arraycopy(quats, 0, tmpQuats, 0, numQuats);
|
436
|
|
437
|
for(int cubit=0; cubit<mNumCubits; cubit++)
|
438
|
if( mRotRow[cubit][ax]==bitLayer )
|
439
|
{
|
440
|
int currQuat = tmpQuats[cubit];
|
441
|
int newQuat = getMultQuat(quat,currQuat);
|
442
|
tmpQuats[cubit] = newQuat;
|
443
|
}
|
444
|
|
445
|
int childIndex = getIndex(tmpQuats);
|
446
|
byte newLevel = mTablebase.retrievePacked(childIndex);
|
447
|
|
448
|
if( ((newLevel-level+1)%3) == 0 )
|
449
|
{
|
450
|
addMove(moves,ax,layer,angle);
|
451
|
index = childIndex;
|
452
|
level = (level==0 ? 2 : (byte)(level-1));
|
453
|
found = true;
|
454
|
}
|
455
|
}
|
456
|
|
457
|
getNextAxisLayerAngleQuat(data);
|
458
|
}
|
459
|
|
460
|
quats = getQuats(index);
|
461
|
|
462
|
if( !found )
|
463
|
{
|
464
|
android.util.Log.e("D", "----> solution error: no move found!");
|
465
|
return null;
|
466
|
}
|
467
|
}
|
468
|
|
469
|
int[][] ret = convertMoves(moves);
|
470
|
return extraInfo(ret,extra);
|
471
|
}
|
472
|
|
473
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
474
|
|
475
|
public void scramble(Random rnd, int depth, int[][] scramble)
|
476
|
{
|
477
|
if( !mInitialized ) return;
|
478
|
|
479
|
int solDepth = 0;
|
480
|
int scraLen = scramble.length;
|
481
|
if( depth>mMinScramble ) depth = mMinScramble;
|
482
|
|
483
|
int[] cornerTwist = new int[4];
|
484
|
for(int i=0; i<4; i++) cornerTwist[i] = (rnd.nextInt(3)-1);
|
485
|
|
486
|
while( solDepth<depth )
|
487
|
{
|
488
|
int randomPosition = rnd.nextInt(mSize-1);
|
489
|
int[][] sol = solution(randomPosition,cornerTwist);
|
490
|
solDepth = sol.length;
|
491
|
|
492
|
if( solDepth>=depth )
|
493
|
{
|
494
|
int num = Math.min(scraLen,solDepth);
|
495
|
|
496
|
for(int i=0; i<num; i++)
|
497
|
{
|
498
|
int[] source = sol[solDepth-1-i];
|
499
|
scramble[i][0] = source[0];
|
500
|
scramble[i][1] = source[1];
|
501
|
scramble[i][2] =-source[2];
|
502
|
}
|
503
|
}
|
504
|
}
|
505
|
}
|
506
|
|
507
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
508
|
|
509
|
public boolean test(int index)
|
510
|
{
|
511
|
if( !mInitialized ) return false;
|
512
|
|
513
|
int[] data = new int[4];
|
514
|
byte level = mTablebase.retrievePacked(index);
|
515
|
int[] quats = getQuats(index);
|
516
|
int numQuats = quats.length;
|
517
|
int[] tmpQuats = new int[numQuats];
|
518
|
|
519
|
data[0]=0;
|
520
|
data[1]=0;
|
521
|
data[2]=1;
|
522
|
data[3]=1;
|
523
|
|
524
|
for(int ax=0; ax<mNumAxis; ax++)
|
525
|
for(int cubit=0; cubit<mNumCubits; cubit++)
|
526
|
mRotRow[cubit][ax] = computeRow(mPosition[cubit],quats[cubit],ax);
|
527
|
|
528
|
for(int s=0; s<mScalingFactor; s++)
|
529
|
{
|
530
|
int ax = data[0];
|
531
|
int layer = data[1];
|
532
|
int quat = data[3];
|
533
|
|
534
|
if( mRotatable[ax][layer] )
|
535
|
{
|
536
|
int bitLayer = (1<<layer);
|
537
|
System.arraycopy(quats, 0, tmpQuats, 0, numQuats);
|
538
|
|
539
|
for(int cubit=0; cubit<mNumCubits; cubit++)
|
540
|
if( mRotRow[cubit][ax]==bitLayer )
|
541
|
{
|
542
|
int currQuat = tmpQuats[cubit];
|
543
|
int newQuat = getMultQuat(quat,currQuat);
|
544
|
tmpQuats[cubit] = newQuat;
|
545
|
}
|
546
|
|
547
|
int childIndex = getIndex(tmpQuats);
|
548
|
byte newLevel = mTablebase.retrievePacked(childIndex);
|
549
|
|
550
|
if( ((newLevel-level+1)%3) == 0 ) return true;
|
551
|
}
|
552
|
|
553
|
getNextAxisLayerAngleQuat(data);
|
554
|
}
|
555
|
|
556
|
return false;
|
557
|
}
|
558
|
|
559
|
///////////////////////////////////////////////////////////////////////////////////////////////////
|
560
|
|
561
|
public void testPruning(int level)
|
562
|
{
|
563
|
for( int supp : PruningTable.SUPPORTED )
|
564
|
{
|
565
|
PruningTable table = new PruningTable(mTablebase, level, supp);
|
566
|
int size = mTablebase.getSize();
|
567
|
|
568
|
StringBuilder sb = new StringBuilder();
|
569
|
int num = 0;
|
570
|
|
571
|
for(int i=0; i<size; i++)
|
572
|
{
|
573
|
if (table.contains(i))
|
574
|
{
|
575
|
if ((num % 10) == 0) sb.append("\n");
|
576
|
num++;
|
577
|
sb.append(i);
|
578
|
sb.append(' ');
|
579
|
}
|
580
|
}
|
581
|
|
582
|
android.util.Log.e("D", "numbers: " + sb);
|
583
|
}
|
584
|
}
|
585
|
}
|