JRootFile.hh

Go to the documentation of this file.

#ifndef __JROOTFILE__
#define __JROOTFILE__
 
#include "TFile.h"
#include "TError.h"
 
#include "JLang/JAccessible.hh"
#include "JLang/JException.hh"
#include "JLang/JThrow.hh"
#include "JLang/JStorage.hh"
 
 
/**
 * \author mdejong
 */
 
namespace JROOT {}
namespace JPP { using namespace JROOT; }
 
namespace JROOT {
 
  using JLANG::JAccessible;
  using JLANG::JStorage;
  using JLANG::JFileOpenException;
  using JLANG::JThrow;
 
 
  /**
   * ROOT file.
   *
   * This class implements the methods is_open() and close() of the JLANG::JAccessible interface.
   */
  class JRootFile :
    public virtual JAccessible,
    public JStorage<TFile>
  {
  protected:
    /**
     * Default constructor.
     * The printing of ROOT errors is suppressed. 
     */
    JRootFile()
    {
      gErrorIgnoreLevel = kError;
    }
 
 
    /**
     * Destructor.
     *
     * The destructor closes the file if it is still open.
     */
    ~JRootFile()
    {
      close();
    }
 
 
  public:
    /**
     * Get file.
     *
     * \return                 pointer to file
     */
    TFile* getFile() const 
    {
      return get();
    }
 
 
    /**
     * Check is file is open.
     *
     * \return                 true if open; else false
     */
    virtual bool is_open() const
    {
      return (is_valid() && getFile()->IsOpen());
    }
 
 
    /**
     * Close file.
     */
    virtual void close()
    {
      if (is_open()) {
        getFile()->Close();
      }
 
      reset();
    }
  };
 
 
  /**
   * ROOT input file.
   *
   * This class implements the method open() of the JLANG::JAccessible interface.
   */
  class JRootInputFile :
    public JRootFile
  {
  public:
    /**
     * Default constructor.
     */
    JRootInputFile() :
      JRootFile()
    {}
 
 
    /**
     * Constructor.
     *
     * \param  file_name      file name
     */
    JRootInputFile(const char* file_name) :
      JRootFile()
    {
      open(file_name);
    }
 
 
    /**
     * Open file.
     * The file is not opened when no file exists with the given name.
     *
     * \param  file_name      file name
     */
    virtual void open(const char* file_name)
    {
      set(TFile::Open(file_name, "READ"));
 
      if (!is_open()) {
        Throw(JFileOpenException(std::string("Error opening file ") + file_name));
      }
    }
  };
 
 
  /**
   * ROOT output file.
   *
   * This class implements the method open() of the JLANG::JAccessible interface.
   */
  class JRootOutputFile :
    public JRootFile
  {
  public:
    /**
     * Default constructor.
     */
    JRootOutputFile() :
      JRootFile()
    {}
 
 
    /**
     * Constructor.
     *
     * \param  file_name      file name
     */
    JRootOutputFile(const char* file_name) :
      JRootFile()
    {
      open(file_name);
    }
 
 
    /**
     * Open file.
     * The file is not opened when a file exists with the given name.
     *
     * \param  file_name      file name
     */
    virtual void open(const char* file_name)
    {
      set(TFile::Open(file_name, "CREATE"));
 
      if (!is_open()) {
        Throw(JFileOpenException(std::string("Error opening file ") + file_name));
      }
    }
 
 
    /**
     * Close file.
     * This method calls the TFile::Write method before closing the file.
     */
    virtual void close()
    {
      if (is_open()) {
        getFile()->Write();
        getFile()->Close();
      }
 
      reset();
    }
  };
}
 
#endif