PndForwardTrackFinderTask.h

Go to the documentation of this file.

#ifndef PndForwardTrackFinderTask_H
#define PndForwardTrackFinderTask_H

#include "FairTask.h"
#include "FairRuntimeDb.h"
#include "FairRunAna.h"
#include "FairHit.h"

#include "PndGeoFtsPar.h"
#include "PndFtsMapCreator.h"
#include "PndFtsCellTrackletGenerator.h"
#include "PndTrackCombiner.h"
#include "PndLine.h"
#include "PndFtsLineApproximator.h"
#include "PndModuleCombiner.h"
#include "PndWayFollower.h"
#include "PndTrackCollection.h"

#include "PndFtsPoint.h"

#include "TClonesArray.h"
#include "TVector3.h"

#include <iostream>
#include <vector>
#include <string>

class PndForwardTrackFinderTask : public FairTask {
 public:
  PndForwardTrackFinderTask() : eventNumber(0)
  {
    PndTrackCombiner::compDist = 0.8;
    PndTrackCombiner::compAngle = 7;
    PndModuleCombiner::compDist = 0.8;
    PndModuleCombiner::compAngle = 7;
    PndTrackCombiner::compQuali = 700;
    PndWayFollower::qualiT = 4000;
    PndModuleCombiner::hitAddT = 3;
  }
  virtual ~PndForwardTrackFinderTask();

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

  virtual void Exec(Option_t *opt);
  virtual void FinishEvent();
  virtual void Finish();

  void saveCombined(vector<PndLineApproximation> combined, Int_t branch, Int_t branchID, Bool_t useOrgHits);
  void saveCorrectedTracklets(map<Int_t, vector<PndFtsExpandedTrackCand>>);
  void saveTrackCollection(vector<PndTrackCollection> coll, Bool_t withOrgHits);

 private:
  Int_t eventNumber;
  // PndGeoFtsPar *fFtsParameters;  // needed for FtsStrawMap
  // TClonesArray *fTubeArray;      // needed for FtsStrawMap
  TClonesArray *fHits;
  FairRootManager *fIoman;
  std::vector<PndTrackCand> fFirstTrackCand;
  string branchName = "FTSHitErr";

  TClonesArray *planes;    // temp
  TClonesArray *planesHit; // temp
  TClonesArray *fFirstTrackCandArray;
  TClonesArray *correctedTracklets;
  TClonesArray *correctedHits;
  TClonesArray *fLayerTracks;
  TClonesArray *fLayerTracksHits;
  TClonesArray *fGlobalTracks;
  TClonesArray *fGlobalTracksHits;
  TClonesArray *fFinalSolution;
  TClonesArray *fTrackCollection;
  TClonesArray *fTrackCollectionHits;

  TClonesArray *rootBranches[5];

  PndFtsCellTrackletGenerator *fPndFtsCellTrackletGenerator;
  PndFtsLineApproximator *fPndFtsLineApproximator;
  PndTrackCombiner *fPndTrackCombiner;
  PndModuleCombiner *fModuleCombiner;
  PndWayFollower *fWayFollower;
  // ClassDef(PndForwardTrackFinderTask,1);

  Int_t combinedTracksHitNum;
  Int_t combinedTracksTrackNum;

  map<Int_t, PndFtsHit *> fOriginalHits;

  void createStatictcs(vector<PndTrackCollection> c, vector<PndLineApproximation> a);
  PndLineApproximation getBest(PndTrackCollection &c, vector<PndLineApproximation> a);
  Int_t getMcId(vector<PndFtsHit *> hits);
  Int_t dists[20];
  Int_t angles[20];

  ClassDef(PndForwardTrackFinderTask, 1);
};

#endif