roclib/src_nxyter/DistFunc.cxx (r4864/r619)

Go to the documentation of this file.

//============================================================================
//============================================================================

#include <cmath>
#include <algorithm>
#include <iostream>

#include "nxyter/DistFunc.h"

//----------------------------------------------------------------------------

nxyter::DistFunc::DistFunc(int cap) :
  fMaxEntries(0),
  fFull(false),
  fSorted(false)
{
  fData.reserve(cap);
}

//----------------------------------------------------------------------------

nxyter::DistFunc::~DistFunc()
{
}

//----------------------------------------------------------------------------

void nxyter::DistFunc::setMaxEntries(int max)
{
  fMaxEntries = max;
  fFull = false;
  if (max > 0) fFull = fData.size() > max;
}

//----------------------------------------------------------------------------

void nxyter::DistFunc::addEntry(float val)
{
  if (fMaxEntries>0 && fData.size() >= fMaxEntries) fFull = true;
  if (fFull) return;
  fData.push_back(val);
  fSorted = false;
}

//----------------------------------------------------------------------------

float nxyter::DistFunc::operator()(float prob)
{
  if (fData.size() == 0) return 0.;
  
  sort();

  if (prob < 0.) prob = 0.;
  if (prob > 1.) prob = 1.;
  
  float ind_f = float(fData.size()-1) * prob;
  int   ind_i = int(std::floor(ind_f));
  float w_i1 = ind_f - ind_i;
  float w_i0 = 1. - w_i1;

  float res;
  if (ind_i == fData.size()-1 || ind_f == 0.) {
    res = fData[ind_i];
  } else {
    res = w_i0 * fData[ind_i] + w_i1 * fData[ind_i+1];
  }

#ifdef never
  printf("+++1");
  for (int i=0; i<fData.size(); i++) printf(" %6.0f", fData[i]);
  printf("\n");

  printf("+++2 p=%6.3f n=%5d i_f=%6.1f i_i=%5d d[i]=%4.0f d[+1]=%4.0f res=%6.1f\n",
         prob, int(fData.size()), ind_f, ind_i, fData[ind_i],
         (ind_i<fData.size()-1) ? fData[ind_i+1] : 0, res);
#endif

  return res;

}

//----------------------------------------------------------------------------

float nxyter::DistFunc::operator[](int ind)
{
  sort();
  return fData.at(ind);
}

//----------------------------------------------------------------------------

float nxyter::DistFunc::getMedian()
{
  return (*this)(0.5);
}

//----------------------------------------------------------------------------

float nxyter::DistFunc::getWidth(float cut)
{
  return (*this)(0.5+cut) - (*this)(0.5-cut);
}

//----------------------------------------------------------------------------

float nxyter::DistFunc::getMin()
{
  if (fData.size() == 0) return 0.;
  return (*this)[0];
}

//----------------------------------------------------------------------------

float nxyter::DistFunc::getMax()
{
  if (fData.size() == 0) return 0.;
  return (*this)[fData.size()-1];
}

//----------------------------------------------------------------------------
  
void nxyter::DistFunc::sort()
{
  if (!fSorted) {
    std::sort(fData.begin(), fData.end());
    fSorted = true;
  }
}