TrackLabeler.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 O2_MID_TRACKLABELER_H
#define O2_MID_TRACKLABELER_H
 
#include <vector>
#include <gsl/gsl>
#include "SimulationDataFormat/MCTruthContainer.h"
#include "SimulationDataFormat/MCCompLabel.h"
#include "DataFormatsMID/Track.h"
#include "DataFormatsMID/Cluster.h"
#include "DataFormatsMID/MCClusterLabel.h"
 
namespace o2
{
namespace mid
{
class TrackLabeler
{
 public:
  void process(gsl::span<const Cluster> clusters, gsl::span<const Track> tracks, const o2::dataformats::MCTruthContainer<MCClusterLabel>& inMCContainer);
 
  const std::vector<MCCompLabel>& getTracksLabels() { return mMCTracksLabels; }
 
  const o2::dataformats::MCTruthContainer<MCClusterLabel>& getTrackClustersLabels() { return mMCTrackClustersLabels; }
 
 private:
  bool areBothSidesFired(const gsl::span<const MCClusterLabel>& labels) const;
  MCCompLabel makeTrackLabel(const Track& track, const o2::dataformats::MCTruthContainer<MCClusterLabel>& inMCContainer) const;
 
  std::vector<MCCompLabel> mMCTracksLabels;                                 
  o2::dataformats::MCTruthContainer<MCClusterLabel> mMCTrackClustersLabels; 
};
} // namespace mid
} // namespace o2
 
#endif /* O2_MID_TRACKLABELER_H */