JPrintAppliedWeights.cc

Go to the documentation of this file.

#include <iostream>
#include <string>
 
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
#include "JLang/JClonable.hh"
#include "JLang/JException.hh"
 
#include "JAAnet/JHead.hh"
#include "JAAnet/JEvtWeightFactor.hh"
#include "JAAnet/JEvtCategoryMap.hh"
#include "JAAnet/JEvtCategoryToolkit.hh"
 
#include "JSirene/JVisibleEnergyToolkit.hh"
 
#include "JSupport/JSupport.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JEvtWeightFileScannerSet.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"
 
 
namespace {
 
  /**
   * Auxiliary class for event-categories without a defined weight.
   */
  class JEvtWeightFactorUndefined final :
    public JLANG::JClonable<JAANET::JEvtWeightFactor, JEvtWeightFactorUndefined>
  {
  public:
    
    /**
     * Constructor.
     */
    JEvtWeightFactorUndefined()
    {}
 
    
    /**
     * Get event-weight factor of given event.
     *
     * \param  evt               event
     * \return                   event-weight factor
     */
    double getFactor(const Evt& evt) const override final
    {
      using namespace JPP;
      
      THROW(JNoValue, "JEvtWeightFactorUndefined::getFactor(): No weight defined.");
    };
 
 
    /**
     * Get properties of this class.
     *
     * \param  eqpars          equation parameters
     */
    JProperties getProperties(const JEquationParameters& eqpars = JEvtWeightFactor::getEquationParameters()) override final
    {
      return JEvtWeightFactorUndefinedHelper(*this, eqpars);
    }
 
 
    /**
     * Get properties of this class.
     *
     * \param  eqpars          equation parameters
     */
    JProperties getProperties(const JEquationParameters& eqpars = JEvtWeightFactor::getEquationParameters()) const override final
    {
      return JEvtWeightFactorUndefinedHelper(*this, eqpars);      
    }
 
 
  private:
 
    /**
     * Auxiliary class for I/O of JEvtWeightFactorUndefined.
     */    
    struct JEvtWeightFactorUndefinedHelper :
      public JEEP::JProperties
    {
      /**
       * Constructor.
       *
       * \param  formula       TFormula-based event-weight factor
       * \param  eqpars        equation parameters 
       */
      template<class JEvtWeightFactorUndefined_t>
      JEvtWeightFactorUndefinedHelper(JEvtWeightFactorUndefined_t& factor,
                                      const JEquationParameters&   eqpars) :
        JProperties(eqpars, 1)
      {
        (*this)[JEvtWeightFactor::getTypeKey()] = "undefined";
      }
    };
  };
}
 
 
/**
 * \file
 * Program for printing the applied event-weights for each header type\n
 * contained in a given set of input files.
 *
 * \author bjjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  
  JMultipleFileScanner_t  inputFiles;
 
  JEvtWeightFactorMap     factors;
  
  int                     debug;
  
  try {
    
    JParser<> zap;
    
    zap['f'] = make_field(inputFiles);
    zap['@'] = make_field(factors)       = JPARSER::initialised();
    zap['d'] = make_field(debug)         = 1;
 
    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }  
 
  
  JEvtWeightFileScannerSet<> scanners(inputFiles);
 
  if (scanners.setEvtWeightFactor(factors) == 0) {
    WARNING("No flux functions set." << endl);
  }
  
  for (JEvtWeightFileScannerSet<>::iterator scanner = scanners.begin(); scanner != scanners.end(); ++scanner) {
    
    const JHead&      header   = scanner->getHeader();
    const JEvtWeight& weighter = scanner->getEvtWeighter();
    
    const JEvtCategorySet categories = getCategories(header);
 
    const JEvtWeightFactorHelper* p = dynamic_cast<const JEvtWeightFactorHelper*>(&weighter);
 
    if (p == NULL) { continue; }
 
    JEvtWeightFactorMap Fmap(factors.getOscProb());
 
    if (*p) {
      
      for (JEvtCategorySet::const_iterator i = categories.cbegin(); i != categories.cend(); ++i) {
        Fmap.insert(*i, *p);
      }
      
    } else {
 
      const JEvtWeightFactorHelper helper(make_shared<JEvtWeightFactorUndefined>());
      
      for (JEvtCategorySet::const_iterator i = categories.cbegin(); i != categories.cend(); ++i) {
        Fmap.insert(*i, helper);
      }
    }
 
    NOTICE("scanner " << distance(scanners.begin(), scanner) << ":" << endl <<
           "======================================================" << endl);
    DEBUG (endl << header << endl << endl);
    NOTICE(Fmap << endl);
  }
  
  return 0;
}