Martin Valas fast mixing classes.

[u/mrichter/AliRoot.git] / ANALYSIS / EventMixing / AliMixEventPool.h

//
// Class AliMixEventPool
//
// AliMixEventPool is used to find 
// similar events
//
// author:
//        Martin Vala (martin.vala@cern.ch)
//

#ifndef ALIMIXEVENTPOOL_H
#define ALIMIXEVENTPOOL_H

#include <TObjArray.h>
#include <TNamed.h>

class TEntryList;
class AliMixEventCutObj;
class AliVEvent;
class AliMixEventPool : public TNamed
{
  public:
    AliMixEventPool(const char* name="mixEventPool", const char* title="Mix event pool");
    AliMixEventPool(const AliMixEventPool& obj);
    virtual ~AliMixEventPool();

    // prints object info
    virtual void      Print(const Option_t* option="") const;

    // inits correctly object
    Int_t       Init();
    
    void        CreateEntryListsRecursivly(Int_t index);
    void        SearchIndexRecursive(Int_t num, Int_t *i, Int_t *d, Int_t &index);
    TEntryList* AddEntryList();
    
    Bool_t      AddEntry(Long64_t entry, AliVEvent* ev);
    TEntryList *FindEntryList(AliVEvent *ev);
    
    void        AddCut(AliMixEventCutObj*cut);
    
    Bool_t      NeedInit() { return (fListOfEntryList.GetEntries() == 0); }
    TObjArray  *GetListOfEntryLists() { return &fListOfEntryList; }   
    TObjArray  *GetListOfEventCuts() { return &fListOfEventCuts; }
    
  private:

    TObjArray   fListOfEntryList;       // list of entry lists
    TObjArray   fListOfEventCuts;       // list of entry lists
    
    Int_t       fBinNumber;             // bin number
    
    AliMixEventPool& operator= (const AliMixEventPool&) { return *this; }

    ClassDef(AliMixEventPool, 1)
};

#endif