[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"
#include "AliESDpid.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),
  fEventTag(0),
  fReadFriends(0),
  fFriendFileName("AliESDfriends.root")
{
  // default constructor
}

//______________________________________________________________________________
AliESDInputHandler::~AliESDInputHandler() 
{
  //  destructor
  if (fRunTag) delete fRunTag;
  delete fESDpid;
}

//______________________________________________________________________________
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), fEventTag(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();

    if (fMixingHandler) fMixingHandler->Init(tree,  opt);

    return kTRUE;
}

//______________________________________________________________________________
Bool_t AliESDInputHandler::BeginEvent(Long64_t entry)
{
    
    // Copy from old to new format if necessary
  static Bool_t called = kFALSE;
  if (!called && fEventCuts && IsUserCallSelectionMask())
     AliInfo(Form("The ESD input handler expects that the first task calls AliESDInputHandler::CheckSelectionMask() %s", fEventCuts->ClassName()));
  AliESD* old = ((AliESDEvent*) fEvent)->GetAliESDOld();
  if (old) {
   ((AliESDEvent*)fEvent)->CopyFromOldESD();
   old->Reset();
  }

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

  if (fMixingHandler) fMixingHandler->BeginEvent(entry);
  if (fUseTags && fRunTag) {
    fEventTag = 0;
    if (entry >= fRunTag->GetNEvents()) {
      AliError(Form("Current event %d does not match max range from run tag: 0-%d", (Int_t)entry, fRunTag->GetNEvents()));
      return kTRUE;
    }
    fEventTag = fRunTag->GetEventTag(entry);   
  }      
  return kTRUE;
}

//______________________________________________________________________________
void AliESDInputHandler::CheckSelectionMask()
{
// This method can be called by a task only if IsUserCallSelectionMask is true.
   if (!fEventCuts || !IsUserCallSelectionMask()) return;
   fIsSelectedResult = fEventCuts->GetSelectionMask((AliESDEvent*)fEvent);
}
   
//______________________________________________________________________________
Bool_t  AliESDInputHandler::FinishEvent()
{
    // Finish the event 
  if(fEvent)fEvent->Reset();
  if (fMixingHandler) fMixingHandler->FinishEvent();
  return kTRUE;
} 

//______________________________________________________________________________
Bool_t AliESDInputHandler::Notify(const char* path)
{
  // Notify a directory change
  static Bool_t firsttime = kFALSE;
  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"));
    if (fFriend) 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) {
    if (fMixingHandler) fMixingHandler->Notify(path);
    return kTRUE;
  }
    
  Bool_t zip = kFALSE;
    
  // Setup the base path
  TString pathName(path);
  Int_t index = pathName.Index("#");
  if (index>=0) {
    zip = kTRUE;
    pathName = pathName(0,index);
  } else {
    pathName = gSystem->DirName(pathName);
  }
  if (fTree->GetCurrentFile()->GetArchive()) zip = kTRUE;
  if (pathName.IsNull()) pathName = "./";  
  printf("AliESDInputHandler::Notify() Path: %s\n", pathName.Data());

  if (fRunTag) {
    fRunTag->Clear();
  } else {
    fRunTag = new AliRunTag();
  }
    
  const char* tagPattern = "ESD.tag.root";
  TString sname;
  TString tagFilename;
  if (zip) {
    TObjArray* arr = fTree->GetCurrentFile()->GetArchive()->GetMembers();
    TIter next(arr);
    TObject *objarchive;
    while ((objarchive = next())) {
      sname = objarchive->GetName();
	   if (sname.Contains(tagPattern)) { 
        tagFilename = pathName;
        if (index>=0) tagFilename += "#";
        else tagFilename += "/";
        tagFilename += sname;
        AliInfo(Form("Tag file found: %s\n", tagFilename.Data()));
        break; // There should be only one such file in the archive
      }//pattern check
    } // archive file loop
  } else {
    void * dirp = gSystem->OpenDirectory(pathName.Data());
    while(1) {
      sname = gSystem->GetDirEntry(dirp);
      if (sname.IsNull()) break;
      if (sname.Contains(tagPattern)) { 
        tagFilename = pathName;
        tagFilename += "/";
        tagFilename += sname;
        AliInfo(Form("Tag file found: %s\n", tagFilename.Data()));
        break;
      }//pattern check
    }//directory loop
    gSystem->FreeDirectory(dirp);
  }
  if (tagFilename.IsNull()) {
    if (firsttime) AliWarning(Form("Tag file not found in directory: %s", pathName.Data()));
    firsttime = kFALSE;
    delete fRunTag; fRunTag = 0;
    return kTRUE;
  }
  TFile *tagfile = TFile::Open(tagFilename);
  if (!tagfile) {
    AliError(Form("Cannot open tag file: %s", tagFilename.Data()));
    delete fRunTag; fRunTag = 0;
    return kTRUE;
  }   
  fTreeT = (TTree*)tagfile->Get("T"); // file is the owner
  if (!fTreeT) {
    AliError(Form("Cannot get tree of tags from file: %s", tagFilename.Data()));
    delete fRunTag; fRunTag = 0;
    return kTRUE;
  }
    
  fTreeT->SetBranchAddress("AliTAG",&fRunTag);
  fTreeT->GetEntry(0);
  delete tagfile;
  // Notify the mixing handler after the tags are loaded
  if (fMixingHandler) fMixingHandler->Notify(path);
  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;
  
}

//______________________________________________________________________________
TObject *AliESDInputHandler::GetStatistics(Option_t *option) const
{
// Get the statistics histogram(s) from the physics selection object. This
// should be called during FinishTaskOutput(). Option can be empty (default
// statistics histogram) or BIN0.
   if (!fEventCuts) return NULL;
   TString opt(option);
   opt.ToUpper();
   if (opt=="BIN0") return fEventCuts->GetStatistics("BIN0");
   else return fEventCuts->GetStatistics("ALL");
}   

//______________________________________________________________________________
void AliESDInputHandler::CreatePIDResponse(Bool_t isMC/*=kFALSE*/)
{
  //
  // create the pid response object if it does not exist yet
  //
  if (fESDpid) return;
  fESDpid=new AliESDpid(isMC);

}
Commit	Line	Data
d3dd3d14	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
933fd60f	23	#include <TTree.h>
1aa09fd1	24	#include <TChain.h>
	25	#include <TFile.h>
	26	#include <TArchiveFile.h>
	27	#include <TObjArray.h>
	28	#include <TSystem.h>
300d5701	29	#include <TString.h>
300d5701	30	#include <TObjString.h>
80e1b60a	31	#include <TProcessID.h>
f0f29480	32	#include <TMap.h>
d3dd3d14	33
	34	#include "AliESDInputHandler.h"
	35	#include "AliESDEvent.h"
620ad672	36	#include "AliESDfriend.h"
b2af62ab	37	#include "AliVCuts.h"
d3dd3d14	38	#include "AliESD.h"
1aa09fd1	39	#include "AliRunTag.h"
1aa09fd1	40	#include "AliEventTag.h"
300d5701	41	#include "AliLog.h"
9006fe9c	42	#include "AliESDpid.h"
d3dd3d14	43
	44	ClassImp(AliESDInputHandler)
	45
c2b6979d	46	static Option_t *gESDDataType = "ESD";
c2b6979d	47
d3dd3d14	48	//______________________________________________________________________________
6989bff3	49	AliESDInputHandler::AliESDInputHandler() :
6989bff3	50	AliInputEventHandler(),
0cd61c1d	51	fEvent(0x0),
620ad672	52	fFriend(0x0),
67675e42	53	fESDpid(0x0),
1aa09fd1	54	fAnalysisType(0),
4d2a7370	55	fNEvents(0),
6ccd727b	56	fHLTEvent(0x0),
	57	fHLTTree(0x0),
	58	fUseHLT(kFALSE),
f0f29480	59	fTagCutSumm(0x0),
1aa09fd1	60	fUseTags(kFALSE),
	61	fChainT(0),
	62	fTreeT(0),
98c482f7	63	fRunTag(0),
cec9e119	64	fEventTag(0),
038879cc	65	fReadFriends(0),
b626cd71	66	fFriendFileName("AliESDfriends.root")
d3dd3d14	67	{
	68	// default constructor
	69	}
	70
	71	//______________________________________________________________________________
	72	AliESDInputHandler::~AliESDInputHandler()
	73	{
1aa09fd1	74	// destructor
cec9e119	75	if (fRunTag) delete fRunTag;
9006fe9c	76	delete fESDpid;
d3dd3d14	77	}
	78
	79	//______________________________________________________________________________
	80	AliESDInputHandler::AliESDInputHandler(const char* name, const char* title):
67675e42	81	AliInputEventHandler(name, title), fEvent(0x0), fFriend(0x0), fESDpid(0x0), fAnalysisType(0),
cec9e119	82	fNEvents(0), fHLTEvent(0x0), fHLTTree(0x0), fUseHLT(kFALSE), fTagCutSumm(0x0), fUseTags(kFALSE), fChainT(0), fTreeT(0), fRunTag(0), fEventTag(0), fReadFriends(0), fFriendFileName("AliESDfriends.root")
d3dd3d14	83	{
e12e402c	84	// Constructor
d3dd3d14	85	}
d3dd3d14	86
aaf2d706	87	//______________________________________________________________________________
1aa09fd1	88	Bool_t AliESDInputHandler::Init(TTree* tree, Option_t* opt)
d3dd3d14	89	{
80ce1b4e	90	//
300d5701	91	// Initialisation necessary for each new tree
80ce1b4e	92	//
1aa09fd1	93	fAnalysisType = opt;
7b2a359f	94	fTree = tree;
300d5701	95
6073f8c9	96	if (!fTree) return kFALSE;
cd843abf	97	// fTree->GetEntry(0);
6ccd727b	98
b481861d	99
efea3f54	100	if (!fEvent) fEvent = new AliESDEvent();
60996693	101	fEvent->ReadFromTree(fTree);
4d2a7370	102	fNEvents = fTree->GetEntries();
b890a10d	103
7b2a359f	104	if (fMixingHandler) fMixingHandler->Init(tree, opt);
b890a10d	105
d3dd3d14	106	return kTRUE;
	107	}
	108
aaf2d706	109	//______________________________________________________________________________
6ccd727b	110	Bool_t AliESDInputHandler::BeginEvent(Long64_t entry)
d3dd3d14	111	{
9b98c4df	112
d3dd3d14	113	// Copy from old to new format if necessary
aaf2d706	114	static Bool_t called = kFALSE;
	115	if (!called && fEventCuts && IsUserCallSelectionMask())
	116	AliInfo(Form("The ESD input handler expects that the first task calls AliESDInputHandler::CheckSelectionMask() %s", fEventCuts->ClassName()));
d7749dec	117	AliESD* old = ((AliESDEvent*) fEvent)->GetAliESDOld();
d7749dec	118	if (old) {
cec9e119	119	((AliESDEvent*)fEvent)->CopyFromOldESD();
cec9e119	120	old->Reset();
d7749dec	121	}
6ccd727b	122
	123	if (fHLTTree) {
	124	fHLTTree->GetEntry(entry);
	125	}
	126
f0ee78b0	127	fNewEvent = kTRUE;
b2af62ab	128	//
	129	// Event selection
	130	//
aaf2d706	131	fIsSelectedResult = 0;
	132	if (fEventCuts && !IsUserCallSelectionMask())
	133	fIsSelectedResult = fEventCuts->GetSelectionMask((AliESDEvent*)fEvent);
620ad672	134	//
	135	// Friends
	136	((AliESDEvent*)fEvent)->SetESDfriend(fFriend);
aaf2d706	137	called = kTRUE;
7b2a359f	138
cec9e119	139	if (fMixingHandler) fMixingHandler->BeginEvent(entry);
	140	if (fUseTags && fRunTag) {
	141	fEventTag = 0;
	142	if (entry >= fRunTag->GetNEvents()) {
	143	AliError(Form("Current event %d does not match max range from run tag: 0-%d", (Int_t)entry, fRunTag->GetNEvents()));
	144	return kTRUE;
	145	}
	146	fEventTag = fRunTag->GetEventTag(entry);
	147	}
d7749dec	148	return kTRUE;
d3dd3d14	149	}
d3dd3d14	150
aaf2d706	151	//______________________________________________________________________________
	152	void AliESDInputHandler::CheckSelectionMask()
	153	{
	154	// This method can be called by a task only if IsUserCallSelectionMask is true.
	155	if (!fEventCuts \|\| !IsUserCallSelectionMask()) return;
b2747f93	156	fIsSelectedResult = fEventCuts->GetSelectionMask((AliESDEvent*)fEvent);
aaf2d706	157	}
	158
	159	//______________________________________________________________________________
e12e402c	160	Bool_t AliESDInputHandler::FinishEvent()
	161	{
	162	// Finish the event
b890a10d	163	if(fEvent)fEvent->Reset();
7b2a359f	164	if (fMixingHandler) fMixingHandler->FinishEvent();
b890a10d	165	return kTRUE;
6989bff3	166	}
6989bff3	167
aaf2d706	168	//______________________________________________________________________________
1aa09fd1	169	Bool_t AliESDInputHandler::Notify(const char* path)
1aa09fd1	170	{
cec9e119	171	// Notify a directory change
	172	static Bool_t firsttime = kFALSE;
	173	AliInfo(Form("Directory change %s \n", path));
	174	//
	175	// Handle the friends first
	176	//
	177	if (!fTree->FindBranch("ESDfriend.") && fReadFriends) {
	178	// Try to add ESDfriend. branch as friend
	179	TString esdTreeFName, esdFriendTreeFName;
	180	esdTreeFName = (fTree->GetCurrentFile())->GetName();
	181	esdFriendTreeFName = esdTreeFName;
	182	esdFriendTreeFName.ReplaceAll("AliESDs.root", fFriendFileName.Data());
	183	TTree* cTree = fTree->GetTree();
	184	if (!cTree) cTree = fTree;
	185	cTree->AddFriend("esdFriendTree", esdFriendTreeFName.Data());
	186	cTree->SetBranchStatus("ESDfriend.", 1);
80ce1b4e	187	fFriend = (AliESDfriend*)(fEvent->FindListObject("AliESDfriend"));
cec9e119	188	if (fFriend) cTree->SetBranchAddress("ESDfriend.", &fFriend);
	189	}
	190	//
	191	//
	192	SwitchOffBranches();
	193	SwitchOnBranches();
	194	fFriend = (AliESDfriend*)(fEvent->FindListObject("AliESDfriend"));
	195	//
	196	if (fUseHLT) {
	197	// Get HLTesdTree from current file
	198	TTree* cTree = fTree;
	199	if (fTree->GetTree()) cTree = fTree->GetTree();
	200	TFile* cFile = cTree->GetCurrentFile();
	201	cFile->GetObject("HLTesdTree", fHLTTree);
50aaaf00	202
cec9e119	203	if (fHLTTree) {
	204	if (!fHLTEvent) fHLTEvent = new AliESDEvent();
	205	fHLTEvent->ReadFromTree(fHLTTree);
50aaaf00	206	}
cec9e119	207	}
50aaaf00	208
cec9e119	209	if (!fUseTags) {
	210	if (fMixingHandler) fMixingHandler->Notify(path);
	211	return kTRUE;
	212	}
1aa09fd1	213
cec9e119	214	Bool_t zip = kFALSE;
1aa09fd1	215
cec9e119	216	// Setup the base path
	217	TString pathName(path);
	218	Int_t index = pathName.Index("#");
	219	if (index>=0) {
	220	zip = kTRUE;
	221	pathName = pathName(0,index);
	222	} else {
	223	pathName = gSystem->DirName(pathName);
	224	}
	225	if (fTree->GetCurrentFile()->GetArchive()) zip = kTRUE;
	226	if (pathName.IsNull()) pathName = "./";
	227	printf("AliESDInputHandler::Notify() Path: %s\n", pathName.Data());
6ccd727b	228
cec9e119	229	if (fRunTag) {
	230	fRunTag->Clear();
	231	} else {
	232	fRunTag = new AliRunTag();
	233	}
1aa09fd1	234
cec9e119	235	const char* tagPattern = "ESD.tag.root";
	236	TString sname;
	237	TString tagFilename;
	238	if (zip) {
	239	TObjArray* arr = fTree->GetCurrentFile()->GetArchive()->GetMembers();
	240	TIter next(arr);
	241	TObject *objarchive;
	242	while ((objarchive = next())) {
	243	sname = objarchive->GetName();
	244	if (sname.Contains(tagPattern)) {
	245	tagFilename = pathName;
	246	if (index>=0) tagFilename += "#";
	247	else tagFilename += "/";
	248	tagFilename += sname;
	249	AliInfo(Form("Tag file found: %s\n", tagFilename.Data()));
	250	break; // There should be only one such file in the archive
	251	}//pattern check
	252	} // archive file loop
	253	} else {
	254	void * dirp = gSystem->OpenDirectory(pathName.Data());
	255	while(1) {
	256	sname = gSystem->GetDirEntry(dirp);
	257	if (sname.IsNull()) break;
	258	if (sname.Contains(tagPattern)) {
	259	tagFilename = pathName;
	260	tagFilename += "/";
	261	tagFilename += sname;
	262	AliInfo(Form("Tag file found: %s\n", tagFilename.Data()));
	263	break;
	264	}//pattern check
	265	}//directory loop
62e57953	266	gSystem->FreeDirectory(dirp);
cec9e119	267	}
	268	if (tagFilename.IsNull()) {
	269	if (firsttime) AliWarning(Form("Tag file not found in directory: %s", pathName.Data()));
	270	firsttime = kFALSE;
	271	delete fRunTag; fRunTag = 0;
1aa09fd1	272	return kTRUE;
cec9e119	273	}
	274	TFile *tagfile = TFile::Open(tagFilename);
	275	if (!tagfile) {
	276	AliError(Form("Cannot open tag file: %s", tagFilename.Data()));
	277	delete fRunTag; fRunTag = 0;
	278	return kTRUE;
	279	}
	280	fTreeT = (TTree*)tagfile->Get("T"); // file is the owner
	281	if (!fTreeT) {
	282	AliError(Form("Cannot get tree of tags from file: %s", tagFilename.Data()));
	283	delete fRunTag; fRunTag = 0;
	284	return kTRUE;
	285	}
	286
	287	fTreeT->SetBranchAddress("AliTAG",&fRunTag);
	288	fTreeT->GetEntry(0);
	289	delete tagfile;
	290	// Notify the mixing handler after the tags are loaded
	291	if (fMixingHandler) fMixingHandler->Notify(path);
	292	return kTRUE;
1aa09fd1	293	}
1aa09fd1	294
aaf2d706	295	//______________________________________________________________________________
c2b6979d	296	Option_t *AliESDInputHandler::GetDataType() const
	297	{
	298	// Returns handled data type.
	299	return gESDDataType;
	300	}
f0f29480	301
aaf2d706	302	//______________________________________________________________________________
f0f29480	303	Int_t AliESDInputHandler::GetNEventAcceptedInFile()
	304	{
	305	// Get number of events in file accepted by the tag cuts
	306	// return -1 if no info is available
	307	if (!fTagCutSumm) {
	308	TList *luo = fTree->GetUserInfo();
	309	if (!luo) {
	310	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	311	return -1;
	312	}
	313	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	314	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	315	if (fTagCutSumm) break;
	316	}
	317	if (!fTagCutSumm) {
	318	AliInfo(Form("No tag summary map in input tree\n"));
	319	return -1;
	320	}
	321	}
	322
	323	TObjString *ostr = 0;
	324	if (fTagCutSumm->FindObject(fTree->GetCurrentFile()->GetName()))
	325	ostr = (TObjString *) fTagCutSumm->GetValue(fTree->GetCurrentFile()->GetName());
	326	else {
	327	AliInfo(Form("No tag cut summary for file %s\n", fTree->GetCurrentFile()->GetName()));
	328	return -1;
	329	}
	330	char *iTagInfo;
	331	iTagInfo = strdup(ostr->GetString().Data());
	332
	333	Int_t iAcc = atoi(strtok(iTagInfo, ","));
	334
	335	AliInfo(Form("Got %i accepted events for file %s", iAcc, fTree->GetCurrentFile()->GetName()));
	336
	337	free(iTagInfo);
	338
	339	return iAcc;
	340	}
aaf2d706	341
aaf2d706	342	//______________________________________________________________________________
f0f29480	343	Int_t AliESDInputHandler::GetNEventRejectedInFile()
	344	{
	345	// Get number of events in file rejected by the tag cuts
	346	// return -1 if no info is available
	347	if (!fTagCutSumm) {
	348	TList *luo = fTree->GetUserInfo();
	349	if (!luo) {
	350	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	351	return -1;
	352	}
	353	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	354	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	355	if (fTagCutSumm) break;
	356	}
	357	if (!fTagCutSumm) {
	358	AliInfo(Form("No tag summary map in input tree\n"));
	359	return -1;
	360	}
	361	}
	362
	363	TObjString *ostr = 0;
	364	if (fTagCutSumm->FindObject(fTree->GetCurrentFile()->GetName()))
	365	ostr = (TObjString *) fTagCutSumm->GetValue(fTree->GetCurrentFile()->GetName());
	366	else {
	367	AliInfo(Form("No tag cut summary for file %s\n", fTree->GetCurrentFile()->GetName()));
	368	return -1;
	369	}
	370	char *iTagInfo;
	371	iTagInfo = strdup(ostr->GetString().Data());
	372
	373	strtok(iTagInfo, ",");
	374	Int_t iRej = atoi(strtok(NULL, ","));
	375
	376	AliInfo(Form("Got %i accepted events for file %s", iRej, fTree->GetCurrentFile()->GetName()));
	377
	378	free(iTagInfo);
	379
	380	return iRej;
	381	}
aaf2d706	382
aaf2d706	383	//______________________________________________________________________________
5e6a3170	384	Bool_t AliESDInputHandler::GetCutSummaryForChain(Int_t aTotal, Int_t aAccepted, Int_t *aRejected)
c334d0d3	385	{
	386	// Get number of events in the full chain
	387	// Count accepted and rejected events
	388	// return kFALSE if no info is available
	389	if (!fTagCutSumm) {
	390	TList *luo = fTree->GetUserInfo();
	391	if (!luo) {
	392	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	393	return kFALSE;
	394	}
	395	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	396	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	397	if (fTagCutSumm) break;
	398	}
	399	if (!fTagCutSumm) {
	400	AliInfo(Form("No tag summary map in input tree\n"));
	401	return kFALSE;
	402	}
	403	}
	404
	405	TMapIter *tIter = new TMapIter(fTagCutSumm);
	406
	407	Int_t iTotList=0, iAccList=0, iRejList=0;
	408
	409	TObject *cobj;
9a587f77	410	while ((cobj = tIter->Next())) {
c334d0d3	411	TObjString kstr = (TObjString ) cobj;
	412	TObjString vstr = (TObjString ) fTagCutSumm->GetValue(kstr->GetString().Data());
	413	// printf("Got object value %s %s\n", kstr->GetString().Data(), vstr->GetString().Data());
	414	char *iTagInfo;
	415	iTagInfo = strdup(vstr->GetString().Data());
	416
	417	Int_t iAcc = atoi(strtok(iTagInfo, ","));
	418	Int_t iRej = atoi(strtok(NULL, ","));
	419
	420	iAccList += iAcc;
	421	iRejList += iRej;
	422	iTotList += (iAcc+iRej);
	423	}
	424
	425	*aTotal = iTotList;
	426	*aAccepted = iAccList;
	427	*aRejected = iRejList;
	428
	429	return kTRUE;
	430	}
	431
aaf2d706	432	//______________________________________________________________________________
c334d0d3	433	Int_t AliESDInputHandler::GetNFilesEmpty()
	434	{
	435	// Count number of files in which all events were de-selected
	436	// For such files Notify() will NOT be called
	437	// return -1 if no info is available
	438	if (!fTagCutSumm) {
	439	TList *luo = fTree->GetUserInfo();
	440	if (!luo) {
	441	AliInfo(Form("No user info in input tree - no tag cut summary\n"));
	442	return -1;
	443	}
	444	for (int iluo=0; iluo<luo->GetEntries(); iluo++) {
	445	fTagCutSumm = dynamic_cast<TMap *>(luo->At(iluo));
	446	if (fTagCutSumm) break;
	447	}
	448	if (!fTagCutSumm) {
	449	AliInfo(Form("No tag summary map in input tree\n"));
	450	return -1;
	451	}
	452	}
	453
	454	TMapIter *tIter = new TMapIter(fTagCutSumm);
	455
	456	Int_t iFilesEmpty = 0;
	457
	458	TObject *cobj;
9a587f77	459	while ((cobj = tIter->Next())) {
c334d0d3	460	TObjString kstr = (TObjString ) cobj;
	461	TObjString vstr = (TObjString ) fTagCutSumm->GetValue(kstr->GetString().Data());
	462	// printf("Got object value %s %s\n", kstr->GetString().Data(), vstr->GetString().Data());
	463	char *iTagInfo;
	464	iTagInfo = strdup(vstr->GetString().Data());
	465
	466	Int_t iAcc = atoi(strtok(iTagInfo, ","));
	467	Int_t iRej = atoi(strtok(NULL, ","));
	468
	469	if ((iAcc == 0) && ((iRej+iAcc)>0))
	470	iFilesEmpty++;
	471	}
	472
	473	return iFilesEmpty;
	474
	475	}
70908d05	476
	477	//______________________________________________________________________________
	478	TObject AliESDInputHandler::GetStatistics(Option_t option) const
	479	{
	480	// Get the statistics histogram(s) from the physics selection object. This
	481	// should be called during FinishTaskOutput(). Option can be empty (default
	482	// statistics histogram) or BIN0.
	483	if (!fEventCuts) return NULL;
	484	TString opt(option);
	485	opt.ToUpper();
	486	if (opt=="BIN0") return fEventCuts->GetStatistics("BIN0");
	487	else return fEventCuts->GetStatistics("ALL");
	488	}
9006fe9c	489
	490	//______________________________________________________________________________
	491	void AliESDInputHandler::CreatePIDResponse(Bool_t isMC/=kFALSE/)
	492	{
	493	//
	494	// create the pid response object if it does not exist yet
	495	//
	496	if (fESDpid) return;
	497	fESDpid=new AliESDpid(isMC);
	498
	499	}
	500