JEvtWeightCorsika.hh

Go to the documentation of this file.

#ifndef __JAANET__JEVTWEIGHTCORSIKA__
#define __JAANET__JEVTWEIGHTCORSIKA__

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

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

#include "JAAnet/JHead.hh"
#include "JAAnet/JHeadToolkit.hh"
#include "JAAnet/JFlux.hh"
#include "JAAnet/JEvtWeight.hh"

/**
 * \author bjung
 */

namespace JAANET {
  
  using JLANG::JClonable;
  using JLANG::JIndexOutOfRange;
  using JLANG::JNullPointerException;

  
  /**
   * Implementation of event weighting for Corsika data.
   */
  struct JEvtWeightCorsika :
    public JClonable<JEvtWeight, JEvtWeightCorsika>,
    public JFluxHelper
  {
    /**
     * Default Constructor.
     */
    JEvtWeightCorsika() :
      JFluxHelper(),
      W(0.0)
    {}


    /** 
     * Constructor.
     *
     * \param  header            header
     */    
    JEvtWeightCorsika(const JHead& header) :
      JFluxHelper()
    {
      configure(header);
    }


    /** 
     * Constructor.
     *
     * \param  header            header
     * \param  flux              flux
     */
    JEvtWeightCorsika(const JHead& header,
                      const JFlux& flux)
    {
      configure(header);
      JFluxHelper::configure(flux);
    }


    /**
     * Copy constructor.
     *
     * \param  object           original object
     */
    JEvtWeightCorsika(const JEvtWeightCorsika& object)
    {
      this->W = object.W;

      if (object.is_valid()) {
        JFluxHelper::configure(*(object.get()));
      }
    }
    
    
    /**
     * Weight configuration.
     *
     * \param  header            header
     */
    void configure(const JHead& header) override
    {
      const double N = header.genvol.numberOfEvents;
      
      if (check(header) && N > 0.0) {
        
        W = 1.0 / N;
        
      } else {

        THROW(JValueOutOfRange, "JEvtWeightCorsika::configure(): Provided header is inconsistent with Corsika.");
      }
    }


    /**
     * Check whether header is consistent with this event weighter.
     *
     * \param  header            header
     * \return                   true if consistent; else false
     */
    bool check(const JHead& header) const override
    {
      return is_gseagen(header);
    }

    
    /**
     * Get weight of given event.
     *
     * \param  evt               event
     * \return                   weight        [Hz]
     */
    double getWeight(const Evt& evt) const override
    {
      const double Nretries = evt.w2list[W2LIST_GSEAGEN_N_RETRIES];
      
      if (is_valid() && evt.w.size() > 1) {
        
        return W * evt.w[1] * getFlux(evt) / (1 + Nretries);
        
      } else if (evt.w.size() > 2) {

        return W * evt.w[2] / (1 + Nretries);
        
      } else {

        if (!is_valid()) {
          THROW(JNullPointerException, "JEvtWeightCorsika::getWeight(): Unspecified flux function.");
        } else {
          THROW(JIndexOutOfRange,      "JEvtWeightCorsika::getWeight(): Empty " << (evt.w.size() < 3 ? "w2-" : "w3-") << "weight.");
        }
      }
    }

    
  private:
    
    double W;
  };
}

#endif