quantile::detail::CentroidsMerger< RealType, IndexType > Class Template Reference

#include <quantile.h>

Public Member Functions
DEVICE	CentroidsMerger (Centroids< RealType, IndexType > *buf, Centroids< RealType, IndexType > *centroids, bool const forward)
DEVICE Centroids< RealType, IndexType > *	getNextCentroid () const
DEVICE bool	hasNext () const
DEVICE void	merge (IndexType const max_count)
DEVICE void	next ()
DEVICE IndexType	prefixSum () const
DEVICE IndexType	totalWeight () const

Private Member Functions
DEVICE void	mergeMinMax ()
DEVICE void	setCurrCentroid ()

Private Attributes
Centroids< RealType, IndexType > *	buf_
Centroids< RealType, IndexType > *	centroids_
Centroids< RealType, IndexType > *	curr_centroid_
IndexType	prefix_sum_ {0}
IndexType const	total_weight_
bool const	forward_

Detailed Description

template<typename RealType, typename IndexType>
class quantile::detail::CentroidsMerger< RealType, IndexType >

Definition at line 134 of file quantile.h.

Constructor & Destructor Documentation

template<typename RealType , typename IndexType >

DEVICE quantile::detail::CentroidsMerger< RealType, IndexType >::CentroidsMerger	(	Centroids< RealType, IndexType > *	buf,
		Centroids< RealType, IndexType > *	centroids,
		bool const	forward
	)

Definition at line 441 of file quantile.h.

References quantile::detail::CentroidsMerger< RealType, IndexType >::buf_, quantile::detail::CentroidsMerger< RealType, IndexType >::centroids_, quantile::detail::CentroidsMerger< RealType, IndexType >::forward_, quantile::detail::CentroidsMerger< RealType, IndexType >::mergeMinMax(), and quantile::detail::CentroidsMerger< RealType, IndexType >::setCurrCentroid().

    : buf_(buf)
    , centroids_(centroids)
    , total_weight_(centroids->totalWeight() + buf->totalWeight())
    , forward_(forward) {
  buf_->resetIndices(forward_);
  centroids_->resetIndices(forward_);
  mergeMinMax();
  setCurrCentroid();
}

Here is the call graph for this function:

Member Function Documentation

template<typename RealType , typename IndexType >

DEVICE Centroids< RealType, IndexType > * quantile::detail::CentroidsMerger< RealType, IndexType >::getNextCentroid

(

)

const

Definition at line 457 of file quantile.h.

                                                            {
  if (buf_->hasNext()) {
    if (centroids_->hasNext()) {
      return (*buf_ < *centroids_) == forward_ ? buf_ : centroids_;
    }
    return buf_;
  } else if (centroids_->hasNext()) {
    return centroids_;
  }
  return nullptr;
}

template<typename RealType , typename IndexType >

DEVICE bool quantile::detail::CentroidsMerger< RealType, IndexType >::hasNext ( ) const

inline

Definition at line 153 of file quantile.h.

References quantile::detail::CentroidsMerger< RealType, IndexType >::buf_, and quantile::detail::CentroidsMerger< RealType, IndexType >::centroids_.

{ return buf_->hasNext() || centroids_->hasNext(); }

template<typename RealType , typename IndexType >

DEVICE void quantile::detail::CentroidsMerger< RealType, IndexType >::merge

(

IndexType const

max_count

)

Definition at line 568 of file quantile.h.

Referenced by quantile::detail::TDigest< RealType, IndexType >::mergeCentroids().

                                                                                 {
  Skipped<RealType, IndexType> skipped;
  while (auto* next_centroid = getNextCentroid()) {
    if (skipped) {
      if (skipped.isDifferentMean(next_centroid)) {
        break;
      } else if (curr_centroid_->mergeIfFits(*next_centroid, max_count)) {
        skipped.merged(next_centroid);
      } else {
        skipped.skipSubsequent(next_centroid);
      }
    } else if (!curr_centroid_->mergeIfFits(*next_centroid, max_count)) {
      skipped.skipFirst(next_centroid);
    }
  }
  if (skipped) {
    skipped.shiftCentroidsAndSetNext();
  }
}

Here is the caller graph for this function:

template<typename RealType , typename IndexType >

DEVICE void quantile::detail::CentroidsMerger< RealType, IndexType >::mergeMinMax ( )

private

Definition at line 597 of file quantile.h.

Referenced by quantile::detail::CentroidsMerger< RealType, IndexType >::CentroidsMerger().

                                                              {
  if (centroids_->max_ < buf_->max_) {
    centroids_->max_ = buf_->max_;
  }
  if (buf_->min_ < centroids_->min_) {
    centroids_->min_ = buf_->min_;
  }
}

Here is the caller graph for this function:

template<typename RealType , typename IndexType >

DEVICE void quantile::detail::CentroidsMerger< RealType, IndexType >::next

(

)

Definition at line 608 of file quantile.h.

                                                       {
  prefix_sum_ += curr_centroid_->currCount();
  setCurrCentroid();
}

template<typename RealType , typename IndexType >

DEVICE IndexType quantile::detail::CentroidsMerger< RealType, IndexType >::prefixSum ( ) const

inline

Definition at line 161 of file quantile.h.

References quantile::detail::CentroidsMerger< RealType, IndexType >::prefix_sum_.

{ return prefix_sum_; }

template<typename RealType , typename IndexType >

DEVICE void quantile::detail::CentroidsMerger< RealType, IndexType >::setCurrCentroid ( )

private

Definition at line 614 of file quantile.h.

Referenced by quantile::detail::CentroidsMerger< RealType, IndexType >::CentroidsMerger().

                                                                  {
  if ((curr_centroid_ = getNextCentroid())) {
    curr_centroid_->moveNextToCurrent();
  }
}

Here is the caller graph for this function:

template<typename RealType , typename IndexType >

DEVICE IndexType quantile::detail::CentroidsMerger< RealType, IndexType >::totalWeight ( ) const

inline

Definition at line 163 of file quantile.h.

References quantile::detail::CentroidsMerger< RealType, IndexType >::total_weight_.

{ return total_weight_; }

Member Data Documentation

template<typename RealType , typename IndexType >

Centroids<RealType, IndexType>* quantile::detail::CentroidsMerger< RealType, IndexType >::buf_

private

Definition at line 135 of file quantile.h.

Referenced by quantile::detail::CentroidsMerger< RealType, IndexType >::CentroidsMerger(), and quantile::detail::CentroidsMerger< RealType, IndexType >::hasNext().

template<typename RealType , typename IndexType >

Centroids<RealType, IndexType>* quantile::detail::CentroidsMerger< RealType, IndexType >::centroids_

private

Definition at line 136 of file quantile.h.

Referenced by quantile::detail::CentroidsMerger< RealType, IndexType >::CentroidsMerger(), and quantile::detail::CentroidsMerger< RealType, IndexType >::hasNext().

template<typename RealType , typename IndexType >

Centroids<RealType, IndexType>* quantile::detail::CentroidsMerger< RealType, IndexType >::curr_centroid_

private

Definition at line 137 of file quantile.h.

template<typename RealType , typename IndexType >

bool const quantile::detail::CentroidsMerger< RealType, IndexType >::forward_

private

Definition at line 140 of file quantile.h.

Referenced by quantile::detail::CentroidsMerger< RealType, IndexType >::CentroidsMerger().

template<typename RealType , typename IndexType >

IndexType quantile::detail::CentroidsMerger< RealType, IndexType >::prefix_sum_ {0}

private

Definition at line 138 of file quantile.h.

Referenced by quantile::detail::CentroidsMerger< RealType, IndexType >::prefixSum().

template<typename RealType , typename IndexType >

IndexType const quantile::detail::CentroidsMerger< RealType, IndexType >::total_weight_

private

Definition at line 139 of file quantile.h.

Referenced by quantile::detail::CentroidsMerger< RealType, IndexType >::totalWeight().

---

The documentation for this class was generated from the following file:

• /home/jenkins-slave/workspace/core-os-doxygen/Shared/quantile.h