// NNUE評価関数の入力特徴量Pの定義

#include "../../../config.h"

#if defined(EVAL_NNUE)

#include "p.h"
#include "index_list.h"

namespace YaneuraOu {
namespace Eval::NNUE::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 Eval::NNUE::Features
} // namespace YaneuraOu

#endif  // defined(EVAL_NNUE)
