Add method to set the bad channel histogram per SM

diff --git a/PWG4/PartCorrBase/AliCalorimeterUtils.h b/PWG4/PartCorrBase/AliCalorimeterUtils.h
index 9b9327c528c4d90e5d8143203c4671e12f72edc6..f22cfabb62ee64069a827d5d53a498656b86ed35 100755 (executable)
--- a/PWG4/PartCorrBase/AliCalorimeterUtils.h
+++ b/PWG4/PartCorrBase/AliCalorimeterUtils.h
@@ -72,7 +72,8 @@ class AliCalorimeterUtils : public TObject {
   
   // Bad channels
   Bool_t IsBadChannelsRemovalSwitchedOn()  const { return fRemoveBadChannels ; }
-  void SwitchOnBadChannelsRemoval ()  {fRemoveBadChannels = kTRUE  ; fEMCALRecoUtils->SwitchOnBadChannelsRemoval(); InitPHOSBadChannelStatusMap();}
+  void SwitchOnBadChannelsRemoval ()  {fRemoveBadChannels = kTRUE  ; fEMCALRecoUtils->SwitchOnBadChannelsRemoval(); 
+    if(!fPHOSBadChannelMap) InitPHOSBadChannelStatusMap();}
   void SwitchOffBadChannelsRemoval()  {fRemoveBadChannels = kFALSE ; fEMCALRecoUtils->SwitchOffBadChannelsRemoval();}
   
   Bool_t IsDistanceToBadChannelRecalculated() const { return  IsDistanceToBadChannelRecalculated()         ; }
@@ -95,6 +96,9 @@ class AliCalorimeterUtils : public TObject {
        if(!fPHOSBadChannelMap) InitPHOSBadChannelStatusMap() ; 
        ((TH2I*)fPHOSBadChannelMap->At(imod))->SetBinContent(iCol,iRow,c);}
     
+  void SetEMCALChannelStatusMap(Int_t iSM , TH2I* h) {fEMCALRecoUtils->SetEMCALChannelStatusMap(iSM,h);}
+  void SetPHOSChannelStatusMap(Int_t imod , TH2I* h) {fPHOSBadChannelMap ->AddAt(h,imod);}
+  
   TH2I * GetEMCALChannelStatusMap(Int_t iSM) const {return fEMCALRecoUtils->GetEMCALChannelStatusMap(iSM);}
   TH2I * GetPHOSChannelStatusMap(Int_t imod) const {return (TH2I*)fPHOSBadChannelMap->At(imod);}