AliFMDMCHitEnergyFitter.h

Go to the documentation of this file.

#ifndef ALIFMDMCHITENERGYFITTER_H
#define ALIFMDMCHITENERGYFITTER_H
#include "AliFMDEnergyFitter.h"
#include "AliFMDFloatMap.h"
#include "TArrayF.h"
class AliMCAuxHandler;
class AliMCEvent;
class AliESDEvent;
class TNtuple;
class TClonesArray;

class AliFMDMCHitEnergyFitter : public AliFMDEnergyFitter
{
public:
  AliFMDMCHitEnergyFitter();
  AliFMDMCHitEnergyFitter(const char* title, Bool_t useTuple=false);
  virtual ~AliFMDMCHitEnergyFitter();

  virtual void CreateOutputObjects(TList* dir);
  virtual Bool_t PreEvent(const AliMCEvent& mcInput);
  virtual Bool_t Event(const AliESDEvent&  esdInput,
           const AliMCEvent&   mcInput,
           AliMCAuxHandler&    handler);
  virtual Bool_t AccumulateHits(const AliMCEvent&   mcInput,
        const TClonesArray& hits);
  virtual Bool_t PostEvent(const AliESDEvent& esdInput);
  
  TNtuple* GetTuple() { return fTuple; }
protected:
  AliFMDMCHitEnergyFitter(const AliFMDMCHitEnergyFitter&);
  AliFMDMCHitEnergyFitter& operator=(const AliFMDMCHitEnergyFitter&);

  struct RingHistos : public AliFMDEnergyFitter::RingHistos
  {
    RingHistos();
    RingHistos(UShort_t d, Char_t r);
    ~RingHistos() {}
    RingHistos(const RingHistos& o);
    RingHistos& operator=(const RingHistos& o);
    TArrayD MakeIncreasingAxis(Int_t nBins, 
             Double_t low,
             Double_t high) const;
    virtual void SetupForData(const TAxis& eAxis, 
            const TAxis& cAxis,
            Double_t     maxDE=10, 
            Int_t        nDEbins=300, 
            Bool_t       useIncrBin=true);
    virtual void FillMC(UShort_t flag, Double_t eta, Double_t mult);
    TObjArray* Fit(TList*           dir, 
       Double_t         lowCut, 
       UShort_t         nParticles,
       UShort_t         minEntries,
       UShort_t         minusBins, 
       Double_t         relErrorCut, 
       Double_t         chi2nuCut,
       Double_t         minWeight,
       Double_t         regCut,
       EResidualMethod  residuals) const;
    
    TH2* fPrimary;   // @f$\Delta@f$ vs @f$\eta@f$ for primaries
    TH2* fSecondary; // @f$\Delta@f$ vs @f$\eta@f$ for second.
    TH2* fKind;      // Particle kind
    ClassDef(RingHistos,1); // Cache of histograms per ring 
  };
  AliFMDEnergyFitter::RingHistos* CreateRingHistos(UShort_t d, Char_t r) const;
  AliFMDFloatMap   fSumPrimary;
  AliFMDFloatMap   fSumSecondary;
  TArrayF          fIp;
  Int_t            fNTrack;
  Int_t            fNPrimary;
  TNtuple*         fTuple;

  ClassDef(AliFMDMCHitEnergyFitter,1);
};

#endif
// Local Variables:
//  mode: C++
// End: