AliEmcalCorrectionCellCombineCollections.h

Go to the documentation of this file.

#ifndef ALIEMCALCORRECTIONCELLCOMBINECOLLECTIONS_H
#define ALIEMCALCORRECTIONCELLCOMBINECOLLECTIONS_H

#include "AliEmcalCorrectionComponent.h"

class AliVCaloCells;

class AliEmcalCorrectionCellCombineCollections : public AliEmcalCorrectionComponent {
 public:
  AliEmcalCorrectionCellCombineCollections();
  virtual ~AliEmcalCorrectionCellCombineCollections();

  // Sets up and runs the task
  Bool_t Initialize();
  void UserCreateOutputObjects();
  void ExecOnce();
  Bool_t Run();

  std::string GetExternalCellsBranchName()                      const { return fExternalCellsBranchName; }
  std::string GetCombinedCellsBranchName()                      const { return fCreatedCellsBranchName; }
  
  void SetExternalCellsBranchName(const char * inputName)             { fExternalCellsBranchName = inputName; }
  void SetCombinedCellsBranchName(const char * inputName)             { fCreatedCellsBranchName = inputName; }

 protected:

  void SetupCombinedCells();
  void CreateCombinedCells();
  void AddCellsToCombinedCellObject(AliVCaloCells * inputCells);
  void AddObjectToEvent(TObject *obj, AliVEvent * event, Bool_t attempt = kFALSE);

  std::string fExternalCellsBranchName; 
  std::string fCreatedCellsBranchName;  
  bool fInitializedCombinedCells;       
  AliVCaloCells *fCombinedCells;        
  
  AliEmcalCorrectionCellCombineCollections(const AliEmcalCorrectionCellCombineCollections &);               // Not implemented
  AliEmcalCorrectionCellCombineCollections &operator=(const AliEmcalCorrectionCellCombineCollections &);    // Not implemented

  // Allows the registration of the class so that it is availble to be used by the correction task.
  static RegisterCorrectionComponent<AliEmcalCorrectionCellCombineCollections> reg;

  ClassDef(AliEmcalCorrectionCellCombineCollections, 1); // EMCal correction to combine cell collections
};

#endif /* AliEmcalCorrectionCellCombineCollections_H */