DCA.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_DCA_H
#define ALICEO2_DCA_H
 
#include "GPUCommonDef.h"
#include "GPUCommonRtypes.h"
 
#ifndef GPUCA_GPUCODE_DEVICE
#include <iosfwd>
#include <array>
#endif
 
 
namespace o2
{
namespace dataformats
{
 
class DCA
{
 
 public:
  GPUdDefault() DCA() = default;
 
  GPUd() DCA(float y, float z, float syy = 0.f, float syz = 0.f, float szz = 0.f)
  {
    set(y, z, syy, syz, szz);
  }
 
  GPUd() void set(float y, float z, float syy, float syz, float szz)
  {
    mY = y;
    mZ = z;
    mCov[0] = syy;
    mCov[1] = syz;
    mCov[2] = szz;
  }
 
  GPUd() void set(float y, float z)
  {
    mY = y;
    mZ = z;
  }
 
  GPUd() void setY(float y) { mY = y; }
  GPUd() void setZ(float z) { mZ = z; }
  GPUd() void setSigmaY2(float v) { mCov[0] = v; }
  GPUd() void setSigmaYZ(float v) { mCov[1] = v; }
  GPUd() void setSigmaZ2(float v) { mCov[2] = v; }
  GPUd() void addCov(const std::array<float, 3>& vadd)
  {
    mCov[0] += vadd[0];
    mCov[1] += vadd[1];
    mCov[2] += vadd[2];
  }
  GPUd() auto getY() const { return mY; }
  GPUd() auto getZ() const { return mZ; }
  GPUd() auto getR2() const { return mY * mY + mZ * mZ; }
  GPUd() auto getSigmaY2() const { return mCov[0]; }
  GPUd() auto getSigmaYZ() const { return mCov[1]; }
  GPUd() auto getSigmaZ2() const { return mCov[2]; }
  GPUd() const auto& getCovariance() const { return mCov; }
  GPUd() float calcChi2() const;
 
  void print() const;
 
 private:
  float mY = 0.f;
  float mZ = 0.f;
  std::array<float, 3> mCov; 
 
  ClassDefNV(DCA, 1);
};
 
#ifndef GPUCA_GPUCODE_DEVICE
std::ostream& operator<<(std::ostream& os, const DCA& d);
#endif
 
} // namespace dataformats
} // namespace o2
 
#endif //ALICEO2_DCA_H