JTreeRecorder.hh

Go to the documentation of this file.

#ifndef __JSUPPORT_JTREERECORDER__
#define __JSUPPORT_JTREERECORDER__
 
#include "JLang/JObjectOutput.hh"
#include "JROOT/JTreeWriterObjectOutput.hh"
#include "JROOT/JRootFile.hh"
 
/**
 * \author mdejong
 */
 
namespace JSUPPORT {}
namespace JPP { using namespace JSUPPORT; }
 
namespace JSUPPORT {
 
  using JLANG::JAccessibleObjectOutput;
  using JROOT::JTreeWriterObjectOutput;
  using JROOT::JRootOutputFile;
 
 
  /**
   * ROOT TTree object output.
   *
   * This class implements the JLANG::JAccessibleObjectOutput interface.
   */
  template<class T>
  class JTreeRecorder :
    public JTreeWriterObjectOutput<T>,
    public JAccessibleObjectOutput<T>,
    public JRootOutputFile
  {
  public:
    /**
     * Default constructor.
     */
    JTreeRecorder() 
    {}
 
 
    /**
     * Constructor.
     *
     * \param  file_name      file name
     */
    JTreeRecorder(const char* file_name) 
    {
      open(file_name);
    }
 
 
    /**
     * Open file.
     *
     * \param  file_name      file name
     */
    virtual void open(const char* file_name) override 
    {
      JRootOutputFile::open(file_name);
 
      if (is_open()) {
        this->SetDirectory(getFile());
      }
    }
 
    
    /**
     * Close file.
     */
    virtual void close() override 
    {
      // Write objects in memory.
 
      if (is_open()) {
        getFile()->Write();
      }
 
      // Detach TTree from TDirectory to avoid deletion of TTree.
 
      this->SetDirectory(0);
 
      // Close file.
 
      if (is_open()) {
        getFile()->Close();
      }
 
      reset();
 
      // Reset TTree.
 
      this->Reset();
    }
  };
}
 
#endif