Vector3D.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_VECTOR3D_H_
#define ALICEO2_TPC_VECTOR3D_H_
 
namespace o2
{
namespace tpc
{
 
 
template <typename DataT = double>
class Vector3D
{
 public:
  Vector3D(const unsigned int nr, const unsigned int nz, const unsigned int nphi) : mNr{nr}, mNz{nz}, mNphi{nphi}, mStorage{nr * nz * nphi} {};
 
  Vector3D() = default;
 
  DataT& operator()(const unsigned int iR, const unsigned int iZ, const unsigned int iPhi)
  {
    return mStorage[getIndex(iR, iZ, iPhi)];
  }
 
  const DataT& operator()(const unsigned int iR, const unsigned int iZ, const unsigned int iPhi) const
  {
    return mStorage[getIndex(iR, iZ, iPhi)];
  }
 
  DataT& operator[](const unsigned int index)
  {
    return mStorage[index];
  }
 
  const DataT& operator[](const unsigned int index) const
  {
    return mStorage[index];
  }
 
  int getIndex(const unsigned int iR, const unsigned int iZ, const unsigned int iPhi) const
  {
    return iR + mNr * (iZ + mNz * iPhi);
  }
 
  void resize(const unsigned int nr, const unsigned int nz, const unsigned int nphi)
  {
    mNr = nr;
    mNz = nz;
    mNphi = nphi;
    mStorage.resize(nr * nz * nphi);
  }
 
  const auto& data() const { return mStorage; }
  auto& data() { return mStorage; }
 
  unsigned int getNr() const { return mNr; }          
  unsigned int getNz() const { return mNz; }          
  unsigned int getNphi() const { return mNphi; }      
  unsigned int size() const { return mStorage.size; } 
 
  auto begin() const { return mStorage.begin(); }
  auto begin() { return mStorage.begin(); }
 
  auto end() const { return mStorage.end(); }
  auto end() { return mStorage.end(); }
 
 private:
  unsigned int mNr{};            
  unsigned int mNz{};            
  unsigned int mNphi{};          
  std::vector<DataT> mStorage{}; 
};
 
} // namespace tpc
} // namespace o2
 
#endif