[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisGrid.h

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

// Author: Mihaela Gheata, 01/09/2008

//==============================================================================
//   AliAnalysisGrid - Base grid utility class. Provides interface for creating
// a personalized JDL, finding and creating a dataset.
//==============================================================================

#ifndef ROOT_TNamed
#include <TNamed.h>
#endif

class TChain;

class AliAnalysisGrid : public TNamed {

public:

enum EPluginRunMode {
   kFull     = 0,
   kTest     = BIT(14),
   kOffline  = BIT(15),
   kSubmit   = BIT(16),
   kMerge    = BIT(17),
   kUseTags  = BIT(18),
   kUseESD   = BIT(19),
   kUseAOD   = BIT(20),
   kUseMC    = BIT(21),
   kUsePars  = BIT(22),
   kDefaultOutputs = BIT(23)
};   
enum EPluginBits {
   kBitMask32  = 0xffffffff,
   kUseCopy    = BIT(0),
   kCheckCopy  = BIT(1),
   kKeepLogs   = BIT(2),
   kClearPackages = BIT(3),
   kUseSubmitPolicy = BIT(4),
   kProofConnectGrid = BIT(5),
   kOneStageMerging = BIT(6),
   kUseMCchain = BIT(7),
   kLocalTest  = BIT(8),
   kMergeAOD   = BIT(9)
};

   AliAnalysisGrid() : TNamed(), fSpecialBits(0) {}
   AliAnalysisGrid(const char *name) : TNamed(name,""), fSpecialBits(0) {}
   virtual ~AliAnalysisGrid() {}
   AliAnalysisGrid(const AliAnalysisGrid& other); 
   AliAnalysisGrid& operator=(const AliAnalysisGrid& other);
// Getters
   virtual EPluginRunMode GetRunMode() const;
// Setters   
   virtual void        AddAdditionalLibrary(const char *name)            = 0;
   virtual void        AddIncludePath(const char *path)                  = 0;
   virtual void        AddRunNumber(Int_t run)                           = 0;
   virtual void        AddRunNumber(const char *run)                     = 0;
   virtual void        AddDataFile(const char *lfn)                      = 0;
   virtual Bool_t      IsSingleOutput() const                            = 0;
   virtual void        SetExecutable(const char *name="analysis.sh")     = 0;
   virtual void        SetArguments(const char *name="")                 = 0;
   virtual void        SetAnalysisMacro(const char *name="myAnalysis.C") = 0;
   virtual void        SetAnalysisSource(const char *name="myAnalysisClass.cxx") = 0;
   virtual void        SetValidationScript(const char *name="validation.sh")     = 0;
   virtual void        SetAdditionalLibs(const char *list)               = 0;
   virtual void        SetPrice(Int_t price=1)                           = 0;
   virtual void        SetJobTag(const char *tag="")                     = 0;
   virtual void        SetSplitMode(const char *type="se")               = 0;
   virtual void        SetSplitMaxInputFileNumber(Int_t nfiles=100)      = 0;
   virtual void        SetAPIVersion(const char *version)                = 0;
   virtual void        SetROOTVersion(const char *version)               = 0;
   virtual void        SetAliROOTVersion(const char *version)            = 0;
   virtual void        SetUser(const char *user)                         = 0;
   virtual void        SetTTL(Int_t ttl=30000)                           = 0;
   virtual void        SetGridWorkingDir(const char *name="workdir")     = 0;
   virtual void        SetGridDataDir(const char *name)                  = 0;
   virtual void        SetDataPattern(const char *pattern)               = 0;
   virtual void        SetDefaultOutputs(Bool_t flag=kTRUE)              = 0;
   virtual void        SetGridOutputDir(const char *name="output")       = 0;
   virtual void        SetOutputArchive(const char *list="log_archive.zip:std*@disk=1 root_archive.zip:*.root@disk=2") = 0;
   virtual void        SetOutputFiles(const char *list)                  = 0;
   virtual void        SetInputFormat(const char *format="xml-single")   = 0;
   virtual void        SetMaxInitFailed(Int_t nfail=5)                   = 0;
   virtual void        SetTerminateFiles(const char *list)               = 0;
   virtual void        SetMergeExcludes(const char *list)                = 0;
   virtual void        SetMergeViaJDL(Bool_t on=kTRUE)                   = 0;
   virtual void        SetMasterResubmitThreshold(Int_t percentage)      = 0;
   virtual void        SetNtestFiles(Int_t nfiles)                       = 0;
   virtual void        SetJDLName(const char *name="analysis.jdl")       = 0;
   virtual void        SetPreferedSE(const char *se)                     = 0;
   virtual void        SetProductionMode(Int_t mode=1)                   = 0;
   virtual void        SetRegisterExcludes(const char *list)             = 0;
   virtual void        SetRunPrefix(const char *prefix)                  = 0;
   virtual void        SetOutputSingleFolder(const char *folder)         = 0;
   virtual void        SetFastReadOption(Bool_t on=kTRUE)                = 0;
   virtual void        SetOverwriteMode(Bool_t on=kTRUE)                 = 0;
   
 // Set run mode.  Can be "full", "test", "offline", "submit" or "merge"
   virtual void        SetRunMode(const char *mode="full");
//Utilities
   static  Bool_t      CreateToken(const char *username=0);
   virtual Bool_t      CreateDataset(const char *pattern)                = 0;
   virtual Bool_t      CreateJDL()                                       = 0;
   virtual void        EnablePackage(const char *package)                = 0;
   virtual Bool_t      MergeOutputs()                                    = 0;
   virtual Bool_t      StartAnalysis(Long64_t nentries=123456789, Long64_t firstentry=0) = 0;
   virtual void        WriteAnalysisFile()                               = 0;
   virtual void        WriteAnalysisMacro()                              = 0;
   virtual void        WriteExecutable()                                 = 0;
   virtual void        WriteValidationScript(Bool_t merge=kFALSE)        = 0;

// Flags
   Bool_t              IsUseCopy() const {return TestSpecialBit(kUseCopy);}
   void                SetUseCopy(Bool_t flag=kTRUE) {SetSpecialBit(kUseCopy,flag);}
   Bool_t              IsCheckCopy() const {return TestSpecialBit(kCheckCopy);}
   void                SetCheckCopy(Bool_t flag=kTRUE) {SetSpecialBit(kCheckCopy,flag);}
   Bool_t              IsKeepLogs() const {return TestSpecialBit(kKeepLogs);}
   void                SetKeepLogs(Bool_t flag=kTRUE) {SetSpecialBit(kKeepLogs,flag);}   
   Bool_t              IsUseSubmitPolicy() const {return TestSpecialBit(kUseSubmitPolicy);}
   void                SetUseSubmitPolicy(Bool_t flag=kTRUE) {SetSpecialBit(kUseSubmitPolicy,flag);}   
   Bool_t              IsOneStageMerging() const {return TestSpecialBit(kOneStageMerging);}
   void                SetOneStageMerging(Bool_t flag) {SetSpecialBit(kOneStageMerging,flag);}
   Bool_t              IsUseMCchain() const {return TestSpecialBit(kUseMCchain);}
   void                SetUseMCchain(Bool_t flag=kTRUE) {SetSpecialBit(kUseMCchain,flag);}
   Bool_t              IsLocalTest() const {return TestSpecialBit(kLocalTest);}
   void                SetLocalTest(Bool_t flag=kTRUE) {SetSpecialBit(kLocalTest,flag);}
   Bool_t              IsMergeAOD() const {return TestSpecialBit(kMergeAOD);}
   virtual void        SetMergeAOD(Bool_t flag=kTRUE) {SetSpecialBit(kMergeAOD,flag);}

// PROOF mode
   virtual void        SetProofCluster(const char *cluster)              = 0;
   virtual void        SetProofDataSet(const char *dataset)              = 0;
   virtual const char *GetProofDataSet() const                           = 0;
   virtual void        SetProofReset(Int_t mode)                         = 0;
   virtual void        SetClearPackages(Bool_t flag=kTRUE) {SetSpecialBit(kClearPackages,flag);}
   virtual void        SetProofConnectGrid(Bool_t flag=kTRUE) {SetSpecialBit(kProofConnectGrid,flag);}
   virtual void        SetNproofWorkers(Int_t nworkers)                  = 0;
   virtual void        SetNproofWorkersPerSlave(Int_t nworkers)          = 0;
   virtual void        SetRootVersionForProof(const char *version)       = 0;
   virtual void        SetAliRootMode(const char *mode)                  = 0;
   virtual void        SetProofProcessOpt(const char *proofOpt="")       = 0;
   virtual TString     GetProofProcessOpt()                              = 0;
   // .txt file containing the list of files to be chained in test mode
   virtual void        SetFileForTestMode(const char *filename)          = 0;
   virtual TChain     *GetChainForTestMode(const char *treeName) const   = 0;

protected:
// Methods
   virtual Bool_t      Connect()                                         = 0;
   virtual void        SetDefaults()                                     = 0;
   void     SetSpecialBit(UInt_t f) { fSpecialBits |= f & kBitMask32; }
   void     ResetSpecialBit(UInt_t f) { fSpecialBits &= ~(f & kBitMask32); }
   void     SetSpecialBit(UInt_t f, Bool_t set) {(set)?SetSpecialBit(f):ResetSpecialBit(f);}
   Bool_t   TestSpecialBit(UInt_t f) const { return (Bool_t) ((fSpecialBits & f) != 0); }
   Int_t    TestSpecialBits(UInt_t f) const { return (Int_t) (fSpecialBits & f); }
   void     InvertSpecialBit(UInt_t f) { fSpecialBits ^= f & kBitMask32; }

protected:
   UInt_t              fSpecialBits; // special bits
  

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