TFriendElement.cxx

Go to the documentation of this file.

// @(#)root/tree:$Id$
// Author: Rene Brun   07/04/2001
 
/*************************************************************************
 * Copyright (C) 1995-2001, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
/** \class TFriendElement
\ingroup tree
 
A TFriendElement TF describes a TTree object TF in a file.
When a TFriendElement TF is added to the list of friends of an
existing TTree T, any variable from TF can be referenced in a query
to T.
 
To add a TFriendElement to an existing TTree T, do:
~~~ {.cpp}
    T.AddFriend("friendTreename","friendTreeFile");
~~~
See TTree::AddFriend for more information.
*/
 
#include "TFriendElement.h"
#include "TTree.h"
#include "TFile.h"
#include "TROOT.h"
#include "TChain.h"
 
ClassImp(TFriendElement);
 
////////////////////////////////////////////////////////////////////////////////
/// Default constructor for a friend element.
 
TFriendElement::TFriendElement() : TNamed()
{
   fFile       = nullptr;
   fTree       = nullptr;
   fOwnFile    = false;
   fParentTree = nullptr;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Create a friend element.
///
/// If treename is of the form "a=b", an alias called "a" is created for
/// treename = "b" by default the alias name is the name of the tree.
 
TFriendElement::TFriendElement(TTree *tree, const char *treename, const char *filename)
    :TNamed(treename,filename)
{
   fFile       = nullptr;
   fTree       = nullptr;
   fOwnFile    = true;
   fParentTree = tree;
   fTreeName   = treename;
   if (treename && strchr(treename,'=')) {
      char *temp = Compress(treename);
      char *equal = strchr(temp,'=');
      if (!equal) return;;
      *equal=0;
      fTreeName = equal+1;
      SetName(temp);
      delete [] temp;
   }
 
   Connect();
}
 
////////////////////////////////////////////////////////////////////////////////
/// Create a friend element.
///
/// If treename is of the form "a=b", an alias called "a" is created for
/// treename = "b" by default the alias name is the name of the tree.
/// The passed TFile is managed by the user (i.e. user must delete the TFile).
 
TFriendElement::TFriendElement(TTree *tree, const char *treename, TFile *file)
    :TNamed(treename,file?file->GetName():"")
{
   fFile       = file;
   fTree       = nullptr;
   fOwnFile    = false;
   fParentTree = tree;
   fTreeName   = treename;
   if (fParentTree && fParentTree->GetDirectory()
       && fParentTree->GetDirectory()->GetFile() == fFile) {
      // The friend and the TTree are in the same file, let's not record
      // the filename.
      SetTitle("");
   }
   if (treename && strchr(treename,'=')) {
      char *temp = Compress(treename);
      char *equal = strchr(temp,'=');
      if (!equal) return;;
      *equal=0;
      fTreeName = equal+1;
      SetName(temp);
      delete [] temp;
   }
 
   Connect();
}
 
////////////////////////////////////////////////////////////////////////////////
/// Create a friend element.
 
TFriendElement::TFriendElement(TTree *tree, TTree* friendtree, const char *alias)
   : TNamed(friendtree?friendtree->GetName():"",
            friendtree
            ? (   friendtree->GetDirectory()
                  ? (    friendtree->GetDirectory()->GetFile()
                         ? friendtree->GetDirectory()->GetFile()->GetName()
                         :  "")
                  : "")
            :  "")
{
   fTree       = friendtree;
   fTreeName   = "";
   fFile       = nullptr;
   fOwnFile    = false;
   fParentTree = tree;
   if (fTree) {
      fTreeName   = fTree->GetName();
      if (fTree->GetDirectory()) fFile = fTree->GetDirectory()->GetFile();
      if (fParentTree && fParentTree->GetDirectory()
          && fParentTree->GetDirectory()->GetFile() == fFile) {
         // The friend and the TTree are in the same file, let's not record
         // the filename.
         SetTitle("");
      }
   } else {
      MakeZombie(); // ROOT-7007
   }
   if (alias && strlen(alias)) {
      char *temp = Compress(alias);
      SetName(temp);
      delete [] temp;
   }
 
   if (fTree)
      fTree->RegisterExternalFriend(this);
   // No need to Connect.
}
 
////////////////////////////////////////////////////////////////////////////////
/// Destructor.  Disconnect from the owning tree if needed.
 
TFriendElement::~TFriendElement()
{
   DisConnect();
}
 
////////////////////////////////////////////////////////////////////////////////
/// Connect file and return TTree.
 
TTree *TFriendElement::Connect()
{
   GetFile();
   auto treePtr = GetTree();
   if (!treePtr) MakeZombie(); // ROOT-7007
   return treePtr;
}
 
////////////////////////////////////////////////////////////////////////////////
/// DisConnect file and TTree.
 
TTree *TFriendElement::DisConnect()
{
   // At this point, if the condition is not meant, fTree is usually already
   // deleted (hence the need for a local bit describing fTree)
   if (fTree)
      fTree->RemoveExternalFriend(this);
   if (fOwnFile) delete fFile;
   fFile = nullptr;
   fTree = nullptr;
   return nullptr;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Return pointer to TFile containing this friend TTree.
 
TFile *TFriendElement::GetFile()
{
   if (fFile || IsZombie()) return fFile;
 
   if (strlen(GetTitle())) {
      TDirectory::TContext ctxt;
      fFile = TFile::Open(GetTitle());
      fOwnFile = true;
   } else {
      TDirectory *dir = fParentTree->GetDirectory();
      if (dir) {
         fFile = dir->GetFile();
         fOwnFile = false;
      }
   }
   if (fFile && fFile->IsZombie()) {
      MakeZombie();
      delete fFile;
      fFile = nullptr;
   }
   return fFile;
}
 
////////////////////////////////////////////////////////////////////////////////
/// Return pointer to friend TTree.
 
TTree *TFriendElement::GetTree()
{
   if (fTree) {
      // In cases where the friend was added by the owning chain and the friend is
      // a chain we recorded the address of the chain but we need to return the address
      // of the underlying current TTree.
      if (TestBit(kFromChain))
         return fTree->GetTree();
      else
         return fTree;
   }
 
   if (GetFile()) {
      fFile->GetObject(GetTreeName(),fTree);
   } else {
      // This could be a memory tree or chain
      fTree = dynamic_cast<TTree*>( gROOT->FindObject(GetTreeName()) );
   }
 
   if (fTree)
      fTree->RegisterExternalFriend(this);
 
   return fTree;
}
 
////////////////////////////////////////////////////////////////////////////////
/// List this friend element.
 
void TFriendElement::ls(Option_t *) const
{
   printf(" Friend Tree: %s in file: %s\n",GetName(),GetTitle());
}
 
////////////////////////////////////////////////////////////////////////////////
/// Forget deleted elements.
 
void TFriendElement::RecursiveRemove(TObject *delobj)
{
   if (delobj == fTree)
      fTree = nullptr;
   if (delobj == fFile)
      fFile = nullptr;
}