AliEmcalContainerUtils.h

Go to the documentation of this file.

#ifndef ALIEMCALCONTAINERUTILS_H
#define ALIEMCALCONTAINERUTILS_H

#include <string>

#include <TObjArray.h>

class AliVEvent;

class AliEmcalContainerUtils {
 public:
  enum InputObject_t {
    kNoDefinedInputObject = -1,    
    kCaloCells = 0,                
    kCluster = 1,                  
    kTrack = 2,                    
  };

  // Utility functions
  static std::string DetermineUseDefaultName(InputObject_t objType, bool esdMode, bool returnObjectType = false);
  static const AliVEvent * GetEvent(const AliVEvent * inputEvent, bool isEmbedding = false);
  static AliVEvent * GetEvent(AliVEvent * inputEvent, bool isEmbedding = false);

#if !(defined(__CINT__) || defined(__MAKECINT__))
  template <class T>
  static inline T * AddContainer(const char *n, TObjArray & collection);
  template <class T>
  static T * GetContainer(Int_t i, const TObjArray & collection);
  template <class T>
  static T * GetContainer(const char *name, const TObjArray & collection);

  // Then to use these functions, add below (for example) to each class (or you can use it directly if desired)
  /*AliMCParticleContainer * AddMCParticleContainer(const char * name) { AddContainer<AliMCPartileContainer>(name, fParticleCollArray); }
  AliTrackContainer * AddTrackContainer(const char * name)        { AddContainer<AliTrackContainer>(name, fParticleCollArray); }
  AliParticleContainer * AddTrackContainer(const char * name)     { AddContainer<AliParticleContainer>(name, fParticleCollArray); }
  */
#endif /* Hide from CINT */
};

#if !(defined(__CINT__) || defined(__MAKECINT__))

template <class T>
inline T * AliEmcalContainerUtils::AddContainer(const char *n, TObjArray & collection)
{
  if (TString(n).IsNull()) return 0;

  T * cont = new T(n);

  collection.Add(cont);

  return cont;
}

template <class T>
inline T * AliEmcalContainerUtils::GetContainer(Int_t i, const TObjArray & collection)
{
  if (i<0 || i>collection.GetEntriesFast()) return 0;
  T *cont = static_cast<T *>(collection.At(i));
  return cont;
}

template <class T>
inline T * AliEmcalContainerUtils::GetContainer(const char *name, const TObjArray & collection)
{
  T *cont = static_cast<T *>(collection.FindObject(name));
  return cont;
}
#endif /* Hide from CINT */

#endif /* AliEmcalContainerUtils.h */