Added loop for extraction of clusters really attached to its track.
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisGrid.h
CommitLineData
c57f56b7 1#ifndef ALIANALYSISGRID_H
2#define ALIANALYSISGRID_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6// Author: Mihaela Gheata, 01/09/2008
7
8//==============================================================================
9// AliAnalysisGrid - Base grid utility class. Provides interface for creating
10// a personalized JDL, finding and creating a dataset.
11//==============================================================================
12
13#ifndef ROOT_TNamed
14#include <TNamed.h>
15#endif
16
3bdcb562 17class TChain;
18
c57f56b7 19class AliAnalysisGrid : public TNamed {
20
21public:
22
23enum EPluginRunMode {
24 kFull = 0,
25 kTest = BIT(14),
26 kOffline = BIT(15),
27 kSubmit = BIT(16),
28 kMerge = BIT(17),
29 kUseTags = BIT(18),
30 kUseESD = BIT(19),
31 kUseAOD = BIT(20),
4e5c5506 32 kUseMC = BIT(21),
bb885a9e 33 kUsePars = BIT(22),
a3e84053 34 kDefaultOutputs = BIT(23)
c57f56b7 35};
348be253 36enum EPluginBits {
37 kBitMask32 = 0xffffffff,
38 kUseCopy = BIT(0),
149d288c 39 kCheckCopy = BIT(1),
3bdcb562 40 kKeepLogs = BIT(2),
41 kClearPackages = BIT(3),
27734f0e 42 kUseSubmitPolicy = BIT(4),
d3b18c4c 43 kProofConnectGrid = BIT(5),
c85cfc0f 44 kOneStageMerging = BIT(6),
21ca8e59 45 kUseMCchain = BIT(7),
46 kLocalTest = BIT(8)
348be253 47};
c57f56b7 48
348be253 49 AliAnalysisGrid() : TNamed(), fSpecialBits(0) {}
50 AliAnalysisGrid(const char *name) : TNamed(name,""), fSpecialBits(0) {}
c57f56b7 51 virtual ~AliAnalysisGrid() {}
52 AliAnalysisGrid(const AliAnalysisGrid& other);
53 AliAnalysisGrid& operator=(const AliAnalysisGrid& other);
54// Getters
55 virtual EPluginRunMode GetRunMode() const;
56// Setters
37d21c01 57 virtual void AddAdditionalLibrary(const char *name) = 0;
f965131e 58 virtual void AddIncludePath(const char *path) = 0;
c57f56b7 59 virtual void AddRunNumber(Int_t run) = 0;
ee75cfc3 60 virtual void AddRunNumber(const char *run) = 0;
c57f56b7 61 virtual void AddDataFile(const char *lfn) = 0;
fe2d7fc2 62 virtual Bool_t IsSingleOutput() const = 0;
c57f56b7 63 virtual void SetExecutable(const char *name="analysis.sh") = 0;
64 virtual void SetArguments(const char *name="") = 0;
65 virtual void SetAnalysisMacro(const char *name="myAnalysis.C") = 0;
66 virtual void SetAnalysisSource(const char *name="myAnalysisClass.cxx") = 0;
d3b18c4c 67 virtual void SetValidationScript(const char *name="validation.sh") = 0;
c57f56b7 68 virtual void SetAdditionalLibs(const char *list) = 0;
69 virtual void SetPrice(Int_t price=1) = 0;
c6cb3634 70 virtual void SetJobTag(const char *tag="") = 0;
c57f56b7 71 virtual void SetSplitMode(const char *type="se") = 0;
72 virtual void SetSplitMaxInputFileNumber(Int_t nfiles=100) = 0;
73 virtual void SetAPIVersion(const char *version) = 0;
74 virtual void SetROOTVersion(const char *version) = 0;
75 virtual void SetAliROOTVersion(const char *version) = 0;
76 virtual void SetUser(const char *user) = 0;
77 virtual void SetTTL(Int_t ttl=30000) = 0;
78 virtual void SetGridWorkingDir(const char *name="workdir") = 0;
79 virtual void SetGridDataDir(const char *name) = 0;
80 virtual void SetDataPattern(const char *pattern) = 0;
bb885a9e 81 virtual void SetDefaultOutputs(Bool_t flag=kTRUE) = 0;
c57f56b7 82 virtual void SetGridOutputDir(const char *name="output") = 0;
149d288c 83 virtual void SetOutputArchive(const char *list="log_archive.zip:std*@disk=1 root_archive.zip:*.root@disk=2") = 0;
c57f56b7 84 virtual void SetOutputFiles(const char *list) = 0;
85 virtual void SetInputFormat(const char *format="xml-single") = 0;
86 virtual void SetMaxInitFailed(Int_t nfail=5) = 0;
d3b18c4c 87 virtual void SetTerminateFiles(const char *list) = 0;
bb885a9e 88 virtual void SetMergeExcludes(const char *list) = 0;
0f389141 89 virtual void SetMergeViaJDL(Bool_t on=kTRUE) = 0;
c57f56b7 90 virtual void SetMasterResubmitThreshold(Int_t percentage) = 0;
bb885a9e 91 virtual void SetNtestFiles(Int_t nfiles) = 0;
c57f56b7 92 virtual void SetJDLName(const char *name="analysis.jdl") = 0;
43da816a 93 virtual void SetPreferedSE(const char *se) = 0;
0f389141 94 virtual void SetProductionMode(Int_t mode=1) = 0;
37d21c01 95 virtual void SetRegisterExcludes(const char *list) = 0;
0f389141 96 virtual void SetRunPrefix(const char *prefix) = 0;
97 virtual void SetOutputSingleFolder(const char *folder) = 0;
98 virtual void SetFastReadOption(Bool_t on=kTRUE) = 0;
99 virtual void SetOverwriteMode(Bool_t on=kTRUE) = 0;
c57f56b7 100
101 // Set run mode. Can be "full", "test", "offline", "submit" or "merge"
102 virtual void SetRunMode(const char *mode="full");
103//Utilities
104 static Bool_t CreateToken(const char *username=0);
105 virtual Bool_t CreateDataset(const char *pattern) = 0;
106 virtual Bool_t CreateJDL() = 0;
4e5c5506 107 virtual void EnablePackage(const char *package) = 0;
c57f56b7 108 virtual Bool_t MergeOutputs() = 0;
5513444a 109 virtual Bool_t StartAnalysis(Long64_t nentries=123456789, Long64_t firstentry=0) = 0;
c57f56b7 110 virtual void WriteAnalysisFile() = 0;
111 virtual void WriteAnalysisMacro() = 0;
112 virtual void WriteExecutable() = 0;
0f389141 113 virtual void WriteValidationScript(Bool_t merge=kFALSE) = 0;
c57f56b7 114
348be253 115// Flags
116 Bool_t IsUseCopy() const {return TestSpecialBit(kUseCopy);}
117 void SetUseCopy(Bool_t flag=kTRUE) {SetSpecialBit(kUseCopy,flag);}
118 Bool_t IsCheckCopy() const {return TestSpecialBit(kCheckCopy);}
119 void SetCheckCopy(Bool_t flag=kTRUE) {SetSpecialBit(kCheckCopy,flag);}
149d288c 120 Bool_t IsKeepLogs() const {return TestSpecialBit(kKeepLogs);}
121 void SetKeepLogs(Bool_t flag=kTRUE) {SetSpecialBit(kKeepLogs,flag);}
3bdcb562 122 Bool_t IsUseSubmitPolicy() const {return TestSpecialBit(kUseSubmitPolicy);}
123 void SetUseSubmitPolicy(Bool_t flag=kTRUE) {SetSpecialBit(kUseSubmitPolicy,flag);}
d3b18c4c 124 Bool_t IsOneStageMerging() const {return TestSpecialBit(kOneStageMerging);}
125 void SetOneStageMerging(Bool_t flag) {SetSpecialBit(kOneStageMerging,flag);}
c85cfc0f 126 Bool_t IsUseMCchain() const {return TestSpecialBit(kUseMCchain);}
127 void SetUseMCchain(Bool_t flag=kTRUE) {SetSpecialBit(kUseMCchain,flag);}
21ca8e59 128 Bool_t IsLocalTest() const {return TestSpecialBit(kLocalTest);}
129 void SetLocalTest(Bool_t flag=kTRUE) {SetSpecialBit(kLocalTest,flag);}
3bdcb562 130
131// PROOF mode
132 virtual void SetProofCluster(const char *cluster) = 0;
133 virtual void SetProofDataSet(const char *dataset) = 0;
134 virtual const char *GetProofDataSet() const = 0;
135 virtual void SetProofReset(Int_t mode) = 0;
136 virtual void SetClearPackages(Bool_t flag=kTRUE) {SetSpecialBit(kClearPackages,flag);}
27734f0e 137 virtual void SetProofConnectGrid(Bool_t flag=kTRUE) {SetSpecialBit(kProofConnectGrid,flag);}
3bdcb562 138 virtual void SetNproofWorkers(Int_t nworkers) = 0;
27734f0e 139 virtual void SetNproofWorkersPerSlave(Int_t nworkers) = 0;
3bdcb562 140 virtual void SetRootVersionForProof(const char *version) = 0;
141 virtual void SetAliRootMode(const char *mode) = 0;
d88e4159 142 virtual void SetProofProcessOpt(const char *proofOpt="") = 0;
143 virtual TString GetProofProcessOpt() = 0;
3bdcb562 144 // .txt file containing the list of files to be chained in test mode
145 virtual void SetFileForTestMode(const char *filename) = 0;
146 virtual TChain *GetChainForTestMode(const char *treeName) const = 0;
348be253 147
c57f56b7 148protected:
348be253 149// Methods
c57f56b7 150 virtual Bool_t Connect() = 0;
348be253 151 virtual void SetDefaults() = 0;
152 void SetSpecialBit(UInt_t f) { fSpecialBits |= f & kBitMask32; }
153 void ResetSpecialBit(UInt_t f) { fSpecialBits &= ~(f & kBitMask32); }
154 void SetSpecialBit(UInt_t f, Bool_t set) {(set)?SetSpecialBit(f):ResetSpecialBit(f);}
155 Bool_t TestSpecialBit(UInt_t f) const { return (Bool_t) ((fSpecialBits & f) != 0); }
156 Int_t TestSpecialBits(UInt_t f) const { return (Int_t) (fSpecialBits & f); }
157 void InvertSpecialBit(UInt_t f) { fSpecialBits ^= f & kBitMask32; }
158
159protected:
160 UInt_t fSpecialBits; // special bits
161
162
163 ClassDef(AliAnalysisGrid, 2) // Base class for GRID utilities
c57f56b7 164};
165#endif