CbmStsEfficiency.h

Go to the documentation of this file.

/* Copyright (C) 2023 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Dario Ramirez [committer] */
 
#ifndef CBMSTSEFFICIENCY_H
#define CBMSTSEFFICIENCY_H
 
#include "CbmDcaVertexFinder.h"
#include "CbmEvent.h"
#include "CbmGlobalTrack.h"
#include "CbmStsAnaBase.h"
#include "CbmStsCluster.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
#include "CbmStsUtils.h"
#include "CbmTrack.h"
 
#include <FairTask.h>
 
#include <TClonesArray.h>
#include <TVector3.h>
 
#include <cstring>
#include <map>
#include <unordered_map>
#include <unordered_set>
 
struct Residual {
  TVector3 mean;
  TVector3 sigma;
  Residual() {}
  Residual(TVector3 m, TVector3 s)
  {
    mean  = m;
    sigma = s;
  }
};

class CbmStsEfficiency : public FairTask, public CbmStsAnaBase {
 public:
  CbmStsEfficiency()  = default;
  ~CbmStsEfficiency() = default;

  CbmStsEfficiency(uint32_t);

  CbmStsEfficiency(uint32_t, double, double, double);
 
  InitStatus Init();
  void Exec(Option_t*);
  void FinishTask();
 
  void SetSensorResidual(int32_t, TVector3, TVector3);
  void SetResidual(std::string);
 
  void SetVertexFinder(CbmDcaVertexFinder*);
 
  bool fDrawOpt{true};
 
 private:
  uint32_t fTestLayer{0};
  std::vector<int32_t> fDUTList;
  double fMaxDisTrgVtx{0.25};
  double fMaxDCATrkVtx{0.25};
  double fDefaultResidual{0.12};
  std::map<int32_t, Residual> fResiduals;
 
 
  CbmDcaVertexFinder* fVertexFinder;
 
  std::vector<CbmGlobalTrack*> fGlbTrks;
  std::map<int32_t, std::vector<CbmStsHit*>> fStsHits;
 
  TClonesArray* fCbmEvtArray{nullptr};
  TClonesArray* fGlbTrkArray{nullptr};
  TClonesArray* fStsTrkArray{nullptr};
  TClonesArray* fRchTrkArray{nullptr};
  TClonesArray* fMchTrkArray{nullptr};
  TClonesArray* fTrdTrkArray{nullptr};
  TClonesArray* fTofTrkArray{nullptr};
  TClonesArray* fStsHitArray{nullptr};
  TClonesArray* fStsCluArray{nullptr};
 
  void BookHistograms();
 
  void Efficiency(uint32_t);

  void ProcessEvent(CbmEvent*);

  void ProcessGlobalTrack(CbmGlobalTrack*);

  void ProcessHit(CbmStsHit*);
 
  void CheckEfficiency();
 
  ClassDef(CbmStsEfficiency, 1);
};
#endif