[u/mrichter/AliRoot.git] / ANALYSIS / 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 "AliDetectorTagCuts.h"
#include "AliLHCTagCuts.h"
#include "AliRunTagCuts.h"
#include "AliXMLCollection.h"

class TTree;

ClassImp(AliTagAnalysis)

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

//___________________________________________________________________________
AliTagAnalysis::AliTagAnalysis(const char* type): 
  TObject(),
  ftagresult(0x0),
  fTagDirName(),
  fChain(0x0),
  fAnalysisType(type)
{
  //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 = 0x0;
  if(fAnalysisType == "ESD") tagPattern = "ESD.tag.root";
  else if(fAnalysisType == "AOD") tagPattern = "AOD.tag.root";
  else AliFatal("Only ESD and AOD type is implemented!!!");

  // 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, AliLHCTagCuts *lhcTagCuts, AliDetectorTagCuts *detTagCuts, 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........"));

  TString fAliceFile;
  if(fAnalysisType == "ESD") fAliceFile = "esdTree";
  else if(fAnalysisType == "AOD") fAliceFile = "aodTree";
  else AliFatal("Only ESD and AOD type is implemented!!!");

  //ESD file chain
  TChain *fESDchain = new TChain(fAliceFile.Data());
  //global entry list
  TEntryList *fGlobalList = new TEntryList();
  
  //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)) {
      if(lhcTagCuts->IsAccepted(tag->GetLHCTag())) {
	if(detTagCuts->IsAccepted(tag->GetDetectorTags())) {
	  TEntryList *fLocalList = new TEntryList();
	  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();
	    fLocalList->SetTreeName(fAliceFile.Data());
	    if(turl!="") fLocalList->SetFileName(turl.Data());
	    else fLocalList->SetFileName(path.Data());
	    if(evTagCuts->IsAccepted(evTag)) fLocalList->Enter(i);
	  }//event loop
	  
	  if(path != "") fESDchain->AddFile(path);
	  else if(turl != "") fESDchain->AddFile(turl);
	  fGlobalList->Add(fLocalList);
	  iAccepted += fLocalList->GetN();
	}//detector tag cuts
      }//lhc tag cuts
    }//run tags cut
  }//tag file loop
  AliInfo(Form("Accepted events: %d",iAccepted));
  fESDchain->SetEntryList(fGlobalList,"ne");
   
  return fESDchain;
}

//___________________________________________________________________________
TChain *AliTagAnalysis::QueryTags(const char *fRunCut, const char *fLHCCut, const char *fDetectorCut, 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........")); 	 
  
  TString fAliceFile;
  if(fAnalysisType == "ESD") fAliceFile = "esdTree";
  else if(fAnalysisType == "AOD") fAliceFile = "aodTree";
  else AliFatal("Only ESD and AOD type is implemented!!!");

  //ESD file chain
  TChain *fESDchain = new TChain(fAliceFile.Data());
  //global entry list
  TEntryList *fGlobalList = new TEntryList();
  
  //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 *fLHCFormula = new TTreeFormula("fLHC",fLHCCut,fChain); 	 
  TTreeFormula *fDetectorFormula = new TTreeFormula("fDetector",fDetectorCut,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(); 	 
      fLHCFormula->UpdateFormulaLeaves(); 	 
      fDetectorFormula->UpdateFormulaLeaves(); 	 
      fEventFormula->UpdateFormulaLeaves(); 	 
      current = fChain->GetTreeNumber(); 	 
    } 	 
    if(fRunFormula->EvalInstance(iTagFiles) == 1) { 	 
      if(fLHCFormula->EvalInstance(iTagFiles) == 1) { 	 
	if(fDetectorFormula->EvalInstance(iTagFiles) == 1) { 	 
	  TEntryList *fLocalList = new TEntryList();
	  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(); 	 
	    fLocalList->SetTreeName(fAliceFile.Data());
	    fLocalList->SetFileName(turl.Data());
	    if(fEventFormula->EvalInstance(i) == 1) fLocalList->Enter(i);
	  }//event loop 	 
	  iAccepted += fLocalList->GetN(); 	 
	  
	  if(path != "") fESDchain->AddFile(path); 	 
	  else if(turl != "") fESDchain->AddFile(turl); 	 
	  fGlobalList->Add(fLocalList);
	  iAccepted += fLocalList->GetN();
	}//detector tag cuts
      }//lhc tag cuts
    }//run tag cut 	 
  }//tag file loop 	 
  AliInfo(Form("Accepted events: %d",iAccepted)); 	 
  fESDchain->SetEntryList(fGlobalList,"ne"); 	 
  
  return fESDchain; 	 
}

//___________________________________________________________________________
Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, AliRunTagCuts *runTagCuts, AliLHCTagCuts *lhcTagCuts, AliDetectorTagCuts *detTagCuts, 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();

  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)) {
      if(lhcTagCuts->IsAccepted(tag->GetLHCTag())) {
	if(detTagCuts->IsAccepted(tag->GetDetectorTags())) {
	  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);
	}//detector tag cuts
      }//lhc tag cuts 
    }//run tag cuts
  }//tag file loop
  collection->Export();

  return kTRUE;
}

//___________________________________________________________________________
Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, const char *fRunCut, const char *fLHCCut, const char *fDetectorCut, 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 *fLHCFormula = new TTreeFormula("fLHC",fLHCCut,fChain); 	 
  TTreeFormula *fDetectorFormula = new TTreeFormula("fDetector",fDetectorCut,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();
      fLHCFormula->UpdateFormulaLeaves();
      fDetectorFormula->UpdateFormulaLeaves();
      fEventFormula->UpdateFormulaLeaves();
      current = fChain->GetTreeNumber();
    }
    if(fRunFormula->EvalInstance(iTagFiles) == 1) {
      if(fLHCFormula->EvalInstance(iTagFiles) == 1) { 	 
	if(fDetectorFormula->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);
	}//detector tag cuts
      }//lhc tag cuts 
    }//run tag cuts
  }//tag file loop
  collection->Export();

  return kTRUE;
}

//___________________________________________________________________________
TChain *AliTagAnalysis::GetInputChain(const char* system, const char *wn) {
  //returns the chain+event list - used in batch sessions
  // this function will be removed once the new root 
  // improvements are committed
  TString fsystem = system;
  Int_t iAccepted = 0;

  TChain *fAnalysisChain = 0;
  if(fAnalysisType == "ESD") fAnalysisChain = new TChain("esdTree");
  else if(fAnalysisType == "AOD") fAnalysisChain = new TChain("aodTree");
  else AliFatal("Only ESD and AOD type is implemented!!!");
  
  //Event list
  TEventList *fEventList = new TEventList();
  AliXMLCollection *collection = AliXMLCollection::Open(wn);

  collection->Reset();
  while (collection->Next()) {
    AliInfo(Form("Adding: %s",collection->GetTURL("")));
    fAnalysisChain->Add(collection->GetTURL(""));
    TEntryList *list = (TEntryList *)collection->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;
  }

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

  return fAnalysisChain;
}

//___________________________________________________________________________
TChain *AliTagAnalysis::GetChainFromCollection(const char* collectionname, const char* treename) {
  //returns the TChain+TEntryList object- used in batch sessions
  TString fAliceFile = treename;
  Int_t iAccepted = 0;
  TChain *fAnalysisChain = 0;
  if(fAliceFile == "esdTree") fAnalysisChain = new TChain("esdTree");
  else if(fAliceFile == "aodTree") fAnalysisChain = new TChain("aodTree");
  else AliFatal("Inconsistent tree name - use esdTree or aodTree!");

  //Event list
  TEntryList *fGlobalList = new TEntryList();
  AliXMLCollection *collection = AliXMLCollection::Open(collectionname);

  collection->Reset();
  while (collection->Next()) {
    AliInfo(Form("Adding: %s",collection->GetTURL("")));
    fAnalysisChain->Add(collection->GetTURL(""));
    TEntryList *list = (TEntryList *)collection->GetEventList("");
    list->SetTreeName(fAliceFile.Data());
    list->SetFileName(collection->GetTURL(""));
    fGlobalList->Add(list);
    iAccepted += list->GetN();
  }

  fAnalysisChain->SetEntryList(fGlobalList,"ne");
  
  AliInfo(Form("Number of selected events: %d",iAccepted));

  return fAnalysisChain;
}
Commit	Line	Data
d08133e6	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>
	25	#include <TFile.h>
	26	#include <TEventList.h>
	27	#include <TEntryList.h>
	28	#include <TTreeFormula.h>
	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"
2ec6a1d3	39	#include "AliDetectorTagCuts.h"
2ec6a1d3	40	#include "AliLHCTagCuts.h"
d08133e6	41	#include "AliRunTagCuts.h"
	42	#include "AliXMLCollection.h"
	43
	44	class TTree;
	45
	46	ClassImp(AliTagAnalysis)
	47
	48	//___________________________________________________________________________
	49	AliTagAnalysis::AliTagAnalysis():
	50	TObject(),
	51	ftagresult(0x0),
	52	fTagDirName(),
6cfbb923	53	fChain(0x0),
6cfbb923	54	fAnalysisType()
d08133e6	55	{
	56	//Default constructor for a AliTagAnalysis
	57	}
	58
6cfbb923	59	//___________________________________________________________________________
	60	AliTagAnalysis::AliTagAnalysis(const char* type):
	61	TObject(),
	62	ftagresult(0x0),
	63	fTagDirName(),
	64	fChain(0x0),
	65	fAnalysisType(type)
	66	{
	67	//constructor for a AliTagAnalysis
	68	}
	69
d08133e6	70	//___________________________________________________________________________
	71	AliTagAnalysis::~AliTagAnalysis() {
	72	//Default destructor for a AliTagAnalysis
	73	}
	74
	75	//___________________________________________________________________________
	76	Bool_t AliTagAnalysis::AddTagsFile(const char *alienUrl) {
	77
	78	// Add a single tags file to the chain
	79
	80	Bool_t rv = kTRUE ;
	81
	82	if (! fChain) fChain = new TChain("T");
	83
	84	TFile *f = TFile::Open(alienUrl,"READ");
	85	fChain->Add(alienUrl);
	86	AliInfo(Form("Chained tag files: %d ",fChain->GetEntries()));
	87	delete f;
	88
	89	if (fChain->GetEntries() == 0 )
	90	rv = kFALSE ;
	91
	92	return rv ;
	93	}
	94
	95	//___________________________________________________________________________
	96	void AliTagAnalysis::ChainLocalTags(const char *dirname) {
	97	//Searches the entries of the provided direcory
	98	//Chains the tags that are stored locally
	99	fTagDirName = dirname;
	100	TString fTagFilename;
	101
	102	if (! fChain) fChain = new TChain("T");
6cfbb923	103	const char * tagPattern = 0x0;
	104	if(fAnalysisType == "ESD") tagPattern = "ESD.tag.root";
	105	else if(fAnalysisType == "AOD") tagPattern = "AOD.tag.root";
	106	else AliFatal("Only ESD and AOD type is implemented!!!");
	107
d08133e6	108	// Open the working directory
	109	void * dirp = gSystem->OpenDirectory(fTagDirName);
	110	const char * name = 0x0;
	111	// Add all files matching pattern to the chain
	112	while((name = gSystem->GetDirEntry(dirp))) {
	113	if (strstr(name,tagPattern)) {
	114	fTagFilename = fTagDirName;
	115	fTagFilename += "/";
	116	fTagFilename += name;
	117
	118	fChain->Add(fTagFilename);
	119	}//pattern check
	120	}//directory loop
	121	AliInfo(Form("Chained tag files: %d ",fChain->GetEntries()));
	122	}
	123
	124
	125	//___________________________________________________________________________
	126	void AliTagAnalysis::ChainGridTags(TGridResult *res) {
	127	//Loops overs the entries of the TGridResult
	128	//Chains the tags that are stored in the GRID
	129	ftagresult = res;
	130	Int_t nEntries = ftagresult->GetEntries();
	131
	132	if (! fChain) fChain = new TChain("T");
	133
	134	TString gridname = "alien://";
	135	TString alienUrl;
	136
	137	for(Int_t i = 0; i < nEntries; i++) {
	138	alienUrl = ftagresult->GetKey(i,"turl");
	139	fChain->Add(alienUrl);
	140	}//grid result loop
	141	}
	142
	143
	144	//___________________________________________________________________________
2ec6a1d3	145	TChain AliTagAnalysis::QueryTags(AliRunTagCuts runTagCuts, AliLHCTagCuts lhcTagCuts, AliDetectorTagCuts detTagCuts, AliEventTagCuts *evTagCuts) {
d08133e6	146	//Queries the tag chain using the defined
	147	//event tag cuts from the AliEventTagCuts object
	148	//and returns a TChain along with the associated TEventList
	149	AliInfo(Form("Querying the tags........"));
b97856f9	150
6cfbb923	151	TString fAliceFile;
	152	if(fAnalysisType == "ESD") fAliceFile = "esdTree";
	153	else if(fAnalysisType == "AOD") fAliceFile = "aodTree";
	154	else AliFatal("Only ESD and AOD type is implemented!!!");
b97856f9	155
d08133e6	156	//ESD file chain
b97856f9	157	TChain *fESDchain = new TChain(fAliceFile.Data());
	158	//global entry list
	159	TEntryList *fGlobalList = new TEntryList();
d08133e6	160
	161	//Defining tag objects
	162	AliRunTag *tag = new AliRunTag;
	163	AliEventTag *evTag = new AliEventTag;
	164	fChain->SetBranchAddress("AliTAG",&tag);
	165
	166	TString guid = 0;
	167	TString turl = 0;
	168	TString path = 0;
	169
	170	Int_t iAccepted = 0;
	171	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
	172	fChain->GetEntry(iTagFiles);
2ec6a1d3	173	if(runTagCuts->IsAccepted(tag)) {
	174	if(lhcTagCuts->IsAccepted(tag->GetLHCTag())) {
	175	if(detTagCuts->IsAccepted(tag->GetDetectorTags())) {
	176	TEntryList *fLocalList = new TEntryList();
	177	Int_t iEvents = tag->GetNEvents();
	178	const TClonesArray *tagList = tag->GetEventTags();
	179	for(Int_t i = 0; i < iEvents; i++) {
	180	evTag = (AliEventTag *) tagList->At(i);
	181	guid = evTag->GetGUID();
	182	turl = evTag->GetTURL();
	183	path = evTag->GetPath();
	184	fLocalList->SetTreeName(fAliceFile.Data());
c51c9b0e	185	if(turl!="") fLocalList->SetFileName(turl.Data());
c51c9b0e	186	else fLocalList->SetFileName(path.Data());
2ec6a1d3	187	if(evTagCuts->IsAccepted(evTag)) fLocalList->Enter(i);
	188	}//event loop
	189
	190	if(path != "") fESDchain->AddFile(path);
	191	else if(turl != "") fESDchain->AddFile(turl);
	192	fGlobalList->Add(fLocalList);
	193	iAccepted += fLocalList->GetN();
	194	}//detector tag cuts
	195	}//lhc tag cuts
d08133e6	196	}//run tags cut
d08133e6	197	}//tag file loop
b97856f9	198	AliInfo(Form("Accepted events: %d",iAccepted));
b97856f9	199	fESDchain->SetEntryList(fGlobalList,"ne");
d08133e6	200
	201	return fESDchain;
	202	}
	203
	204	//___________________________________________________________________________
2ec6a1d3	205	TChain AliTagAnalysis::QueryTags(const char fRunCut, const char fLHCCut, const char fDetectorCut, const char *fEventCut) {
d08133e6	206	//Queries the tag chain using the defined
	207	//event tag cuts from the AliEventTagCuts object
	208	//and returns a TChain along with the associated TEventList
	209	AliInfo(Form("Querying the tags........"));
	210
6cfbb923	211	TString fAliceFile;
	212	if(fAnalysisType == "ESD") fAliceFile = "esdTree";
	213	else if(fAnalysisType == "AOD") fAliceFile = "aodTree";
	214	else AliFatal("Only ESD and AOD type is implemented!!!");
b97856f9	215
	216	//ESD file chain
	217	TChain *fESDchain = new TChain(fAliceFile.Data());
	218	//global entry list
	219	TEntryList *fGlobalList = new TEntryList();
d08133e6	220
	221	//Defining tag objects
	222	AliRunTag *tag = new AliRunTag;
	223	AliEventTag *evTag = new AliEventTag;
	224	fChain->SetBranchAddress("AliTAG",&tag);
	225
	226	TString guid = 0;
	227	TString turl = 0;
	228	TString path = 0;
	229
	230	TTreeFormula *fRunFormula = new TTreeFormula("fRun",fRunCut,fChain);
2ec6a1d3	231	TTreeFormula *fLHCFormula = new TTreeFormula("fLHC",fLHCCut,fChain);
	232	TTreeFormula *fDetectorFormula = new TTreeFormula("fDetector",fDetectorCut,fChain);
	233	TTreeFormula *fEventFormula = new TTreeFormula("fEvent",fEventCut,fChain);
d08133e6	234
	235	Int_t current = -1;
	236	Int_t iAccepted = 0;
	237	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
	238	fChain->GetEntry(iTagFiles);
	239	if (current != fChain->GetTreeNumber()) {
	240	fRunFormula->UpdateFormulaLeaves();
2ec6a1d3	241	fLHCFormula->UpdateFormulaLeaves();
2ec6a1d3	242	fDetectorFormula->UpdateFormulaLeaves();
d08133e6	243	fEventFormula->UpdateFormulaLeaves();
	244	current = fChain->GetTreeNumber();
	245	}
	246	if(fRunFormula->EvalInstance(iTagFiles) == 1) {
2ec6a1d3	247	if(fLHCFormula->EvalInstance(iTagFiles) == 1) {
	248	if(fDetectorFormula->EvalInstance(iTagFiles) == 1) {
	249	TEntryList *fLocalList = new TEntryList();
	250	Int_t iEvents = fEventFormula->GetNdata();
	251	const TClonesArray *tagList = tag->GetEventTags();
	252	for(Int_t i = 0; i < iEvents; i++) {
	253	evTag = (AliEventTag *) tagList->At(i);
	254	guid = evTag->GetGUID();
	255	turl = evTag->GetTURL();
	256	path = evTag->GetPath();
	257	fLocalList->SetTreeName(fAliceFile.Data());
	258	fLocalList->SetFileName(turl.Data());
	259	if(fEventFormula->EvalInstance(i) == 1) fLocalList->Enter(i);
	260	}//event loop
	261	iAccepted += fLocalList->GetN();
	262
	263	if(path != "") fESDchain->AddFile(path);
	264	else if(turl != "") fESDchain->AddFile(turl);
	265	fGlobalList->Add(fLocalList);
	266	iAccepted += fLocalList->GetN();
	267	}//detector tag cuts
	268	}//lhc tag cuts
d08133e6	269	}//run tag cut
d08133e6	270	}//tag file loop
b97856f9	271	AliInfo(Form("Accepted events: %d",iAccepted));
b97856f9	272	fESDchain->SetEntryList(fGlobalList,"ne");
d08133e6	273
	274	return fESDchain;
	275	}
	276
	277	//___________________________________________________________________________
2ec6a1d3	278	Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, AliRunTagCuts runTagCuts, AliLHCTagCuts lhcTagCuts, AliDetectorTagCuts detTagCuts, AliEventTagCuts evTagCuts) {
d08133e6	279	//Queries the tag chain using the defined
	280	//event tag cuts from the AliEventTagCuts object
	281	//and returns a XML collection
	282	AliInfo(Form("Creating the collection........"));
	283
	284	AliXMLCollection *collection = new AliXMLCollection();
	285	collection->SetCollectionName(name);
	286	collection->WriteHeader();
	287
d08133e6	288	TString guid = 0x0;
	289	TString turl = 0x0;
	290	TString lfn = 0x0;
	291
	292	//Defining tag objects
	293	AliRunTag *tag = new AliRunTag;
	294	AliEventTag *evTag = new AliEventTag;
	295	fChain->SetBranchAddress("AliTAG",&tag);
	296
	297	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
	298	//Event list
	299	TEntryList *fList = new TEntryList();
	300	fChain->GetEntry(iTagFiles);
2ec6a1d3	301	if(runTagCuts->IsAccepted(tag)) {
	302	if(lhcTagCuts->IsAccepted(tag->GetLHCTag())) {
	303	if(detTagCuts->IsAccepted(tag->GetDetectorTags())) {
	304	Int_t iEvents = tag->GetNEvents();
	305	const TClonesArray *tagList = tag->GetEventTags();
	306	for(Int_t i = 0; i < iEvents; i++) {
	307	evTag = (AliEventTag *) tagList->At(i);
	308	guid = evTag->GetGUID();
	309	turl = evTag->GetTURL();
	310	lfn = turl(8,turl.Length());
	311	if(evTagCuts->IsAccepted(evTag)) fList->Enter(i);
	312	}//event loop
	313	collection->WriteBody(iTagFiles+1,guid,lfn,turl,fList);
	314	}//detector tag cuts
	315	}//lhc tag cuts
d08133e6	316	}//run tag cuts
	317	}//tag file loop
	318	collection->Export();
	319
	320	return kTRUE;
	321	}
	322
	323	//___________________________________________________________________________
2ec6a1d3	324	Bool_t AliTagAnalysis::CreateXMLCollection(const char* name, const char fRunCut, const char fLHCCut, const char fDetectorCut, const char fEventCut) {
d08133e6	325	//Queries the tag chain using the defined
	326	//event tag cuts from the AliEventTagCuts object
	327	//and returns a XML collection
	328	AliInfo(Form("Creating the collection........"));
	329
	330	AliXMLCollection *collection = new AliXMLCollection();
	331	collection->SetCollectionName(name);
	332	collection->WriteHeader();
	333
	334	TString guid = 0x0;
	335	TString turl = 0x0;
	336	TString lfn = 0x0;
	337
	338	//Defining tag objects
	339	AliRunTag *tag = new AliRunTag;
	340	AliEventTag *evTag = new AliEventTag;
	341	fChain->SetBranchAddress("AliTAG",&tag);
	342
	343	TTreeFormula *fRunFormula = new TTreeFormula("fRun",fRunCut,fChain);
2ec6a1d3	344	TTreeFormula *fLHCFormula = new TTreeFormula("fLHC",fLHCCut,fChain);
2ec6a1d3	345	TTreeFormula *fDetectorFormula = new TTreeFormula("fDetector",fDetectorCut,fChain);
d08133e6	346	TTreeFormula *fEventFormula = new TTreeFormula("fEvent",fEventCut,fChain);
	347
	348	Int_t current = -1;
	349	for(Int_t iTagFiles = 0; iTagFiles < fChain->GetEntries(); iTagFiles++) {
	350	//Event list
	351	TEntryList *fList = new TEntryList();
	352	fChain->GetEntry(iTagFiles);
	353	if (current != fChain->GetTreeNumber()) {
	354	fRunFormula->UpdateFormulaLeaves();
2ec6a1d3	355	fLHCFormula->UpdateFormulaLeaves();
2ec6a1d3	356	fDetectorFormula->UpdateFormulaLeaves();
d08133e6	357	fEventFormula->UpdateFormulaLeaves();
	358	current = fChain->GetTreeNumber();
	359	}
	360	if(fRunFormula->EvalInstance(iTagFiles) == 1) {
2ec6a1d3	361	if(fLHCFormula->EvalInstance(iTagFiles) == 1) {
	362	if(fDetectorFormula->EvalInstance(iTagFiles) == 1) {
	363	Int_t iEvents = fEventFormula->GetNdata();
	364	const TClonesArray *tagList = tag->GetEventTags();
	365	for(Int_t i = 0; i < iEvents; i++) {
	366	evTag = (AliEventTag *) tagList->At(i);
	367	guid = evTag->GetGUID();
	368	turl = evTag->GetTURL();
	369	lfn = turl(8,turl.Length());
	370	if(fEventFormula->EvalInstance(i) == 1) fList->Enter(i);
	371	}//event loop
	372	collection->WriteBody(iTagFiles+1,guid,lfn,turl,fList);
	373	}//detector tag cuts
	374	}//lhc tag cuts
d08133e6	375	}//run tag cuts
	376	}//tag file loop
	377	collection->Export();
	378
	379	return kTRUE;
	380	}
	381
	382	//___________________________________________________________________________
	383	TChain AliTagAnalysis::GetInputChain(const char system, const char *wn) {
	384	//returns the chain+event list - used in batch sessions
	385	// this function will be removed once the new root
	386	// improvements are committed
	387	TString fsystem = system;
	388	Int_t iAccepted = 0;
6cfbb923	389
88fb0edc	390	TChain *fAnalysisChain = 0;
	391	if(fAnalysisType == "ESD") fAnalysisChain = new TChain("esdTree");
	392	else if(fAnalysisType == "AOD") fAnalysisChain = new TChain("aodTree");
6cfbb923	393	else AliFatal("Only ESD and AOD type is implemented!!!");
6cfbb923	394
d08133e6	395	//Event list
	396	TEventList *fEventList = new TEventList();
	397	AliXMLCollection *collection = AliXMLCollection::Open(wn);
	398
	399	collection->Reset();
	400	while (collection->Next()) {
	401	AliInfo(Form("Adding: %s",collection->GetTURL("")));
6cfbb923	402	fAnalysisChain->Add(collection->GetTURL(""));
d08133e6	403	TEntryList list = (TEntryList )collection->GetEventList("");
	404	for(Int_t i = 0; i < list->GetN(); i++) fEventList->Enter(iAccepted+list->GetEntry(i));
	405
	406	if(fsystem == "pp") iAccepted += 100;
	407	else if(fsystem == "PbPb") iAccepted += 1;
	408	}
	409
6cfbb923	410	fAnalysisChain->SetEventList(fEventList);
d08133e6	411
	412	AliInfo(Form("Number of selected events: %d",fEventList->GetN()));
	413
6cfbb923	414	return fAnalysisChain;
d08133e6	415	}
	416
	417	//___________________________________________________________________________
	418	TChain AliTagAnalysis::GetChainFromCollection(const char collectionname, const char* treename) {
	419	//returns the TChain+TEntryList object- used in batch sessions
	420	TString fAliceFile = treename;
	421	Int_t iAccepted = 0;
88fb0edc	422	TChain *fAnalysisChain = 0;
	423	if(fAliceFile == "esdTree") fAnalysisChain = new TChain("esdTree");
	424	else if(fAliceFile == "aodTree") fAnalysisChain = new TChain("aodTree");
d08133e6	425	else AliFatal("Inconsistent tree name - use esdTree or aodTree!");
6cfbb923	426
d08133e6	427	//Event list
	428	TEntryList *fGlobalList = new TEntryList();
	429	AliXMLCollection *collection = AliXMLCollection::Open(collectionname);
	430
	431	collection->Reset();
	432	while (collection->Next()) {
	433	AliInfo(Form("Adding: %s",collection->GetTURL("")));
	434	fAnalysisChain->Add(collection->GetTURL(""));
	435	TEntryList list = (TEntryList )collection->GetEventList("");
	436	list->SetTreeName(fAliceFile.Data());
	437	list->SetFileName(collection->GetTURL(""));
	438	fGlobalList->Add(list);
	439	iAccepted += list->GetN();
	440	}
	441
	442	fAnalysisChain->SetEntryList(fGlobalList,"ne");
	443
	444	AliInfo(Form("Number of selected events: %d",iAccepted));
	445
	446	return fAnalysisChain;
	447	}