AliAnalysisTaskEMCALAlig.h

Go to the documentation of this file.

#ifndef AliAnalysisTaskEMCALAlig_H
#define AliAnalysisTaskEMCALAlig_H

//_________________________________________________________________________
// Matching residual for electrons on EMCal.
// Both default residual (until the surface) and a improved one, using
// electron mass hypotehsis and propagation until the cluster, are available
//_________________________________________________________________________


#include "AliAnalysisTaskEmcal.h"
#include "TObject.h"

class AliEMCALRecoUtils;
class AliPIDResponse;
class TH1F;
class TH2F;
class AliEMCALGeometry;
class TTree;

class ElectronForAlignment : public TObject {
public:
    
    //General Track properties
    Short_t charge;
    Float_t pt;
    Float_t pz;
    Float_t eta_track;
    Float_t phi_track;
    
    //cluster properties
    Float_t energy;
    Float_t M20;
    Float_t M02;
    Float_t eta_cluster;
    Float_t phi_cluster;
    UShort_t super_module_number;
    
    //mathing properties using default matcher
    Float_t x_resitual_def;
    Float_t y_resitual_def;
    Float_t z_resitual_def;
    Float_t phi_resitual_def;
    Float_t eta_resitual_def;
    
    //mathing properties using electron mass
    Float_t x_resitual_e;
    Float_t y_resitual_e;
    Float_t z_resitual_e;
    Float_t phi_resitual_e;
    Float_t eta_resitual_e;
    
    //PID properties
    Float_t n_sigma_electron_TPC;
    
    ElectronForAlignment()
    {
        //General Track properties
        charge = -9;
        pt = -999;
        pz = -999;
        eta_track = -999;
        phi_track = -999;
        
        //cluster properties
        energy = -999;
        M20 = -999;
        M02 = -999;
        eta_cluster = -999;
        phi_cluster = -999;
        
        //mathing properties using default matcher
        x_resitual_def = -999;
        y_resitual_def = -999;
        z_resitual_def = -999;
        phi_resitual_def = -999;
        eta_resitual_def = -999;
        
        //mathing properties using electron mass
        x_resitual_e = -999;
        y_resitual_e = -999;
        z_resitual_e = -999;
        phi_resitual_e = -999;
        eta_resitual_e = -999;
        
        super_module_number = 99;
        //PID properties
        n_sigma_electron_TPC = -999;
    }
    
    void Reset()
    {
        //General Track properties
        charge = -9;
        pt = -999;
        pz = -999;
        eta_track = -999;
        phi_track = -999;
        
        //cluster properties
        energy = -999;
        M20 = -999;
        M02 = -999;
        eta_cluster = -999;
        phi_cluster = -999;
        
        //mathing properties using default matcher
        x_resitual_def = -999;
        y_resitual_def = -999;
        z_resitual_def = -999;
        phi_resitual_def = -999;
        eta_resitual_def = -999;
        
        //mathing properties using electron mass
        x_resitual_e = -999;
        y_resitual_e = -999;
        z_resitual_e = -999;
        phi_resitual_e = -999;
        eta_resitual_e = -999;
        
        super_module_number = 99;
        //PID properties
        n_sigma_electron_TPC = -999;
    }
    
    //Default Initializer
    
    ClassDef(ElectronForAlignment, 2);
    
};


class AliAnalysisTaskEMCALAlig : public AliAnalysisTaskEmcal {
public:
    
    AliAnalysisTaskEMCALAlig();
    AliAnalysisTaskEMCALAlig(const char *name);
    virtual ~AliAnalysisTaskEMCALAlig();
    
    void UserCreateOutputObjects();
    void Terminate(Option_t *option);
    void SetSuffix(TString suff) { fTreeSuffix = suff;};
    
protected:
    void                        ExecOnce();
    Bool_t                      FillHistograms();
    Bool_t                      Run();
    
    void                        DoTrackLoop();
    
    
    AliEMCALRecoUtils *fEMCALRecoUtils; 
    AliEMCALGeometry *fEMCALGeo; 
    AliPIDResponse *fPIDResponse; 
    
    ElectronForAlignment fElectronInformation; 
    TTree* fElectronTree; 
    TString fTreeSuffix; // Suffix for tree name

    
private:
    AliAnalysisTaskEMCALAlig(const AliAnalysisTaskEMCALAlig&)           ; // not implemented
    AliAnalysisTaskEMCALAlig &operator=(const AliAnalysisTaskEMCALAlig&); // not implemented
    
  ClassDef(AliAnalysisTaskEMCALAlig, 3);

};
#endif