PIDBase.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_TRD_PIDBASE_H
#define O2_TRD_PIDBASE_H
 
#include "Rtypes.h"
#include "DataFormatsTRD/PID.h"
#include "DataFormatsTRD/TrackTRD.h"
#include "DataFormatsTRD/Tracklet64.h"
#include "DataFormatsTRD/Constants.h"
#include "TRDBase/PadCalibrationsAliases.h"
#include "TRDPID/PIDParameters.h"
#include "Framework/ProcessingContext.h"
#include "DataFormatsGlobalTracking/RecoContainer.h"
 
#include <gsl/span>
#include <memory>
 
namespace o2
{
namespace trd
{
 
class PIDBase
{
 public:
  virtual ~PIDBase() = default;
  PIDBase(PIDPolicy policy) : mPolicy(policy) {}
 
  virtual void init(o2::framework::ProcessingContext& pc) = 0;
 
  virtual float process(const TrackTRD& trk, const o2::globaltracking::RecoContainer& input, bool isTPCTRD) const = 0;
 
  void setLocalGainFactors(const LocalGainFactor* localGain) { mLocalGain = localGain; }
 
 protected:
  std::array<float, constants::NCHARGES> getCharges(const Tracklet64& tracklet, const int layer, const TrackTRD& trk, const o2::globaltracking::RecoContainer& input, float snp, float tgl) const noexcept;
 
  const TRDPIDParams& mParams{TRDPIDParams::Instance()}; 
  const PIDPolicy mPolicy;                               
 
 private:
  std::array<float, constants::NCHARGES> correctCharges(const Tracklet64& trklt, float snp, float tgl) const noexcept;
 
  // correction factors
  const LocalGainFactor* mLocalGain; 
 
  ClassDef(PIDBase, 1);
};
 
std::unique_ptr<PIDBase> getTRDPIDPolicy(PIDPolicy policy);
 
} // namespace trd
} // namespace o2
 
#endif