updating UE leading track analysis (Sara & Jan Fiete)

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

diff --git a/STEER/TTreeStream.cxx b/STEER/TTreeStream.cxx
index 445f1fb5825bec547d2dafa2470deded8a23f338..0e2013362015f315d4369e9cb1a99afa7277a81a 100644 (file)
--- a/STEER/TTreeStream.cxx
+++ b/STEER/TTreeStream.cxx
@@ -325,7 +325,8 @@ Int_t TTreeStream::CheckIn(Char_t type, void *pointer)
   //
   // Insert object of given type
   //
-  if (!fElements) fElements = new TObjArray(1000);
+  if (!fElements) fElements = new TObjArray(10000);
+  if (fElements->GetSize()<=fCurrentIndex) fElements->Expand(fCurrentIndex*2);
   TTreeDataElement* element = (TTreeDataElement*)fElements->At(fCurrentIndex);
   if (!element) {
     element = new TTreeDataElement(type);
@@ -400,8 +401,8 @@ void TTreeStream::BuildTree(){
   //
   // Build the Tree
   //
-  if (fTree->GetEntries()>0) return;
-  fTree = new TTree(GetName(),GetName());
+  if (fTree && fTree->GetEntries()>0) return;
+  if (!fTree)  fTree = new TTree(GetName(),GetName());
   Int_t entries = fElements->GetEntriesFast();  
   fBranches = new TObjArray(entries);
   
@@ -468,7 +469,7 @@ TTreeStream & TTreeStream::Endl()
 }
 
 
-TTreeStream  &TTreeStream::operator<<(Char_t *name)
+TTreeStream  &TTreeStream::operator<<(const Char_t *name)
 {
   //
   // Endl