mirror of
https://github.com/HChaZZY/Stockfish.git
synced 2025-12-23 18:46:59 +08:00
98 lines
3.3 KiB
C++
98 lines
3.3 KiB
C++
// NNUE評価関数の入力特徴量HalfRelativeKPの定義
|
|
|
|
#if defined(EVAL_NNUE)
|
|
|
|
#include "half_relative_kp.h"
|
|
#include "index_list.h"
|
|
|
|
namespace Eval {
|
|
|
|
namespace NNUE {
|
|
|
|
namespace Features {
|
|
|
|
// 玉の位置とBonaPieceから特徴量のインデックスを求める
|
|
template <Side AssociatedKing>
|
|
inline IndexType HalfRelativeKP<AssociatedKing>::MakeIndex(
|
|
Square sq_k, BonaPiece p) {
|
|
constexpr IndexType W = kBoardWidth;
|
|
constexpr IndexType H = kBoardHeight;
|
|
const IndexType piece_index = (p - fe_hand_end) / SQUARE_NB;
|
|
const Square sq_p = static_cast<Square>((p - fe_hand_end) % SQUARE_NB);
|
|
const IndexType relative_file = file_of(sq_p) - file_of(sq_k) + (W / 2);
|
|
const IndexType relative_rank = rank_of(sq_p) - rank_of(sq_k) + (H / 2);
|
|
return H * W * piece_index + H * relative_file + relative_rank;
|
|
}
|
|
|
|
// 駒の情報を取得する
|
|
template <Side AssociatedKing>
|
|
inline void HalfRelativeKP<AssociatedKing>::GetPieces(
|
|
const Position& pos, Color perspective,
|
|
BonaPiece** pieces, Square* sq_target_k) {
|
|
*pieces = (perspective == BLACK) ?
|
|
pos.eval_list()->piece_list_fb() :
|
|
pos.eval_list()->piece_list_fw();
|
|
const PieceNumber target = (AssociatedKing == Side::kFriend) ?
|
|
static_cast<PieceNumber>(PIECE_NUMBER_KING + perspective) :
|
|
static_cast<PieceNumber>(PIECE_NUMBER_KING + ~perspective);
|
|
*sq_target_k = static_cast<Square>(((*pieces)[target] - f_king) % SQUARE_NB);
|
|
}
|
|
|
|
// 特徴量のうち、値が1であるインデックスのリストを取得する
|
|
template <Side AssociatedKing>
|
|
void HalfRelativeKP<AssociatedKing>::AppendActiveIndices(
|
|
const Position& pos, Color perspective, IndexList* active) {
|
|
// コンパイラの警告を回避するため、配列サイズが小さい場合は何もしない
|
|
if (RawFeatures::kMaxActiveDimensions < kMaxActiveDimensions) return;
|
|
|
|
BonaPiece* pieces;
|
|
Square sq_target_k;
|
|
GetPieces(pos, perspective, &pieces, &sq_target_k);
|
|
for (PieceNumber i = PIECE_NUMBER_ZERO; i < PIECE_NUMBER_KING; ++i) {
|
|
if (pieces[i] >= fe_hand_end) {
|
|
if (pieces[i] != Eval::BONA_PIECE_ZERO) {
|
|
active->push_back(MakeIndex(sq_target_k, pieces[i]));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// 特徴量のうち、一手前から値が変化したインデックスのリストを取得する
|
|
template <Side AssociatedKing>
|
|
void HalfRelativeKP<AssociatedKing>::AppendChangedIndices(
|
|
const Position& pos, Color perspective,
|
|
IndexList* removed, IndexList* added) {
|
|
BonaPiece* pieces;
|
|
Square sq_target_k;
|
|
GetPieces(pos, perspective, &pieces, &sq_target_k);
|
|
const auto& dp = pos.state()->dirtyPiece;
|
|
for (int i = 0; i < dp.dirty_num; ++i) {
|
|
if (dp.pieceNo[i] >= PIECE_NUMBER_KING) continue;
|
|
const auto old_p = static_cast<BonaPiece>(
|
|
dp.changed_piece[i].old_piece.from[perspective]);
|
|
if (old_p >= fe_hand_end) {
|
|
if (old_p != Eval::BONA_PIECE_ZERO) {
|
|
removed->push_back(MakeIndex(sq_target_k, old_p));
|
|
}
|
|
}
|
|
const auto new_p = static_cast<BonaPiece>(
|
|
dp.changed_piece[i].new_piece.from[perspective]);
|
|
if (new_p >= fe_hand_end) {
|
|
if (new_p != Eval::BONA_PIECE_ZERO) {
|
|
added->push_back(MakeIndex(sq_target_k, new_p));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
template class HalfRelativeKP<Side::kFriend>;
|
|
template class HalfRelativeKP<Side::kEnemy>;
|
|
|
|
} // namespace Features
|
|
|
|
} // namespace NNUE
|
|
|
|
} // namespace Eval
|
|
|
|
#endif // defined(EVAL_NNUE)
|