GPUTPCGrid.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 GPUTPCGRID_H
#define GPUTPCGRID_H
 
#include "GPUTPCDef.h"
 
namespace o2::gpu
{
class GPUTPCGrid
{
 public:
  GPUd() void CreateEmpty();
  GPUd() void Create(float yMin, float yMax, float zMin, float zMax, int32_t ny, int32_t nz);
 
  GPUd() int32_t GetBin(float Y, float Z) const;
  GPUd() int32_t GetBinBounded(float Y, float Z) const;
  GPUd() void GetBin(float Y, float Z, int32_t* const bY, int32_t* const bZ) const;
  GPUd() void GetBinArea(float Y, float Z, float dy, float dz, int32_t& bin, int32_t& ny, int32_t& nz) const;
 
  GPUd() uint32_t N() const { return mN; }
  GPUd() uint32_t Ny() const { return mNy; }
  GPUd() uint32_t Nz() const { return mNz; }
  GPUd() float YMin() const { return mYMin; }
  GPUd() float YMax() const { return mYMax; }
  GPUd() float ZMin() const { return mZMin; }
  GPUd() float ZMax() const { return mZMax; }
  GPUd() float StepYInv() const { return mStepYInv; }
  GPUd() float StepZInv() const { return mStepZInv; }
 
 private:
  friend class GPUTPCNeighboursFinder;
 
  uint32_t mNy;    //* N bins in Y
  uint32_t mNz;    //* N bins in Z
  uint32_t mN;     //* total N bins
  float mYMin;     //* minimal Y value
  float mYMax;     //* maximal Y value
  float mZMin;     //* minimal Z value
  float mZMax;     //* maximal Z value
  float mStepYInv; //* inverse bin size in Y
  float mStepZInv; //* inverse bin size in Z
};
} // namespace o2::gpu
 
#endif // GPUTPCGRID_H