JTreeWriterObjectOutput.hh

Go to the documentation of this file.

#ifndef __JROOT__JTREEWRITEROBJECTOUTPUT__
#define __JROOT__JTREEWRITEROBJECTOUTPUT__

#include "JLang/JObjectOutput.hh"
#include "JROOT/JTreeWriter.hh"


/**
 * \author mdejong
 */

namespace JROOT {}
namespace JPP { using namespace JROOT; }

namespace JROOT {

  using JLANG::JObjectOutput;


  /**
   * JTreeWriter object output.
   *
   * This class implements the JLANG::JObjectOutput interface.
   */
  template<class T>
  class JTreeWriterObjectOutput :
    public virtual JObjectOutput<T>
  {
  protected:
    /**
     * Default constructor.
     */
    JTreeWriterObjectOutput() :
      out(new JTreeWriter<T>(getTreeParameters<T>())),
      del(true)
    {}


    /**
     * Destructor.
     */
    ~JTreeWriterObjectOutput()
    {
      if (del) {
        delete out;
      }
    }


    /**
     * Get TreeWriter.
     *
     * \return            TreeWriter
     */
    JTreeWriter<T>& getTreeWriter() 
    {
      return *out;
    }


    /**
     * Set directory.
     *
     * \param  dir        pointer to directory
     */
    void SetDirectory(TDirectory* dir)
    {
      out->SetDirectory(dir);

      del = (dir == NULL);
    }


    JTreeWriter<T>* out;
    bool            del;  // delete TTree on destruction 

  public:
    /**
     * Object output.
     *
     * \param  object     object
     * \return            true if OK; else false
     */
    virtual bool put(const T& object)
    {
      return (out->Write(object) > 0);
    }
  };
}

#endif