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

/**************************************************************************
 * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * 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$ */

//-------------------------------------------------------------------------
//     Event handler for ESD input 
//     Author: Andreas Morsch, CERN
//-------------------------------------------------------------------------

#include <TTree.h>
#include <TChain.h>
#include <TFile.h>
#include <TArchiveFile.h>
#include <TObjArray.h>
#include <TSystem.h>
#include <TString.h>
#include <TObjString.h>
#include <TProcessID.h>
#include <TMap.h>

#include "AliESDInputHandler.h"
#include "AliESDEvent.h"
#include "AliESDfriend.h"
#include "AliVCuts.h"
#include "AliESD.h"
#include "AliRunTag.h"
#include "AliEventTag.h"
#include "AliLog.h"

ClassImp(AliESDInputHandler)

static Option_t *gESDDataType = "ESD";

//______________________________________________________________________________
AliESDInputHandler::AliESDInputHandler() :
  AliInputEventHandler(),
  fEvent(0x0),
  fFriend(0x0),
  fESDpid(0x0),
  fAnalysisType(0),
  fNEvents(0),
  fHLTEvent(0x0),
  fHLTTree(0x0),
  fUseHLT(kFALSE),
  fTagCutSumm(0x0),
  fUseTags(kFALSE),
  fChainT(0),
  fTreeT(0),
  fRunTag(0),
  fReadFriends(0),
  fFriendFileName("AliESDfriends.root")
{
  // default constructor
}

//______________________________________________________________________________
AliESDInputHandler::~AliESDInputHandler() 
{
  //  destructor
}

//______________________________________________________________________________
AliESDInputHandler::AliESDInputHandler(const char* name, const char* title):
    AliInputEventHandler(name, title), fEvent(0x0), fFriend(0x0), fESDpid(0x0), fAnalysisType(0),
    fNEvents(0),  fHLTEvent(0x0), fHLTTree(0x0), fUseHLT(kFALSE), fTagCutSumm(0x0), fUseTags(kFALSE), fChainT(0), fTreeT(0), fRunTag(0), fReadFriends(0), fFriendFileName("AliESDfriends.root")
{
    // Constructor
}

Bool_t AliESDInputHandler::Init(TTree* tree,  Option_t* opt)
{
    //
    // Initialisation necessary for each new tree 
    // 
    fAnalysisType = opt;
    fTree         = tree;
    
    if (!fTree) return kFALSE;
    fTree->GetEntry(0);
    

    if (!fEvent) fEvent = new AliESDEvent();
    fEvent->ReadFromTree(fTree);
    fNEvents = fTree->GetEntries();
    return kTRUE;
}

Bool_t AliESDInputHandler::BeginEvent(Long64_t entry)
{
    
    // Copy from old to new format if necessary
  AliESD* old = ((AliESDEvent*) fEvent)->GetAliESDOld();
  if (old) {
	((AliESDEvent*)fEvent)->CopyFromOldESD();
	old->Reset();
  }

  if (fHLTTree) {
      fHLTTree->GetEntry(entry);
  }
  
  fNewEvent = kTRUE;
  //
  // Event selection
  // 
  if (fEventCuts)
    fIsSelectedResult = fEventCuts->GetSelectionMask((AliESDEvent*)fEvent); 
  //
  // Friends
  ((AliESDEvent*)fEvent)->SetESDfriend(fFriend);
  return kTRUE;
}

Bool_t  AliESDInputHandler::FinishEvent()
{
    // Finish the event 
    if(fEvent)fEvent->Reset();
    return kTRUE;
} 

Bool_t AliESDInputHandler::Notify(const char* path)
{
    // Notify a directory change
    AliInfo(Form("Directory change %s \n", path));
    //
    // Handle the friends first
    //
    if (!fTree->FindBranch("ESDfriend.") && fReadFriends) {
      // Try to add ESDfriend. branch as friend
      TString esdTreeFName, esdFriendTreeFName;    
      esdTreeFName = (fTree->GetCurrentFile())->GetName();
      esdFriendTreeFName = esdTreeFName;
      esdFriendTreeFName.ReplaceAll("AliESDs.root", fFriendFileName.Data());
      
      TTree* cTree = fTree->GetTree();
      if (!cTree) cTree = fTree;
      
      cTree->AddFriend("esdFriendTree", esdFriendTreeFName.Data());
      cTree->SetBranchStatus("ESDfriend.", 1);
      fFriend = (AliESDfriend*)(fEvent->FindListObject("AliESDfriend"));
      cTree->SetBranchAddress("ESDfriend.", &fFriend);
    } 
    //
    //
    SwitchOffBranches();
    SwitchOnBranches();
    fFriend = (AliESDfriend*)(fEvent->FindListObject("AliESDfriend"));
    

    //
    if (fUseHLT) {
	// Get HLTesdTree from current file
	TTree* cTree = fTree;
	if (fTree->GetTree()) cTree = fTree->GetTree();
	TFile* cFile = cTree->GetCurrentFile();
	cFile->GetObject("HLTesdTree", fHLTTree);
	
	if (fHLTTree) {
	  if (!fHLTEvent) fHLTEvent = new AliESDEvent();
	  fHLTEvent->ReadFromTree(fHLTTree);
	}
    }


    if (!fUseTags) return (kTRUE);
    
    Bool_t zip = kFALSE;
    
    TString fileName(path);
    if(fileName.Contains("#AliESDs.root")){
	zip = kTRUE;
    } 
    else if (fileName.Contains("AliESDs.root")){
	fileName.ReplaceAll("AliESDs.root", "");
    }
    else if(fileName.Contains("#AliAOD.root")){
	zip = kTRUE;
    }
    else if(fileName.Contains("AliAOD.root")){
	fileName.ReplaceAll("AliAOD.root", "");
    }
    else if(fileName.Contains("#galice.root")){
	// For running with galice and kinematics alone...
	zip = kTRUE;
    }
    else if(fileName.Contains("galice.root")){
	// For running with galice and kinematics alone...
	fileName.ReplaceAll("galice.root", "");
    }

    
    TString pathName("./");
    if (fileName.Length() != 0) {
	pathName = fileName;
    }
    
    printf("AliESDInputHandler::Notify() Path: %s\n", pathName.Data());

    if (fRunTag) {
	fRunTag->Clear();
    } else {
	fRunTag = new AliRunTag();
    }
    
    delete fTreeT; fTreeT = 0;
    
    if (fChainT) {
	delete fChainT;
	fChainT = 0;
    }
    
    if (!fChainT) {
	fChainT = new TChain("T");
    }
    

    const char* tagPattern = "ESD.tag.root";
    const char* name = 0x0;
    TString tagFilename;
    if (zip) {
	TFile* file = fTree->GetCurrentFile();
	TArchiveFile* arch = file->GetArchive();
	TObjArray* arr = arch->GetMembers();
	TIter next(arr);
	
	while ((file = (TFile*) next())) {
	    name = file->GetName();
	    if (strstr(name,tagPattern)) { 
		tagFilename = pathName.Data();
		tagFilename += "#";
		tagFilename += name;
		fChainT->Add(tagFilename);  
		AliInfo(Form("Adding %s to tag chain \n", tagFilename.Data()));
	    }//pattern check
	} // archive file loop
    } else {
	void * dirp = gSystem->OpenDirectory(pathName.Data());
	while((name = gSystem->GetDirEntry(dirp))) {
	    if (strstr(name,tagPattern)) { 
		tagFilename = pathName.Data();
		tagFilename += "/";
		tagFilename += name;
		fChainT->Add(tagFilename);  
		AliInfo(Form("Adding %s to tag chain \n", tagFilename.Data()));
	    }//pattern check
	}//directory loop
    }
    fChainT->SetBranchAddress("AliTAG",&fRunTag);
    fChainT->GetEntry(0);
    return kTRUE;
}


Option_t *AliESDInputHandler::GetDataType() const
{
// Returns handled data type.
   return gESDDataType;
}

Int_t AliESDInputHandler::GetNEventAcceptedInFile()
{
  // Get number of events in file accepted by the tag cuts
  // return -1 if no info is available
  if (!fTagCutSumm) {
    TList *luo = fTree->GetUserInfo();
    if (!luo) {
      AliInfo(Form("No user info in input tree - no tag cut summary\n"));
      return -1;
    }
    for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
      fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
      if (fTagCutSumm) break;
    }
    if (!fTagCutSumm) {
      AliInfo(Form("No tag summary map in input tree\n"));
      return -1;
    }
  }

  TObjString *ostr = 0;
  if (fTagCutSumm->FindObject(fTree->GetCurrentFile()->GetName()))
    ostr = (TObjString *) fTagCutSumm->GetValue(fTree->GetCurrentFile()->GetName());
  else {
    AliInfo(Form("No tag cut summary for file %s\n", fTree->GetCurrentFile()->GetName()));
    return -1;
  }
  char *iTagInfo;
  iTagInfo = strdup(ostr->GetString().Data());

  Int_t iAcc = atoi(strtok(iTagInfo, ","));
  
  AliInfo(Form("Got %i accepted events for file %s", iAcc,  fTree->GetCurrentFile()->GetName()));
  
  free(iTagInfo);

  return iAcc;
}
Int_t AliESDInputHandler::GetNEventRejectedInFile()
{
  // Get number of events in file rejected by the tag cuts
  // return -1 if no info is available
  if (!fTagCutSumm) {
    TList *luo = fTree->GetUserInfo();
    if (!luo) {
      AliInfo(Form("No user info in input tree - no tag cut summary\n"));
      return -1;
    }
    for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
      fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
      if (fTagCutSumm) break;
    }
    if (!fTagCutSumm) {
      AliInfo(Form("No tag summary map in input tree\n"));
      return -1;
    }
  }

  TObjString *ostr = 0;
  if (fTagCutSumm->FindObject(fTree->GetCurrentFile()->GetName()))
    ostr = (TObjString *) fTagCutSumm->GetValue(fTree->GetCurrentFile()->GetName());
  else {
    AliInfo(Form("No tag cut summary for file %s\n", fTree->GetCurrentFile()->GetName()));
    return -1;
  }
  char *iTagInfo;
  iTagInfo = strdup(ostr->GetString().Data());

  strtok(iTagInfo, ",");
  Int_t iRej = atoi(strtok(NULL, ","));
  
  AliInfo(Form("Got %i accepted events for file %s", iRej,  fTree->GetCurrentFile()->GetName()));
  
  free(iTagInfo);

  return iRej;
}
Bool_t AliESDInputHandler::GetCutSummaryForChain(Int_t *aTotal, Int_t *aAccepted, Int_t *aRejected)
{
  // Get number of events in the full chain
  // Count accepted and rejected events
  // return kFALSE if no info is available
  if (!fTagCutSumm) {
    TList *luo = fTree->GetUserInfo();
    if (!luo) {
      AliInfo(Form("No user info in input tree - no tag cut summary\n"));
      return kFALSE;
    }
    for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
      fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
      if (fTagCutSumm) break;
    }
    if (!fTagCutSumm) {
      AliInfo(Form("No tag summary map in input tree\n"));
      return kFALSE;
    }
  }
  
  TMapIter *tIter = new TMapIter(fTagCutSumm);
  
  Int_t iTotList=0, iAccList=0, iRejList=0;

  TObject *cobj;
  while ((cobj = tIter->Next())) {
    TObjString *kstr = (TObjString *) cobj;
    TObjString *vstr = (TObjString *) fTagCutSumm->GetValue(kstr->GetString().Data());
    //    printf("Got object value %s %s\n", kstr->GetString().Data(), vstr->GetString().Data());
    char *iTagInfo;
    iTagInfo = strdup(vstr->GetString().Data());
    
    Int_t iAcc = atoi(strtok(iTagInfo, ","));
    Int_t iRej = atoi(strtok(NULL, ","));
    
    iAccList += iAcc;
    iRejList += iRej;
    iTotList += (iAcc+iRej);
  }

  *aTotal = iTotList;
  *aAccepted = iAccList;
  *aRejected = iRejList;

  return kTRUE;
}

Int_t AliESDInputHandler::GetNFilesEmpty()
{
  // Count number of files in which all events were de-selected
  // For such files Notify() will NOT be called
  // return -1 if no info is available
  if (!fTagCutSumm) {
    TList *luo = fTree->GetUserInfo();
    if (!luo) {
      AliInfo(Form("No user info in input tree - no tag cut summary\n"));
      return -1;
    }
    for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
      fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
      if (fTagCutSumm) break;
    }
    if (!fTagCutSumm) {
      AliInfo(Form("No tag summary map in input tree\n"));
      return -1;
    }
  }
  
  TMapIter *tIter = new TMapIter(fTagCutSumm);
  
  Int_t iFilesEmpty = 0;

  TObject *cobj;
  while ((cobj = tIter->Next())) {
    TObjString *kstr = (TObjString *) cobj;
    TObjString *vstr = (TObjString *) fTagCutSumm->GetValue(kstr->GetString().Data());
    //    printf("Got object value %s %s\n", kstr->GetString().Data(), vstr->GetString().Data());
    char *iTagInfo;
    iTagInfo = strdup(vstr->GetString().Data());
    
    Int_t iAcc = atoi(strtok(iTagInfo, ","));
    Int_t iRej = atoi(strtok(NULL, ","));
    
    if ((iAcc == 0) && ((iRej+iAcc)>0))
      iFilesEmpty++;
  }

  return iFilesEmpty;
  
}
Commit	Line	Data
	1	/**************************************************************************
	2	* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/* $Id$ */
	17
	18	//-------------------------------------------------------------------------
	19	// Event handler for ESD input
	20	// Author: Andreas Morsch, CERN
	21	//-------------------------------------------------------------------------
	22
	23	#include <TTree.h>
	24	#include <TChain.h>
	25	#include <TFile.h>
	26	#include <TArchiveFile.h>
	27	#include <TObjArray.h>
	28	#include <TSystem.h>
	29	#include <TString.h>
	30	#include <TObjString.h>
	31	#include <TProcessID.h>
	32	#include <TMap.h>
	33
	34	#include "AliESDInputHandler.h"
	35	#include "AliESDEvent.h"
	36	#include "AliESDfriend.h"
	37	#include "AliVCuts.h"
	38	#include "AliESD.h"
	39	#include "AliRunTag.h"
	40	#include "AliEventTag.h"
	41	#include "AliLog.h"
	42
	43	ClassImp(AliESDInputHandler)
	44
	45	static Option_t *gESDDataType = "ESD";
	46
	47	//______________________________________________________________________________
	48	AliESDInputHandler::AliESDInputHandler() :
	49	AliInputEventHandler(),
	50	fEvent(0x0),
	51	fFriend(0x0),
	52	fESDpid(0x0),
	53	fAnalysisType(0),
	54	fNEvents(0),
	55	fHLTEvent(0x0),
	56	fHLTTree(0x0),
	57	fUseHLT(kFALSE),
	58	fTagCutSumm(0x0),
	59	fUseTags(kFALSE),
	60	fChainT(0),
	61	fTreeT(0),
	62	fRunTag(0),
	63	fReadFriends(0),
	64	fFriendFileName("AliESDfriends.root")
	65	{
	66	// default constructor
	67	}
	68
	69	//______________________________________________________________________________
	70	AliESDInputHandler::~AliESDInputHandler()
	71	{
	72	// destructor
	73	}
	74
	75	//______________________________________________________________________________
	76	AliESDInputHandler::AliESDInputHandler(const char* name, const char* title):
	77	AliInputEventHandler(name, title), fEvent(0x0), fFriend(0x0), fESDpid(0x0), fAnalysisType(0),
	78	fNEvents(0), fHLTEvent(0x0), fHLTTree(0x0), fUseHLT(kFALSE), fTagCutSumm(0x0), fUseTags(kFALSE), fChainT(0), fTreeT(0), fRunTag(0), fReadFriends(0), fFriendFileName("AliESDfriends.root")
	79	{
	80	// Constructor
	81	}
	82
	83	Bool_t AliESDInputHandler::Init(TTree* tree, Option_t* opt)
	84	{
	85	//
	86	// Initialisation necessary for each new tree
	87	//
	88	fAnalysisType = opt;
	89	fTree = tree;
	90
	91	if (!fTree) return kFALSE;
	92	fTree->GetEntry(0);
	93
	94
	95	if (!fEvent) fEvent = new AliESDEvent();
	96	fEvent->ReadFromTree(fTree);
	97	fNEvents = fTree->GetEntries();
	98	return kTRUE;
	99	}
	100
	101	Bool_t AliESDInputHandler::BeginEvent(Long64_t entry)
	102	{
	103
	104	// Copy from old to new format if necessary
	105	AliESD* old = ((AliESDEvent*) fEvent)->GetAliESDOld();
	106	if (old) {
	107	((AliESDEvent*)fEvent)->CopyFromOldESD();
	108	old->Reset();
	109	}
	110
	111	if (fHLTTree) {
	112	fHLTTree->GetEntry(entry);
	113	}
	114
	115	fNewEvent = kTRUE;
	116	//
	117	// Event selection
	118	//
	119	if (fEventCuts)
	120	fIsSelectedResult = fEventCuts->GetSelectionMask((AliESDEvent*)fEvent);
	121	//
	122	// Friends
	123	((AliESDEvent*)fEvent)->SetESDfriend(fFriend);
	124	return kTRUE;
	125	}
	126
	127	Bool_t AliESDInputHandler::FinishEvent()
	128	{
	129	// Finish the event
	130	if(fEvent)fEvent->Reset();
	131	return kTRUE;
	132	}
	133
	134	Bool_t AliESDInputHandler::Notify(const char* path)
	135	{
	136	// Notify a directory change
	137	AliInfo(Form("Directory change %s \n", path));
	138	//
	139	// Handle the friends first
	140	//
	141	if (!fTree->FindBranch("ESDfriend.") && fReadFriends) {
	142	// Try to add ESDfriend. branch as friend
	143	TString esdTreeFName, esdFriendTreeFName;
	144	esdTreeFName = (fTree->GetCurrentFile())->GetName();
	145	esdFriendTreeFName = esdTreeFName;
	146	esdFriendTreeFName.ReplaceAll("AliESDs.root", fFriendFileName.Data());
	147
	148	TTree* cTree = fTree->GetTree();
	149	if (!cTree) cTree = fTree;
	150
	151	cTree->AddFriend("esdFriendTree", esdFriendTreeFName.Data());
	152	cTree->SetBranchStatus("ESDfriend.", 1);
	153	fFriend = (AliESDfriend*)(fEvent->FindListObject("AliESDfriend"));
	154	cTree->SetBranchAddress("ESDfriend.", &fFriend);
	155	}
	156	//
	157	//
	158	SwitchOffBranches();
	159	SwitchOnBranches();
	160	fFriend = (AliESDfriend*)(fEvent->FindListObject("AliESDfriend"));
	161
	162
	163	//
	164	if (fUseHLT) {
	165	// Get HLTesdTree from current file
	166	TTree* cTree = fTree;
	167	if (fTree->GetTree()) cTree = fTree->GetTree();
	168	TFile* cFile = cTree->GetCurrentFile();
	169	cFile->GetObject("HLTesdTree", fHLTTree);
	170
	171	if (fHLTTree) {
	172	if (!fHLTEvent) fHLTEvent = new AliESDEvent();
	173	fHLTEvent->ReadFromTree(fHLTTree);
	174	}
	175	}
	176
	177
	178
	179
	180	if (!fUseTags) return (kTRUE);
	181
	182	Bool_t zip = kFALSE;
	183
	184	TString fileName(path);
	185	if(fileName.Contains("#AliESDs.root")){
	186	zip = kTRUE;
	187	}
	188	else if (fileName.Contains("AliESDs.root")){
	189	fileName.ReplaceAll("AliESDs.root", "");
	190	}
	191	else if(fileName.Contains("#AliAOD.root")){
	192	zip = kTRUE;
	193	}
	194	else if(fileName.Contains("AliAOD.root")){
	195	fileName.ReplaceAll("AliAOD.root", "");
	196	}
	197	else if(fileName.Contains("#galice.root")){
	198	// For running with galice and kinematics alone...
	199	zip = kTRUE;
	200	}
	201	else if(fileName.Contains("galice.root")){
	202	// For running with galice and kinematics alone...
	203	fileName.ReplaceAll("galice.root", "");
	204	}
	205
	206
	207	TString pathName("./");
	208	if (fileName.Length() != 0) {
	209	pathName = fileName;
	210	}
	211
	212	printf("AliESDInputHandler::Notify() Path: %s\n", pathName.Data());
	213
	214	if (fRunTag) {
	215	fRunTag->Clear();
	216	} else {
	217	fRunTag = new AliRunTag();
	218	}
	219
	220	delete fTreeT; fTreeT = 0;
	221
	222	if (fChainT) {
	223	delete fChainT;
	224	fChainT = 0;
	225	}
	226
	227	if (!fChainT) {
	228	fChainT = new TChain("T");
	229	}
	230
	231
	232
	233	const char* tagPattern = "ESD.tag.root";
	234	const char* name = 0x0;
	235	TString tagFilename;
	236	if (zip) {
	237	TFile* file = fTree->GetCurrentFile();
	238	TArchiveFile* arch = file->GetArchive();
	239	TObjArray* arr = arch->GetMembers();
	240	TIter next(arr);
	241
	242	while ((file = (TFile*) next())) {
	243	name = file->GetName();
	244	if (strstr(name,tagPattern)) {
	245	tagFilename = pathName.Data();
	246	tagFilename += "#";
	247	tagFilename += name;
	248	fChainT->Add(tagFilename);
	249	AliInfo(Form("Adding %s to tag chain \n", tagFilename.Data()));
	250	}//pattern check
	251	} // archive file loop
	252	} else {
	253	void * dirp = gSystem->OpenDirectory(pathName.Data());
	254	while((name = gSystem->GetDirEntry(dirp))) {
	255	if (strstr(name,tagPattern)) {
	256	tagFilename = pathName.Data();
	257	tagFilename += "/";
	258	tagFilename += name;
	259	fChainT->Add(tagFilename);
	260	AliInfo(Form("Adding %s to tag chain \n", tagFilename.Data()));
	261	}//pattern check
	262	}//directory loop
	263	}
	264	fChainT->SetBranchAddress("AliTAG",&fRunTag);
	265	fChainT->GetEntry(0);
	266	return kTRUE;
	267	}
	268
	269
	270
	271	Option_t *AliESDInputHandler::GetDataType() const
	272	{
	273	// Returns handled data type.
	274	return gESDDataType;
	275	}
	276
	277	Int_t AliESDInputHandler::GetNEventAcceptedInFile()
	278	{
	279	// Get number of events in file accepted by the tag cuts
	280	// return -1 if no info is available
	281	if (!fTagCutSumm) {
	282	TList *luo = fTree->GetUserInfo();
	283	if (!luo) {
	284	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	285	return -1;
	286	}
	287	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	288	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	289	if (fTagCutSumm) break;
	290	}
	291	if (!fTagCutSumm) {
	292	AliInfo(Form("No tag summary map in input tree\n"));
	293	return -1;
	294	}
	295	}
	296
	297	TObjString *ostr = 0;
	298	if (fTagCutSumm->FindObject(fTree->GetCurrentFile()->GetName()))
	299	ostr = (TObjString *) fTagCutSumm->GetValue(fTree->GetCurrentFile()->GetName());
	300	else {
	301	AliInfo(Form("No tag cut summary for file %s\n", fTree->GetCurrentFile()->GetName()));
	302	return -1;
	303	}
	304	char *iTagInfo;
	305	iTagInfo = strdup(ostr->GetString().Data());
	306
	307	Int_t iAcc = atoi(strtok(iTagInfo, ","));
	308
	309	AliInfo(Form("Got %i accepted events for file %s", iAcc, fTree->GetCurrentFile()->GetName()));
	310
	311	free(iTagInfo);
	312
	313	return iAcc;
	314	}
	315	Int_t AliESDInputHandler::GetNEventRejectedInFile()
	316	{
	317	// Get number of events in file rejected by the tag cuts
	318	// return -1 if no info is available
	319	if (!fTagCutSumm) {
	320	TList *luo = fTree->GetUserInfo();
	321	if (!luo) {
	322	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	323	return -1;
	324	}
	325	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	326	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	327	if (fTagCutSumm) break;
	328	}
	329	if (!fTagCutSumm) {
	330	AliInfo(Form("No tag summary map in input tree\n"));
	331	return -1;
	332	}
	333	}
	334
	335	TObjString *ostr = 0;
	336	if (fTagCutSumm->FindObject(fTree->GetCurrentFile()->GetName()))
	337	ostr = (TObjString *) fTagCutSumm->GetValue(fTree->GetCurrentFile()->GetName());
	338	else {
	339	AliInfo(Form("No tag cut summary for file %s\n", fTree->GetCurrentFile()->GetName()));
	340	return -1;
	341	}
	342	char *iTagInfo;
	343	iTagInfo = strdup(ostr->GetString().Data());
	344
	345	strtok(iTagInfo, ",");
	346	Int_t iRej = atoi(strtok(NULL, ","));
	347
	348	AliInfo(Form("Got %i accepted events for file %s", iRej, fTree->GetCurrentFile()->GetName()));
	349
	350	free(iTagInfo);
	351
	352	return iRej;
	353	}
	354	Bool_t AliESDInputHandler::GetCutSummaryForChain(Int_t aTotal, Int_t aAccepted, Int_t *aRejected)
	355	{
	356	// Get number of events in the full chain
	357	// Count accepted and rejected events
	358	// return kFALSE if no info is available
	359	if (!fTagCutSumm) {
	360	TList *luo = fTree->GetUserInfo();
	361	if (!luo) {
	362	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	363	return kFALSE;
	364	}
	365	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	366	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	367	if (fTagCutSumm) break;
	368	}
	369	if (!fTagCutSumm) {
	370	AliInfo(Form("No tag summary map in input tree\n"));
	371	return kFALSE;
	372	}
	373	}
	374
	375	TMapIter *tIter = new TMapIter(fTagCutSumm);
	376
	377	Int_t iTotList=0, iAccList=0, iRejList=0;
	378
	379	TObject *cobj;
	380	while ((cobj = tIter->Next())) {
	381	TObjString kstr = (TObjString ) cobj;
	382	TObjString vstr = (TObjString ) fTagCutSumm->GetValue(kstr->GetString().Data());
	383	// printf("Got object value %s %s\n", kstr->GetString().Data(), vstr->GetString().Data());
	384	char *iTagInfo;
	385	iTagInfo = strdup(vstr->GetString().Data());
	386
	387	Int_t iAcc = atoi(strtok(iTagInfo, ","));
	388	Int_t iRej = atoi(strtok(NULL, ","));
	389
	390	iAccList += iAcc;
	391	iRejList += iRej;
	392	iTotList += (iAcc+iRej);
	393	}
	394
	395	*aTotal = iTotList;
	396	*aAccepted = iAccList;
	397	*aRejected = iRejList;
	398
	399	return kTRUE;
	400	}
	401
	402	Int_t AliESDInputHandler::GetNFilesEmpty()
	403	{
	404	// Count number of files in which all events were de-selected
	405	// For such files Notify() will NOT be called
	406	// return -1 if no info is available
	407	if (!fTagCutSumm) {
	408	TList *luo = fTree->GetUserInfo();
	409	if (!luo) {
	410	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	411	return -1;
	412	}
	413	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	414	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	415	if (fTagCutSumm) break;
	416	}
	417	if (!fTagCutSumm) {
	418	AliInfo(Form("No tag summary map in input tree\n"));
	419	return -1;
	420	}
	421	}
	422
	423	TMapIter *tIter = new TMapIter(fTagCutSumm);
	424
	425	Int_t iFilesEmpty = 0;
	426
	427	TObject *cobj;
	428	while ((cobj = tIter->Next())) {
	429	TObjString kstr = (TObjString ) cobj;
	430	TObjString vstr = (TObjString ) fTagCutSumm->GetValue(kstr->GetString().Data());
	431	// printf("Got object value %s %s\n", kstr->GetString().Data(), vstr->GetString().Data());
	432	char *iTagInfo;
	433	iTagInfo = strdup(vstr->GetString().Data());
	434
	435	Int_t iAcc = atoi(strtok(iTagInfo, ","));
	436	Int_t iRej = atoi(strtok(NULL, ","));
	437
	438	if ((iAcc == 0) && ((iRej+iAcc)>0))
	439	iFilesEmpty++;
	440	}
	441
	442	return iFilesEmpty;
	443
	444	}