PndTrackDraw.h

Go to the documentation of this file.

#ifndef PNDTRACKDRAW_H
#define PNDTRACKDRAW_H

#include "FairTask.h"
#include "TEveTrackPropagator.h"
#include "TEveElement.h"

#include "TEveArrow.h"
#include "TString.h"
#include "FairGeanePro.h"

#include "FairRootManager.h"

#include "PndTrackCandDraw.h"
#include <vector>
class TEveTrackList;
class FairEventManager;
class TObjArray;

class PndTrackDraw : public FairTask {

 public:
  explicit PndTrackDraw(Bool_t propagation = kFALSE);

  explicit PndTrackDraw(const char *name, Bool_t propagation = kFALSE, Int_t iVerbose = 1);

  virtual ~PndTrackDraw();

  void SetVerbose(Int_t iVerbose) { fVerbose = iVerbose; }
  virtual void Exec(Option_t *option);
  virtual InitStatus Init();

  void Reset();
  virtual Double_t GetTimeWindowPlus() { return fTimeWindowPlus; }
  virtual Double_t GetTimeWindowMinus() { return fTimeWindowMinus; }

  virtual void SetTimeWindowMinus(Double_t val);
  virtual void SetTimeWindowPlus(Double_t val);

 protected:
  TEveTrackList *GetTrGroup(Int_t pid);
  void PropagateTrack(FairTrackParP &trackPar, Int_t pidHypo, Int_t color = -1, TEveElement *group = 0);
  void PropagateTrackHelix(FairTrackParP &trackPar, Int_t pidHypo, Int_t color = -1, TEveElement *group = 0);

 private:
  TClonesArray *fPndTrackList = nullptr; 
  TEveTrackPropagator *fTrPr = nullptr;
  FairEventManager *fEventManager = nullptr; 
  TObjArray *fEveTrList = nullptr;
  TString fEvent;                   
  TEveTrackList *fTrList = nullptr; 

  // TEveElementList *fTrackCont;
  FairGeanePro *fPro = nullptr;
  // FairRootManager* fManager;
  PndTrackCandDraw *fTrackCandDraw = nullptr;
  std::vector<TEveArrow *> fArrows; 
  Bool_t fDoPropagation;

  TEveElementList *fListOfTracks = nullptr;

  Double_t MinEnergyLimit = -1.0;
  Double_t MaxEnergyLimit = -1.0;
  Double_t PEnergy = -1.0;

  BinaryFunctor *fStartFunctor = nullptr;
  BinaryFunctor *fStopFunctor = nullptr;

  Double_t fTimeWindowPlus = -1.0;
  Double_t fTimeWindowMinus = -1.0;
  Double_t fStartTime = -1.0;
  Bool_t fUseEventTime;

  ClassDef(PndTrackDraw, 1);
};

#endif