AliMergeableCollection.h

Go to the documentation of this file.

#ifndef ALIMERGEABLECOLLECTION_H
#define ALIMERGEABLECOLLECTION_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
* See cxx source for full Copyright notice                               */

// $Id: AliMergeableCollection.h 50593 2011-07-14 17:42:28Z martinez $


#include "TString.h"
#include "TFolder.h"
#include "TIterator.h"
#include "TCollection.h"
#include <map>
#include <string>

class TMap;
class AliMergeableCollectionIterator;
class AliMergeableCollectionProxy;
class TH1;
class TH2;
class TProfile;
class THashList;
class TCollection;

class AliMergeableCollection : public TFolder
{
  friend class AliMergeableCollectionIterator; // our iterator class
  friend class AliMergeableCollectionProxy; // out proxy class
  
public:

  AliMergeableCollection(const char* name="", const char* title="");
  virtual ~AliMergeableCollection();

  virtual AliMergeableCollection* Clone(const char* name="") const;
  
  Bool_t Attach(AliMergeableCollection* mc, const char* identifier, Bool_t pruneFirstIfAlreadyExists=kFALSE);
  
  Bool_t Adopt(TObject* obj);
  Bool_t Adopt(const char* identifier, TObject* obj);
  
  virtual void Browse(TBrowser* b);
  
  virtual void Clear(Option_t *option="") { Delete(option); }
  
  virtual TObject* FindObject(const char* fullIdentifier) const;

  virtual TObject* FindObject(const TObject* object) const;

  virtual void Delete(Option_t *option="");
  
  virtual Int_t NumberOfObjects() const;

  virtual Int_t NumberOfKeys() const;

  TObject* GetObject(const char* fullIdentifier) const;
  TObject* GetObject(const char* identifier, const char* objectName) const;

  TH1* Histo(const char* fullIdentifier) const;
  TH1* Histo(const char* identifier, const char* objectName) const;

  TH1* H1(const char* fullIdentifier) const { return Histo(fullIdentifier); }
  TH1* H1(const char* identifier, const char* objectName) const { return Histo(identifier,objectName); }

  TH2* H2(const char* fullIdentifier) const;
  TH2* H2(const char* identifier, const char* objectName) const;

  TProfile* Prof(const char* fullIdentifier) const;
  TProfile* Prof(const char* identifier, const char* objectName) const;

  virtual AliMergeableCollectionProxy* CreateProxy(const char* identifier, Bool_t createIfNeeded=kFALSE);
  
  virtual TIterator* CreateIterator(Bool_t dir = kIterForward) const;
  
  virtual TList* CreateListOfKeys(Int_t index) const;
  
  virtual TList* CreateListOfObjectNames(const char* identifier) const;
  
  using TFolder::Remove;
  
  virtual TObject* Remove(const char* fullIdentifier);
  
  Int_t RemoveByType(const char* typeName);
  
  TString GetKey(const char* identifier, Int_t index, Bool_t idContainsObjName = kFALSE) const;
  TString GetIdentifier(const char* fullIdentifier) const;
  TString GetObjectName(const char* fullIdentifier) const;
  
  void Print(Option_t *option="") const;
  
  void ClearMessages();
  void PrintMessages(const char* prefix="") const;
  
  Long64_t Merge(TCollection* list);
  
  AliMergeableCollection* Project(const char* identifier) const;
  
  UInt_t EstimateSize(Bool_t show=kFALSE) const;
  
  void ShowEmptyObjects(Bool_t show=kTRUE) {
    fMustShowEmptyObject = show;
  }
  
  void PruneEmptyObjects();
  
  Int_t Prune(const char* identifier);
  
  static Bool_t MergeObject(TObject* baseObject, TObject* objToAdd);
  
  TObject* GetSum(const char* idPattern) const;
  
  Bool_t IsEmptyObject(TObject* obj) const;
  
  static void CorrectIdentifier(TString& sidentifier);
  
private:
  
  AliMergeableCollection(const AliMergeableCollection& rhs);
  AliMergeableCollection& operator=(const AliMergeableCollection& rhs);
  
  TH1* HistoWithAction(const char* identifier, TObject* o, const TString& action) const;

  Bool_t InternalAdopt(const char* identifier, TObject* obj);
  
  TString InternalDecode(const char* fullIdentifier, Int_t index) const;
  
  TObject* InternalObject(const char* identifier, const char* objectName) const;
  
public:
  TObjArray* SortAllIdentifiers() const;

  TString NormalizeName(const char* identifier, const char* action) const;
  
  TMap* Map() const;

private:
  
  mutable TMap* fMap; 
  Bool_t fMustShowEmptyObject; 
  mutable Int_t fMapVersion; 
  mutable std::map<std::string,int> fMessages; 
  
  ClassDef(AliMergeableCollection,4) 
};

class AliMergeableCollectionIterator : public TIterator
{  
public:
  virtual ~AliMergeableCollectionIterator();
  
  AliMergeableCollectionIterator(const AliMergeableCollection* hcol, Bool_t direction=kIterForward);
  AliMergeableCollectionIterator& operator=(const TIterator &rhs);
  
  const TCollection *GetCollection() const { return 0x0; }

  TObject* Next();
  
  void Reset();
  
private:
  const AliMergeableCollection* fkMergeableCollection; // Mergeable objects collection being iterated
  TIterator* fMapIterator; // Iterator for the internal map
  TIterator* fHashListIterator; // Iterator for the current hash list
  Bool_t fDirection; // forward or reverse
  
  AliMergeableCollectionIterator() : fkMergeableCollection(0x0), fMapIterator(0x0), fHashListIterator(0x0), fDirection(kIterForward) {}
  
  AliMergeableCollectionIterator& operator=(const AliMergeableCollectionIterator &rhs);
  AliMergeableCollectionIterator(const AliMergeableCollectionIterator &iter);
    
  ClassDef(AliMergeableCollectionIterator,0)  // Mergeable object collection iterator
};

class AliMergeableCollectionProxy : public TFolder
{
  friend class AliMergeableCollection;
  
protected:
  AliMergeableCollectionProxy(AliMergeableCollection& oc, THashList& list);
  
public:
  
  TObject* GetObject(const char* objectName) const;
  
  TH1* Histo(const char* objectName) const;
  
  TH1* H1(const char* objectName) const { return Histo(objectName); }
  
  TH2* H2(const char* objectName) const;
  
  TProfile* Prof(const char* objectName) const;

  void Print(Option_t* opt="") const;
  
  Bool_t Adopt(TObject* obj);

  Bool_t Adopt(const char* identifier, TObject* obj);

  virtual TIterator* CreateIterator(Bool_t dir = kIterForward) const;

private:
  AliMergeableCollection& fOC;
  THashList& fList;

  ClassDef(AliMergeableCollectionProxy,0)  // Mergeable object collection proxy

};

#endif