JWeightFileScanner.hh

Go to the documentation of this file.

#ifndef __JSUPPORT__JWEIGHTFILESCANNER__ 
#define __JSUPPORT__JWEIGHTFILESCANNER__ 

#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"

#include "JAAnet/JHead.hh"
#include "JAAnet/JWeightEventToolkit.hh"
#include "JAAnet/JWeightEventHelper.hh"

#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"

/**
 * \author bjung
 */

namespace JSUPPORT {

  using JAANET::JWeightEvent;
  using JAANET::JWeightEventHelper;
  
  
  /**
   * Template file scanner with event weight.
   */
  template<class JFileScanner_t = JMultipleFileScanner<Evt> >
  struct JWeightFileScanner :
    public JWeightEventHelper,
    public JFileScanner_t
  {
    /**
     * Default constructor.
     */
    JWeightFileScanner()
    {}
    
    
    /**
     * Constructor.
     *
     * \param  weighter          event weighter
     */
    JWeightFileScanner(const JWeightEvent& weighter)
    {
      reset(weighter);
    }
    
    
    /**
     * Reset file scanner and event weighter.
     *
     * \param  weighter          event weighter
     */
    void reset(const JWeightEvent& weighter)
    {
      JWeightEventHelper::configure(weighter);
      JFileScanner_t    ::clear();
    }
    
    
    /**
     * Put list of files.
     *
     * \param  input              file list
     * \return                    number of added files
     */
    size_t put(const JFileScanner_t& input)
    {
      size_t n = 0;
      
      for (typename JFileScanner_t::const_iterator i = input.begin(); i != input.end(); ++i) {
        n += size_t(this->put(*i));
      }
      
      return n;
    }
    
    
    /**
     * Put file.
     *
     * \param  input              file name
     * \return                    true if successfully added; else false.
     */
    bool put(const std::string& input)
    {
      using namespace JPP;

      const JHead head = JSUPPORT::getHeader(input);
      
      if (this->check(head)) {

        JWeightEventHelper::add(head);
        JFileScanner_t    ::push_back(input);

        return true;
        
      } else {

        return false;
      }
    }
  };
}

#endif