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