Added hit distributions from all events from MC and data in preparation for sharing...

[u/mrichter/AliRoot.git] / FMD / analysis / AliFMDAnalysisTaskSharing.h

#ifndef ALIFMDANALYSISTASKSHARING_H
#define ALIFMDANALYSISTASKSHARING_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
#include "AliAnalysisTask.h"
#include "TH1F.h"
#include "TObjString.h"
#include "AliESDFMD.h"
#include "TTree.h"
#include "AliESDEvent.h"
#include "AliFMDFloatMap.h"
class TChain;
class AliAODEvent;
class AliESDVertex;

/**
 * Do the sharing correction. 
 * @ingroup FMD_ana
 * 
 * 
 */
class AliFMDAnalysisTaskSharing : public AliAnalysisTask
{
 public:
    AliFMDAnalysisTaskSharing();
    AliFMDAnalysisTaskSharing(const char* name, Bool_t SE = kTRUE);
    virtual ~AliFMDAnalysisTaskSharing() {;}
 AliFMDAnalysisTaskSharing(const AliFMDAnalysisTaskSharing& o) : AliAnalysisTask(),
      fDebug(o.fDebug),
      fESD(o.fESD),
      // fOutputESD(),
      foutputESDFMD(o.foutputESDFMD),
      fEnergy(o.fEnergy),
      fNstrips(o.fNstrips),
      fSharedThis(o.fSharedThis),
      fSharedPrev(o.fSharedPrev),
      fDiagList(),
      fStandalone(o.fStandalone),
      fEsdVertex(o.fEsdVertex),
      fStatus(o.fStatus)   {}
    AliFMDAnalysisTaskSharing& operator=(const AliFMDAnalysisTaskSharing&) { return *this; }
    
    // Implementation of interface methods
    virtual void ConnectInputData(Option_t *option = "");
    virtual void CreateOutputObjects();
    virtual void Init() {}
    virtual void LocalInit() {Init();}
    virtual void Exec(Option_t */*option*/);
    virtual void Terminate(Option_t* /* option*/) {}
    virtual void SetDebugLevel(Int_t level) {fDebug = level;}
    Float_t GetMultiplicityOfStrip(Float_t mult, Float_t eta, Float_t Eprev, Float_t Enext, UShort_t   det, Char_t  ring, UShort_t sec, UShort_t strip);
    // void GetVertex(Double_t* vertexXYZ) ;
    void SetFMDData(AliESDFMD* fmd) {foutputESDFMD = fmd;}
    void SetOutputList(TList* outlist) {fDiagList = outlist;}
    void SetVertex(AliESDVertex* vertex) {fEsdVertex = vertex;}
    void SetInputESD(AliESDEvent* esd) {fESD = esd;}
    Bool_t GetEventStatus() {return fStatus;}
 private:
    Float_t Eta2Theta(Float_t eta);
    Double_t EtaFromStrip(UShort_t det, Char_t ring, UShort_t sector, UShort_t strip, Double_t zvtx);
    void ProcessPrimary();
    
    Int_t         fDebug;        //  Debug flag
    AliESDEvent*  fESD;          //! ESD
    // AliESDEvent   fOutputESD;
    AliESDFMD*    foutputESDFMD;
    Float_t       fEnergy;
    Int_t         fNstrips;
    Bool_t        fSharedThis;
    Bool_t        fSharedPrev;
    TList*        fDiagList;
    Bool_t        fStandalone;
    AliESDVertex* fEsdVertex;
    Bool_t        fStatus;
    AliFMDFloatMap fLastTrackByStrip;
    
    ClassDef(AliFMDAnalysisTaskSharing, 0); // Analysis task for FMD analysis
};
 
#endif
// Local Variables:
//  mode: C++
// End Variables;