CbmStsResolution.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 CBMSTSRESOLUTION_H
#define CBMSTSRESOLUTION_H
 
#include "CbmEvent.h"
#include "CbmGlobalTrack.h"
#include "CbmStsAnaBase.h"
#include "CbmStsCluster.h"
#include "CbmStsHit.h"
#include "CbmStsUtils.h"
 
#include <FairTask.h>
 
#include <TClonesArray.h>
 
#include <cstring>
#include <map>
#include <unordered_map>
#include <unordered_set>
 

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

  CbmStsResolution(double);
 
  InitStatus Init();
  void Exec(Option_t*);
  void Finish();
 
  void SetInputType(InputType type) { fInputType = type; };
 
  void EnableRefSensor(int32_t);
 
  void SetMinTrkHitQ(double q) { fTrkHitQmin = q; }
 
  void TrackUseTwoSts() { tracking_use_two_sts_ = true; }
 
 private:
  double fImpactParMax{-1};
  double fTrkHitQmin{-1};
 
  std::vector<CbmGlobalTrack*> fGlbTrks;
  std::map<int32_t, std::vector<CbmStsHit*>> fStsHits;
 
  InputType fInputType = InputType::kEventMode;
 
  bool tracking_use_two_sts_{false};
 
  std::vector<int32_t> fActiveRefSensors;
 
  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 ProcessEvent(CbmEvent*);

  void ProcessHit(CbmStsHit*);
 
  void ProcessGlobalTrack(CbmGlobalTrack* trk);
 
  void BuildResidual();
 
  ClassDef(CbmStsResolution, 1);
};
#endif