ellpack_page.h

 
#ifndef XGBOOST_DATA_ELLPACK_PAGE_H_
#define XGBOOST_DATA_ELLPACK_PAGE_H_
 
#include <memory>  // for unique_ptr
 
#include "../common/hist_util.h"  // for HistogramCuts
#include "xgboost/context.h"      // for Context
#include "xgboost/data.h"         // for DMatrix, BatchParam
 
namespace xgboost {
class EllpackPageImpl;
class EllpackPage {
 public:
  EllpackPage();
  explicit EllpackPage(Context const* ctx, DMatrix* dmat, const BatchParam& param);
 
  ~EllpackPage();
 
  EllpackPage(EllpackPage&& that);
 
  [[nodiscard]] size_t Size() const;
 
  void SetBaseRowId(std::size_t row_id);
 
  [[nodiscard]] const EllpackPageImpl* Impl() const { return impl_.get(); }
  EllpackPageImpl* Impl() { return impl_.get(); }
 
  [[nodiscard]] common::HistogramCuts& Cuts();
  [[nodiscard]] common::HistogramCuts const& Cuts() const;
 
 private:
  std::unique_ptr<EllpackPageImpl> impl_;
};
}  // namespace xgboost
#endif  // XGBOOST_DATA_ELLPACK_PAGE_H_