TrackBasedCalib.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_TRACKBASEDCALIB_H
#define O2_TRD_TRACKBASEDCALIB_H
 
#include "DataFormatsTRD/TrackTRD.h"
#include "DataFormatsTRD/Tracklet64.h"
#include "DataFormatsTRD/CalibratedTracklet.h"
#include "DataFormatsTRD/AngularResidHistos.h"
#include "DataFormatsTRD/GainCalibHistos.h"
#include "DataFormatsTRD/NoiseCalibration.h"
#include "TRDBase/PadCalibrationsAliases.h"
#include "DetectorsBase/Propagator.h"
#include "GPUTRDRecoParam.h"
 
#include "Rtypes.h"
 
#include <gsl/span>
 
namespace o2
{
 
namespace globaltracking
{
class RecoContainer;
}
 
namespace trd
{
 
class TrackBasedCalib
{
  using MatCorrType = o2::base::Propagator::MatCorrType;
 
 public:
  TrackBasedCalib() = default;
  TrackBasedCalib(const TrackBasedCalib&) = delete;
  ~TrackBasedCalib() = default;
 
  void init();
 
  void setInput(const o2::globaltracking::RecoContainer& input);
 
  void setNoiseMapMCM(const NoiseStatusMCM* map) { mNoiseCalib = map; };
 
  // Set the local gain factors with values from the ccdb
  void setLocalGainFactors(const LocalGainFactor* localGain) { mLocalGain = localGain; }
 
  void setApplyShift(bool f) { mApplyShift = f; }
 
  void reset();
 
  void calculateAngResHistos();
 
  int doTrdOnlyTrackFits(gsl::span<const TrackTRD>& tracks);
 
  void calculateGainCalibObjs();
 
  int filldEdx(gsl::span<const TrackTRD>& tracks, bool isTPCTRD);
 
  bool propagateAndUpdate(TrackTRD& trk, int iLayer, bool doUpdate) const;
 
  const AngularResidHistos& getAngResHistos() const { return mAngResHistos; }
  const auto& getGainCalibHistos() const { return mGainCalibHistos; }
 
 private:
  float mMaxSnp{o2::base::Propagator::MAX_SIN_PHI};  
  float mMaxStep{o2::base::Propagator::MAX_STEP};    
  MatCorrType mMatCorr{MatCorrType::USEMatCorrNONE}; 
  o2::gpu::GPUTRDRecoParam mRecoParam;               
  AngularResidHistos mAngResHistos;                  
  std::vector<int> mGainCalibHistos;                 
  float bz;                                          
  bool mApplyShift{true};
 
  // input arrays which should not be modified since they are provided externally
  gsl::span<const TrackTRD> mTracksInITSTPCTRD;        
  gsl::span<const TrackTRD> mTracksInTPCTRD;           
  gsl::span<const Tracklet64> mTrackletsRaw;           
  gsl::span<const CalibratedTracklet> mTrackletsCalib; 
  gsl::span<const o2::tpc::TrackTPC> mTracksTPC;       
  gsl::span<const o2::dataformats::TrackTPCITS> mTracksITSTPC;
 
  // corrections from ccdb, some need to be loaded only once hence an init flag
  const LocalGainFactor* mLocalGain{nullptr}; 
  const NoiseStatusMCM* mNoiseCalib{nullptr}; 
 
  ClassDefNV(TrackBasedCalib, 1);
};
 
} // namespace trd
} // namespace o2
 
#endif // O2_TRD_TRACKBASEDCALIB_H