PndLmdGeaneTask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                   PndLmdGeaneTask header file                 -----
// -----                  Created 20/03/07  by R.Kliemt                -----
// -----                  modified for Lmd by M. Michel & A.Karavdina  -----
// -------------------------------------------------------------------------

#ifndef PndLmdGeaneTask_H
#define PndLmdGeaneTask_H

// framework includes
#include "FairTask.h"
#include "TH1.h"
//#include "GFTrackCand.h"
#include "FairGeanePro.h"
#include "FairTrackParP.h"
#include "TMatrixD.h"
#include "TNtuple.h"
#include "TVector3.h"
//#include "PndSdsGeoHandling.h"
#include <map>
#include <vector>
#include "PndGeoHandling.h"
#include "PndMultiField.h"
class TClonesArray;
class PndSdsCluster;
// class TGeant3;//GEANE uses Geant3!
// class TGeant4;

class PndLmdGeaneTask : public FairTask {
 public:
  PndLmdGeaneTask();

  PndLmdGeaneTask(Double_t pBeam, TVector3 IP, bool is_prefiltered = true);

  ~PndLmdGeaneTask();

  virtual void SetParContainers();
  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  virtual void Finish();

  FairTrackParP *PropToPlane(FairTrackParP *fStartPst, double zpos, int dir,
                             bool &isProp); // propagation between planes
 private:
  FairGeanePro *fPro;
  PndGeoHandling *fGeoH;
  int fEventNr;
  bool fUseMVDPoint;
  Double_t fPbeam;
  Int_t fPDGid;
  TVector3 vtx;
  FairField *pndField;
  std::map<int, std::vector<int>> fTrackPixHitIdMap;   // Track -> PixHitId
  std::map<int, std::vector<int>> fTrackStripHitIdMap; // Track -> StripHitId

  std::string track_branch_name;

  // Input Data ----------
  TClonesArray *fTracks; // Real tracks

  // Output Data -------
  TClonesArray *fTrackParGeane;
  TClonesArray *fTrackParIni;
  TClonesArray *fTrackParFinal;
  TClonesArray *fDetName;

  //----------------------------------------

  void Register();

  void Reset();

  void ProduceHits();

  ClassDef(PndLmdGeaneTask, 2);
};

#endif