AliEmcalCopyCollection.h

Go to the documentation of this file.

#ifndef ALIEMCALCOPYCOLLECTION_H
#define ALIEMCALCOPYCOLLECTION_H

class AliVEvent;

#include "AliAnalysisTaskSE.h"
#include "AliEmcalCorrectionTask.h"

class AliEmcalCopyCollection : public AliAnalysisTaskSE {
 public:
  AliEmcalCopyCollection();
  AliEmcalCopyCollection(std::string name, AliEmcalCorrectionTask::InputObject_t inputObjectType, std::string collectionToCopyName, std::string newCollectionName, bool isEmbedding);
  virtual ~AliEmcalCopyCollection() {};

  // Methods from AliAnalysisTaskSE
  void UserCreateOutputObjects();
  void UserExec(Option_t * option);

  // Get methods
  AliEmcalCorrectionTask::InputObject_t GetInputObjectType() const { return fInputObjectType; }
  std::string GetCollectionToCopyName() const { return fCollectionToCopyName; }
  std::string GetNewCollectionName() const { return fNewCollectionName; }
  bool GetIsEmbedding() const { return fIsEmbedding; }

  // Set methods
  void SetInputObjectType(const AliEmcalCorrectionTask::InputObject_t inputObjectType) { fInputObjectType = inputObjectType; }
  void SetCollectionToCopyName(const std::string collectionToCopyName) { fCollectionToCopyName = collectionToCopyName; }
  void SetNewCollectionName(const std::string newCollectionName) { fNewCollectionName = newCollectionName; }
  void SetIsEmbedding(bool isEmbedding) { fIsEmbedding = isEmbedding; }

  // Add Task
  static AliEmcalCopyCollection* AddTaskEmcalCopyCollection(AliEmcalCorrectionTask::InputObject_t inputObjectType = AliEmcalCorrectionTask::kNoDefinedInputObject,
                          const char * collectionToCopyName = "",
                          const char * newCollectionName = "",
                          bool isEmbedding = false);

 protected:
  void                        CreateNewObjectBranch();
  void                        NewBranch();
  void                        CopyBranchToNewObject();
  void                        CopyClusters(TClonesArray *orig, TClonesArray *dest);

  AliEmcalCorrectionTask::InputObject_t fInputObjectType;  
  std::string fCollectionToCopyName;                       
  std::string fNewCollectionName;                          
  bool fIsEmbedding;                                       
  bool fEventInitialized;                                  
  bool fIsEsd;                                             
  AliVEvent * fEvent;                                      

  ClassDef(AliEmcalCopyCollection, 1); // Copy cell, cluster, or track collections
};

#endif /* ALIEMCALCOPYCOLLECTION_H */