CbmStsRecoBeamSpot.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 CBMSTSRECOBEAMSPOT_H
#define CBMSTSRECOBEAMSPOT_H
 
#include "CbmEvent.h"
#include "CbmGlobalTrack.h"
#include "CbmStsAnaBase.h"
#include "CbmStsCluster.h"
#include "CbmStsHit.h"
#include "CbmStsTrack.h"
#include "CbmStsUtils.h"
#include "CbmTarget.h"
#include "CbmTrack.h"
 
#include <FairTask.h>
 
#include <TClonesArray.h>
#include <TVector3.h>
 
#include <cstring>
#include <map>
#include <unordered_map>
#include <unordered_set>

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

 
  void AddTarget(CbmTarget* target = nullptr);

  void AddTarget(std::string trg_name = "", CbmTarget* target = nullptr);
 
  void Exec(Option_t*);
  InitStatus Init();
  void FinishTask();
 
 private:
  /* Beam Spot sampling range
   * It is dynamically caculate when multiple targets planes are provided
   */
  double fSampleRangeXmin{-10};  // cm
  double fSampleRangeXmax{+10};  // cm
  double fSampleRangeYmin{-10};  // cm
  double fSampleRangeYmax{+10};  // cm
  double fSampleRangeZmin{-10};  // cm
  double fSampleRangeZmax{+10};  // cm
 
  /* Bin width for Beam Spot 2D histograms */
  double fSampleBinSizeX{0.01};
  double fSampleBinSizeY{0.01};
  double fSampleBinSizeZ{0.01};
 
  std::vector<CbmStsTrack*> fStsTrks;
  std::map<int32_t, std::vector<CbmStsHit*>> fStsHits;
 
  std::map<std::string, CbmTarget*> fTargets;
 
 
  TClonesArray* fCbmEvtArray{nullptr};
  TClonesArray* fStsTrkArray{nullptr};
  TClonesArray* fStsHitArray{nullptr};
  TClonesArray* fStsCluArray{nullptr};

  void BeamSpotReco();
 
  void BookHistograms();

  TVector3 ExtrapolateTrackTo(CbmPixelHit*, CbmPixelHit*, CbmTarget*);

  void ProcessEvent(CbmEvent*);

  void ProcessStsTrack(CbmGlobalTrack*);

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