JMultiParticleFlux.cc File Reference

Example program for scanning event-weights of Monte Carlo files
containing a given set of primaries one primary. More...

#include <iostream>
#include <iomanip>
#include <string>
#include <set>
#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"
#include "JLang/JException.hh"
#include "JAAnet/JAAnetToolkit.hh"
#include "JAAnet/JFluxFunction.hh"
#include "JAAnet/JMultiParticleFlux.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JLimit.hh"
#include "JSupport/JWeightFileScannerSet.hh"
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
#include "Jeep/JProperties.hh"

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Detailed Description

Example program for scanning event-weights of Monte Carlo files
containing a given set of primaries one primary.

The list of possible options for the flux includes:

   -@ zero      <type>
   -@ flat      <type>  <value>
   -@ powerlaw  <type>  <normalisation>  <spectral index>

Author

bjung

Definition in file JMultiParticleFlux.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 218 of file JMultiParticleFlux.cc.

{
  using namespace std;
  using namespace JPP;

  JMultipleFileScanner_t               inputFiles;
  JLimit                               numberOfEvents;

  vector<int>                          zeroFluxes;
  map<int, JFlatFlux>                  flatFluxes;
  map<int, JPowerLawFlux>              powerlawFluxes;

  int                                  debug;

  try {

    JProperties fluxMaps;

    fluxMaps["zero"]     = zeroFluxes;
    fluxMaps["flat"]     = flatFluxes;
    fluxMaps["powerlaw"] = powerlawFluxes;
    
    JParser<> zap;
    
    zap['f'] = make_field(inputFiles);
    zap['@'] = make_field(fluxMaps)           = JPARSER::initialised();
    zap['n'] = make_field(numberOfEvents)     = JLimit::max();
    zap['d'] = make_field(debug)              = 1;

    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }

  // Sanity checks

  for (map<int, JPowerLawFlux>::const_iterator i = powerlawFluxes.cbegin(); i != powerlawFluxes.cend(); ++i) {
    if (abs(i->first) != TRACK_TYPE_NUE  &&
        abs(i->first) != TRACK_TYPE_NUMU &&
        abs(i->first) != TRACK_TYPE_NUTAU) {
      FATAL("Particle type " << i->first << " does not correspond to a neutrino." << endl);
    }
  }  

  // Create multi-particle flux function

  JMultiParticleFlux multiFlux;  
  
  for (vector<int>::const_iterator i = zeroFluxes.cbegin(); i != zeroFluxes.cend(); ++i) {
    multiFlux.insert(*i, make_fluxFunction(zeroFlux));
  }

  for (map<int, JFlatFlux>::const_iterator i = flatFluxes.cbegin(); i != flatFluxes.cend(); ++i) {
    multiFlux.insert(i->first, make_fluxFunction(i->second));
  }

  for (map<int, JPowerLawFlux>::const_iterator i = powerlawFluxes.cbegin(); i != powerlawFluxes.cend(); ++i) {
    multiFlux.insert(i->first, make_fluxFunction(i->second));
  }

  
  // Set event weighter
  
  JWeightFileScannerSet<> scanners(inputFiles, numberOfEvents);

  size_t n = scanners.setFlux(multiFlux);

  if (n == 0) {
    WARNING("No file found containing all given primaries; Flux function not set." << endl);
  }
  
  
  // Scan events
  
  for (JWeightFileScannerSet<>::iterator scanner = scanners.begin(); scanner != scanners.end(); ++scanner) {
    
    if (scanner->simul.size() > 0) {
      STATUS("Scanning " << scanner->simul[0].program << " files..." << endl);
    }

    STATUS(LEFT(15) << "event" << RIGHT(15) << "weight" << endl);
    
    while (scanner->hasNext()) {
      
      const Evt*   event  = scanner->next();
      const double weight = scanner->getWeight(*event);
      
      STATUS(LEFT      (15)    << scanner->getCounter() <<
             SCIENTIFIC(15, 3) << weight << '\r'); DEBUG(endl);
    }
  }

  return 0;
}