PWG3/muon/AliCFMuonResTask1.h

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 //-----------------------------------------------------------------------
  17 // Author : R. Vernet, Consorzio Cometa - Catania (it)
  18 //-----------------------------------------------------------------------
  19 // Modification done by X. Lopez - LPC Clermont (fr)
  20 //-----------------------------------------------------------------------
  21
  22 #ifndef ALICFMUONRESTASK1_H
  23 #define ALICFMUONRESTASK1_H
  24
  25 #include "AliAnalysisTaskSE.h"
  26
  27 class TH1I;
  28 class TParticle ;
  29 class TLorentzVector ;
  30 class TFile ;
  31 class AliStack ;
  32 class AliCFManager;
  33 class AliESDtrack;
  34 class AliVParticle;
  35
  36 class AliCFMuonResTask1 : public AliAnalysisTaskSE {
  37   public:
  38
  39   AliCFMuonResTask1();
  40   AliCFMuonResTask1(const Char_t* name);
  41   AliCFMuonResTask1& operator= (const AliCFMuonResTask1& c);
  42   AliCFMuonResTask1(const AliCFMuonResTask1& c);
  43   virtual ~AliCFMuonResTask1();
  44
  45   // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
  46   void     UserExec(Option_t *option);
  47   void     Terminate(Option_t *);
  48
  49   // CORRECTION FRAMEWORK RELATED FUNCTIONS
  50   void           SetCFManager(AliCFManager* const io) {fCFManager = io;}   // global correction manager
  51   AliCFManager * GetCFManager() const {return fCFManager;}           // get corr manager
  52   void           SetQAList(TList* const list) {fQAHistList = list;}
  53
  54   // Data types
  55   Bool_t IsReadAODData()   const {return fReadAODData;}
  56   void   SetReadAODData   (Bool_t flag=kTRUE) {fReadAODData=flag;}
  57
  58  protected:
  59
  60   Bool_t          fReadAODData ;   // flag for AOD/ESD input files
  61   AliCFManager   *fCFManager   ;   // pointer to the CF manager
  62   TList          *fQAHistList  ;   // list of QA histograms
  63   TH1I           *fHistEventsProcessed; // simple histo for monitoring the number of events processed
  64   Double_t        fNevt        ;   // event counter
  65
  66   Float_t Imass(Float_t e1, Float_t px1, Float_t py1, Float_t pz1,
  67                 Float_t e2, Float_t px2, Float_t py2, Float_t p2) const;
  68   Float_t Rap(Float_t e, Float_t pz) const;
  69   Float_t Phideg(Float_t phi) const;
  70
  71   Double_t CostCS(Double_t px1, Double_t py1, Double_t pz1, Double_t e1, Double_t charge1,
  72                         Double_t px2, Double_t py2, Double_t pz2, Double_t e2, Double_t energy);
  73   Double_t CostHE(Double_t px1, Double_t py1, Double_t pz1, Double_t e1, Double_t charge1,
  74                         Double_t px2, Double_t py2, Double_t pz2, Double_t e2);
  75   Double_t PhiCS(Double_t px1, Double_t py1, Double_t pz1, Double_t e1, Double_t charge1,
  76                        Double_t px2, Double_t py2, Double_t pz2, Double_t e2, Double_t energy);
  77   Double_t PhiHE(Double_t px1, Double_t py1, Double_t pz1, Double_t e1, Double_t charge1,
  78                        Double_t px2, Double_t py2, Double_t pz2, Double_t e2, Double_t energy);
  79
  80   ClassDef(AliCFMuonResTask1,1);
  81 };
  82
  83 #endif