DataContainer3D.h

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.
 
 
#ifndef ALICEO2_TPC_DATACONTAINER3D_H_
#define ALICEO2_TPC_DATACONTAINER3D_H_
 
#include "Rtypes.h"
#include <vector>
 
class TFile;
class TTree;
 
namespace o2
{
namespace tpc
{
 
template <class T>
class RegularGrid3D;
 
 
template <typename DataT = double>
struct DataContainer3D {
 
  DataContainer3D(unsigned short nZ, unsigned short nR, unsigned short nPhi) : mZVertices{nZ}, mRVertices{nR}, mPhiVertices{nPhi}, mData(nZ * nR * nPhi){};
 
  DataContainer3D() = default;
 
  const DataT& operator[](size_t i) const { return mData[i]; }
  DataT& operator[](size_t i) { return mData[i]; }
 
  const auto& getData() const { return mData; }
  auto& getData() { return mData; }
 
  const DataT& operator()(size_t iz, size_t ir, size_t iphi) const { return mData[getDataIndex(iz, ir, iphi)]; }
 
  DataT& operator()(size_t iz, size_t ir, size_t iphi) { return mData[getDataIndex(iz, ir, iphi)]; }
 
  DataT interpolate(const DataT z, const DataT r, const DataT phi, const o2::tpc::RegularGrid3D<DataT>& grid) const;
 
  size_t getDataIndex(const size_t iz, const size_t ir, const size_t iphi) const { return (iz + mZVertices * (ir + iphi * mRVertices)); }
 
  size_t getNDataPoints() const { return mData.size(); }
 
  unsigned short getNZ() const { return mZVertices; }
 
  unsigned short getNR() const { return mRVertices; }
 
  unsigned short getNPhi() const { return mPhiVertices; }
 
  static void setAliases(TTree* tree);
 
  static void setAliasesForDump(TTree* tree);
 
  static size_t getIndexZ(size_t index, const int nz, const int nr, const int nphi);
  size_t getIndexZ(size_t index) const { return getIndexZ(index, getNZ(), getNR(), getNPhi()); }
 
  static size_t getIndexR(size_t index, const int nz, const int nr, const int nphi);
  size_t getIndexR(size_t index) const { return getIndexR(index, getNZ(), getNR(), getNPhi()); }
 
  static size_t getIndexPhi(size_t index, const int nz, const int nr, const int nphi);
  size_t getIndexPhi(size_t index) const { return getIndexPhi(index, getNZ(), getNR(), getNPhi()); }
 
  void setGrid(unsigned short nZ, unsigned short nR, unsigned short nPhi, const bool resize);
 
  template <typename DataTOut = DataT>
  int writeToFile(TFile& outf, const char* name = "data") const;
 
  int writeToFile(std::string_view file, std::string_view option, std::string_view name = "data", const int nthreads = 1) const;
 
  template <typename DataTIn = DataT>
  bool initFromFile(TFile& inpf, const char* name = "data");
 
  bool initFromFile(std::string_view file, std::string_view name = "data", const int nthreads = 1);
 
  inline static DataContainer3D<DataT>* loadFromFile(TFile& inpf, const char* name = "data");
 
  static void dumpSlice(std::string_view treename, std::string_view fileIn, std::string_view fileOut, std::string_view option, std::pair<unsigned short, unsigned short> rangeiR, std::pair<unsigned short, unsigned short> rangeiZ, std::pair<unsigned short, unsigned short> rangeiPhi, const int nthreads = 1);
 
  static void dumpInterpolation(std::string_view treename, std::string_view fileIn, std::string_view fileOut, std::string_view option, std::pair<float, float> rangeR, std::pair<float, float> rangeZ, std::pair<float, float> rangePhi, const int nR, const int nZ, const int nPhi, const int nthreads = 1);
 
  static bool getVertices(std::string_view treename, std::string_view fileIn, unsigned short& nR, unsigned short& nZ, unsigned short& nPhi);
 
  void print() const;
 
  DataContainer3D<DataT>& operator*=(const DataT value);
  DataContainer3D<DataT>& operator+=(const DataContainer3D<DataT>& other);
  DataContainer3D<DataT>& operator*=(const DataContainer3D<DataT>& other);
  DataContainer3D<DataT>& operator-=(const DataContainer3D<DataT>& other);
 
 private:
  unsigned short mZVertices{};   
  unsigned short mRVertices{};   
  unsigned short mPhiVertices{}; 
  std::vector<DataT> mData{};    
 
  static auto getDataSlice(const std::vector<DataT>& data, size_t entries, const size_t values_per_entry, ULong64_t entry);
 
  ClassDefNV(DataContainer3D, 1)
};
 
} // namespace tpc
} // namespace o2
 
#endif