mirror of
https://github.com/peterosterlund2/droidfish.git
synced 2024-12-05 01:35:50 +01:00
83 lines
2.3 KiB
C++
83 lines
2.3 KiB
C++
|
/*
|
||
|
|
DroidFish - An Android chess program.
|
||
|
|
Copyright (C) 2011-2012 Peter Österlund, peterosterlund2@gmail.com
|
||
|
|
|
||
|
|
This program 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 3 of the License, or
|
||
|
|
(at your option) any later version.
|
||
|
|
|
||
|
|
This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
*/
|
||
|
|
#include "RtbProbe.h"
|
||
|
|
#include "tbprobe.hpp"
|
||
|
|
#include <algorithm>
|
||
|
|
|
||
|
|
|
||
|
|
static bool initOk = false;
|
||
|
|
|
||
|
|
JNIEXPORT jboolean
|
||
|
|
JNICALL Java_org_petero_droidfish_gtb_RtbProbe_init(
|
||
|
|
JNIEnv* env, jclass cls, jstring jTbPath)
|
||
|
|
{
|
||
|
|
initOk = false;
|
||
|
|
const char* tbPath = (*env).GetStringUTFChars(jTbPath, NULL);
|
||
|
|
if (!tbPath)
|
||
|
|
return false;
|
||
|
|
std::string rtbPath(tbPath);
|
||
|
|
(*env).ReleaseStringUTFChars(jTbPath, tbPath);
|
||
|
|
|
||
|
|
TBProbe::initialize(rtbPath);
|
||
|
|
initOk = true;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
JNIEXPORT void
|
||
|
|
JNICALL Java_org_petero_droidfish_gtb_RtbProbe_probe(
|
||
|
|
JNIEnv* env, jobject ths, jbyteArray jSquares, jboolean wtm,
|
||
|
|
jint epSq, jint castleMask,
|
||
|
|
jint halfMoveClock, jint fullMoveCounter,
|
||
|
|
jintArray result)
|
||
|
|
{
|
||
|
|
if ((*env).GetArrayLength(result) < 2)
|
||
|
|
return;
|
||
|
|
|
||
|
|
jint res[2];
|
||
|
|
res[0] = 1000;
|
||
|
|
res[1] = 1000;
|
||
|
|
(*env).SetIntArrayRegion(result, 0, 2, res);
|
||
|
|
|
||
|
|
if (!initOk)
|
||
|
|
return;
|
||
|
|
|
||
|
|
const int len = (*env).GetArrayLength(jSquares);
|
||
|
|
if (len != 64)
|
||
|
|
return;
|
||
|
|
|
||
|
|
Position pos;
|
||
|
|
jbyte* jbPtr = (*env).GetByteArrayElements(jSquares, NULL);
|
||
|
|
for (int i = 0; i < 64; i++)
|
||
|
|
pos.setPiece(i, jbPtr[i]);
|
||
|
|
(*env).ReleaseByteArrayElements(jSquares, jbPtr, 0);
|
||
|
|
|
||
|
|
pos.setWhiteMove(wtm);
|
||
|
|
pos.setEpSquare(epSq);
|
||
|
|
pos.setCastleMask(castleMask);
|
||
|
|
pos.setHalfMoveClock(halfMoveClock);
|
||
|
|
pos.setFullMoveCounter(fullMoveCounter);
|
||
|
|
|
||
|
|
int score;
|
||
|
|
if (TBProbe::rtbProbeWDL(pos, score))
|
||
|
|
res[0] = score;
|
||
|
|
if (TBProbe::rtbProbeDTZ(pos, score))
|
||
|
|
res[1] = score;
|
||
|
|
|
||
|
|
(*env).SetIntArrayRegion(result, 0, 2, res);
|
||
|
|
}
|