Modified macros for TOF analysis of K* in pA for possibility to use daughter's pt...

[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / AliRsnLoop.h

#ifndef ALIRSNLOOP_H
#define ALIRSNLOOP_H

//
// Base class to implement any computation within the RSN package.
// It contains only an array of output objects which must derive
// from AliRsnOutput.
// Its core functions ar Init() and DoLoop() which must be
// overloaded by any class which inherits from this.
//

#include "TNamed.h"
#include "TObjArray.h"

#include "AliRsnListOutput.h"
#include "AliRsnValue.h"
#include "AliRsnCutSet.h"

class TList;
class AliRsnEvent;
class AliRsnListOutput;
class AliRsnDaughterSelector;

class AliRsnLoop : public TNamed {
public:

   enum EOut {
      kH1,
      kHSparse,
      kNtuple
   };

   AliRsnLoop(const char *name = "default", Bool_t isMixed = kFALSE);
   AliRsnLoop(const AliRsnLoop &copy);
   AliRsnLoop &operator=(const AliRsnLoop &copy);
   ~AliRsnLoop();

   void           SetMixed(Bool_t yn = kTRUE)     {fIsMixed = yn;}
   void           SetEventCuts(AliRsnCutSet *set) {fEventCuts = set;}
   Bool_t         IsMixed() const                 {return fIsMixed;}
   AliRsnCutSet  *GetEventCuts()                  {return fEventCuts;}
   Bool_t         OkEvent(AliRsnEvent *rsn);

   virtual void   AddOutput(TObject *output);
   virtual void   Print(Option_t *option = "") const;
   virtual Bool_t Init(const char *prefix, TList *list);
   virtual Int_t  DoLoop(AliRsnEvent *main, AliRsnDaughterSelector *smain, AliRsnEvent *mix = 0, AliRsnDaughterSelector *smix = 0);

protected:

   Bool_t         fIsMixed;    //  flag to know if the loop works with event mixing
   AliRsnCutSet  *fEventCuts;  //  event cuts
   TClonesArray   fOutputs;    //  output object definitions

private:

   ClassDef(AliRsnLoop, 1)
};

#endif