// Author: Markus Oldenburg, CERN
//-------------------------------------------------------------------------
+#include <TTree.h>
+
#include "AliAODEvent.h"
#include "AliAODHeader.h"
#include "AliAODTrack.h"
ClassImp(AliAODEvent)
+// definition of std AOD member names
+ const char* AliAODEvent::fAODListName[kAODListN] = {"header",
+ "tracks",
+ "vertices",
+ "v0s",
+ "tracklets",
+ "jets",
+ "emcalCells",
+ "phosCells",
+ "caloClusters",
+ "fmdClusters",
+ "pmdClusters"
+};
//______________________________________________________________________________
AliAODEvent::AliAODEvent() :
+ AliVEvent(),
fAODObjects(new TList()),
- fHeader((AliAODHeader*)fAODObjects->At(0)),
- fTracks((TClonesArray*)fAODObjects->At(1)),
- fVertices((TClonesArray*)fAODObjects->At(2)),
- fClusters((TClonesArray*)fAODObjects->At(3)),
- fJets((TClonesArray*)fAODObjects->At(4))
+ fHeader(0),
+ fTracks(0),
+ fVertices(0),
+ fV0s(0),
+ fTracklets(0),
+ fJets(0),
+ fEmcalCells(0),
+ fPhosCells(0),
+ fCaloClusters(0),
+ fFmdClusters(0),
+ fPmdClusters(0)
{
// default constructor
}
//______________________________________________________________________________
AliAODEvent::~AliAODEvent()
{
- // destructor
-
- delete fAODObjects;
+// destructor
+ delete fAODObjects;
}
//______________________________________________________________________________
void AliAODEvent::AddObject(TObject* obj)
{
- // Add an object to the list of object.
+ // Add an object to the list of objects.
// Please be aware that in order to increase performance you should
// refrain from using TObjArrays (if possible). Use TClonesArrays, instead.
}
//______________________________________________________________________________
-TObject *AliAODEvent::GetObject(const char *objName) const
+void AliAODEvent::RemoveObject(TObject* obj)
+{
+ // Removes an object from the list of objects.
+
+ fAODObjects->Remove(obj);
+}
+
+//______________________________________________________________________________
+TObject *AliAODEvent::FindListObject(const char *objName)
{
// Return the pointer to the object with the given name.
AddObject(new AliAODHeader());
AddObject(new TClonesArray("AliAODTrack", 0));
AddObject(new TClonesArray("AliAODVertex", 0));
- AddObject(new TClonesArray("AliAODCluster", 0));
+ AddObject(new TClonesArray("AliAODv0", 0));
+ AddObject(new AliAODTracklets());
AddObject(new TClonesArray("AliAODJet", 0));
+ AddObject(new AliAODCaloCells());
+ AddObject(new AliAODCaloCells());
+ AddObject(new TClonesArray("AliAODCaloCluster", 0));
+ AddObject(new TClonesArray("AliAODFmdCluster", 0));
+ AddObject(new TClonesArray("AliAODPmdCluster", 0));
+ // set names
+ SetStdNames();
// read back pointers
GetStdContent();
- // set names
- fTracks->SetName("tracks");
- fVertices->SetName("vertices");
- fClusters->SetName("clusters");
- fJets->SetName("jets");
-
+ return;
}
//______________________________________________________________________________
-void AliAODEvent::GetStdContent() const
+void AliAODEvent::SetStdNames()
+{
+ // introduce the standard naming
+
+ if(fAODObjects->GetEntries()==kAODListN){
+ for(int i = 0;i < fAODObjects->GetEntries();i++){
+ TObject *fObj = fAODObjects->At(i);
+ if(fObj->InheritsFrom("TNamed")){
+ ((TNamed*)fObj)->SetName(fAODListName[i]);
+ }
+ else if(fObj->InheritsFrom("TClonesArray")){
+ ((TClonesArray*)fObj)->SetName(fAODListName[i]);
+ }
+ }
+ }
+ else{
+ printf("%s:%d SetStdNames() Wrong number of Std Entries \n",(char*)__FILE__,__LINE__);
+ }
+}
+
+//______________________________________________________________________________
+void AliAODEvent::GetStdContent()
{
// set pointers for standard content
- fHeader = (AliAODHeader*)fAODObjects->At(0);
- fTracks = (TClonesArray*)fAODObjects->At(1);
- fVertices = (TClonesArray*)fAODObjects->At(2);
- fClusters = (TClonesArray*)fAODObjects->At(3);
- fJets = (TClonesArray*)fAODObjects->At(4);
+ fHeader = (AliAODHeader*)fAODObjects->FindObject("header");
+ fTracks = (TClonesArray*)fAODObjects->FindObject("tracks");
+ fVertices = (TClonesArray*)fAODObjects->FindObject("vertices");
+ fV0s = (TClonesArray*)fAODObjects->FindObject("v0s");
+ fTracklets = (AliAODTracklets*)fAODObjects->FindObject("tracklets");
+ fJets = (TClonesArray*)fAODObjects->FindObject("jets");
+ fEmcalCells = (AliAODCaloCells*)fAODObjects->FindObject("emcalCells");
+ fPhosCells = (AliAODCaloCells*)fAODObjects->FindObject("phosCells");
+ fCaloClusters = (TClonesArray*)fAODObjects->FindObject("caloClusters");
+ fFmdClusters = (TClonesArray*)fAODObjects->FindObject("fmdClusters");
+ fPmdClusters = (TClonesArray*)fAODObjects->FindObject("pmdClusters");
}
//______________________________________________________________________________
-void AliAODEvent::ResetStd(Int_t trkArrSize, Int_t vtxArrSize)
+void AliAODEvent::ResetStd(Int_t trkArrSize,
+ Int_t vtxArrSize,
+ Int_t v0ArrSize,
+ Int_t jetSize,
+ Int_t caloClusSize,
+ Int_t fmdClusSize,
+ Int_t pmdClusSize)
{
- // deletes content of standard arrays and resets size
+ // deletes content of standard arrays and resets size
+
fTracks->Delete();
if (trkArrSize > fTracks->GetSize())
fTracks->Expand(trkArrSize);
fVertices->Delete();
if (vtxArrSize > fVertices->GetSize())
fVertices->Expand(vtxArrSize);
+
+ fV0s->Delete();
+ if (v0ArrSize > fV0s->GetSize())
+ fV0s->Expand(v0ArrSize);
+
+ fJets->Delete();
+ if (jetSize > fJets->GetSize())
+ fJets->Expand(jetSize);
+
+ fCaloClusters->Delete();
+ if (caloClusSize > fCaloClusters->GetSize())
+ fCaloClusters->Expand(caloClusSize);
+
+ fFmdClusters->Delete();
+ if (fmdClusSize > fFmdClusters->GetSize())
+ fFmdClusters->Expand(fmdClusSize);
+
+ fPmdClusters->Delete();
+ if (pmdClusSize > fPmdClusters->GetSize())
+ fPmdClusters->Expand(pmdClusSize);
+
+ // Reset the tracklets
+ fTracklets->DeleteContainer();
+ fPhosCells->DeleteContainer();
+ fEmcalCells->DeleteContainer();
+
}
+void AliAODEvent::ClearStd()
+{
+ // clears the standard arrays
+ fTracks ->Clear();
+ fVertices ->Clear();
+ fV0s ->Clear();
+ fTracklets ->DeleteContainer();
+ fJets ->Delete();
+ fEmcalCells ->DeleteContainer();
+ fPhosCells ->DeleteContainer();
+ fCaloClusters ->Clear();
+ fFmdClusters ->Clear();
+ fPmdClusters ->Clear();
+}
+
+//______________________________________________________________________________
+Int_t AliAODEvent::GetMuonTracks(TRefArray *muonTracks) const
+{
+ // fills the provided TRefArray with all found muon tracks
+
+ muonTracks->Clear();
+
+ AliAODTrack *track = 0;
+ for (Int_t iTrack = 0; iTrack < GetNTracks(); iTrack++) {
+ if ((track = GetTrack(iTrack))->IsMuonTrack()) {
+ muonTracks->Add(track);
+ }
+ }
+
+ return muonTracks->GetEntriesFast();
+}
+void AliAODEvent::ReadFromTree(TTree *tree)
+{
+ // connects aod event to tree
+
+ if(!tree){
+ Printf("%s %d AliAODEvent::ReadFromTree() Zero Pointer to Tree \n",(char*)__FILE__,__LINE__);
+ return;
+ }
+ // load the TTree
+ if(!tree->GetTree())tree->LoadTree(0);
+
+ // Try to find AliAODEvent
+ AliAODEvent *aodEvent = 0;
+ aodEvent = (AliAODEvent*)tree->GetTree()->GetUserInfo()->FindObject("AliAODEvent");
+ if(aodEvent){
+ // Check if already connected to tree
+ TList* connectedList = (TList*) (tree->GetUserInfo()->FindObject("AODObjectsConnectedToTree"));
+ if (connectedList) {
+ // If connected use the connected list if objects
+ fAODObjects->Delete();
+ fAODObjects = connectedList;
+ GetStdContent();
+ return;
+ }
+ // Connect to tree
+ // prevent a memory leak when reading back the TList
+ delete fAODObjects;
+ fAODObjects = 0;
+ // create a new TList from the UserInfo TList...
+ // copy constructor does not work...
+ fAODObjects = (TList*)(aodEvent->GetList()->Clone());
+ fAODObjects->SetOwner(kFALSE);
+ if(fAODObjects->GetEntries()<kAODListN){
+ printf("%s %d AliAODEvent::ReadFromTree() TList contains less than the standard contents %d < %d \n",
+ (char*)__FILE__,__LINE__,fAODObjects->GetEntries(),kAODListN);
+ }
+ // set the branch addresses
+ TIter next(fAODObjects);
+ TNamed *el;
+ while((el=(TNamed*)next())){
+ TString bname(el->GetName());
+ // check if branch exists under this Name
+ TBranch *br = tree->GetBranch(bname.Data());
+ if(br){
+ tree->SetBranchAddress(bname.Data(),fAODObjects->GetObjectRef(el));
+ }
+ else{
+ br = tree->GetBranch(Form("%s.",bname.Data()));
+ if(br){
+ tree->SetBranchAddress(Form("%s.",bname.Data()),fAODObjects->GetObjectRef(el));
+ }
+ else{
+ printf("%s %d AliAODEvent::ReadFromTree() No Branch found with Name %s or %s. \n",
+ (char*)__FILE__,__LINE__,bname.Data(),bname.Data());
+ }
+ }
+ }
+
+ GetStdContent();
+ // when reading back we are not owner of the list
+ // must not delete it
+ fAODObjects->SetOwner(kFALSE);
+ fAODObjects->SetName("AODObjectsConnectedToTree");
+ // we are not owner of the list objects
+ // must not delete it
+ tree->GetUserInfo()->Add(fAODObjects);
+ }// no aodEvent
+ else {
+ // we can't get the list from the user data, create standard content
+ // and set it by hand
+ CreateStdContent();
+ TIter next(fAODObjects);
+ TNamed *el;
+ while((el=(TNamed*)next())){
+ TString bname(el->GetName());
+ tree->SetBranchAddress(bname.Data(),fAODObjects->GetObjectRef(el));
+ }
+ GetStdContent();
+ // when reading back we are not owner of the list
+ // must not delete it
+ fAODObjects->SetOwner(kFALSE);
+ }
+}
+
+//______________________________________________________________________________
+void AliAODEvent::Print(Option_t *) const
+{
+ // Something meaningful should be implemented here.
+
+ return;
+}
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff