PID.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_PID_H
#define AliceO2_TPC_QC_PID_H
 
#include <vector>
#include <unordered_map>
#include <memory>
#include <string_view>
 
// root includes
#include "TH1.h"
// Decapricated to be removed in next PR
#include "TCanvas.h"
 
// o2 includes
#include "DataFormatsTPC/Defs.h"
 
namespace o2
{
namespace tpc
{
 
class TrackTPC;
 
namespace qc
{
 
class PID
{
 public:
  PID() = default;
 
  bool processTrack(const o2::tpc::TrackTPC& track, size_t nTracks);
 
  // dummy version to make it compatible with old QC version
  bool processTrack(const o2::tpc::TrackTPC& track);
 
  void initializeHistograms();
 
  void resetHistograms();
 
  void dumpToFile(std::string filename);
 
  // To set the elementary track cuts
  void setPIDCuts(int minnCls = 60, float absTgl = 1., float mindEdxTot = 10.0,
                  float maxdEdxTot = 70., float minpTPC = 0.05, float maxpTPC = 20., float minpTPCMIPs = 0.45, float maxpTPCMIPs = 0.55, bool turnOffHistosForAsync = false, bool getdEdxVspHypoHist = false)
  {
    mCutMinnCls = minnCls;
    mCutAbsTgl = absTgl;
    mCutMindEdxTot = mindEdxTot;
    mCutMaxdEdxTot = maxdEdxTot;
    mCutMinpTPC = minpTPC;
    mCutMaxpTPC = maxpTPC;
    mCutMinpTPCMIPs = minpTPCMIPs;
    mCutMaxpTPCMIPs = maxpTPCMIPs;
    mTurnOffHistosForAsync = turnOffHistosForAsync;
    mGetdEdxVspHypoHist = getdEdxVspHypoHist;
  }
 
  // Decapricated to be removed in next PR
  void setCreateCanvas(int createCanvas = 1)
  {
    mCreateCanvas = createCanvas;
  }
 
  std::unordered_map<std::string_view, std::vector<std::unique_ptr<TH1>>>& getMapOfHisto() { return mMapHist; }
  const std::unordered_map<std::string_view, std::vector<std::unique_ptr<TH1>>>& getMapOfHisto() const { return mMapHist; }
 
  // Decapricated to be removed in next PR
  std::unordered_map<std::string_view, std::vector<std::unique_ptr<TCanvas>>>& getMapOfCanvas() { return mMapCanvas; }
  TCanvas* getSeparationPowerCanvas() { return mSeparationPowerCanvas.get(); }
  const std::unordered_map<std::string_view, std::vector<std::unique_ptr<TCanvas>>>& getMapOfCanvas() const { return mMapCanvas; }
 
 private:
  int mCutMinnCls = 60;                // minimum N clusters
  float mCutAbsTgl = 1.f;              // AbsTgl max cut
  float mCutMindEdxTot = 10.f;         // dEdxTot min value
  float mCutMaxdEdxTot = 70.f;         // dEdxTot max value
  float mCutMinpTPC = 0.05f;           // pTPC min value
  float mCutMaxpTPC = 20.f;            // pTPC max value
  float mCutMinpTPCMIPs = 0.45f;       // pTPC min value for MIPs
  float mCutMaxpTPCMIPs = 0.55f;       // pTPC max value for MIPs
  bool mTurnOffHistosForAsync = false; // Decide whether to turn off some histograms for async to reduce memory
  bool mGetdEdxVspHypoHist = false;    // Decide whether to generate the EdxVspHypo histograms
  // Decapricated to be removed in next PR
  bool mCreateCanvas = true; // Decide whether to create the TCanvas Object as it cannot be merged
 
  std::unordered_map<std::string_view, std::vector<std::unique_ptr<TH1>>> mMapHist;
  // Decapricated to be removed in next PR
  // Map for Canvases to be published
  std::unordered_map<std::string_view, std::vector<std::unique_ptr<TCanvas>>> mMapCanvas;
  // Map for Histograms which will be put onto the canvases, and not published separately
  std::unordered_map<std::string_view, std::vector<std::unique_ptr<TH1>>> mMapHistCanvas;
  std::unique_ptr<TCanvas> mSeparationPowerCanvas;
 
  ClassDefNV(PID, 1)
};
} // namespace qc
} // namespace tpc
} // namespace o2
 
#endif