JEvtWeightFileScanner.hh

Go to the documentation of this file.

#ifndef __JSUPPORT__JEVTWEIGHTFILESCANNER__ 
#define __JSUPPORT__JEVTWEIGHTFILESCANNER__ 
 
#include <set>
#include <vector>
#include <string>
#include <algorithm>
 
#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"
 
#include "JLang/JException.hh"
 
#include "JAAnet/JHead.hh"
 
#include "JAAnet/JEvtCategory.hh"
#include "JAAnet/JEvtCategoryHelper.hh"
#include "JAAnet/JEvtCategoryToolkit.hh"
#include "JAAnet/JEvtCategoryMap.hh"
 
#include "JAAnet/JEvtWeight.hh"
#include "JAAnet/JEvtWeightHelper.hh"
#include "JAAnet/JEvtWeightToolkit.hh"
 
#include "JAAnet/JEvtWeightFactor.hh"
#include "JAAnet/JEvtWeightFactorHelper.hh"
#include "JAAnet/JFluxHelper.hh"
 
#include "JSupport/JSupport.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"
 
 
/**
 * \author bjung
 */
 
namespace JSUPPORT {}
namespace JPP { using namespace JSUPPORT; }
 
namespace JSUPPORT {
 
  using JAANET::JHead;
  
  using JAANET::JEvtCategoryHelper;
  using JAANET::JEvtCategorySet;
  using JAANET::JEvtCategoryMap;
  
  using JAANET::JEvtWeight;
  using JAANET::JEvtWeightHelper;
 
  using JAANET::JEvtWeightFactorHelper;
  using JAANET::JFluxHelper;
  
  
  /**
   * Event-weighter-associated file organiser.
   */
  struct JEvtWeightFileScanner :
    public JEvtWeightHelper,
    public JMultipleFileScanner_t
  {
    typedef JMultipleFileScanner_t        input_type;
    
    
    /**
     * Default constructor.
     */
    JEvtWeightFileScanner() :
      JEvtWeightHelper(),
      JMultipleFileScanner_t()
    {}
    
    
    /**
     * Constructor.
     *
     * \param  weighter          event weighter
     */
    JEvtWeightFileScanner(const JEvtWeight& weighter) :
      JEvtWeightHelper(),
      JMultipleFileScanner_t()
    {
      reset(weighter);
    }
    
    
    /**
     * Constructor.
     *
     * \param  input             input
     */
    JEvtWeightFileScanner(const input_type& input) :
      JEvtWeightHelper(),
      JMultipleFileScanner_t()
    {
      using namespace JPP;
      
      const JHead&      header   = JSUPPORT::getHeader(input);
      const JEvtWeight& weighter = getEventWeighter(header);
      
      reset(weighter);
      put  (input);
    }
    
    
    /**
     * Reset file scanner and event weighter.
     *
     * \param  weighter          event weighter
     */
    void reset(const JEvtWeight& weighter)
    {
      JEvtWeightHelper      ::configure(weighter);
      JMultipleFileScanner_t::clear();
    }
    
    
    /**
     * Put files.
     *
     * \param  input              input files
     * \return                    number of added files
     */
    size_t put(const input_type& input)
    {
      size_t n = 0;
      
      for (typename input_type::const_iterator i = input.begin(); i != input.end(); ++i) {
        n += size_t(this->put(*i));
      }
      
      return n;
    }
    
    
    /**
     * Put file.
     *
     * \param  input              input file
     * \return                    true if successfully added; else false.
     */
    bool put(const std::string& input)
    {
      using namespace JPP;
 
      JHead head = JSUPPORT::getHeader(input);
 
      head.createUUID(); // Ensure valid UUID
      
      if (this->check(head)) {
 
        JEvtWeightHelper      ::add(head);
        JMultipleFileScanner_t::push_back(input);
 
        return true;
        
      } else {
 
        return false;
      }
    }
 
 
    /**
     * Set event-weight factor for the event-weighter associated with this file scanner.
     *
     * The template argument refers to the desired helper class for an event-weight factor\n
     * (c.f. `JAANET::JEvtWeightFactorHelper`, `JAANET::JFluxHelper` or `JAANET::JDiffuseFluxHelper`).
     *
     * \param  factor                      event-weight factor
     * \return                             true if event-weight factor set successfully; else false
     */
    template<class JEvtWeightFactorHelper_t>
    bool setEvtWeightFactor(const JEvtWeightFactorHelper_t& factor)
    {
      JEvtWeightFactorHelper_t* helper = dynamic_cast<JEvtWeightFactorHelper_t*>(this->get());
        
      if (helper != NULL) {
          
        helper->configure(factor);
          
        return true;
        
      } else {
        
        return false;
      }
    }
    
 
    /**
     * Set event-weight factor corresponding to a given event category.
     *
     * \param  category                    event category
     * \param  factor                      event-weight factor
     * \return                             true if event-weight factor set successfully; else false
     */
    bool setEvtWeightFactor(const JEvtCategoryHelper&     category,
                            const JEvtWeightFactorHelper& factor)
    {
      const JHead& header = this->getHeader();
      
      return (category.match(header) ? setEvtWeightFactor(factor) : false);
    }
 
 
    /**
     * Set event-weight factor corresponding to a given set of event categories.
     * 
     * Note that the given event-weight factor will only be assigned\n
     * if the header matches < b>all< /b> of the specified event categories.
     *
     * \param  categories                  set of event categories
     * \param  factor                      event-weight factor
     * \return                             true if event-weight factor set successfully; else false
     */
    bool setEvtWeightFactor(const JEvtCategorySet&        categories,
                            const JEvtWeightFactorHelper& factor)
    {
      using namespace std;
      using namespace JPP;
      
      bool matching = (!categories.empty());
      
      const JHead& header = this->getHeader();
      
      const JEvtCategorySet cats = getCategories(header);
      
      for (JEvtCategorySet::const_iterator i = categories.cbegin(); i != categories.cend() && matching; ++i) {
        
        JEvtCategorySet::const_iterator j = std::find(cats.cbegin(), cats.cend(), *i);
        
        matching = (j != cats.cend());
      }
 
      return (matching ? setEvtWeightFactor(factor) : false);
    }
 
 
    /**
     * Set event-weight factor according to a given map between event categories and event-weight factors.
     *
     * The template argument refers either to the class `JAANET::JEvtWeightFactorHelper` or `JAANET::JFluxHelper`.
     *
     * Note that the given event-weight factors will only be assigned\n
     * if the header matches < b>all< /b> of the event categories associated with the given map.
     *
     * \param  object                     map between event categories and event-weight factors
     * \return                            true if event-weight factors are set successfully; else false
     */
    template<class JEvtWeightFactorHelper_t>
    bool setEvtWeightFactor(const JEvtCategoryMap<JEvtWeightFactorHelper_t>& object)
    {
      using namespace std;
      using namespace JPP;
 
      typedef JEvtCategoryMap<JEvtWeightFactorHelper_t>   JEvtCategoryMap_t;
      
      
      bool matching = (!object.empty());
      
      const JHead& header = this->getHeader();
      
      const JEvtCategorySet cats = getCategories(header);
      
      for (typename JEvtCategoryMap_t::const_iterator i = object.cbegin(); i != object.cend() && matching; ++i) {
        
        JEvtCategorySet::const_iterator j = std::find(cats.cbegin(), cats.cend(), i->first);
        
        matching = (j != cats.cend());
      }
      
      return (matching ? setEvtWeightFactor(object) : false);
    }
 
 
    /**
     * Set flux function for the event-weighter associated with this file scanner.
     *
     * \param  flux                        flux function
     * \return                             true if flux function set successfully; else false
     */
    bool setFlux(const JFluxHelper& flux)
    {
      return setEvtWeightFactor(flux);
    }
    
 
    /**
     * Set flux function corresponding to a given event category.
     *
     * \param  category                    event category
     * \param  flux                        flux function
     * \return                             true if flux function set successfully; else false
     */
    bool setFlux(const JEvtCategoryHelper& category,
                 const JFluxHelper&        flux)
    {
      return setEvtWeightFactor(category, flux);
    }
 
 
    /**
     * Set flux function corresponding to a given set of event categories.
     * 
     * Note that the given flux function will only be assigned\n
     * if the header matches < b>all< /b> of the specified event categories.
     *
     * \param  categories                  set of event categories
     * \param  flux                        flux function
     * \return                             true if event-weight factor set successfully; else false
     */
    bool setFlux(const JEvtCategorySet& categories,
                 const JFluxHelper&     flux)
    {
      return setEvtWeightFactor(categories, flux);
    }
 
 
    /**
     * Set flux function according to a given map between event categories and flux functions.
     * 
     * Note that the given flux functions will only be assigned\n
     * if the header matches < b>all< /b> of the event categories associated with the given map.
     *
     * \param  object                     map between event categories and flux functions
     * \return                            true if event-weight factors are set successfully; else false
     */
    bool setFlux(const JEvtCategoryMap<JFluxHelper>& object)
    {
      return setEvtWeightFactor<JFluxHelper>(object);
    }
 
  private:
    using JEvtWeightHelper::reset;
  };
}
 
#endif