CbmKF.h

Go to the documentation of this file.

/* Copyright (C) 2006-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Denis Bertini [committer], Florian Uhlig, Sergey Gorbunov */
 
#ifndef CBMKF_H
#define CBMKF_H
 
#include "CbmKFMaterial.h"
#include "FairField.h"
#include "FairTask.h"
#include "TGeoNode.h"
 
#include <iostream>
#include <map>
#include <vector>
 
class CbmKFTrackInterface;
class CbmKFVertexInterface;
class FairGeoNode;
class TObjArray;
 
class CbmKF : public FairTask {
 
 public:
  CbmKF(const char* name = "KF", Int_t iVerbose = 1);
  ~CbmKF();
 
  static CbmKF* Instance() { return fInstance; }
 
 
  void SetParContainers();
  InitStatus ReInit();
  InitStatus Init();
 
 
  Int_t GetMaterialIndex(Int_t uid);
 
  Int_t Propagate(Double_t* T, Double_t* C, Double_t z_out, Double_t QP0);
 
  Int_t PassMaterial(CbmKFTrackInterface& track, Double_t& QP0, Int_t ifst, Int_t ilst);
  Int_t PassMaterialBetween(CbmKFTrackInterface& track, Double_t& QP0, Int_t ifst, Int_t ilst);
  Int_t PassMaterialBetween(CbmKFTrackInterface& track, Double_t& QP0, CbmKFHit* fst, CbmKFHit* lst);
 
 
  std::vector<CbmKFMaterial*> vMaterial;
 
  std::vector<CbmKFTube> vMvdMaterial;
  std::vector<CbmKFTube> vStsMaterial;
  std::vector<CbmKFTube> vMuchMaterial;
  std::vector<CbmKFTube> vMuchDetectors;
  std::vector<CbmKFTube> vRichMaterial;
  std::vector<CbmKFWall> vTrdMaterial;
  std::vector<CbmKFTube> vTargets;
  std::vector<CbmKFCone> vPipe;
 
  std::vector<CbmKFTube> vPassiveTube;
  std::vector<CbmKFWall> vPassiveWall;
  std::vector<CbmKFBox> vPassiveBox;
 
 
  FairField* GetMagneticField() { return fMagneticField; }
 
  std::map<Int_t, Int_t> MvdStationIDMap;
  std::map<Int_t, Int_t> StsStationIDMap;
  std::map<Int_t, Int_t> TrdStationIDMap;
  std::map<Int_t, Int_t> MuchMCID2StationMap;
  std::map<Int_t, Int_t> MuchStation2MCIDMap;
 
  Int_t GetMethod() { return fMethod; }
 
  int GetNMvdStations() const { return CbmKF::Instance()->vMvdMaterial.size(); }
 
  const std::array<float, 3> GetTargetPosition();
 
 private:
  static CbmKF* fInstance;
 
  FairField* fMagneticField;
 
 
  Int_t fMethod; /* 0 = straight line,
                    1 = AnalyticLight
                    2 = Runge-Kutta 4 order, 
                 */
 
  std::map<Int_t, Int_t> fMaterialID2IndexMap;
 
  Int_t ReadTube(CbmKFTube& tube, FairGeoNode* node);
  CbmKFMaterial* ReadPassive(FairGeoNode* node);
  void GetTargetInfo();
 
 
  void FindTargetNode(TString& targetPath, TGeoNode*& targetNode);
 
  std::array<float, 3> fTargetXYZ;
 
 
 private:
  CbmKF(const CbmKF&);
  void operator=(const CbmKF&);
 
 public:
  ClassDef(CbmKF, 1);
};
 
#endif /* !CBMKF_H */