CbmKF.h

Go to the documentation of this file.

 
#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(); }
 
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 loop_over_nodes(TObjArray* nodes);
 
  TGeoNode* fTarget {nullptr};
 
private:
  CbmKF(const CbmKF&);
  void operator=(const CbmKF&);
 
public:
  ClassDef(CbmKF, 1);
};
 
#endif /* !CBMKF_H */