PrimaryVertexExt.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_PRIMARYVERTEX_EXT_H
#define ALICEO2_PRIMARYVERTEX_EXT_H
 
#include "ReconstructionDataFormats/PrimaryVertex.h"
#include "ReconstructionDataFormats/GlobalTrackID.h"
#include "SimulationDataFormat/MCEventLabel.h"
 
namespace o2
{
namespace dataformats
{
 
// extended primary vertex info
 
struct PrimaryVertexExt : public PrimaryVertex {
  using PrimaryVertex::PrimaryVertex;
  std::array<uint16_t, o2::dataformats::GlobalTrackID::Source::NSources> nSrc{};   // N contributors for each source type
  std::array<uint16_t, o2::dataformats::GlobalTrackID::Source::NSources> nSrcA{};  // N associated and passing cuts for each source type
  std::array<uint16_t, o2::dataformats::GlobalTrackID::Source::NSources> nSrcAU{}; // N ambgous associated and passing cuts for each source type
  o2::MCEventLabel mcLb{};
  double FT0Time = -1.;                                                            // time of closest FT0 trigger
  float FT0A = -1;                                                               // amplitude of closest FT0 A side
  float FT0C = -1;                                                               // amplitude of closest FT0 C side
  int VtxID = -1;                                                                // original vtx ID
 
  int getNSrc(int i) const { return nSrc[i]; }
  int getNSrcA(int i) const { return nSrcA[i]; }
  int getNSrcAU(int i) const { return nSrcAU[i]; }
 
#ifndef GPUCA_GPUCODE_DEVICE
  void print() const;
  std::string asString() const;
#endif
 
  ClassDefNV(PrimaryVertexExt, 7);
};
 
#ifndef GPUCA_GPUCODE_DEVICE
std::ostream& operator<<(std::ostream& os, const o2::dataformats::PrimaryVertexExt& v);
#endif
 
} // namespace dataformats
 
namespace framework
{
template <typename T>
struct is_messageable;
template <>
struct is_messageable<o2::dataformats::PrimaryVertexExt> : std::true_type {
};
} // namespace framework
 
} // namespace o2
#endif