JTreeReader.hh

Go to the documentation of this file.

#ifndef __JROOT__JTREEREADER__
#define __JROOT__JTREEREADER__
 
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wall"
#include "TFile.h"
#include "TTree.h"
#include "TDictionary.h"
#pragma GCC diagnostic pop
 
#include "JROOT/JRootAddress.hh"
#include "JROOT/JTreeParameters.hh"
#include "JLang/JPointer.hh"
#include "JROOT/JTreeBranchAddresses.hh"
 
/**
 * \file
 * TTree reading for template data type.
 * \author mdejong
 */
namespace JROOT {}
namespace JPP { using namespace JROOT; }
 
namespace JROOT {
 
  using JLANG::JPointer;
  
 
  /**
   * Interface for TTree reading.
   */
  struct JAbstractTreeReader : 
    public JPointer<TTree>
  {
    /**
     * Load TTree from given file.
     *
     * \param  file          pointer to file
     * \return               true when TTree correctly loaded; else false
     */
    virtual bool load(TFile* file) = 0;
 
 
    /**
     * Get ROOT dictionary.
     *
     * \return               dictionary.
     */
    virtual const TDictionary* getDictionary() = 0;
  };
 
 
  /**
   * Auxiliary class for template TTree reading.
   */
  template<class T>
  class JTreeReader : 
    public JAbstractTreeReader,
    public JTreeParameters,
    public JTreeBranchAddresses<T>
  {
  public:
    /**
     * Constructor.
     *
     * Note that the default TTree parameters are obtained using method JROOT::getTreeParameters.
     *
     * \param  parameters    parameters of TTree
     */
    JTreeReader(const JTreeParameters& parameters = JROOT::getTreeParameters<T>()) :
      JTreeParameters(parameters)
    {}
 
 
    /**
     * Load TTree from given file.
     *
     * This methods releases the pending memory allocated from previous data reading.
     *
     * \param  file          pointer to file
     * \return               true when TTree correctly loaded; else false
     */
    virtual bool load(TFile* file) override 
    {
      this->release();
 
      if (file != NULL) {
 
        this->set(dynamic_cast<TTree*>(file->Get(this->getTreeName())));
      
        if (this->is_valid()) {
 
         this->setBranchAddress(*(this->get()));
 
          return true;
        }
      }
 
      return false;
    }
 
 
    /**
     * Get ROOT dictionary.
     *
     * \return               dictionary.
     */
    virtual const TDictionary* getDictionary() override 
    {
      return TDictionary::GetDictionary(T::Class_Name());
    }
  };
}
 
#endif