PndBarrelTrackFinder.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

//* $Id: */

// -------------------------------------------------------------------------
// -----                    PndBarrelTrackFinder header file                 -----
// -----                  Created 05/12/2010 by R. Karabowicz          -----
// -------------------------------------------------------------------------

#ifndef PNDBARRELTRACKFINDER_H
#define PNDBARRELTRACKFINDER_H 1

#include "PndPersistencyTask.h"
#include "FairHit.h"
#include "FairTSBufferFunctional.h"

#include "TStopwatch.h"
#include "PndGeoSttPar.h"

#include <list>
#include <map>
#include <vector>

class TEllipse;
class TCanvas;
class TClonesArray;
class TLatex;
class PndGemDigiPar;
class TH1F;

struct TrackParameter {
  Double_t x;
  Double_t y;
  Double_t r;
  Double_t z_phi;
  Int_t n;
  // x,y,r parameters of the circle,
  //  //  p=atan(x/y),
  // z_phi - z increase divided by phi angle increase,
  //  //  p_1=atan(x_1/y_1) of the first hit,
  // n='goodness' of parameter
};

struct TrackBasis {
  std::vector<TrackParameter> trackPars;
  std::vector<FairHit *> trackHits;
  std::vector<Int_t> trackHitD;
  std::vector<Int_t> trackHitN;
  Double_t meanX;
  Double_t meanY;
  Double_t meanR;
  Double_t meanZ_Phi;
};

class PndBarrelTrackFinder : public PndPersistencyTask {

 public:
  PndBarrelTrackFinder();

  PndBarrelTrackFinder(Int_t iVerbose);

  PndBarrelTrackFinder(const char *name, Int_t iVerbose);

  virtual ~PndBarrelTrackFinder();

  virtual void Exec(Option_t *opt);

  void UseMvdSttGem(const Bool_t useMvd, const Bool_t useStt, const Bool_t useGem);

  void DrawDetails(Bool_t tb = kTRUE) { fDrawDetails = tb; }
  void DrawHistos(Bool_t tb = kTRUE) { fDrawHistos = tb; }

  void SetRunTimeBased(Bool_t val = kTRUE) { fRunTimeBased = val; }

 private:
  Bool_t fRunTimeBased;
  StopTime *fFunctor;
  double fStopTimeValue;

  Bool_t fDrawDetails;
  Bool_t fDrawHistos;
  TEllipse *fEllipse[1000];
  TCanvas *fCanvas;
  TLatex *fStageDesc;
  TH1F *fClonesXDiff;
  TH1F *fClonesYDiff;
  TH1F *fClonesRDiff;
  TH1F *fClonesZ_PhiDiff;
  TH1F *fClonesXDiffRel;
  TH1F *fClonesYDiffRel;
  TH1F *fClonesRDiffRel;
  TH1F *fClonesZ_PhiDiffRel;

  TH1F *fhHitDist[5][3];

  std::vector<TrackBasis> fTracksVector;

  std::vector<Int_t> fHitDetId;
  std::vector<Int_t> fHitDetNo;
  // previous hits
  std::vector<FairHit *> fHitVector;
  std::vector<Int_t> fHitVectDI;
  std::vector<Int_t> fHitVectHN;

  Bool_t fIncludeDet[4];
  TClonesArray *fHitArray[4];
  TString fDetName[4];
  Int_t fDetType[4];
  TString fHitArrayName[4];

  TClonesArray *fBarrelTrackArray; 
  TClonesArray *fBarrelTrackCandArray;

  Int_t fTNofEvents; 
  Int_t fTNofTracks;

  double fRadToMomConversion;

  Double_t fMaximalDist;
  Double_t fReasonableDist;
  Double_t fMaximalZ_PhiD;
  Double_t fReasonableZ_PhiD;
  Double_t fMaximalRadDiff;
  Double_t fMaximalPhiDiff;

  Bool_t MatchHitToTrack(FairHit *thisHit, Int_t detId, Int_t hitNo, Int_t trackNo);
  Bool_t MatchSkewedSttHitTT(FairHit *thisHit, Int_t detId, Int_t hitNo, Int_t trackNo);
  Bool_t MatchParallelSttHitTT(FairHit *thisHit, Int_t detId, Int_t hitNo, Int_t trackNo);
  Bool_t MatchHitWithZInfoTT(FairHit *thisHit, Int_t detId, Int_t hitNo, Int_t trackNo);

  Bool_t ExtractMeanRPhiFromTrack(Int_t trackNo);
  Bool_t ExtractMeanZ_PhiFromTrack(Int_t trackNo);

  Bool_t MatchHitToHit(FairHit *thisHit, Int_t detId, Int_t hitNo, Int_t prevHNo);
  void RemoveHitFromPreviousHits(Int_t hitNo);

  Bool_t HitBelongsToTrack(Int_t detId, Int_t hitNo, Int_t trackNo);

  void AddHitToPreviousHits(FairHit *thisHit, Int_t detId, Int_t hitNo);

  void DrawTracks();
  void PrintTracks();
  Int_t CleanTracks();
  Int_t WriteTracks();

  void RemoveShortTracks();
  void RemoveCloneTracks();

  Bool_t FindCircPar(Double_t *c1, Double_t *c2, Double_t *c3, Int_t cno, Double_t *cl);
  Double_t FindCircDist(Double_t *c1, Double_t *c2);
  Int_t FindInterestingRegions(Double_t *circ, Double_t *tube, Double_t *reg);

  Double_t CalcPhi(Double_t x, Double_t y);
  Double_t CalcZ_Phi(Double_t *circ, Double_t hx, Double_t hy, Double_t hz);

  virtual void SetParContainers();
 //  CHECK added
  TClonesArray *fTubeArray;
  PndGeoSttPar *fSttParameters; //  CHECK added
  Int_t fGeoType;

  virtual InitStatus Init();

  virtual InitStatus ReInit();

  void Reset();

  virtual void Finish();

  ClassDef(PndBarrelTrackFinder, 1);
};

#endif