quantile::detail::Centroids< RealType, IndexType > Struct Template Reference

#include <quantile.h>

Collaboration diagram for quantile::detail::Centroids< RealType, IndexType >:

Public Member Functions
	Centroids ()=default
DEVICE	Centroids (RealType *sums, IndexType *counts, IndexType const size)
DEVICE	Centroids (VectorView< RealType > sums, VectorView< IndexType > counts)
DEVICE void	appendAndSortCurrent (Centroids &buff)
DEVICE IndexType	capacity () const
DEVICE void	clear ()
DEVICE IndexType	currCount () const
DEVICE RealType	currMean () const
DEVICE bool	hasCurr () const
DEVICE bool	hasNext () const
DEVICE RealType	mean (IndexType const i) const
DEVICE bool	mergeIfFits (Centroids &centroid, IndexType const max_count)
DEVICE void	moveNextToCurrent ()
DEVICE IndexType	nextCount () const
DEVICE RealType	nextSum () const
DEVICE bool	operator< (Centroids const &b) const
DEVICE void	push_back (RealType const value, RealType const count)
DEVICE void	resetIndices (bool const forward)
DEVICE size_t	size () const
DEVICE IndexType	totalWeight () const

Public Attributes
IndexType	curr_idx_
IndexType	next_idx_
int	inc_
VectorView< RealType >	sums_
VectorView< IndexType >	counts_
RealType	max_ {-infinity}
RealType	min_ {infinity}

Static Public Attributes
static constexpr RealType	infinity = std::numeric_limits<RealType>::infinity()
static constexpr RealType	nan = std::numeric_limits<RealType>::quiet_NaN()

Friends
template<typename RealType2 , typename IndexType2 >
std::ostream &	operator<< (std::ostream &, Centroids< RealType2, IndexType2 > const &)

Detailed Description

template<typename RealType, typename IndexType>
struct quantile::detail::Centroids< RealType, IndexType >

Definition at line 56 of file quantile.h.

Constructor & Destructor Documentation

template<typename RealType, typename IndexType>

quantile::detail::Centroids< RealType, IndexType >::Centroids ( )

default

template<typename RealType, typename IndexType>

DEVICE quantile::detail::Centroids< RealType, IndexType >::Centroids ( RealType *  sums, IndexType *  counts, IndexType const  size  )

inline

Definition at line 69 of file quantile.h.

      : sums_(sums, size, size), counts_(counts, size, size) {}

template<typename RealType, typename IndexType>

DEVICE quantile::detail::Centroids< RealType, IndexType >::Centroids ( VectorView< RealType >  sums, VectorView< IndexType >  counts  )

inline

Definition at line 72 of file quantile.h.

      : sums_(sums), counts_(counts) {}

Member Function Documentation

template<typename RealType , typename IndexType >

DEVICE void quantile::detail::Centroids< RealType, IndexType >::appendAndSortCurrent

(

Centroids< RealType, IndexType > &

buff

)

Definition at line 321 of file quantile.h.

References VectorView< T >::begin(), gpu_enabled::copy(), quantile::detail::Centroids< RealType, IndexType >::counts_, quantile::detail::Centroids< RealType, IndexType >::curr_idx_, gpu_enabled::reverse(), VectorView< T >::size(), quantile::detail::Centroids< RealType, IndexType >::size(), gpu_enabled::sort(), and quantile::detail::Centroids< RealType, IndexType >::sums_.

                                                                                {
  if (inc_ == -1 && curr_idx_ != 0) {
    // Shift data to the left left by curr_idx_.
    // Prefer to copy, but thrust::copy doesn't support overlapping ranges.
    // Reverse instead, which gets sorted below.
    // gpu_enabled::copy(sums_.begin() + curr_idx_, sums_.end(), sums_.begin());
    // gpu_enabled::copy(counts_.begin() + curr_idx_, counts_.end(), counts_.begin());
    gpu_enabled::reverse(sums_.begin(), sums_.end());
    gpu_enabled::reverse(counts_.begin(), counts_.end());
  }
  // Shift VectorViews to the right by buff.curr_idx_.
  IndexType const offset = inc_ == 1 ? 0 : buff.curr_idx_;
  IndexType const buff_size =
      inc_ == 1 ? buff.curr_idx_ + 1 : buff.size() - buff.curr_idx_;
  VectorView<RealType> buff_sums(buff.sums_.begin() + offset, buff_size, buff_size);
  VectorView<IndexType> buff_counts(buff.counts_.begin() + offset, buff_size, buff_size);
  // Copy buff into sums_ and counts_.
  IndexType const curr_size = inc_ == 1 ? curr_idx_ + 1 : size() - curr_idx_;
  IndexType const total_size = curr_size + buff_sums.size();
  assert(total_size <= sums_.capacity());  // TODO proof of inequality
  sums_.resize(total_size);
  gpu_enabled::copy(buff_sums.begin(), buff_sums.end(), sums_.begin() + curr_size);
  assert(total_size <= counts_.capacity());
  counts_.resize(total_size);
  gpu_enabled::copy(buff_counts.begin(), buff_counts.end(), counts_.begin() + curr_size);

  // Sort sums_ and counts_ by (mean, -count).
  double_sort::Iterator<RealType, IndexType> const begin(sums_.begin(), counts_.begin());
  double_sort::Iterator<RealType, IndexType> const end(sums_.end(), counts_.end());
  gpu_enabled::sort(begin, end, OrderByMeanAscCountDesc<RealType, IndexType>());
}

Here is the call graph for this function:

template<typename RealType, typename IndexType>

DEVICE IndexType quantile::detail::Centroids< RealType, IndexType >::capacity ( ) const

inline

Definition at line 77 of file quantile.h.

References VectorView< T >::capacity(), and quantile::detail::Centroids< RealType, IndexType >::sums_.

{ return sums_.capacity(); }

Here is the call graph for this function:

template<typename RealType, typename IndexType>

DEVICE void quantile::detail::Centroids< RealType, IndexType >::clear ( )

inline

Definition at line 79 of file quantile.h.

References VectorView< T >::clear(), quantile::detail::Centroids< RealType, IndexType >::counts_, quantile::detail::Centroids< RealType, IndexType >::infinity, quantile::detail::Centroids< RealType, IndexType >::max_, quantile::detail::Centroids< RealType, IndexType >::min_, and quantile::detail::Centroids< RealType, IndexType >::sums_.

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

                      {
    sums_.clear();
    counts_.clear();
    max_ = -infinity;
    min_ = infinity;
  }

Here is the call graph for this function:

Here is the caller graph for this function:

template<typename RealType, typename IndexType>

DEVICE IndexType quantile::detail::Centroids< RealType, IndexType >::currCount ( ) const

inline

Definition at line 86 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::counts_, and quantile::detail::Centroids< RealType, IndexType >::curr_idx_.

{ return counts_[curr_idx_]; }

template<typename RealType, typename IndexType>

DEVICE RealType quantile::detail::Centroids< RealType, IndexType >::currMean ( ) const

inline

Definition at line 88 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::curr_idx_, and quantile::detail::Centroids< RealType, IndexType >::mean().

{ return mean(curr_idx_); }

Here is the call graph for this function:

template<typename RealType, typename IndexType>

DEVICE bool quantile::detail::Centroids< RealType, IndexType >::hasCurr ( ) const

inline

Definition at line 90 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::curr_idx_, and quantile::detail::Centroids< RealType, IndexType >::size().

{ return curr_idx_ < size(); }

Here is the call graph for this function:

template<typename RealType, typename IndexType>

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

inline

Definition at line 92 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::next_idx_, and quantile::detail::Centroids< RealType, IndexType >::size().

{ return next_idx_ < size(); }

Here is the call graph for this function:

template<typename RealType, typename IndexType>

DEVICE RealType quantile::detail::Centroids< RealType, IndexType >::mean ( IndexType const  i ) const

inline

Definition at line 94 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::counts_, and quantile::detail::Centroids< RealType, IndexType >::sums_.

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

{ return sums_[i] / counts_[i]; }

Here is the caller graph for this function:

template<typename RealType , typename IndexType >

DEVICE bool quantile::detail::Centroids< RealType, IndexType >::mergeIfFits	(	Centroids< RealType, IndexType > &	centroid,
		IndexType const	max_count
	)

Definition at line 355 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::inc_, quantile::detail::Centroids< RealType, IndexType >::next_idx_, quantile::detail::Centroids< RealType, IndexType >::nextCount(), and quantile::detail::Centroids< RealType, IndexType >::nextSum().

                                                                                   {
  if (counts_[curr_idx_] + centroid.nextCount() <= max_count) {
    sums_[curr_idx_] += centroid.nextSum();
    counts_[curr_idx_] += centroid.nextCount();
    centroid.next_idx_ += centroid.inc_;
    return true;
  }
  return false;
}

Here is the call graph for this function:

template<typename RealType , typename IndexType >

DEVICE void quantile::detail::Centroids< RealType, IndexType >::moveNextToCurrent

(

)

Definition at line 367 of file quantile.h.

                                                              {
  curr_idx_ += inc_;
  if (curr_idx_ != next_idx_) {
    sums_[curr_idx_] = sums_[next_idx_];
    counts_[curr_idx_] = counts_[next_idx_];
  }
  next_idx_ += inc_;
}

template<typename RealType, typename IndexType>

DEVICE IndexType quantile::detail::Centroids< RealType, IndexType >::nextCount ( ) const

inline

Definition at line 101 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::counts_, and quantile::detail::Centroids< RealType, IndexType >::next_idx_.

Referenced by quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::isDifferentMean(), quantile::detail::Centroids< RealType, IndexType >::mergeIfFits(), quantile::detail::Centroids< RealType, IndexType >::operator<(), and quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::skipFirst().

{ return counts_[next_idx_]; }

Here is the caller graph for this function:

template<typename RealType, typename IndexType>

DEVICE RealType quantile::detail::Centroids< RealType, IndexType >::nextSum ( ) const

inline

Definition at line 103 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::next_idx_, and quantile::detail::Centroids< RealType, IndexType >::sums_.

Referenced by quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::isDifferentMean(), quantile::detail::Centroids< RealType, IndexType >::mergeIfFits(), quantile::detail::Centroids< RealType, IndexType >::operator<(), and quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::skipFirst().

{ return sums_[next_idx_]; }

Here is the caller graph for this function:

template<typename RealType, typename IndexType>

DEVICE bool quantile::detail::Centroids< RealType, IndexType >::operator< ( Centroids< RealType, IndexType > const &  b ) const

inline

Definition at line 107 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::nextCount(), and quantile::detail::Centroids< RealType, IndexType >::nextSum().

                                                  {
    // lhs < rhs is same as nextMean() < b.nextMean() without division
    RealType const lhs = nextSum() * b.nextCount();
    RealType const rhs = b.nextSum() * nextCount();
    return lhs < rhs || (lhs == rhs && b.nextCount() < nextCount());
  }

Here is the call graph for this function:

template<typename RealType, typename IndexType>

DEVICE void quantile::detail::Centroids< RealType, IndexType >::push_back ( RealType const  value, RealType const  count  )

inline

Definition at line 114 of file quantile.h.

References quantile::detail::Centroids< RealType, IndexType >::counts_, VectorView< T >::push_back(), and quantile::detail::Centroids< RealType, IndexType >::sums_.

                                                                    {
    sums_.push_back(value);
    counts_.push_back(count);
  }

Here is the call graph for this function:

template<typename RealType , typename IndexType >

DEVICE void quantile::detail::Centroids< RealType, IndexType >::resetIndices

(

bool const

forward

)

Definition at line 377 of file quantile.h.

                                                                           {
  if (forward) {
    inc_ = 1;
    curr_idx_ = ~IndexType(0);
  } else {
    inc_ = -1;
    curr_idx_ = size();
  }
  static_assert(std::is_unsigned<IndexType>::value,
                "IndexType must be an unsigned type.");
  next_idx_ = curr_idx_ + inc_;
}

template<typename RealType, typename IndexType>

DEVICE size_t quantile::detail::Centroids< RealType, IndexType >::size ( ) const

inline

Definition at line 121 of file quantile.h.

References VectorView< T >::size(), and quantile::detail::Centroids< RealType, IndexType >::sums_.

Referenced by quantile::detail::Centroids< RealType, IndexType >::appendAndSortCurrent(), quantile::detail::Centroids< RealType, IndexType >::hasCurr(), quantile::detail::Centroids< RealType, IndexType >::hasNext(), and quantile::detail::operator<<().

{ return sums_.size(); }

Here is the call graph for this function:

Here is the caller graph for this function:

template<typename RealType, typename IndexType>

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

inline

Definition at line 123 of file quantile.h.

References gpu_enabled::accumulate(), VectorView< T >::begin(), quantile::detail::Centroids< RealType, IndexType >::counts_, and VectorView< T >::end().

                                       {
    return gpu_enabled::accumulate(counts_.begin(), counts_.end(), IndexType(0));
  }

Here is the call graph for this function:

Friends And Related Function Documentation

template<typename RealType, typename IndexType>

template<typename RealType2 , typename IndexType2 >

std::ostream& operator<< ( std::ostream &  , Centroids< RealType2, IndexType2 > const &    )

friend

Member Data Documentation

template<typename RealType, typename IndexType>

VectorView<IndexType> quantile::detail::Centroids< RealType, IndexType >::counts_

Definition at line 61 of file quantile.h.

Referenced by quantile::detail::Centroids< RealType, IndexType >::appendAndSortCurrent(), quantile::detail::Centroids< RealType, IndexType >::clear(), quantile::detail::Centroids< RealType, IndexType >::currCount(), quantile::detail::Centroids< RealType, IndexType >::mean(), quantile::detail::Centroids< RealType, IndexType >::nextCount(), quantile::detail::operator<<(), quantile::detail::Centroids< RealType, IndexType >::push_back(), and quantile::detail::Centroids< RealType, IndexType >::totalWeight().

template<typename RealType, typename IndexType>

IndexType quantile::detail::Centroids< RealType, IndexType >::curr_idx_

Definition at line 57 of file quantile.h.

Referenced by quantile::detail::Centroids< RealType, IndexType >::appendAndSortCurrent(), quantile::detail::Centroids< RealType, IndexType >::currCount(), quantile::detail::Centroids< RealType, IndexType >::currMean(), quantile::detail::Centroids< RealType, IndexType >::hasCurr(), and quantile::detail::operator<<().

template<typename RealType, typename IndexType>

int quantile::detail::Centroids< RealType, IndexType >::inc_

Definition at line 59 of file quantile.h.

Referenced by quantile::detail::Centroids< RealType, IndexType >::mergeIfFits(), quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::skipFirst(), and quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::skipSubsequent().

template<typename RealType, typename IndexType>

constexpr RealType quantile::detail::Centroids< RealType, IndexType >::infinity = std::numeric_limits<RealType>::infinity()

static

Definition at line 62 of file quantile.h.

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

template<typename RealType, typename IndexType>

RealType quantile::detail::Centroids< RealType, IndexType >::max_ {-infinity}

Definition at line 64 of file quantile.h.

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

template<typename RealType, typename IndexType>

RealType quantile::detail::Centroids< RealType, IndexType >::min_ {infinity}

Definition at line 65 of file quantile.h.

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

template<typename RealType, typename IndexType>

constexpr RealType quantile::detail::Centroids< RealType, IndexType >::nan = std::numeric_limits<RealType>::quiet_NaN()

static

Definition at line 63 of file quantile.h.

template<typename RealType, typename IndexType>

IndexType quantile::detail::Centroids< RealType, IndexType >::next_idx_

Definition at line 58 of file quantile.h.

Referenced by quantile::detail::Centroids< RealType, IndexType >::hasNext(), quantile::detail::Centroids< RealType, IndexType >::mergeIfFits(), quantile::detail::Centroids< RealType, IndexType >::nextCount(), quantile::detail::Centroids< RealType, IndexType >::nextSum(), quantile::detail::operator<<(), quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::skipFirst(), and quantile::detail::anonymous_namespace{quantile.h}::Skipped< RealType, IndexType >::skipSubsequent().

template<typename RealType, typename IndexType>

VectorView<RealType> quantile::detail::Centroids< RealType, IndexType >::sums_

Definition at line 60 of file quantile.h.

Referenced by quantile::detail::Centroids< RealType, IndexType >::appendAndSortCurrent(), quantile::detail::Centroids< RealType, IndexType >::capacity(), quantile::detail::Centroids< RealType, IndexType >::clear(), quantile::detail::Centroids< RealType, IndexType >::mean(), quantile::detail::TDigest< RealType, IndexType >::mergeSorted(), quantile::detail::Centroids< RealType, IndexType >::nextSum(), quantile::detail::operator<<(), quantile::detail::Centroids< RealType, IndexType >::push_back(), and quantile::detail::Centroids< RealType, IndexType >::size().

---

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

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