Update documentation (SPD, SSD, multiplicity)

[u/mrichter/AliRoot.git] / TRD / Cal / AliTRDCalPIDRefMaker.h

#ifndef ALITRDCALPIDREFMAKER_H
#define ALITRDCALPIDREFMAKER_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
//  TRD calibration class for building reference data for PID                //
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#ifndef ROOT_TObject
#include "TObject.h"
#endif

class TH1;
class TH2;
class TH3;
class TPrincipal;
class TLinearFitter;

class AliTRDCalPIDRefMaker : public TObject {

public:
  AliTRDCalPIDRefMaker();
  AliTRDCalPIDRefMaker(const AliTRDCalPIDRefMaker &ref);
  ~AliTRDCalPIDRefMaker();
  AliTRDCalPIDRefMaker& operator=(const AliTRDCalPIDRefMaker &ref);

         Bool_t   BuildLQReferences(Char_t *filename="TRDpidLQ.root", Char_t *dir =".");
         Bool_t   BuildNNReferences(Char_t *filename="TRDpidNN.root", Char_t *dir =".");
 
  static Double_t Estimate2D2(TH2 *h, Float_t &x, Float_t &y);
  static Double_t Estimate2D1(TH2 *h, Float_t &x, Float_t &y, Float_t &dCT, Float_t &rmin, Float_t &rmax);
  //     Double_t Estimate3D2(TH3 *h, Float_t &x, Float_t &y, Float_t &z);

private:
         Int_t    CheckProdDirTree(Char_t *dir=".");
         void     Prepare2D();
         void     Reset();
         void     SaveReferences(const Int_t mom, const char *fn);
 

private:
         TPrincipal    *fPrinc[5];  // Used for principal component analysis
  static TLinearFitter *fFitter2D2; // Working object for linear fitter
  static TLinearFitter *fFitter2D1; // Working object for linear fitter
         TH2           *fH2dEdx[5]; // dE/dx data holders
         TH1           *fH1TB[2];   // Max time bin data holders

  ClassDef(AliTRDCalPIDRefMaker, 2) // Reference histograms builder

};

#endif