BmnStsVectorFinderV9.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                 BmnStsVectorFinder header file                -----
// -----  Created 3/12/21  by A.Zinchenko, D.Zinchenko, R.Zinchenko    -----
// -------------------------------------------------------------------------
 
/*** BmnStsVectorFinder.h
 *@author A.Zinchenko <Alexander.Zinchenko@jinr.ru>
 **
 ** Vector Finder in BM@N tracker
 **/
 
#ifndef BMNSTSVECTORFINDERV9_H
#define BMNSTSVECTORFINDERV9_H 1
 
#include "BmnGemStripStationSet.h"
#include "BmnSiliconStationSet.h"
#include "BmnStsKFTrackFitter.h"
#include "BmnToCbmHitConverter.h"   //   AZ-230626
#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 <fstream>
#include <set>
#include <string>
#include <unordered_map>
 
// typedef std::pair<Double_t,std::map<Int_t,Int_t> > candvec;
 
class BmnStsVectorFinderV9 : 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)
        {
            ;
        }
    };
 
    /** Default constructor **/
    BmnStsVectorFinderV9();
 
    /** Destructor **/
    ~BmnStsVectorFinderV9();
 
    /** Virtual method Init **/
    virtual InitStatus Init();
 
    /** Virtual method Exec **/
    virtual void Exec(Option_t* opt);
 
    /** Virtual method FinishEvent **/
    virtual void FinishEvent();
 
    /** Virtual method Finish **/
    virtual void Finish();
 
    void SetMatBudgetFileName(TString s) { fMatBudgetFileName = s; }
    void SetNofSiStations(int nsi) { fNSi = nsi; }                           // AZ-051022
    int GetNofSiStations() const { return fNSi; }                            // AZ-280725
    void SetOutBranch(TString outBranch) { fOutBranch = outBranch; }         // AZ-041125
    void SetExactSel(int trID) { fExactSel = trID; }                         // AZ-251225
    TClonesArray* GetVectorArray() { return fVectorArray; }                  // AZ-230626
    void SetKillHitsTxP(int ikill = 1) { fKillHitsTxP = ikill; }             // AZ-230626
    void SetTracksToKill(TClonesArray* tracks) { fTracksToKill = tracks; }   // AZ-230626
 
  private:
    // void ReadMatBudget();
    void ReadTxInfo(TString fileName);
    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);
    }
    // AZ-080623 void BuildTrackCand();
    void FillHitInfo();   // AZ-080623
    void BuildDoublets();
    int FindBin(int ista, Float_t binw, Float_t tx);
    Float_t BinCenterTx(int ista, int ibin, Float_t binw);
    void DTxWindow(int ista2, Float_t tx, int idu, Float_t curvCut, Float_t* dtx);
    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, int ix = 0);
    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, int mult2);   // AZ-300922
    void InitTMVA2();                                               // AZ-221022
    Float_t TMVAOutput2(candvec& aaa1);                             // AZ-241022
    void InitTMVAtracks();                                          // AZ-040525
    void TMVAOutputTra();                                           // AZ-040525
    void GoToTarget();                                              // AZ-240625
    void KillHitsTxP();                                             // AZ-230626
 
    /** Input / output array of CbmStsHits **/
    TClonesArray *fClusArray[4], *fHitArray, *fTrackArray, *fDigiMatches, *fStsPoints, *fSilPoints, *fVspPoints;
    TClonesArray* fVectorArray;   // output tracks
    int fEvent;
    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
    // AZ-080623 std::multimap<Double_t,Int_t> fmapPhx[19], fmapTy[19], fmapX[19], fmapY[19];
    std::multimap<Double_t, Int_t> fmapTx[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::unordered_map<std::string, int> fMap2Out;   // AZ-210922 - doublets for writing out
    // AZ-240623 std::map<std::string,int> *fMap2OutPtr; //AZ-210922 - doublets for writing out
    std::unordered_map<std::string, int> fMap3Out;   // AZ-260922 - triplets for writing out
    // AZ-240623 std::map<std::string,int> *fMap3OutPtr; //AZ-260922 - triplets for writing out
    std::ofstream fStream2out;   // 240623
    std::ofstream fStream3out;   // 240623
    // 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
    // AZ-240626 Int_t *fNhitsMin, fNskips[20];
    // AZ-240626 Double_t *fdTanX, *fdTanY, *fdTanY3, *fdTanXB0, *fPhiXZ, *fTanXmax, *fZmean, *fPTcut, *fCurvSta;
    int fNskips[20];
    vector<int> fNhitsMin;
    vector<Double_t> fdTanX, fdTanY, fdTanY3, fdTanXB0, fPhiXZ, fTanXmax, fZmean, fPTcut, fCurvSta;
 
    // CbmStsKFTrackFitter fitter;
    BmnStsKFTrackFitter fitter;
    std::set<string> fCandSet[19];
 
    // AZ-070623 - binned hits in Tx
    std::unordered_multimap<int, int> fTxBins[19];
    std::unordered_map<int, std::unordered_map<int, std::vector<Float_t>>> fStatData;
    Float_t fTxStep;
    Float_t fTxBinw;
 
    TString fMatBudgetFileName;
    // std::map<Double_t,TProfile2D*> fMatHistos;
    int fUseTMVA;           // AZ-300922
    int fTMVAtracks;        // AZ-040525
    int fNSi;               // AZ-051022
    Float_t fVarTMVA[20];   // AZ-300922
    // TMVA::Reader* fReaderTMVA3;   // AZ-300922
    std::map<int, TMVA::Reader*> fReadersTMVA3;   // AZ-080525 - for triplets
    TMVA::Reader* fReaderTMVA2;                   // AZ-221022
    std::map<int, TMVA::Reader*> fTmvaReaders;    // AZ-040525 - for tracks
    // std::ofstream *fOutf; // AZ-290525
 
    TString fGemConfigFile;
    TString fSilConfigFile;
    TString fOutBranch;
    int fKillHitsTxP;                        // AZ-230626
    TClonesArray* fTracksToKill;             // AZ-230626
    BmnToCbmHitConverter* fHitConverter;     // AZ-230626
    BmnGemStripStationSet* fGemStationSet;   //! Entire GEM detector
    BmnSiliconStationSet* fSilStationSet;    //! Entire SILICON detector
 
    BmnStsVectorFinderV9(const BmnStsVectorFinderV9&);
    BmnStsVectorFinderV9& operator=(const BmnStsVectorFinderV9&);
 
    ClassDef(BmnStsVectorFinderV9, 0);
};
 
#endif