BmnStsVectorFinder.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 BmnStsVectorFinder header file                -----
// -----  Created 3/12/21  by A.Zinchenko, D.Zinchenko, R.Zinchenko    -----
// -------------------------------------------------------------------------
 
#ifndef BMNSTSVECTORFINDER_H
#define BMNSTSVECTORFINDER_H 1
 
#include "BmnStsKFTrackFitter.h"
#include "CbmStsKFTrackFitter.h"
#include "CbmStsTrack.h"
 
#include "FairTask.h"
 
#include <TClonesArray.h>
#include <TMVA/Reader.h> //AZ-300922
#include <TProfile2D.h>
#include <TVector3.h>
 
#include <set>
#include <string>
#include <unordered_map>
 
//typedef std::pair<Double_t,std::map<Int_t,Int_t> > candvec;
 
class BmnStsVectorFinder : public FairTask
{
 
 public:
 
  struct hitinfo {
    hitinfo() {}
    hitinfo (TVector3 v3, Double_t phxi, Double_t txi, Double_t tyi) : xyz(v3), phx(phxi), tx(txi), ty(tyi), used(0) {;}
    TVector3 xyz;
    Double_t phx;
    Double_t tx;
    Double_t ty;
    int used;
  };
 
  struct candvec {
    Int_t nskips;
    Int_t idmaxP;
    Double_t momxz;
    Double_t ty;
    Double_t lengxz;
    //Double_t varx;
    Double_t tx;
    std::map<Int_t,Int_t> stahit;
    std::string code;
    FairTrackParam param;
    Double_t chi2;
    candvec() {}
    candvec (Double_t momi, std::map<Int_t,Int_t> stahi) : nskips(0), momxz(momi), stahit(stahi) {;}
  };
  
  BmnStsVectorFinder();
 
  ~BmnStsVectorFinder();
 
  virtual InitStatus Init();
 
  virtual void Exec(Option_t* opt);
 
  virtual void FinishEvent();
  
  virtual void Finish();
 
  void SetMatBudgetFileName (TString s) { fMatBudgetFileName = s; }
  void SetNofSiStations(int nsi) { fNSi = nsi; } //AZ-051022
 
 private:
 
  //void ReadMatBudget(); 
  Int_t ExcludeHits(Int_t minHits, TClonesArray *trArray);
  std::set<Int_t> GetHitId(CbmStsHit *hit, Int_t &idmaxP);
  std::set<Int_t> GetHitIdBmn(CbmStsHit *hit, Int_t &idmaxP);
  std::set<Int_t> GetHitId(Int_t hitInd, Int_t &idmaxP) {
    return GetHitId((CbmStsHit*)fHitArray->UncheckedAt(hitInd), idmaxP); }
  void BuildTrackCand();
  void BuildDoublets();
  Bool_t CheckVarx(Double_t dx, Double_t dz, Double_t tx, Double_t distxz, Double_t& varx); //AZ-310722
  void BuildTriplets();
  void BuildTracks();
  void ExtendTrack(candvec cand);
  Double_t FitTrack(candvec &cand);
  Double_t FilterHit(candvec &cand, CbmStsTrack &track);
  void MakeStsTrack(candvec &cand, std::string &hitcode, CbmStsTrack &track);
  void ExtendTracks(Int_t ista);
  TVector3 Parabola(TVector3& pos0, TVector3& pos1, TVector3& pos2);
  void FitTracks();
  void RemoveDoubles();
  Bool_t AreTracksDoubles(CbmStsTrack &tr1, CbmStsTrack &tr2);
  void RemoveFakes();
  void ExcludeFakes();
  Double_t DxVsMom(Int_t ista, candvec &aaa);
  Double_t Proxim(Double_t ang1, Double_t ang2);
  //AZ-280722 Double_t LinearFit(CbmStsTrack *tr, Int_t newtr, Double_t &ty);
  Double_t LinearFit(candvec &cand, candvec &cand2, CbmStsTrack *tr, Int_t newtr, Double_t &ty); //AZ-280722
  Double_t Curv3(candvec &cand1, candvec &cand2, candvec &cand3, int newtr3);
  std::set<int> KalmanWindow(candvec &cand, int hitIndx);
  std::string MakeCode(candvec &cand);
  void PrintHits(candvec &cand);
  void InitTMVA3(); //AZ-300922
  Float_t TMVAOutput3(candvec &aaa1, candvec &aaa2); //AZ-300922
  void InitTMVA2(); //AZ-221022
  Float_t TMVAOutput2(candvec &aaa1); //AZ-241022
   
  TClonesArray *fClusArray[4], *fHitArray, *fTrackArray, *fDigiMatches, *fStsPoints, *fSilPoints;
  TClonesArray *fVectorArray; // output tracks
  Int_t fPass; // pass number
  Int_t fNsta; // number of stations
  Int_t fNbranches; // max number of branches in a triplet from one doublet
  Int_t fExact; // exact match - for debug
  Int_t fExactSel; // exact match for selected ID - for debug
  Int_t discarded; // number of discarded 3-hit candidates with bad chi2, debug
  int fOut; //AZ-210922 - flag for writing out
  TVector3 fXyzv; // mean vertex position
  Double_t fBy; // max. field in Tesla
  std::multimap<Double_t,Int_t> fmapPhx[19], fmapTy[19], fmapX[19], fmapY[19];
  std::vector<candvec> fSeedVec[19]; // 
  std::vector<candvec> fCandVec[19]; // 
  std::set<string> fCandCodes[19];
  //std::multimap<Int_t,candvec> fCandMap2[19]; // 
  //std::multimap<Int_t,candvec> fCandMap3[19]; // 
 
  std::vector<candvec> fCandVec2[19]; // doublets
  std::map<std::string,int> fMap2Out; //AZ-210922 - doublets for writing out
  std::map<std::string,int> *fMap2OutPtr; //AZ-210922 - doublets for writing out
  std::map<std::string,int> fMap3Out; //AZ-260922 - triplets for writing out
  std::map<std::string,int> *fMap3OutPtr; //AZ-260922 - triplets for writing out
  //AZ-130722 std::multimap<int,int> fMap2[19]; // hit index -> index in fCandVec2
  std::unordered_multimap<int,int> fMap2[19]; //AZ-130722 hit index -> index in fCandVec2
 
  std::vector<candvec> fCandVec3[19]; // triplets
  //AZ-130722 std::multimap<int,int> fMap3[19]; // hit index -> index in fCandVec3
  std::unordered_multimap<int,int> fMap3[19]; //AZ-130722 hit index -> index in fCandVec3
  //AZ-130722 std::map<std::string,int> fMapCode3[19]; // triplet code -> index in fCandVec3
  std::unordered_map<std::string,int> fMapCode3[19]; //AZ-130722 triplet code -> index in fCandVec3
 
  std::multimap<Double_t,CbmStsTrack> fTracks; // 
  //AZ-130722 std::map<Int_t,hitinfo> fmapHits;
  std::unordered_map<Int_t,hitinfo> fmapHits; //AZ-130722
  //AZ-130722 std::multimap<Int_t,Int_t> fHit2id, fClusMaps[2];
  std::unordered_multimap<Int_t,Int_t> fHit2id, fClusMaps[2]; //AZ-130722
  Int_t *fNhitsMin, fNskips[20];
   Double_t *fdTanX, *fdTanY, *fdTanY3, *fdTanXB0, *fPhiXZ, *fTanXmax, *fZmean, *fPTcut, *fCurvSta;
  //CbmStsKFTrackFitter fitter;
  BmnStsKFTrackFitter fitter;
  std::set<string> fCandSet[19];
 
  TString fMatBudgetFileName;
  //std::map<Double_t,TProfile2D*> fMatHistos;
  int fUseTMVA; //AZ-300922
  int fNSi; //AZ-051022
  Float_t fVarTMVA[20]; //AZ-300922
  TMVA::Reader *fReaderTMVA3; //AZ-300922
  TMVA::Reader *fReaderTMVA2; //AZ-221022
  
  BmnStsVectorFinder(const BmnStsVectorFinder&);
  BmnStsVectorFinder& operator=(const BmnStsVectorFinder&);
 
  ClassDef(BmnStsVectorFinder,0);
 
};
 
#endif