Tracks.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_TPC_QC_TRACKS_H
#define AliceO2_TPC_QC_TRACKS_H
 
#include <vector>
#include <string_view>
#include <string>
#include <unordered_map>
#include <memory>
 
// root includes
#include "TH1.h"
#include "TH1F.h"
#include "TH2F.h"
 
namespace o2
{
namespace tpc
{
 
class TrackTPC;
 
namespace qc
{
 
class Tracks
{
 public:
  // default constructor
  Tracks() = default;
 
  bool processTrack(const o2::tpc::TrackTPC& track);
 
  void initializeHistograms();
 
  void resetHistograms();
 
  void processEndOfCycle();
 
  void dumpToFile(std::string_view filename);
 
  // To set the elementary track cuts
  void setTrackCuts(float AbsEta = 1.,
                    int nClusterCut = 60, float dEdxTot = 20, float cutPtForDCAr = 1.5, float samplingFractionDCAr = 0.1, bool turnOffHistosForAsync = false, float cutMaxAbsDCAr = 0.1, bool useCutMaxAbsDCArOnHistos = false)
  {
    mCutAbsEta = AbsEta;
    mCutMinnCls = nClusterCut;
    mCutMindEdxTot = dEdxTot;
    mCutMinPtDCAr = cutPtForDCAr;
    mSamplingFractionDCAr = samplingFractionDCAr;
    mTurnOffHistosForAsync = turnOffHistosForAsync;
    mCutMaxAbsDCAr = cutMaxAbsDCAr;
    mUseCutMaxAbsDCArOnHistos = useCutMaxAbsDCArOnHistos;
  }
 
  std::unordered_map<std::string, std::unique_ptr<TH1>>& getMapHist() { return mMapHist; }
  const std::unordered_map<std::string, std::unique_ptr<TH1>>& getMapHist() const { return mMapHist; }
 
 private:
  float mCutAbsEta = 1.f;                 // Eta cut
  int mCutMinnCls = 60;                   // minimum N clusters
  float mCutMindEdxTot = 20.f;            // dEdxTot min value
  float mCutMinPtDCAr = 1.5f;             // minimum pT for DCAr plots DCAr vs. phi, eta, nCluster
  float mSamplingFractionDCAr = 0.1f;     // sampling rate for calculation of DCAr
  bool mTurnOffHistosForAsync = false;    // Decide whether to turn off some histograms for async to reduce memory
  float mCutMaxAbsDCAr = 1.f;             // maximum DCAr
  bool mUseCutMaxAbsDCArOnHistos = false; // Decide whether to use the cut on maximum DCAr for the histograms
 
  std::unordered_map<std::string, std::unique_ptr<TH1>> mMapHist;
  std::vector<TH1F> mHist1D{};      
  std::vector<TH2F> mHist2D{};      
  std::vector<TH1F> mHistRatio1D{}; 
 
  ClassDefNV(Tracks, 2)
};
} // namespace qc
} // namespace tpc
} // namespace o2
 
#endif