PndLineApproximation.h

Go to the documentation of this file.

/*
 * PndLineApproximation.h
 *
 *  Created on: Jun 6, 2016
 *      Author: kibellus
 */

#ifndef PNDTOOLS_PNDFORWARDTRACKFINDER_PNDLINEAPPROXIMATION_H_
#define PNDTOOLS_PNDFORWARDTRACKFINDER_PNDLINEAPPROXIMATION_H_

#include "PndLine.h"
#include "PndFtsHit.h"
#include "PndTrack.h"
#include "PndTrackCand.h"
#include "FairTrackParP.h"
#include "TVector2.h"
#include "TVector3.h"
#include "TClonesArray.h"
#include <vector>

using namespace std;

class PndLineApproximation {
 public:
  PndLineApproximation(vector<PndFtsHit *> hits) : fLineApprox(linearRegression(hits)), fCorrectedHits(hits), hasYInfo(kTRUE) {}
  PndLineApproximation() {}
  PndLineApproximation(PndLine lineApprox, vector<PndFtsHit *> correctedHits) : fLineApprox(lineApprox), fCorrectedHits(correctedHits) {}
  PndLineApproximation(PndLine lineApprox, vector<PndFtsHit *> ch1, vector<PndFtsHit *> ch2);
  virtual ~PndLineApproximation();
  PndLine &getLine() { return fLineApprox; }
  void setLine(PndLine l) { fLineApprox = l; }
  vector<PndFtsHit *> getHits() { return fCorrectedHits; }
  void setHits(vector<PndFtsHit *> h) { fCorrectedHits = h; }
  PndLineApproximation newApproximation(PndLineApproximation &approx2);
  PndLine linearRegression(vector<PndFtsHit *> hits);
  PndLine linearRegressionXZ(vector<PndFtsHit *> hits);
  PndLine linearRegressionZX(vector<PndFtsHit *> hits);
  PndLine linearRegressionYZ(vector<PndFtsHit *> hits);
  PndLine linearRegressionZY(vector<PndFtsHit *> hits);
  PndTrack plot(Double_t zVal1, Double_t zVal2, TClonesArray *hitArr);
  void correctHits3DAndAdd(vector<PndFtsHit *> correctedHits);
  void setYInfo(Bool_t i) { hasYInfo = i; }
  Bool_t HasYInfo() { return hasYInfo; }

 private:
  TVector2 linearRegression(Double_t matrix[2][3]);
  PndLine fLineApprox;
  vector<PndFtsHit *> fCorrectedHits;
  PndFtsHit *copyHitWithNewPosition(PndFtsHit *h, TVector3 pos);
  Bool_t hasYInfo;
};

#endif /* PNDTOOLS_PNDFORWARDTRACKFINDER_PNDLINEAPPROXIMATION_H_ */