JEvtCategoryToolkit.hh

Go to the documentation of this file.

#ifndef __JAANET__JEVTCATEGORYTOOLKIT__
#define __JAANET__JEVTCATEGORYTOOLKIT__
 
#include <set>
#include <iostream>
 
#include "JLang/JToken.hh"
 
#include "Jeep/JProperties.hh"
 
#include "JAAnet/JHead.hh"
#include "JAAnet/JHeadToolkit.hh"
 
#include "JAAnet/JNeutrinoInteractionCategory.hh"
#include "JAAnet/JMuonBundleCategory.hh"
#include "JAAnet/JEvtCategoryHelper.hh"
 
 
/**
 * \author bjung
 */
 
namespace JAANET {}
namespace JPP { using namespace JAANET; }
 
namespace JAANET {
 
 
  /**
   * Container for a set of event categories.
   *
   * Input syntax as follows:
   * <pre>
   *       <PDG type> [& <PDG type>]...
   *       <key>=<value>[,<key>=<value>]...[ & <key>=<value>[,<key>=<value>]...]...
   * </pre>
   * where
   *  - <PDG type> refers   to a PDG identifier;
   *  - <key>               to one of the data members of a data structure derived from `JAANET::JEvtCategory`;
   *  - <value>             and to the corresponding value.
   * Multiple event categories can be specified, separated by the '&' character.
   */
  struct JEvtCategorySet :
    public std::set<JEvtCategoryHelper>
  {
    /**
     * Default constructor.
     */
    JEvtCategorySet()
    {}
 
 
    /**
     * Stream input.
     *
     * \param  in                 input stream
     * \param  categories         set of event categories
     * \return                    input stream
     */
    friend inline std::istream& operator>>(std::istream& in, JEvtCategorySet& categories)
    {
      using namespace std;
      using namespace JPP;
 
      for (JToken<'&'> token; in >> token; ) {
 
        JEvtCategoryHelper category;
        
        istringstream iss(token);
        iss >> category;
 
        if (!fail(iss)) { categories.insert(category); }
      }
 
      return in;
    }
 
 
    /**
     * Stream output.
     *
     * \param  out                output stream
     * \param  categories         set of event categories
     * \return                    output stream
     */
    friend inline std::ostream& operator<<(std::ostream& out, const JEvtCategorySet& categories)
    {
      using namespace std;
      using namespace JPP;
 
      if (categories.size() > 0) {
        out << *categories.cbegin();
      }
      
      for (iterator i = next(categories.cbegin()); i != categories.cend(); ++i) {
        out << " & " << *i;
      }
 
      return out;
    }
  };
 
 
  /**
   * Auxiliary method to retrieve the unique event categories corresponding to a MC header.
   *
   * \param  header               MC header
   * \return                      vector of unique event categories
   */
  inline JEvtCategorySet getCategories(const JHead& header)
  {
    using namespace std;
    using namespace JPP;
    
    JEvtCategorySet categories;
    
    if        (is_gseagen(header)) {
 
      const JNeutrinoInteractionCategory cat(header);
      const JEvtCategoryHelper           helper(cat);
 
      categories.insert(helper);
 
      for (vector<JAANET::flux>::const_iterator i = header.flux.cbegin(); i != header.flux.cend(); ++i) {
        
        const JNeutrinoInteractionCategory cat2(header, i->type);
        const JEvtCategoryHelper           helper2(cat2);
 
        categories.insert(helper2);
      }
      
    } else if (is_mupage(header) || is_corsika(header)) {
 
      const JMuonBundleCategory cat(header);
      const JEvtCategoryHelper  helper(cat);
      
      categories.insert(helper);
    }
 
    return categories;
  }
}
 
#endif