JHeadWriter.hh

Go to the documentation of this file.

#ifndef __JAANET__JHEADWRITER__
#define __JAANET__JHEADWRITER__
 
#include "km3net-dataformat/offline/Head.hh"
 
#include "JLang/JNullStream.hh"
#include "JROOT/JRootDictionary.hh"
#include "JROOT/JRootStreamer.hh"
#include "JAAnet/JHead.hh"
#include "JAAnet/JAAnetDictionary.hh"
 
 
/**
 * \author mdejong
 */
 
namespace JAANET {
 
  using JLANG::JEquation;
  using JLANG::JEquationParameters;
  using JROOT::JRootDictionary_t;
  using JROOT::JRootWriter;
 
  
  /**
   * Implementation for Head output of JHead objects with ROOT dictionary.
   */
  struct JHeadWriter :
    public JRootWriter
  {
    using JRootWriter::put;
 
    /**
     * Constructor.
     *
     * \param  header          header output
     * \param  parameters      equation parameters
     * \param  dictionary      dictionary
     */
    JHeadWriter(Head& header,
                const JEquationParameters& parameters,
                const JRootDictionary_t&   dictionary) :
      JRootWriter(JLANG::null, parameters, dictionary),
      header(header)
    {}
 
 
    /**
     * Write equation.
     *
     * \param  equation        equation
     * \return                 this ROOT writer
     */
    virtual JRootWriter& put(const JEquation& equation) override 
    {
      using namespace std;
 
      string key = equation.getKey();
 
      for (int i = 1; header.find(key) != header.end(); ++i) {
        key = getTag(equation.getKey(), i);
      }
 
      header[key] = equation.getValue();
      
      return *this;
    }
 
  private:
    Head& header;
  }; 
}
 
#endif