TRDGlobalTrackingSpec.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_GLOBALTRACKING
#define O2_TRD_GLOBALTRACKING
 
#include "Framework/DataProcessorSpec.h"
#include "Framework/Task.h"
#include "TStopwatch.h"
#include "TRDBase/GeometryFlat.h"
#include "GPUO2Interface.h"
#include "GPUTRDTracker.h"
#include "ReconstructionDataFormats/GlobalTrackID.h"
#include "DataFormatsGlobalTracking/RecoContainer.h"
#include "DataFormatsTRD/TrackTRD.h"
#include "DataFormatsTRD/TrackTriggerRecord.h"
#include "DataFormatsTRD/PID.h"
#include "TRDPID/PIDBase.h"
#include "SimulationDataFormat/MCCompLabel.h"
#include "SimulationDataFormat/ConstMCTruthContainer.h"
#include <memory>
#include "DetectorsBase/GRPGeomHelper.h"
#include "TPCCalibration/VDriftHelper.h"
#include "TPCCalibration/CorrectionMapsLoader.h"
#include "GPUO2InterfaceRefit.h"
#include "TPCFastTransform.h"
#include "TRDBase/RecoParam.h"
#include "DataFormatsTPC/TrackTPC.h"
#include "DataFormatsITS/TrackITS.h"
#include "DataFormatsITSMFT/TrkClusRef.h"
#include "DataFormatsITSMFT/TopologyDictionary.h"
 
#ifdef ENABLE_UPGRADES
#include "ITS3Reconstruction/TopologyDictionary.h"
#endif
 
namespace o2
{
namespace trd
{
 
class TRDGlobalTracking : public o2::framework::Task
{
 public:
  TRDGlobalTracking(bool useMC, bool withPID, PIDPolicy policy, std::shared_ptr<o2::globaltracking::DataRequest> dataRequest, std::shared_ptr<o2::base::GRPGeomRequest> gr, const o2::tpc::CorrectionMapsLoaderGloOpts& sclOpts,
                    o2::dataformats::GlobalTrackID::mask_t src, bool trigRecFilterActive, bool strict) : mUseMC(useMC), mWithPID(withPID), mDataRequest(dataRequest), mGGCCDBRequest(gr), mTrkMask(src), mTrigRecFilter(trigRecFilterActive), mStrict(strict), mPolicy(policy)
  {
    mTPCCorrMapsLoader.setLumiScaleType(sclOpts.lumiType);
    mTPCCorrMapsLoader.setLumiScaleMode(sclOpts.lumiMode);
  }
  ~TRDGlobalTracking() override = default;
  void init(o2::framework::InitContext& ic) final;
  void fillMCTruthInfo(const TrackTRD& trk, o2::MCCompLabel lblSeed, std::vector<o2::MCCompLabel>& lblContainerTrd, std::vector<o2::MCCompLabel>& lblContainerMatch, const o2::dataformats::MCTruthContainer<o2::MCCompLabel>* trkltLabels) const;
  void fillTrackTriggerRecord(const std::vector<TrackTRD>& tracks, std::vector<TrackTriggerRecord>& trigRec, const gsl::span<const o2::trd::TriggerRecord>& trackletTrigRec) const;
  void run(o2::framework::ProcessingContext& pc) final;
  void finaliseCCDB(o2::framework::ConcreteDataMatcher& matcher, void* obj) final;
  bool refitITSTPCTRDTrack(TrackTRD& trk, float timeTRD, o2::globaltracking::RecoContainer* recoCont);
  bool refitTPCTRDTrack(TrackTRD& trk, float timeTRD, o2::globaltracking::RecoContainer* recoCont);
  bool refitTRDTrack(TrackTRD& trk, float& chi2, bool inwards, bool tpcSA);
  void endOfStream(o2::framework::EndOfStreamContext& ec) final;
 
 private:
  void updateTimeDependentParams(o2::framework::ProcessingContext& pc);
 
  o2::gpu::GPUTRDTracker* mTracker{nullptr};                     
  o2::gpu::GPUReconstruction* mRec{nullptr};                     
  o2::gpu::GPUChainTracking* mChainTracking{nullptr};            
  std::unique_ptr<GeometryFlat> mFlatGeo{nullptr};               
  bool mUseMC{false};                                            
  bool mWithPID{false};                                          
  float mTPCTBinMUS{.2f};                                        
  float mTPCTBinMUSInv{1.f / mTPCTBinMUS};                       
  float mTPCVdrift{2.58f};                                       
  float mTPCTDriftOffset{0.f};                                   
  std::shared_ptr<o2::globaltracking::DataRequest> mDataRequest; 
  std::shared_ptr<o2::base::GRPGeomRequest> mGGCCDBRequest;
  o2::tpc::VDriftHelper mTPCVDriftHelper{};
  o2::tpc::CorrectionMapsLoader mTPCCorrMapsLoader{};
  o2::dataformats::GlobalTrackID::mask_t mTrkMask; 
  bool mTrigRecFilter{false};                      
  bool mStrict{false};                             
  TStopwatch mTimer;
  // temporary members -> should go into processor (GPUTRDTracker or additional refit processor?)
  std::unique_ptr<o2::gpu::GPUO2InterfaceRefit> mTPCRefitter;         
  const o2::tpc::ClusterNativeAccess* mTPCClusterIdxStruct = nullptr; 
  RecoParam mRecoParam;                                               
  gsl::span<const Tracklet64> mTrackletsRaw;                          
  gsl::span<const CalibratedTracklet> mTrackletsCalib;                
  gsl::span<const o2::tpc::TrackTPC> mTPCTracksArray;                 
  gsl::span<const o2::its::TrackITS> mITSTracksArray;                 
  gsl::span<const o2::itsmft::TrkClusRef> mITSABRefsArray;            
  gsl::span<const int> mITSTrackClusIdx;                              
  gsl::span<const int> mITSABTrackClusIdx;                            
  std::vector<o2::BaseCluster<float>> mITSClustersArray;              
  const o2::itsmft::TopologyDictionary* mITSDict = nullptr;           
#ifdef ENABLE_UPGRADES
  const o2::its3::TopologyDictionary* mIT3Dict = nullptr; 
#endif
  std::array<float, 5> mCovDiagInner{}; 
  std::array<float, 5> mCovDiagOuter{}; 
  // PID
  PIDPolicy mPolicy{PIDPolicy::DEFAULT}; 
  std::unique_ptr<PIDBase> mBase;        
};
 
framework::DataProcessorSpec getTRDGlobalTrackingSpec(bool useMC, o2::dataformats::GlobalTrackID::mask_t src, bool trigRecFilterActive, bool strict /* = false*/, bool withPID /* = false*/, PIDPolicy policy /* = PIDPolicy::DEFAULT*/, const o2::tpc::CorrectionMapsLoaderGloOpts& sclOpts);
 
} // namespace trd
} // namespace o2
 
#endif /* O2_TRD_TRACKLETREADER */