FlatHisto2D.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include "CommonDataFormat/FlatHisto2D.h"
#include <TH1F.h>
#include <TH2F.h>
 
namespace o2
{
namespace dataformats
{
 
using namespace o2::dataformats;
 
template <typename T>
FlatHisto2D<T>::FlatHisto2D(uint32_t nbx, T xmin, T xmax, uint32_t nby, T ymin, T ymax)
{
  init(nbx, xmin, xmax, nby, ymin, ymax);
}
 
template <typename T>
FlatHisto2D<T>::FlatHisto2D(const FlatHisto2D& src)
{
  mContainer = src.mContainer;
  init(gsl::span<const T>(mContainer.data(), mContainer.size()));
}
 
template <typename T>
FlatHisto2D<T>& FlatHisto2D<T>::operator=(const FlatHisto2D<T>& rhs)
{
  if (this == &rhs) {
    return *this;
  }
  if (!rhs.canFill() && rhs.getNBins()) { // was initialized
    throw std::runtime_error("trying to copy read-only histogram");
  }
  if (rhs.getNBins()) {
    mContainer = rhs.mContainer;
    init(gsl::span<const T>(mContainer.data(), mContainer.size()));
  }
  return *this;
}
 
template <typename T>
void FlatHisto2D<T>::adoptExternal(const gsl::span<const T> ext)
{
  assert(ext.size() > NServiceSlots);
  mContainer.clear();
  mContainerView = ext;
  init(mContainerView);
}
 
template <typename T>
void FlatHisto2D<T>::add(const FlatHisto2D& other)
{
  if (!(getNBinsX() == other.getNBinsX() && getXMin() == other.getXMin() && getXMax() == other.getXMax() &&
        getNBinsY() == other.getNBinsY() && getYMin() == other.getYMin() && getYMax() == other.getYMax() &&
        canFill())) {
    throw std::runtime_error("adding incompatible histos or destination histo is const");
  }
  for (uint32_t i = getNBins(); i--;) {
    mDataPtr[i] += other.mDataPtr[i];
  }
}
 
template <typename T>
void FlatHisto2D<T>::subtract(const FlatHisto2D& other)
{
  if (!(getNBinsX() == other.getNBinsX() && getXMin() == other.getXMin() && getXMax() == other.getXMax() &&
        getNBinsY() == other.getNBinsY() && getYMin() == other.getYMin() && getYMax() == other.getYMax() &&
        canFill())) {
    throw std::runtime_error("subtracting incompatible histos or destination histo is const");
  }
  for (uint32_t i = getNBins(); i--;) {
    mDataPtr[i] -= other.mDataPtr[i];
  }
}
 
template <typename T>
T FlatHisto2D<T>::getSum() const
{
  T sum = 0;
  for (uint32_t i = getNBins(); i--;) {
    sum += getBinContent(i);
  }
  return sum;
}
 
template <typename T>
void FlatHisto2D<T>::init(const gsl::span<const T> ext)
{ // when reading from file, need to call this method to make it operational
  assert(ext.size() > NServiceSlots);
  mContainerView = ext;
  mDataPtr = const_cast<T*>(&ext[NServiceSlots]);
  mNBinsX = (uint32_t)ext[NBinsX];
  mNBinsY = (uint32_t)ext[NBinsY];
  mXMin = ext[XMin];
  mXMax = ext[XMax];
  mYMin = ext[YMin];
  mYMax = ext[YMax];
  mBinSizeX = ext[BinSizeX];
  mBinSizeY = ext[BinSizeY];
  mBinSizeXInv = 1. / mBinSizeX;
  mBinSizeYInv = 1. / mBinSizeY;
}
 
template <typename T>
void FlatHisto2D<T>::init(uint32_t nbx, T xmin, T xmax, uint32_t nby, T ymin, T ymax)
{ // when reading from file, need to call this method to make it operational
  assert(nbx > 0 && xmin < xmax);
  assert(nby > 0 && ymin < ymax);
  mContainer.resize(nbx * nby + NServiceSlots, 0.);
  mContainer[NBinsX] = nbx;
  mContainer[NBinsY] = nby;
  mContainer[XMin] = xmin;
  mContainer[XMax] = xmax;
  mContainer[YMin] = ymin;
  mContainer[YMax] = ymax;
  mContainer[BinSizeX] = (xmax - xmin) / nbx;
  mContainer[BinSizeY] = (ymax - ymin) / nby;
  init(gsl::span<const T>(mContainer.data(), mContainer.size()));
}
 
template <typename T>
std::unique_ptr<TH2F> FlatHisto2D<T>::createTH2F(const std::string& name) const
{
  auto h = std::make_unique<TH2F>(name.c_str(), name.c_str(), getNBinsX(), getXMin(), getXMax(), getNBinsY(), getYMin(), getYMax());
  for (uint32_t i = getNBinsX(); i--;) {
    for (uint32_t j = getNBinsY(); j--;) {
      auto w = getBinContent(i, j);
      if (w) {
        h->SetBinContent(i + 1, j + 1, w);
      }
    }
  }
  return std::move(h);
}
 
template <typename T>
std::unique_ptr<TH1F> FlatHisto2D<T>::createSliceYTH1F(uint32_t binX, const std::string& name) const
{
  auto h = std::make_unique<TH1F>(name.c_str(), name.c_str(), getNBinsY(), getYMin(), getYMax());
  if (binX < getNBinsX()) {
    for (uint32_t i = getNBinsY(); i--;) {
      h->SetBinContent(i + 1, getBinContent(binX, i));
    }
  }
  return h;
}
 
template <typename T>
std::unique_ptr<TH1F> FlatHisto2D<T>::createSliceXTH1F(uint32_t binY, const std::string& name) const
{
  auto h = std::make_unique<TH1F>(name.c_str(), name.c_str(), getNBinsX(), getXMin(), getXMax());
  if (binY < getNBinsY()) {
    for (uint32_t i = getNBinsX(); i--;) {
      h->SetBinContent(i + 1, getBinContent(i, binY));
    }
  }
  return h;
}
 
template class FlatHisto2D<float>;
template class FlatHisto2D<double>;
 
} // namespace dataformats
} // namespace o2