TrivialVertexer.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_ITS_TRIVIALVERTEXER_H
#define ALICEO2_ITS_TRIVIALVERTEXER_H
 
#include <array>
 
#include "Rtypes.h" // for TrivialVertexer::Class, Double_t, ClassDef, etc
 
class TFile;
class TTree;
class FairMCEventHeader;
 
namespace o2
{
namespace itsmft
{
class Cluster;
}
} // namespace o2
 
namespace o2
{
class MCCompLabel;
namespace dataformats
{
template <typename T>
class MCTruthContainer;
}
namespace its
{
class TrivialVertexer
{
  using Cluster = o2::itsmft::Cluster;
  using Label = o2::MCCompLabel;
 
 public:
  TrivialVertexer();
  ~TrivialVertexer();
 
  TrivialVertexer(const TrivialVertexer&) = delete;
  TrivialVertexer& operator=(const TrivialVertexer&) = delete;
 
  Bool_t openInputFile(const Char_t*);
 
  void process(const std::vector<Cluster>& clusters, std::vector<std::array<Double_t, 3>>& vertices);
  void setMCTruthContainer(const o2::dataformats::MCTruthContainer<o2::MCCompLabel>* truth) { mClsLabels = truth; }
 
 private:
  const o2::dataformats::MCTruthContainer<o2::MCCompLabel>* mClsLabels = nullptr; // Cluster MC labels
 
  TFile* mFile = nullptr;
  TTree* mTree = nullptr;
  FairMCEventHeader* mHeader = nullptr;
};
} // namespace its
} // namespace o2
 
#endif /* ALICEO2_ITS_TRIVIALVERTEXER_H */