#ifndef _OTB_TRISKELE_AVERAGE_ATTRIBUTES_TPP
#define _OTB_TRISKELE_AVERAGE_ATTRIBUTES_TPP

using namespace boost::chrono;

template<typename PixelT>
inline
AverageAttributes::AverageAttributes (const Tree &tree, const Raster<PixelT> &raster, const AreaAttributes &areaAttributes)
  : CompAttribute<double> (tree) {
  compute (raster, areaAttributes);
}

inline
AverageAttributes::~AverageAttributes () {
}

template<typename PixelT>
inline void
AverageAttributes::compute (const Raster<PixelT> &raster, const AreaAttributes &areaAttributes) {
  auto start = high_resolution_clock::now ();
  const PixelT *pixels = raster.getPixels ();
  const DimImg *areas = areaAttributes.getValues ();
  computeSameCompLevel ([this, &pixels, &areas] (const DimImg &parentId) {
      double tmpAverage = 0.;
      tree.forEachChildTI (parentId, [this, &tmpAverage, &pixels, &areas] (const bool &isLeaf, const DimImg &childId) {
	  tmpAverage += isLeaf ? pixels[childId] : (CompAttribute<double>::values [childId] * areas[childId]);
	});
      CompAttribute<double>::values [parentId] = tmpAverage/areas[parentId];
    });
  globalTreeStats.addTime (averageStats, duration_cast<duration<double> > (high_resolution_clock::now ()-start).count ());
}

#endif // _OTB_TRISKELE_AVERAGE_ATTRIBUTES_TPP