[u/mrichter/AliRoot.git] / STEER / AliTagAnalysis.cxx

/**************************************************************************
 * Author: Panos Christakoglou.                                           *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//-----------------------------------------------------------------
//           AliTagAnalysis class
//   This is the class to deal with the tag analysis
//   Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
//-----------------------------------------------------------------

//ROOT
#include <TSystem.h>
#include <TChain.h>
#include <TFile.h>
#include <TEventList.h>
#include <TEntryList.h>
#include <TTreeFormula.h>

//ROOT-AliEn
#include <TGridResult.h>

#include "AliLog.h"

#include "AliRunTag.h"
#include "AliEventTag.h"
#include "AliTagAnalysis.h"
#include "AliEventTagCuts.h"
#include "AliRunTagCuts.h"
#include "AliXMLCollection.h"

class TTree;

ClassImp(AliTagAnalysis)

//___________________________________________________________________________
AliTagAnalysis::AliTagAnalysis(): 
  TObject(),
  ftagresult(0x0),
  fTagDirName(),
  fChain(0x0)
{
  //Default constructor for a AliTagAnalysis
}

//___________________________________________________________________________
AliTagAnalysis::~AliTagAnalysis() {
//Default destructor for a AliTagAnalysis
}

//___________________________________________________________________________
Bool_t  AliTagAnalysis::AddTagsFile(const char *alienUrl) {

  // Add a single tags file to the chain

  Bool_t rv = kTRUE ;

  if (! fChain) fChain = new TChain("T");

  TFile *f = TFile::Open(alienUrl,"READ");
  fChain->Add(alienUrl);
  AliInfo(Form("Chained tag files: %d ",fChain->GetEntries()));
  delete f;

  if (fChain->GetEntries() == 0 )
    rv = kFALSE ;

  return rv ;
}

//___________________________________________________________________________
void AliTagAnalysis::ChainLocalTags(const char *dirname) {
  //Searches the entries of the provided direcory
  //Chains the tags that are stored locally
  fTagDirName = dirname;
  TString fTagFilename;
  
  if (! fChain)  fChain = new TChain("T");
  
  const char * tagPattern = "tag.root";
  // Open the working directory
  void * dirp = gSystem->OpenDirectory(fTagDirName);
  const char * name = 0x0;
  // Add all files matching *pattern* to the chain
  while((name = gSystem->GetDirEntry(dirp))) {
    if (strstr(name,tagPattern)) { 
      fTagFilename = fTagDirName;
      fTagFilename += "/";
      fTagFilename += name;
	  	
      fChain->Add(fTagFilename);  
    }//pattern check
  }//directory loop
  AliInfo(Form("Chained tag files: %d ",fChain->GetEntries()));
}


//___________________________________________________________________________
void AliTagAnalysis::ChainGridTags(TGridResult *res) {
  //Loops overs the entries of the TGridResult
  //Chains the tags that are stored in the GRID
  ftagresult = res;
  Int_t nEntries = ftagresult->GetEntries();
 
  if (! fChain)  fChain = new TChain("T");

  TString gridname = "alien://";
  TString alienUrl;
 
  for(Int_t i = 0; i < nEntries; i++) {
    alienUrl = ftagresult->GetKey(i,"turl");
    fChain->Add(alienUrl);
  }//grid result loop  
}


//___________________________________________________________________________
TChain *AliTagAnalysis::QueryTags(AliRunTagCuts *RunTagCuts, AliEventTagCuts *EvTagCuts) {
  //Queries the tag chain using the defined 
  //event tag cuts from the AliEventTagCuts object
  //and returns a TChain along with the associated TEventList
  AliInfo(Form("Querying the tags........"));
  
  //ESD file chain
  TChain *fESDchain = new TChain("esdTree");
  //Event list
  TEventList *fEventList = new TEventList();
  
  //Defining tag objects
  AliRunTag *tag = new AliRunTag;
  AliEventTag *evTag = new AliEventTag;
  fChain->SetBranchAddress("AliTAG",&tag);

  TString guid = 0;
  TString turl = 0;
  TString path = 0;

  Int_t iAccepted = 0;
  for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
    fChain->GetEntry(iTagFiles);
    if(RunTagCuts->IsAccepted(tag)) {
      Int_t iEvents = tag->GetNEvents();
      const TClonesArray *tagList = tag->GetEventTags();
      for(Int_t i = 0; i < iEvents; i++) {
	evTag = (AliEventTag *) tagList->At(i);
	guid = evTag->GetGUID(); 
	turl = evTag->GetTURL(); 
	path = evTag->GetPath();
	if(EvTagCuts->IsAccepted(evTag)) fEventList->Enter(iAccepted+i);
      }//event loop
      iAccepted += iEvents;
    
      if(path != "") fESDchain->AddFile(path);
      else if(turl != "") fESDchain->AddFile(turl);
    }//run tags cut
  }//tag file loop
  AliInfo(Form("Accepted events: %d",fEventList->GetN()));
  fESDchain->SetEventList(fEventList);
   
  return fESDchain;
}

//___________________________________________________________________________
TChain *AliTagAnalysis::QueryTags(const char *fRunCut, const char *fEventCut) { 	 
  //Queries the tag chain using the defined 	 
  //event tag cuts from the AliEventTagCuts object 	 
  //and returns a TChain along with the associated TEventList 	 
  AliInfo(Form("Querying the tags........")); 	 
  
  //ESD file chain 	 
  TChain *fESDchain = new TChain("esdTree"); 	 
  //Event list 	 
  TEventList *fEventList = new TEventList(); 	 
  
  //Defining tag objects 	 
  AliRunTag *tag = new AliRunTag; 	 
  AliEventTag *evTag = new AliEventTag; 	 
  fChain->SetBranchAddress("AliTAG",&tag); 	 
  
  TString guid = 0; 	 
  TString turl = 0; 	 
  TString path = 0; 	 
  
  TTreeFormula *fRunFormula = new TTreeFormula("fRun",fRunCut,fChain); 	 
  TTreeFormula *fEventFormula = new TTreeFormula("fEvent",fEventCut,fChain); 	 
  
  Int_t current = -1; 	 
  Int_t iAccepted = 0; 	 
  for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) { 	 
    fChain->GetEntry(iTagFiles); 	 
    if (current != fChain->GetTreeNumber()) { 	 
      fRunFormula->UpdateFormulaLeaves(); 	 
      fEventFormula->UpdateFormulaLeaves(); 	 
      current = fChain->GetTreeNumber(); 	 
    } 	 
    if(fRunFormula->EvalInstance(iTagFiles) == 1) { 	 
      Int_t iEvents = fEventFormula->GetNdata(); 	 
      const TClonesArray *tagList = tag->GetEventTags(); 	 
      for(Int_t i = 0; i < iEvents; i++) { 	 
	evTag = (AliEventTag *) tagList->At(i); 	 
	guid = evTag->GetGUID(); 	 
	turl = evTag->GetTURL(); 	 
	path = evTag->GetPath(); 	 
	if(fEventFormula->EvalInstance(i) == 1) fEventList->Enter(iAccepted+i); 	 
      }//event loop 	 
      iAccepted += iEvents; 	 
      
      if(path != "") fESDchain->AddFile(path); 	 
      else if(turl != "") fESDchain->AddFile(turl); 	 
    }//run tag cut 	 
  }//tag file loop 	 
  AliInfo(Form("Accepted events: %d",fEventList->GetN())); 	 
  fESDchain->SetEventList(fEventList); 	 
  
  return fESDchain; 	 
}

//___________________________________________________________________________
Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, AliRunTagCuts *RunTagCuts, AliEventTagCuts *EvTagCuts) {
  //Queries the tag chain using the defined 
  //event tag cuts from the AliEventTagCuts object
  //and returns a XML collection
  AliInfo(Form("Creating the collection........"));

  AliXMLCollection *collection = new AliXMLCollection();
  collection->SetCollectionName(name);
  collection->WriteHeader();

  //Event list
  //TEntryList *fEventList = new TEntryList();
  TString guid = 0x0;
  TString turl = 0x0;
  TString lfn = 0x0;
  
  //Defining tag objects
  AliRunTag *tag = new AliRunTag;
  AliEventTag *evTag = new AliEventTag;
  fChain->SetBranchAddress("AliTAG",&tag);

  for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
    //Event list
    TEntryList *fList = new TEntryList();
    fChain->GetEntry(iTagFiles);
    if(RunTagCuts->IsAccepted(tag)) {
      Int_t iEvents = tag->GetNEvents();
      const TClonesArray *tagList = tag->GetEventTags();
      for(Int_t i = 0; i < iEvents; i++) {
	evTag = (AliEventTag *) tagList->At(i);
	guid = evTag->GetGUID(); 
	turl = evTag->GetTURL(); 
	lfn = turl(8,turl.Length());
	if(EvTagCuts->IsAccepted(evTag)) fList->Enter(i);
      }//event loop
      collection->WriteBody(iTagFiles+1,guid,lfn,turl,fList);
    }//run tag cuts
  }//tag file loop
  collection->Export();

  return kTRUE;
}

//___________________________________________________________________________
Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, const char *fRunCut, const char *fEventCut) {
  //Queries the tag chain using the defined 
  //event tag cuts from the AliEventTagCuts object
  //and returns a XML collection
  AliInfo(Form("Creating the collection........"));

  AliXMLCollection *collection = new AliXMLCollection();
  collection->SetCollectionName(name);
  collection->WriteHeader();

  TString guid = 0x0;
  TString turl = 0x0;
  TString lfn = 0x0;
  
  //Defining tag objects
  AliRunTag *tag = new AliRunTag;
  AliEventTag *evTag = new AliEventTag;
  fChain->SetBranchAddress("AliTAG",&tag);

  TTreeFormula *fRunFormula = new TTreeFormula("fRun",fRunCut,fChain);
  TTreeFormula *fEventFormula = new TTreeFormula("fEvent",fEventCut,fChain);

  Int_t current = -1;
  for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
    //Event list
    TEntryList *fList = new TEntryList();
    fChain->GetEntry(iTagFiles);
    if (current != fChain->GetTreeNumber()) {
      fRunFormula->UpdateFormulaLeaves();
      fEventFormula->UpdateFormulaLeaves();
      current = fChain->GetTreeNumber();
    }
    if(fRunFormula->EvalInstance(iTagFiles) == 1) {
      Int_t iEvents = fEventFormula->GetNdata();
      const TClonesArray *tagList = tag->GetEventTags();
      for(Int_t i = 0; i < iEvents; i++) {
	evTag = (AliEventTag *) tagList->At(i);
	guid = evTag->GetGUID(); 
	turl = evTag->GetTURL(); 
	lfn = turl(8,turl.Length());
	if(fEventFormula->EvalInstance(i) == 1) fList->Enter(i);
      }//event loop
      collection->WriteBody(iTagFiles+1,guid,lfn,turl,fList);
    }//run tag cuts
  }//tag file loop
  collection->Export();

  return kTRUE;
}

//___________________________________________________________________________
TChain *AliTagAnalysis::GetInputChain(const char* system, const char*global, const char *wn) {
  //returns the chain+event list - used in batch sessions
  TString fsystem = system;
  Int_t iAccepted = 0;
  //ESD file chain
  TChain *fESDchain = new TChain("esdTree");
  //Event list
  TEventList *fEventList = new TEventList();
  AliXMLCollection *collection1 = AliXMLCollection::Open(global);
  AliXMLCollection *collection2 = AliXMLCollection::Open(wn);
  collection1->OverlapCollection(collection2);
  collection1->Reset();
  while (collection1->Next()) {
    AliInfo(Form("Adding: %s",collection1->GetTURL("")));
    fESDchain->Add(collection1->GetTURL(""));
    TEntryList *list = (TEntryList *)collection1->GetEventList("");
    for(Int_t i = 0; i < list->GetN(); i++) fEventList->Enter(iAccepted+list->GetEntry(i));

    if(fsystem == "pp") iAccepted += 100;
    else if(fsystem == "PbPb") iAccepted += 1;
  }

  fESDchain->SetEventList(fEventList);
  
  AliInfo(Form("Number of selected events: %d",fEventList->GetN()));

  return fESDchain;
}
Commit	Line	Data
c7e89ea3	1	/**************************************************************************
	2	* Author: Panos Christakoglou. *
	3	* Contributors are mentioned in the code where appropriate. *
	4	* *
	5	* Permission to use, copy, modify and distribute this software and its *
	6	* documentation strictly for non-commercial purposes is hereby granted *
	7	* without fee, provided that the above copyright notice appears in all *
	8	* copies and that both the copyright notice and this permission notice *
	9	* appear in the supporting documentation. The authors make no claims *
	10	* about the suitability of this software for any purpose. It is *
	11	* provided "as is" without express or implied warranty. *
	12	**************************************************************************/
	13
	14	/* $Id$ */
	15
	16	//-----------------------------------------------------------------
	17	// AliTagAnalysis class
	18	// This is the class to deal with the tag analysis
	19	// Origin: Panos Christakoglou, UOA-CERN, Panos.Christakoglou@cern.ch
	20	//-----------------------------------------------------------------
	21
	22	//ROOT
	23	#include <TSystem.h>
	24	#include <TChain.h>
2f514327	25	#include <TFile.h>
c7e89ea3	26	#include <TEventList.h>
2f514327	27	#include <TEntryList.h>
5b7be856	28	#include <TTreeFormula.h>
c7e89ea3	29
	30	//ROOT-AliEn
	31	#include <TGridResult.h>
	32
	33	#include "AliLog.h"
	34
	35	#include "AliRunTag.h"
	36	#include "AliEventTag.h"
	37	#include "AliTagAnalysis.h"
	38	#include "AliEventTagCuts.h"
a3acd4e8	39	#include "AliRunTagCuts.h"
b6003316	40	#include "AliXMLCollection.h"
c7e89ea3	41
2f514327	42	class TTree;
2f514327	43
c7e89ea3	44	ClassImp(AliTagAnalysis)
c7e89ea3	45
54ac820d	46	//___________________________________________________________________________
	47	AliTagAnalysis::AliTagAnalysis():
	48	TObject(),
fe12e09c	49	ftagresult(0x0),
	50	fTagDirName(),
	51	fChain(0x0)
c7e89ea3	52	{
54ac820d	53	//Default constructor for a AliTagAnalysis
c7e89ea3	54	}
c7e89ea3	55
54ac820d	56	//___________________________________________________________________________
	57	AliTagAnalysis::~AliTagAnalysis() {
	58	//Default destructor for a AliTagAnalysis
c7e89ea3	59	}
c7e89ea3	60
6e2eae7e	61	//___________________________________________________________________________
	62	Bool_t AliTagAnalysis::AddTagsFile(const char *alienUrl) {
	63
	64	// Add a single tags file to the chain
	65
	66	Bool_t rv = kTRUE ;
	67
fc9e60bc	68	if (! fChain) fChain = new TChain("T");
6e2eae7e	69
	70	TFile *f = TFile::Open(alienUrl,"READ");
	71	fChain->Add(alienUrl);
	72	AliInfo(Form("Chained tag files: %d ",fChain->GetEntries()));
	73	delete f;
	74
	75	if (fChain->GetEntries() == 0 )
	76	rv = kFALSE ;
	77
	78	return rv ;
	79	}
	80
54ac820d	81	//___________________________________________________________________________
54ac820d	82	void AliTagAnalysis::ChainLocalTags(const char *dirname) {
c7e89ea3	83	//Searches the entries of the provided direcory
	84	//Chains the tags that are stored locally
	85	fTagDirName = dirname;
	86	TString fTagFilename;
	87
fc9e60bc	88	if (! fChain) fChain = new TChain("T");
c7e89ea3	89
f2313c5b	90	const char * tagPattern = "tag.root";
c7e89ea3	91	// Open the working directory
	92	void * dirp = gSystem->OpenDirectory(fTagDirName);
	93	const char * name = 0x0;
	94	// Add all files matching pattern to the chain
54ac820d	95	while((name = gSystem->GetDirEntry(dirp))) {
	96	if (strstr(name,tagPattern)) {
	97	fTagFilename = fTagDirName;
	98	fTagFilename += "/";
	99	fTagFilename += name;
c7e89ea3	100
54ac820d	101	fChain->Add(fTagFilename);
54ac820d	102	}//pattern check
54ac820d	103	}//directory loop
c7e89ea3	104	AliInfo(Form("Chained tag files: %d ",fChain->GetEntries()));
	105	}
	106
	107
54ac820d	108	//___________________________________________________________________________
54ac820d	109	void AliTagAnalysis::ChainGridTags(TGridResult *res) {
c7e89ea3	110	//Loops overs the entries of the TGridResult
	111	//Chains the tags that are stored in the GRID
	112	ftagresult = res;
	113	Int_t nEntries = ftagresult->GetEntries();
	114
fc9e60bc	115	if (! fChain) fChain = new TChain("T");
c7e89ea3	116
	117	TString gridname = "alien://";
	118	TString alienUrl;
	119
54ac820d	120	for(Int_t i = 0; i < nEntries; i++) {
54ac820d	121	alienUrl = ftagresult->GetKey(i,"turl");
54ac820d	122	fChain->Add(alienUrl);
54ac820d	123	}//grid result loop
c7e89ea3	124	}
	125
	126
54ac820d	127	//___________________________________________________________________________
a3acd4e8	128	TChain AliTagAnalysis::QueryTags(AliRunTagCuts RunTagCuts, AliEventTagCuts *EvTagCuts) {
c7e89ea3	129	//Queries the tag chain using the defined
c7e89ea3	130	//event tag cuts from the AliEventTagCuts object
b6003316	131	//and returns a TChain along with the associated TEventList
c7e89ea3	132	AliInfo(Form("Querying the tags........"));
c7e89ea3	133
df16faa3	134	//ESD file chain
	135	TChain *fESDchain = new TChain("esdTree");
	136	//Event list
	137	TEventList *fEventList = new TEventList();
c7e89ea3	138
	139	//Defining tag objects
	140	AliRunTag *tag = new AliRunTag;
	141	AliEventTag *evTag = new AliEventTag;
	142	fChain->SetBranchAddress("AliTAG",&tag);
	143
d021d0d7	144	TString guid = 0;
	145	TString turl = 0;
	146	TString path = 0;
c7e89ea3	147
df16faa3	148	Int_t iAccepted = 0;
df16faa3	149	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
54ac820d	150	fChain->GetEntry(iTagFiles);
a3acd4e8	151	if(RunTagCuts->IsAccepted(tag)) {
	152	Int_t iEvents = tag->GetNEvents();
	153	const TClonesArray *tagList = tag->GetEventTags();
	154	for(Int_t i = 0; i < iEvents; i++) {
	155	evTag = (AliEventTag *) tagList->At(i);
	156	guid = evTag->GetGUID();
	157	turl = evTag->GetTURL();
	158	path = evTag->GetPath();
	159	if(EvTagCuts->IsAccepted(evTag)) fEventList->Enter(iAccepted+i);
	160	}//event loop
	161	iAccepted += iEvents;
d021d0d7	162
a3acd4e8	163	if(path != "") fESDchain->AddFile(path);
	164	else if(turl != "") fESDchain->AddFile(turl);
	165	}//run tags cut
54ac820d	166	}//tag file loop
df16faa3	167	AliInfo(Form("Accepted events: %d",fEventList->GetN()));
df16faa3	168	fESDchain->SetEventList(fEventList);
c7e89ea3	169
df16faa3	170	return fESDchain;
c7e89ea3	171	}
c7e89ea3	172
5b7be856	173	//___________________________________________________________________________
	174	TChain AliTagAnalysis::QueryTags(const char fRunCut, const char *fEventCut) {
	175	//Queries the tag chain using the defined
	176	//event tag cuts from the AliEventTagCuts object
	177	//and returns a TChain along with the associated TEventList
	178	AliInfo(Form("Querying the tags........"));
	179
	180	//ESD file chain
	181	TChain *fESDchain = new TChain("esdTree");
	182	//Event list
	183	TEventList *fEventList = new TEventList();
	184
	185	//Defining tag objects
	186	AliRunTag *tag = new AliRunTag;
	187	AliEventTag *evTag = new AliEventTag;
	188	fChain->SetBranchAddress("AliTAG",&tag);
	189
	190	TString guid = 0;
	191	TString turl = 0;
	192	TString path = 0;
	193
	194	TTreeFormula *fRunFormula = new TTreeFormula("fRun",fRunCut,fChain);
	195	TTreeFormula *fEventFormula = new TTreeFormula("fEvent",fEventCut,fChain);
	196
	197	Int_t current = -1;
	198	Int_t iAccepted = 0;
	199	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
	200	fChain->GetEntry(iTagFiles);
	201	if (current != fChain->GetTreeNumber()) {
	202	fRunFormula->UpdateFormulaLeaves();
	203	fEventFormula->UpdateFormulaLeaves();
	204	current = fChain->GetTreeNumber();
	205	}
	206	if(fRunFormula->EvalInstance(iTagFiles) == 1) {
	207	Int_t iEvents = fEventFormula->GetNdata();
	208	const TClonesArray *tagList = tag->GetEventTags();
	209	for(Int_t i = 0; i < iEvents; i++) {
	210	evTag = (AliEventTag *) tagList->At(i);
	211	guid = evTag->GetGUID();
	212	turl = evTag->GetTURL();
	213	path = evTag->GetPath();
	214	if(fEventFormula->EvalInstance(i) == 1) fEventList->Enter(iAccepted+i);
	215	}//event loop
	216	iAccepted += iEvents;
	217
	218	if(path != "") fESDchain->AddFile(path);
	219	else if(turl != "") fESDchain->AddFile(turl);
	220	}//run tag cut
	221	}//tag file loop
	222	AliInfo(Form("Accepted events: %d",fEventList->GetN()));
	223	fESDchain->SetEventList(fEventList);
	224
	225	return fESDchain;
	226	}
	227
b6003316	228	//___________________________________________________________________________
a3acd4e8	229	Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, AliRunTagCuts RunTagCuts, AliEventTagCuts EvTagCuts) {
b6003316	230	//Queries the tag chain using the defined
	231	//event tag cuts from the AliEventTagCuts object
	232	//and returns a XML collection
	233	AliInfo(Form("Creating the collection........"));
	234
	235	AliXMLCollection *collection = new AliXMLCollection();
	236	collection->SetCollectionName(name);
	237	collection->WriteHeader();
	238
6e2eae7e	239	//Event list
6e2eae7e	240	//TEntryList *fEventList = new TEntryList();
b6003316	241	TString guid = 0x0;
c6e443c8	242	TString turl = 0x0;
eb771b73	243	TString lfn = 0x0;
b6003316	244
	245	//Defining tag objects
	246	AliRunTag *tag = new AliRunTag;
	247	AliEventTag *evTag = new AliEventTag;
	248	fChain->SetBranchAddress("AliTAG",&tag);
	249
	250	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
5b7be856	251	//Event list
5b7be856	252	TEntryList *fList = new TEntryList();
b6003316	253	fChain->GetEntry(iTagFiles);
a3acd4e8	254	if(RunTagCuts->IsAccepted(tag)) {
	255	Int_t iEvents = tag->GetNEvents();
	256	const TClonesArray *tagList = tag->GetEventTags();
	257	for(Int_t i = 0; i < iEvents; i++) {
	258	evTag = (AliEventTag *) tagList->At(i);
	259	guid = evTag->GetGUID();
	260	turl = evTag->GetTURL();
eb771b73	261	lfn = turl(8,turl.Length());
5b7be856	262	if(EvTagCuts->IsAccepted(evTag)) fList->Enter(i);
	263	}//event loop
	264	collection->WriteBody(iTagFiles+1,guid,lfn,turl,fList);
	265	}//run tag cuts
	266	}//tag file loop
	267	collection->Export();
	268
	269	return kTRUE;
	270	}
	271
	272	//___________________________________________________________________________
	273	Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, const char fRunCut, const char fEventCut) {
	274	//Queries the tag chain using the defined
	275	//event tag cuts from the AliEventTagCuts object
	276	//and returns a XML collection
	277	AliInfo(Form("Creating the collection........"));
	278
	279	AliXMLCollection *collection = new AliXMLCollection();
	280	collection->SetCollectionName(name);
	281	collection->WriteHeader();
2f514327	282
5b7be856	283	TString guid = 0x0;
	284	TString turl = 0x0;
	285	TString lfn = 0x0;
	286
	287	//Defining tag objects
	288	AliRunTag *tag = new AliRunTag;
	289	AliEventTag *evTag = new AliEventTag;
	290	fChain->SetBranchAddress("AliTAG",&tag);
	291
	292	TTreeFormula *fRunFormula = new TTreeFormula("fRun",fRunCut,fChain);
	293	TTreeFormula *fEventFormula = new TTreeFormula("fEvent",fEventCut,fChain);
	294
	295	Int_t current = -1;
	296	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
	297	//Event list
	298	TEntryList *fList = new TEntryList();
	299	fChain->GetEntry(iTagFiles);
	300	if (current != fChain->GetTreeNumber()) {
	301	fRunFormula->UpdateFormulaLeaves();
	302	fEventFormula->UpdateFormulaLeaves();
	303	current = fChain->GetTreeNumber();
	304	}
	305	if(fRunFormula->EvalInstance(iTagFiles) == 1) {
	306	Int_t iEvents = fEventFormula->GetNdata();
	307	const TClonesArray *tagList = tag->GetEventTags();
	308	for(Int_t i = 0; i < iEvents; i++) {
	309	evTag = (AliEventTag *) tagList->At(i);
	310	guid = evTag->GetGUID();
	311	turl = evTag->GetTURL();
	312	lfn = turl(8,turl.Length());
	313	if(fEventFormula->EvalInstance(i) == 1) fList->Enter(i);
a3acd4e8	314	}//event loop
5b7be856	315	collection->WriteBody(iTagFiles+1,guid,lfn,turl,fList);
a3acd4e8	316	}//run tag cuts
b6003316	317	}//tag file loop
	318	collection->Export();
	319
	320	return kTRUE;
	321	}
c7e89ea3	322
df122a89	323	//___________________________________________________________________________
2f514327	324	TChain AliTagAnalysis::GetInputChain(const char system, const charglobal, const char wn) {
	325	//returns the chain+event list - used in batch sessions
	326	TString fsystem = system;
	327	Int_t iAccepted = 0;
	328	//ESD file chain
	329	TChain *fESDchain = new TChain("esdTree");
df122a89	330	//Event list
df122a89	331	TEventList *fEventList = new TEventList();
2f514327	332	AliXMLCollection *collection1 = AliXMLCollection::Open(global);
	333	AliXMLCollection *collection2 = AliXMLCollection::Open(wn);
	334	collection1->OverlapCollection(collection2);
	335	collection1->Reset();
	336	while (collection1->Next()) {
	337	AliInfo(Form("Adding: %s",collection1->GetTURL("")));
	338	fESDchain->Add(collection1->GetTURL(""));
	339	TEntryList list = (TEntryList )collection1->GetEventList("");
	340	for(Int_t i = 0; i < list->GetN(); i++) fEventList->Enter(iAccepted+list->GetEntry(i));
	341
	342	if(fsystem == "pp") iAccepted += 100;
	343	else if(fsystem == "PbPb") iAccepted += 1;
	344	}
df122a89	345
2f514327	346	fESDchain->SetEventList(fEventList);
	347
	348	AliInfo(Form("Number of selected events: %d",fEventList->GetN()));
df122a89	349
2f514327	350	return fESDchain;
df122a89	351	}