mirror of
https://github.com/sockspls/badfish
synced 2025-07-12 12:09:14 +00:00
46 lines
1.4 KiB
C++
46 lines
1.4 KiB
C++
// NNUE評価関数の入力特徴量Pの定義
|
|
|
|
#if defined(EVAL_NNUE)
|
|
|
|
#include "p.h"
|
|
#include "index_list.h"
|
|
|
|
namespace Eval {
|
|
|
|
namespace NNUE {
|
|
|
|
namespace Features {
|
|
|
|
// 特徴量のうち、値が1であるインデックスのリストを取得する
|
|
void P::AppendActiveIndices(
|
|
const Position& pos, Color perspective, IndexList* active) {
|
|
// コンパイラの警告を回避するため、配列サイズが小さい場合は何もしない
|
|
if (RawFeatures::kMaxActiveDimensions < kMaxActiveDimensions) return;
|
|
|
|
const BonaPiece* pieces = (perspective == BLACK) ?
|
|
pos.eval_list()->piece_list_fb() :
|
|
pos.eval_list()->piece_list_fw();
|
|
for (PieceNumber i = PIECE_NUMBER_ZERO; i < PIECE_NUMBER_KING; ++i) {
|
|
active->push_back(pieces[i]);
|
|
}
|
|
}
|
|
|
|
// 特徴量のうち、一手前から値が変化したインデックスのリストを取得する
|
|
void P::AppendChangedIndices(
|
|
const Position& pos, Color perspective,
|
|
IndexList* removed, IndexList* added) {
|
|
const auto& dp = pos.state()->dirtyPiece;
|
|
for (int i = 0; i < dp.dirty_num; ++i) {
|
|
if (dp.pieceNo[i] >= PIECE_NUMBER_KING) continue;
|
|
removed->push_back(dp.changed_piece[i].old_piece.from[perspective]);
|
|
added->push_back(dp.changed_piece[i].new_piece.from[perspective]);
|
|
}
|
|
}
|
|
|
|
} // namespace Features
|
|
|
|
} // namespace NNUE
|
|
|
|
} // namespace Eval
|
|
|
|
#endif // defined(EVAL_NNUE)
|