GPUTRDGeometry.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 GPUTRDGEOMETRY_H
#define GPUTRDGEOMETRY_H
 
#include "GPUCommonDef.h"
 
class TObjArray;
#include "GPUDef.h"
#include "TRDBase/GeometryFlat.h"
#include "TRDBase/PadPlane.h"
#include "DataFormatsTRD/Constants.h"
#include "GPUCommonTransform3D.h"
 
namespace o2::gpu
{
 
class GPUTRDpadPlane : private o2::trd::PadPlane
{
 public:
  GPUd() float GetTiltingAngle() const { return getTiltingAngle(); }
  GPUd() float GetRowSize(int32_t row) const { return getRowSize(row); }
  GPUd() float GetColSize(int32_t col) const { return getColSize(col); }
  GPUd() float GetRow0() const { return getRow0(); }
  GPUd() float GetCol0() const { return getCol0(); }
  GPUd() float GetRowEnd() const { return getRowEnd(); }
  GPUd() float GetColEnd() const { return getColEnd(); }
  GPUd() float GetRowPos(int32_t row) const { return getRowPos(row); }
  GPUd() float GetColPos(int32_t col) const { return getColPos(col); }
  GPUd() float GetNrows() const { return getNrows(); }
  GPUd() float GetNcols() const { return getNcols(); }
  GPUd() int32_t GetPadRowNumber(double z) const { return getPadRowNumber(z); }
};
 
class GPUTRDGeometry : private o2::trd::GeometryFlat
{
 public:
  GPUd() static bool CheckGeometryAvailable() { return true; }
 
  // Make sub-functionality available directly in GPUTRDGeometry
  GPUd() float GetPadPlaneWidthIPad(int32_t det) const { return getPadPlane(det)->getWidthIPad(); }
  GPUd() float GetPadPlaneRowPos(int32_t layer, int32_t stack, int32_t row) const { return getPadPlane(layer, stack)->getRowPos(row); }
  GPUd() float GetPadPlaneRowSize(int32_t layer, int32_t stack, int32_t row) const { return getPadPlane(layer, stack)->getRowSize(row); }
  GPUd() int32_t GetGeomManagerVolUID(int32_t det, int32_t modId) const { return 0; }
 
  // Base functionality of Geometry
  GPUd() float GetTime0(int32_t layer) const { return getTime0(layer); }
  GPUd() float GetCol0(int32_t layer) const { return getCol0(layer); }
  GPUd() float GetCdrHght() const { return cdrHght(); }
  GPUd() int32_t GetLayer(int32_t det) const { return getLayer(det); }
  GPUd() bool CreateClusterMatrixArray() const { return false; }
  GPUd() float AnodePos() const { return anodePos(); }
  GPUd() const Transform3D* GetClusterMatrix(int32_t det) const { return getMatrixT2L(det); }
  GPUd() int32_t GetDetector(int32_t layer, int32_t stack, int32_t sector) const { return getDetector(layer, stack, sector); }
  GPUd() const GPUTRDpadPlane* GetPadPlane(int32_t layer, int32_t stack) const { return (GPUTRDpadPlane*)getPadPlane(layer, stack); }
  GPUd() const GPUTRDpadPlane* GetPadPlane(int32_t detector) const { return (GPUTRDpadPlane*)getPadPlane(detector); }
  GPUd() int32_t GetSector(int32_t det) const { return getSector(det); }
  GPUd() int32_t GetStack(int32_t det) const { return getStack(det); }
  GPUd() int32_t GetStack(float z, int32_t layer) const { return getStack(z, layer); }
  GPUd() float GetAlpha() const { return getAlpha(); }
  GPUd() bool IsHole(int32_t la, int32_t st, int32_t se) const { return isHole(la, st, se); }
  GPUd() int32_t GetRowMax(int32_t layer, int32_t stack, int32_t sector) const { return getRowMax(layer, stack, sector); }
  GPUd() bool ChamberInGeometry(int32_t det) const { return chamberInGeometry(det); }
 
  static constexpr int32_t kNstack = o2::trd::constants::NSTACK;
};
} // namespace o2::gpu
 
#endif // GPUTRDGEOMETRY_H