EdbPositionAlignment.h

Go to the documentation of this file.

#ifndef ROOT_EdbPositionAlignment
#define ROOT_EdbPositionAlignment
 
//#include <TNtuple.h>
//#include <TString.h>
#include "EdbBrick.h"
#include "EdbPattern.h"
#include "EdbAffine.h"
#include "EdbCell1.h"
#include "EdbPatCell2.h"
 
class EdbPositionAlignment : public TObject
{
public:
  // service objects:
  EdbPatCell2 ePC1, ePC2;        
  TObjArray   eComb1, eComb2;    
  TObjArray   eTracks;           
  TObjArray   eSegCouples;       
  EdbPatCell2 ePC;               
 
  // input parameters:
  float   eX0, eY0;              
 
  Float_t eXcell, eYcell;        
  Float_t eDTXmax,eDTYmax;       
  Float_t eBinX, eBinY;          
 
  Float_t eDXmin,  eDYmin;       
  Float_t eDTXmin, eDTYmin;      
  Float_t eRpeak;                
 
  // z-selection block: 
  Float_t eZ1from,eZ1to;         
  Float_t eZ2from,eZ2to;         
  Int_t   eNZ1step, eNZ2step;    
 
 // results of the z-selection:
  Float_t eZ1peak, eZ2peak;     
  Float_t eXpeak,eYpeak;        
  Int_t   eNpeak;               
  EdbAffine2D *eAff;            
  EdbH2       eHpeak;           
  EdbH2       eHDZ;             
 
  TString     eSmoothKernel;    
 
  TTree   *ePosTree;            
 
  // linking:
  Float_t eS0x,eS0y,eS0tx,eS0ty;    
  Float_t eWbaseMin;                
  Float_t eChi2Max;                 
  
  // shrinkage-selection block: 
  Float_t eShr1from,eShr1to;        
  Float_t eShr2from,eShr2to;        
  Int_t   eNShr1step, eNShr2step;   
  // results of the shrinkage selection:
  EdbH2       eHShr1;             
  EdbH2       eHShr2;             
 
 
  Bool_t      eDoRot;             
  EdbH1       eRot;               
                                     
public:
  EdbPositionAlignment();
  virtual ~EdbPositionAlignment();
 
  bool  ActivatePosTree(const char *name="postree");
  int   WritePosTree();
  int   ReadPosTree(TTree *tree,  EdbPattern *p=0, EdbPattern *p1=0, EdbPattern *p2=0, TEventList *lst=0 );
  int   FillPosTree( float dz1, float dz2, int flag); //{ return FillPosTree(eComb1,eComb2,dz1,dz2,flag); }
  //int   FillPosTree(TObjArray &arr1,TObjArray &arr2, float dz1, float dz2, int flag);
  //int   FillPosTree(EdbPattern &p1, EdbPattern &p2, float dz1, float dz2, int flag);
  void  PrintStat();
  void  PrintSummary();
 
  int   Align();
 
  int   SelectZone( float min[2], float max[2], TObjArray &arr1, TObjArray &arr2, float maxDens );
  int   SelectZoneSide( EdbPatCell2 &pc, float min[2], float max[2], TObjArray &arr, int nmax=kMaxInt );
  static int   SelectBestComptons( TObjArray &a,  TObjArray &arr, int nmax);
 
  int   FillArrays(EdbPattern &p1 ,  EdbPattern &p2);
  int   FillArrays(TObjArray &arr1,  TObjArray &arr2, float xymin[2], float xymax[2] );
  int   SpotsFilterOut(int nmax);
  int   DoubletsFilterOut(bool checkview=true);
  int   DoubletsFilterOutSide(EdbPatCell2 &pc, bool checkview=true);
  int   FillCombinations();
  int   FillCombinations(EdbPatCell2 &pc1, EdbPatCell2 &pc2, float dx, float dy, float dtx, float dty);
 
  int   SelectNarrowPeakDXDY(float dr, EdbH2 &hxy);
  void  PositionPlotA(EdbH2 &hd, float DR=0, TObjArray *arr1=0, TObjArray *arr2=0);
  void  FindDiff(TObjArray &arr1, TObjArray &arr2, EdbPatCell2 &pc1, EdbPatCell2 &pc2, 
         float &dx, float &dy, float &dtx, float &dty );
  void  FindDiff12(float &dx, float &dy, float &dtx, float &dty );
  void  FindDiff10(float &dx, float &dy, float &dtx, float &dty );
  void  FindDiff20(float &dx, float &dy, float &dtx, float &dty );
  void  AlignWithTracks(int nang=3, int npos=0);
  void  FindCorrections(EdbPattern &pat1, EdbPattern &pat2, float DZ, bool doShrink);
 
  void  PositionPlot(float dz1, float dz2, EdbH2 &hd);
  void  PositionPlot(EdbH2 &hd);
  void  Zselection(EdbH2 &hd);
 
  void  ShrinkageSelection(float dzbase);
  int   ShrinkageSelectionSide(EdbPatCell2 &pc1, EdbPatCell2 &pc2, EdbH2 &hshr, int nstep, float deltaShr);
  int   Link();
  //int   Link(EdbPattern &p1, EdbPattern &p2);
 
  float Xcorr(EdbSegP &s, float dz) {return s.X() + dz*s.TX();}
  float Ycorr(EdbSegP &s, float dz) {return s.Y() + dz*s.TY();}
  float Xcorr(EdbSegP &s, float dz, float dx) {return dx + s.X() + dz*s.TX();}
  float Ycorr(EdbSegP &s, float dz, float dy) {return dy + s.Y() + dz*s.TY();}
 
  int SelectPeak(EdbPattern &p1,EdbPattern &p2, float dx, float dy, float dz1, float dz2 ); 
  int SelectPeak(EdbPattern &p1,EdbPattern &p2, float dx=10, float dy=10 )  
    { return SelectPeak(p1,p2,dx,dy,eZ1peak,eZ2peak); } 
 
  void ResetPeak() { eZ1peak=eZ2peak=eXpeak=eYpeak=0; eNpeak=0; }
 
  void RankCouples0();
  void RankCouples();
  void RankCouples( TObjArray &arr1,TObjArray &arr2, TObjArray &arrtr );
  void RankCouplesFast( TObjArray &arrtr );
 
  void ResetPositionCorr() { ePC1.eDX = ePC1.eDY = ePC2.eDX = ePC2.eDY = 0;}
  void ResetAngularCorr()  { ePC1.eDTX = ePC1.eDTY = ePC2.eDTX = ePC2.eDTY = 0;}
 
  //  void ApplyCorrections(EdbPlateP &plate);
 
  int  WideSearchXY(EdbPattern &pat1, EdbPattern &pat2, EdbH2 &hxy, float dz, float xmin, float xmax, float ymin, float ymax);
 
  void SaveAsTree(EdbPattern &pat, const char *name);
 
 
  ClassDef(EdbPositionAlignment,1)  // new alignment class developed mainly for compton search
};
#endif /* ROOT_EdbPositionAlignment */