removing obsolete cout

[u/mrichter/AliRoot.git] / PWG1 / AliIntSpotEstimator.h

diff --git a/PWG1/AliIntSpotEstimator.h b/PWG1/AliIntSpotEstimator.h
index 585b055aa90ff501ee97ddb495855cad62305d35..eb00e024047da4a0fa1dfcb89deb20ba466e4a73 100644 (file)
--- a/PWG1/AliIntSpotEstimator.h
+++ b/PWG1/AliIntSpotEstimator.h
@@ -8,6 +8,7 @@
 #include <TAxis.h>
 
 class TH1;
+class TNtuple;
 class TCanvas;
 class AliESDEvent;
 class AliESDtrack;
@@ -19,36 +20,42 @@ class AliIntSpotEstimator : public TNamed {
  public:
   //
   AliIntSpotEstimator(Bool_t initDef=kFALSE);
-  AliIntSpotEstimator(const char* name, Double_t outcut=1e-4,
+  AliIntSpotEstimator(const char* name, Double_t outcut=1e-4,Int_t ntrIP=2,
                      Int_t nPhiBins=12,Int_t nestb=500,
                      Double_t estmin=-2e-2,Double_t estmax=6e-2,
                      Int_t ntrBins=10,Int_t ntMn=2,Int_t ntMx=32,
-                     Int_t nPBins=14,Double_t pmn=0.2,Double_t pmx=3.);
+                     Int_t nPBins=14,Double_t pmn=0.2,Double_t pmx=3., Bool_t ntuple=kFALSE);
   ~AliIntSpotEstimator();
   AliIntSpotEstimator &operator += (const AliIntSpotEstimator &src);  
   //
   void          InitEstimators(Int_t nPhiBins=12,Int_t nestb=500,
                               Double_t estmin=-2e-2,Double_t estmax=6e-2,
                               Int_t ntrBins=10,Int_t ntMn=2,Int_t ntMx=32,
-                              Int_t nPBins=14,Double_t pmn=0.2,Double_t pmx=3.);
+                              Int_t nPBins=14,Double_t pmn=0.2,Double_t pmx=3.,Bool_t ntuple=kFALSE);
   //
   Bool_t        ProcessEvent(const AliESDEvent* esd, const AliESDVertex* vtx=0);
   Bool_t        ProcessEvent(const TObjArray* tracks);
   //
-  Double_t      GetIPCenter(Int_t id)                 const  {return fIPCenterStat>0 ? fIPCenter[id]/fIPCenterStat:0;}
-  Double_t      GetIPSigma(Int_t phibin=0)            const;
-  Double_t      GetVtxSigma(int ntr)                  const;
-  Double_t      GetDCASigma(double p)                 const;
+  Double_t      GetIPCenter(Int_t id,Double_t *err=0)              const;
+  Double_t      GetIPCenIni(Int_t id)                              const  {return fIPCenIni[id];}
+  Double_t      GetIPSigma(Int_t phibin=0,Double_t *err=0)         const;
+  Double_t      GetVtxSigma(int ntr, Double_t *err=0)              const;
+  Double_t      GetDCASigma(double p, Double_t *err=0)             const;
   //
-  Int_t         GetEventsAccepted()                   const  {return fIPCenterStat;} 
-  Int_t         GetEventsProcessed()                  const  {return fEvProc;} 
+  Int_t         GetEventsAccepted()                                const  {return fIPCenterStat;} 
+  Int_t         GetEventsProcessed()                               const  {return fEvProc;} 
+  Int_t         GetMinTracksForIP()                                const  {return fMinTracksForIP;}
   //
-  void          SetOutlierCut(Double_t v=1e-4)               {fOutlierCut = v;}
+  void          SetOutlierCut(Double_t v=1e-4)                            {fOutlierCut = v;}
+  void          SetIPCenIni(Double_t *xyz)                                {for (int i=3;i--;) fIPCenIni[i] = xyz[i];}
+  void          SetMinTracksForIP(Int_t ntr=2)                            {fMinTracksForIP = ntr>2 ? ntr : 2;}
   //
-  TH2F*         GetHistoIP()                          const  {return fEstimIP;}
-  TH2F*         GetHistoVtx()                         const  {return fEstimVtx;}
-  TH2F*         GetHistoTrc()                         const  {return fEstimTrc;}
-  AliVertexerTracks* GetVertexer()                    const  {return fVertexer;}
+  TH2F*         GetHistoIP()                                       const  {return fEstimIP;}
+  TH2F*         GetHistoVtx()                                      const  {return fEstimVtx;}
+  TH2F*         GetHistoTrc()                                      const  {return fEstimTrc;}
+  TH2F*         GetHistoVtxXY()                                    const  {return fHVtxXY;}
+  TNtuple*      GetNtuple()                                        const  {return (TNtuple*)fNtuple;}
+  AliVertexerTracks* GetVertexer()                                 const  {return fVertexer;}
   //
   Int_t         GetNPhiBins()    const {return !IsValid() ? 0:fEstimIP->GetXaxis()->GetNbins();}
   Int_t         GetNTrackBins()  const {return !IsValid() ? 0:fEstimVtx->GetXaxis()->GetNbins();}
@@ -62,7 +69,7 @@ class AliIntSpotEstimator : public TNamed {
   virtual void  Print(Option_t *opt="")               const;
   virtual void  Clear(Option_t *opt="");
   virtual Long64_t Merge(TCollection *coll);
-  static Double_t CalcMean(TH1* histo, Double_t ctfact);
+  static Double_t CalcMean(TH1* histo, Double_t ctfact,Double_t *err=0);
   //
  protected:
   Bool_t        IsValid()       const {return fEstimIP!=0;}
@@ -80,12 +87,17 @@ class AliIntSpotEstimator : public TNamed {
   //
   Int_t         fEvProc;                         // number of events processed
   Int_t         fIPCenterStat;                   // number of events used for fIPCenter
+  Int_t         fMinTracksForIP;                 // account IP estimator only for vertices with >= tracks
   Double_t      fOutlierCut;                     // cut on outliers
-  Double_t      fIPCenter[3];                    // mean IP position XYZ
+  Double_t      fIPCenIni[3];                    // mean IP position XYZ (initial)
+  Double_t      fIPCenter[3];                    // IP position XYZ
+  Double_t      fIPCen2[3];                      // IP position XYZ^2
   //
   TH2F         *fEstimIP;                        // distribution of IP estimator
   TH2F         *fEstimVtx;                       // distribution of VTRes estimator
   TH2F         *fEstimTrc;                       // distribution of DCA res estimator
+  TH2F         *fHVtxXY;                         // XY histo
+  TNtuple      *fNtuple;                         //! optional ntuple with dca's
   //
   AliVertexerTracks* fVertexer;                  //! vertex fitter
   TObjArray    *fTracks;                         //! storage for processed tracks