median_avg.hpp

/* median_avg.hpp */
/* $Id$ */
#ifndef __median_avg_hpp
#define __median_avg_hpp

#include <set>
//#include <stdio.h>


template <typename T>
T medmean_avg(const std::multiset<T> &values, double reject)
{
  // nothing to do with empty data
  if (values.size() == 0)
    return 0;
  
  // fix reject value to be [0,1]
  if (reject < 0.)
    reject = 0.;
  else if (reject > 1.)
    reject = 1.;
  
  // calc median average value
  size_t id = 0;
  size_t N = 0;
  size_t first = (size_t)(0.5*reject*values.size());
  size_t last = values.size() - first;
  T avg_val = 0;
  //printf("medmean_avg: first=%zu, last=%zu (%zu/%zu)\n", 
  // first, last, last - first, values.size());
  for (typename std::multiset<T>::iterator it = values.begin(); 
       it != values.end(); 
       ++it,++id)
    {
      if (id < first || 
      id >= last)
    continue;
      
      //printf ("medmean_avg: val[%zu]=%g\n", id, *it);
      avg_val += (*it);
      N++;
    }
  
  return (N>0)?(avg_val/N):0;
}

#endif /* median_avg.hpp */