1 #ifndef ALIANALYSISDATACONTAINER_H
2 #define ALIANALYSISDATACONTAINER_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
7 // Author: Andrei Gheata, 31/05/2006
9 //==============================================================================
10 // AliAnalysysDataContainer - Container of data of arbitrary type deriving
11 // from TObject used for analysis. A container must be connected to the
12 // output data slot of a single analysis task (producer) , but also as
13 // input slot for possibly several other tasks (consumers). The connected
14 // slots must enforce the same data type as the container (or a derived type).
15 // A container becomes the owner of the contained data once this was produced.
16 //==============================================================================
29 class AliAnalysisTask;
30 class AliAnalysisDataWrapper;
33 class AliAnalysisDataContainer : public TNamed {
41 AliAnalysisDataContainer();
42 AliAnalysisDataContainer(const AliAnalysisDataContainer &cont);
43 AliAnalysisDataContainer(const char *name, TClass *type);
44 virtual ~AliAnalysisDataContainer();
47 AliAnalysisDataContainer &operator=(const AliAnalysisDataContainer &cont);
49 TObject *GetData() const {return fData;}
50 const char *GetFileName() const {return fFileName.Data();}
51 TClass *GetType() const;
52 AliAnalysisTask *GetProducer() const {return fProducer;}
53 TObjArray *GetConsumers() const {return fConsumers;}
54 virtual void GetEntry(Long64_t ientry);
56 void ResetDataReady() {fDataReady = kFALSE;}
57 virtual Bool_t SetData(TObject *data, Option_t *option="");
58 void SetDataOwned(Bool_t flag) {fOwnedData = flag;}
59 void SetFileName(const char *filename) {fFileName = filename;}
60 void SetProducer(AliAnalysisTask *prod, Int_t islot);
61 void AddConsumer(AliAnalysisTask *cons, Int_t islot);
64 AliAnalysisDataWrapper *ExportData() const;
65 void ImportData(AliAnalysisDataWrapper *pack);
66 // Container status checking
67 Bool_t IsDataReady() const {return fDataReady;}
68 Bool_t IsOwnedData() const {return fOwnedData;}
69 Bool_t ClientsExecuted() const;
70 Bool_t HasConsumers() const {return (fConsumers != 0);}
71 Bool_t HasProducer() const {return (fProducer != 0);}
73 virtual Long64_t Merge(TCollection *list);
74 // Send a notify signal to the container
75 virtual void NotifyChange(ENotifyMessage /*type*/) {;}
76 // Print connected tasks/status
77 void PrintContainer(Option_t *option="all", Int_t indent=0) const;
80 void SetType(TClass *type) {fType = type;}
83 Bool_t fDataReady; // Flag that data is ready
84 Bool_t fOwnedData; // Flag data ownership
85 TString fFileName; // File storing the data
86 TObject *fData; // Contained data
87 TClass *fType; //! Type of contained data
88 AliAnalysisTask *fProducer; // Analysis task to which the slot belongs
89 TObjArray *fConsumers; // List of consumers of the data
91 ClassDef(AliAnalysisDataContainer,1) // Class describing a data container for analysis
94 //==============================================================================
95 // AliAnalysysDataWrapper - A basic wrapper for exchanging via the network
96 // the data held by AliAnalysisDataContainer between the master and the client
98 //==============================================================================
100 class AliAnalysisDataWrapper : public TNamed {
103 AliAnalysisDataWrapper() : TNamed(), fData(NULL) {}
104 AliAnalysisDataWrapper(TObject *data) : TNamed(), fData(data) {}
105 AliAnalysisDataWrapper(const AliAnalysisDataWrapper &other)
106 : TNamed(other), fData(other.fData) {}
107 virtual ~AliAnalysisDataWrapper() {}
110 AliAnalysisDataWrapper &operator=(const AliAnalysisDataWrapper &other);
112 TObject *Data() const {return fData;}
114 virtual Long64_t Merge(TCollection *list);
117 TObject *fData; // Wrapped data
119 ClassDef(AliAnalysisDataWrapper, 1) // Data wrapper class for exchange via the net