JEvtWeightFactorMultiParticle.hh

Go to the documentation of this file.

#ifndef __JAANET__JEVTWEIGHTFACTORMULTIPARTICLE__
#define __JAANET__JEVTWEIGHTFACTORMULTIPARTICLE__

#include <map>

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

#include "JLang/JClonable.hh"
#include "JLang/JPredicate.hh"
#include "JLang/JException.hh"

#include "JAAnet/JEvtWeightFactor.hh"
#include "JAAnet/JEvtWeightFactorHelper.hh"
#include "JAAnet/JAAnetToolkit.hh"

/**
 * \author bjung
 */

namespace JAANET {
  
  using JLANG::JClonable;
  using JLANG::JValueOutOfRange;

  /**
   * Implementation of event-weight factor for multiple particle types.
   */
  struct JEvtWeightFactorMultiParticle :
    public JClonable<JEvtWeightFactor, JEvtWeightFactorMultiParticle>,
    public std::map<int, JEvtWeightFactorHelper>
  {
    typedef std::map<int, JEvtWeightFactorHelper>      map_type;
    
    /**
     * Default constructor.
     */
    JEvtWeightFactorMultiParticle() :
      map_type()
    {}

    
    /**
     * Constructor.
     *
     * \param  type                   particle PDG code
     * \param  factor                 event-weight factor
     * \param  args                   remaining pairs of PDG codes and event-weight factors
     */
    template<class ...Args>
    JEvtWeightFactorMultiParticle(const int               type,
                                  const JEvtWeightFactor& factor,
                                  const Args&          ...args)
    {
      insert(type, factor, args...);
    }


    /**
     * Insert pair of particle code and event-weight factor.
     *
     * \param  type                    particle PDG code
     * \param  factor                  event-weight factor
     */ 
    void insert(const int               type,
                const JEvtWeightFactor& factor)
    {
      map_type::insert(std::make_pair(type, JEvtWeightFactorHelper(factor)));
    }
    

    /**
     * Insert pairs of particle codes and event-weight factors.
     *
     * \param  type                    particle PDG code
     * \param  factor                  event-weight factor
     * \param  args                    remaining pairs of PDG codes and event-weight factors
     */ 
    template<class ...Args>
    void insert(const int               type,
                const JEvtWeightFactor& factor,
                const Args&          ...args)
    {
      insert(type, factor);
      insert(args...);
    }


    /**
     * Get event-weight factor of given event.
     *
     * \param  evt               event
     * \return                   weight-factor for given event
     */
    double getFactor(const Evt& evt) const override
    {
      using namespace std;
      using namespace JPP;

      vector<Trk>::const_iterator primary = find_if(evt.mc_trks.cbegin(),
                                                    evt.mc_trks.cend(),
                                                    make_predicate(&Trk::is_primary, true));
      
      if (primary != evt.mc_trks.cend()) {
        
        return this->at(primary->type).getFactor(evt);
        
      } else if (has_neutrino(evt)) {
        
        const Trk& neutrino = get_neutrino(evt);

        return this->at(neutrino.type).getFactor(evt);
        
      } else {
        
        THROW(JValueOutOfRange, "JEvtWeightFactorMultiParticle::getFactor(): No primary track found for event " << evt.id << ".");
      }
    }
  };
}

#endif