PndAnalysisForwardTask.h

Go to the documentation of this file.

/*
 * PndAnalysisForwardTask.h
 *
 *  Created on: Jul 15, 2016
 *      Author: kibellus
 */

#ifndef PNDTOOLS_PNDFORWARDTRACKFINDER_PNDANALYSISFORWARDTASK_H_
#define PNDTOOLS_PNDFORWARDTRACKFINDER_PNDANALYSISFORWARDTASK_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 "PndFtsPoint.h"

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

#include <iostream>
#include <vector>
#include <fstream>

class PndAnalysisForwardTask : public FairTask {
 public:
  PndAnalysisForwardTask() : eventNum(0)
  {
    for (int i = 0; i < 10; i++) {
      hitArray[i] = 0;
      momentumArray[i] = 0;
      trackCountArray[i] = 0;
      hitArray2[i] = 0;
      momentumArray2[i] = 0;
      trackCountArray2[i] = 0;
    }
  }
  virtual ~PndAnalysisForwardTask();

  virtual InitStatus Init();

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

  map<Int_t, vector<PndFtsHit *>> getMcTracks();
  map<Int_t, vector<PndFtsHit *> *> getReconstructedTracks();
  void analyzeMCTracks(vector<PndFtsHit *> hits, map<Int_t, vector<PndFtsHit *> *> tracks, Int_t trackID, Int_t trackCount);
  vector<Int_t> analyzeMCTrack(vector<PndFtsHit *> mcHits, vector<PndFtsHit *> *reconstructedHits);
  void anaSpecialCases(map<Int_t, vector<PndFtsHit *>> mcTracks);

 private:
  FairRootManager *fIoman;
  TClonesArray *fHits;
  TClonesArray *fSolution;
  TClonesArray *fMCTracks;
  TClonesArray *fIdealTrack;
  Int_t eventNum;

  Int_t completeClean = 0;
  Int_t completeUnclean = 0;
  Int_t incompleteClean = 0;
  Int_t incompleteUnclean = 0;
  Int_t notEnoughFound = 0;
  Int_t notFound = 0;

  map<Int_t, Int_t> mcToHitCount;
  map<Int_t, Double_t> momentum;

  Int_t hitArray[10];
  Double_t momentumArray[10];
  Int_t trackCountArray[10];

  Int_t hitArray2[10];
  Double_t momentumArray2[10];
  Int_t trackCountArray2[10];

  Int_t oneHitPerLayer = 0;
  Int_t moreThan3Hits = 0;
  Int_t twoTracksOneTube = 0;

  ClassDef(PndAnalysisForwardTask, 1);
};

#endif /* PNDTOOLS_PNDFORWARDTRACKFINDER_PNDANALYSISFORWARDTASK_H_ */