Changeset 1080 for trunk/MagicSoft/Mars/mbase/MReadTree.cc

Timestamp:

11/15/01 11:07:21 (23 years ago)

Author:

tbretz

Message:

*** empty log message ***

File:

• trunk/MagicSoft/Mars/mbase/MReadTree.cc (modified) (22 diffs)

trunk/MagicSoft/Mars/mbase/MReadTree.cc

@@ -16 +16 @@
 !
 !
-!   Author(s): Thomas Bretz  12/2000 (tbretz@uni-sw.gwdg.de)
+!   Author(s): Thomas Bretz  12/2000 <mailto:tbretz@uni-sw.gwdg.de>
 !
 !   Copyright: MAGIC Software Development, 2000-2001
@@ -63 +63 @@
 
 #include "MTime.h"
+#include "MFilter.h"
 #include "MParList.h"
 #include "MTaskList.h"
@@ -86 +87 @@
     fTitle = title ? title : "Task to loop over all events in one single tree";
 
-    fVetoList = new TOrdCollection;
+    fVetoList = new TList;
     fVetoList->SetOwner();
 
-    fNotify = new TOrdCollection;
+    fNotify = new TList;
 
     //
@@ -96 +97 @@
     fChain = new TChain(tname);
     fChain->SetNotify(this);
+
+    // root 3.02:
+    // In TChain::Addfile remove the limitation that the file name must contain
+    // the string ".root". ".root" is necessary only in case one wants to specify
+    // a Tree in a subdirectory of a Root file with eg, the format:
 
     if (fname)
@@ -183 +189 @@
         return;
 
-    *fLog << "Branch choosing method enabled (only enabled branches are read)." << endl;
+    *fLog << inf << "Branch choosing method enabled (only enabled branches are read)." << endl;
     fChain->SetBranchStatus("*", kFALSE);
     fBranchChoosing = kTRUE;
@@ -223 +229 @@
     //
     fChain->SetBranchStatus(name, status);
-    *fLog << (status ? "Enabled" : "Disabled");
+    *fLog << inf << (status ? "Enabled" : "Disabled");
     *fLog << " subbranch '" << name << "'." << endl;
 }
@@ -231 +237 @@
 // Set the status of all branches in the list to status.
 //
-void MReadTree::SetBranchStatus(const TOrdCollection *list, Bool_t status)
+void MReadTree::SetBranchStatus(const TList *list, Bool_t status)
 {
     //
@@ -247 +253 @@
 //  This is the implementation of the Auto Enabling Scheme.
 //  For more information see MTask::AddBranchToList.
-//  This function loops over all tasks in the tasklist and enables
-//  all branches which are requested by the tasks.
-//
-//  To enable 'unknown' branches which are not in the Branchlist of
+//  This function loops over all tasks and its filters in the tasklist
+//  and enables all branches which are requested by the tasks and its
+//  filters.
+//
+//  To enable 'unknown' branches which are not in the branchlist of
 //  the tasks you can call EnableBranch
 //
@@ -267 +274 @@
     if (!tlist)
     {
-        *fLog << "Cannot use auto enabeling scheme for branches. 'MTaskList' not found." << endl;
+        *fLog << warn << "Cannot use auto enabeling scheme for branches. 'MTaskList' not found." << endl;
         return;
     }
 
     //
-    // Loop over all tasks in the task list.
+    // Loop over all tasks iand its filters n the task list.
     //
     MTask *task;
     TIter NextTask(tlist->GetList());
     while ((task=(MTask*)NextTask()))
+    {
         SetBranchStatus(task->GetListOfBranches(), kTRUE);
+        const MFilter *filter = task->GetFilter();
+
+        if (filter)
+            SetBranchStatus(filter->GetListOfBranches(), kTRUE);
+
+    }
 }
 
@@ -317 +331 @@
     if (!fNumEntries)
     {
-        *fLog << dbginf << "No entries found in file(s)." << endl;
+        *fLog << warn << dbginf << "No entries found in file(s)." << endl;
         return kFALSE;
     }
@@ -324 +338 @@
     // output logging information
     //
-    *fLog << fNumEntries << " entries found in file(s)." << endl;
+    *fLog << inf << fNumEntries << " entries found in file(s)." << endl;
 
     //
@@ -354 +368 @@
         if (fVetoList->FindObject(oname))
         {
-            *fLog << "Master branch " << bname << " has veto... skipped." << endl;
+            *fLog << inf << "Master branch " << bname << " has veto... skipped." << endl;
             DisableSubBranches(branch);
             continue;
@@ -377 +391 @@
             // we cannot proceed reading this branch
             //
-            *fLog << dbginf << "Warning: Class '" << oname << "' not existing in dictionary. Branch skipped." << endl;
+            *fLog << warn << dbginf << "Warning: Class '" << oname << "' not existing in dictionary. Branch skipped." << endl;
             DisableSubBranches(branch);
             continue;
@@ -387 +401 @@
         //
         fChain->SetBranchAddress(bname, pcont);
-        *fLog << "Master branch address " << bname << " setup for reading." << endl;
+        *fLog << inf << "Master branch address " << bname << " setup for reading." << endl;
 
         //*fLog << "Branch " << bname << " autodel: " << (int)branch->IsAutoDelete() << endl;
@@ -395 +409 @@
     }
 
-    *fLog << "MReadTree setup " << num << " master branches addresses." << endl;
+    *fLog << inf << "MReadTree setup " << num << " master branches addresses." << endl;
 
     //
@@ -410 +424 @@
 
     return kTRUE;
+}
+
+// --------------------------------------------------------------------------
+//
+//  Set the ready to save flag of all containers which branchaddresses are
+//  set for. This is necessary to copy data.
+//
+void MReadTree::SetReadyToSave(Bool_t flag)
+{
+    TIter Next(fChain->GetStatus());
+
+    TChainElement *element = NULL;
+    while ((element=(TChainElement*)Next()))
+    {
+        //
+        // Check whether the branch is enabled
+        //
+        if (!element->GetStatus())
+            continue;
+
+        //
+        // Get the pointer to the pointer of the corresponding container
+        //
+        MParContainer **pcont = (MParContainer**)element->GetBaddress();
+
+        //
+        // Check whether the pointer is not NULL
+        //
+        if (!pcont || !*pcont)
+            continue;
+
+        //
+        // Set the ready to save status of the container.
+        //
+        (*pcont)->SetReadyToSave(flag);
+    }
+
+    //
+    // Set the ready to save status of this task (used?), too
+    //
+    MTask::SetReadyToSave(flag);
 }
 
@@ -421 +476 @@
 Bool_t MReadTree::Process()
 {
+    SetReadyToSave();
     return fChain->GetEntry(fNumEntry++) != 0;
 }
@@ -430 +486 @@
 Bool_t MReadTree::GetEvent()
 {
+    SetReadyToSave();
     return fChain->GetEntry(fNumEntry) != 0;
 }
@@ -442 +499 @@
     if (fNumEntry-dec >= fNumEntries)
     {
-        *fLog << "MReadTree::DecEventNum: WARNING - Event " << fNumEntry << "-";
+        *fLog << warn << "MReadTree::DecEventNum: WARNING - Event " << fNumEntry << "-";
         *fLog << dec << "=" << (Int_t)fNumEntry-dec << " out of Range." << endl;
         return kFALSE;
@@ -460 +517 @@
     if (fNumEntry+inc >= fNumEntries)
     {
-        *fLog << "MReadTree::IncEventNum: WARNING - Event " << fNumEntry << "+";
+        *fLog << warn << "MReadTree::IncEventNum: WARNING - Event " << fNumEntry << "+";
         *fLog << inc << "=" << (Int_t)fNumEntry+inc << " out of Range." << endl;
         return kFALSE;
@@ -480 +537 @@
     if (nr >= fNumEntries)
     {
-        *fLog << "MReadTree::SetEventNum: WARNING - " << nr << " out of Range." << endl;
+        *fLog << warn << "MReadTree::SetEventNum: WARNING - " << nr << " out of Range." << endl;
         return kFALSE;
     }
@@ -522 +579 @@
     fNotify->Add(obj);
 }
+
+void MReadTree::Print(Option_t *o) const
+{
+    *fLog << all << GetDescriptor() << dec << endl;
+    *fLog << setfill('-') << setw(strlen(GetDescriptor())) << "" << endl;
+    *fLog << " Files:" << endl;
+
+    int i = 0;
+    TIter Next(fChain->GetListOfFiles());
+    TObject *obj = NULL;
+    while ((obj=Next()))
+        *fLog << " " << i++ << ") " << obj->GetName() << endl;
+
+    *fLog << " Entries: " << fNumEntries << endl;
+    *fLog << " Next Entry: " << fNumEntry << endl;
+}