HarpoKalmanNew.h

Go to the documentation of this file.

 #ifndef HARPOKALMANNEW_H
#define HARPOKALMANNEW_H

//
// HarpoKalmanNew
//
// Virtual class for Kalman filter in the HARPO geometry
//

#include "TObject.h"
#include "HarpoAnalyse.h"
#include "HarpoEvent.h"
#include "HarpoRecoEvent.h"
#include "TProfile.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TCanvas.h"
#include "TPolyLine.h"
#include "TMatrixD.h"
#include "TGraphErrors.h"
#include "TRootEmbeddedCanvas.h"

#define NTRACK 100

#include <vector>

class HarpoKalmanNew : public HarpoAnalyse {

 public:
  
  virtual ~HarpoKalmanNew() {}

  HarpoKalmanNew() {}

  void   Init();
  void   Save(char *mode=NULL);
  void   process(); 
  void   print(); 

  TH1F* hNcl;
  /* TGraph* GetGraph(Int_t i, Int_t plane = 0) {  */
  /*   /\* if(plane<0 || plane>1) return 0; *\/ */
  /*   /\* if(fId[i][plane]<0) return 0; *\/ */
  /*   /\* if(i>=0 && i<fNtr[plane]) return fGraphs[plane][fId[i][plane]]; *\/ */
  /*   /\*    //if(i>=0 && i<fNtr[plane]) return fGraphs[plane][i]; *\/ */
  /*   /\* else return 0; *\/ */
  /*   return fGraphs[plane][i]; */
  /* } */
  Int_t GetNtr(Int_t plane){if(plane<0 || plane>1) return 0; return fNtr[plane];}
  /* Int_t GetId(Int_t i,Int_t plane) {return fId[i][plane];} */
  Int_t GetNclCommon(TClonesArray* clArray, Int_t plane, Int_t iTr1, Int_t iTr2);

  void SetCanvas(TVirtualPad* c, Int_t plane){fCanvas[plane] = c;}
  void SetHist(TH2F* h, Int_t plane){fHist[plane] = h;}

  void DisplayAnalysis(TRootEmbeddedCanvas* ecTab, TGListBox* infobox);

  Int_t      InitPlane(TClonesArray* clArray, Int_t plane);
  std::vector<TMatrixD> NextStep(TMatrixD X, TMatrixD C, Int_t Ntr, TClonesArray* clArray, TArrayI* arr, Int_t ncl, Int_t plane, Int_t color, Int_t fill,Int_t smooth, Int_t q, Bool_t finder = kTRUE);

  void SetLambdaMin(Double_t val){fLambdaMin = val;}
  void SetLambdaMax(Double_t val){fLambdaMax = val;}
  void SetQmin(Double_t val){fQmin = val;}
  void SetWidthMinT(Double_t val){fWidthMinT = val;}
  void SetWidthMinC(Double_t val){fWidthMinC = val;}
  void SetWidthMax(Double_t val){fWidthMax = val;}
  void SetNclMin(Int_t val){fNclMin = val;} 
  void SetNclMin2(Int_t val){fNclMin2 = val;} 
  void SetMaxSlopeType(Double_t val){fMaxSlopeType = val;}
  void SetResKalman(Double_t val){fResKalman = val;}
  void SetScatKalman(Double_t val){fScatKalman = val;}
  void SetResFinder(Double_t val){fResFinder = val;}
  void SetScatFinder(Double_t val){fScatFinder = val;}


 protected:

/* //TArrayI*  */
/* void FindNext(TMatrixD X, TMatrixD C, Double_t angle, Double_t angle2, Int_t Ntr, TClonesArray* clArray, TArrayI* arr, Int_t ncl, Int_t plane, Int_t color, Int_t fill,Int_t smooth, Int_t q); */
/* //TArrayI*  */
/*   void FindNextClosest(TMatrixD X, TMatrixD C, Double_t angle, Double_t angle2, Int_t Ntr, TClonesArray* clArray, TArrayI* arr, Int_t ncl, Int_t plane, Int_t color, Int_t fill,Int_t smooth, Int_t q); */

  Double_t GetResolution(HarpoRecoClusters* cl);


  Int_t FindClosestNeighbour(TMatrixD X, Int_t iTr, TClonesArray* clArray, Int_t plane, Int_t color, Int_t fill, Bool_t finder = kTRUE);





  Int_t Ccl[NCHAN][20];
  Int_t NCcl[NCHAN];
  Int_t Tcl[NADC][20];
  Int_t NTcl[NADC];
  Int_t fNtr[2];

  Int_t fNclTrack[NTRACK];
  TGraph* fGraph[2][NTRACK]; 
  TGraph* fGraph2[2][NTRACK]; 


  Double_t fStartPointX[NTRACK][10];
  Double_t fStartPointZ[NTRACK][10];
  //  Double_t fStartAngle[NTRACK][10];
  Double_t fStartDirX[NTRACK][10];
  Double_t fStartDirZ[NTRACK][10];
  Double_t fEndPointX[NTRACK][10];
  Double_t fEndPointZ[NTRACK][10];
  Double_t fEndDirX[NTRACK][10];
  Double_t fEndDirZ[NTRACK][10];
  Int_t fStartIndex[NTRACK];
  Int_t fEndIndex[NTRACK];
  /* Int_t fId[NTRACK][2]; */
  /* Double_t fQcommon[NTRACK][NTRACK]; */

  //  Double_t fMapTmp[NCHAN][NADC];
  ULong64_t fMapTmp[NCHAN][NADC];
  Double_t fQtot;
  Double_t fQused;

  Int_t used[4000];
  Int_t reused[4000];
  Int_t fTrack[4000];
  TVirtualPad* fCanvas[2];
  TH2F* fHist[2];

  TH1F* hChi2;
  TH1F* hDistTracks;
  TH1F* hDistMinTracks;
  TH1F*  hThetaTracks;
  TH2F*  hThetaTracksNtr;
  TH2F*  hThetaDist[10];

  TH2F*   fHistQQtestVsDist;
  TH2F*   fHistQQmin;

  /* TMatrixD H; */
  /* TMatrixD Ht; */
  /* TMatrixD Q; */
  /* TMatrixD G; */
  /* TMatrixD V; */
  /* TMatrixD I; */
  /* TMatrixD Corig; */
  
  Double_t fLambdaMin;
  Double_t fLambdaMax;
  Double_t fQmin;
  Double_t fWidthMinT;
  Double_t fWidthMinC;
  Double_t fWidthMax;
  Int_t fNclMin; 
  Int_t fNclMin2; 
  Double_t fMaxSlopeType;
  Double_t  fResKalman;
  Double_t  fScatKalman;
  Double_t  fResFinder;
  Double_t fScatFinder;

  Int_t nEvents;

  ClassDef(HarpoKalmanNew,1)  //Run AnalysePrint
};

#endif