lengthpartitioner.cpp

来自「dysii is a C++ library for distributed p」· C++ 代码 · 共 41 行

#include "LengthPartitioner.hpp"

using namespace indii::ml::aux;

LengthPartitioner::~LengthPartitioner() {
  //
}

bool LengthPartitioner::init(DiracMixturePdf* p,
    const std::vector<unsigned int>& is) {
  /* pre-condition */
  assert (is.size() >= 2);

  unsigned int i, j;
  vector lower(p->get(is[0]));
  vector upper(p->get(is[0]));
  vector length(p->getDimensions());
  
  /* calculate bounds */
  for (i = 1; i < is.size(); i++) {
    for (j = 0; j < p->getDimensions(); j++) {
      vector& x = p->get(is[i]);
      if (x(j) < lower(j)) {
        lower(j) = x(j);
      } else if (x(j) > upper(j)) {
        upper(j) = x(j);
      }
    }
  }

  /* select longest dimension */
  noalias(length) = upper - lower;
  this->index = index_norm_inf(length);   

  /* split on midpoint of selected dimension */
  this->value = (lower(index) + upper(index)) / 2.0;
  
  return length(this->index) > 0.0;
}