CbmStsCorrelation.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 CBMSTSCORRELATION_H
#define CBMSTSCORRELATION_H
 
#include "CbmEvent.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 CbmStsCorrelation : public FairTask, public CbmStsAnaBase {
 public:
  CbmStsCorrelation()  = default;
  ~CbmStsCorrelation() = default;
 
  InitStatus Init();
  void Exec(Option_t*);
  void Finish();
 
 private:
  std::map<int32_t, std::vector<CbmStsHit*>> fStsHits;
 
  TClonesArray* fCbmEvtArray;
  TClonesArray* fStsHitArray;
  TClonesArray* fStsCluArray;
 
  void BookHistograms();

  void ProcessEvent(CbmEvent*);

  void ProcessHit(CbmStsHit*);

  void BuildCorrelation();
 
  ClassDef(CbmStsCorrelation, 1);
};
#endif