JPlotOscProb.cc

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <string>

#include "JLang/JVectorize.hh"

#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

#include "JOscProb/JOscChannel.hh"
#include "JOscProb/JOscParameters.hh"
#include "JOscProb/JOscProbInterpolator2D.hh"

#include "JTools/JAbstractHistogram.hh"

#include "TFile.h"
#include "TH2D.h"


/**
 * \file
 * Example program to plot oscillation probabilties interpolated in energy and zenith-angle.
 *
 * \author bjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  typedef JAbstractHistogram<Double_t>          JHistogram_t;

  string                  inputFile;
  string                  outputFile;
  JOscParameters          parameters;
  JHistogram_t            X;
  JHistogram_t            Y;  
  JOscChannel             channel;
  int                     debug;

  try {
    
    JParser<> zap;

    zap['f'] = make_field(inputFile,  "8D oscillation probability table file");
    zap['o'] = make_field(outputFile, "output file");    
    zap['x'] = make_field(X,          "histogram binning in log10(E)"); 
    zap['y'] = make_field(Y,          "histogram binning in cos(theta)");   
    zap['@'] = make_field(parameters, "oscillation parameters") = JPARSER::initialised();
    zap['#'] = make_field(channel,    "Oscillation channel")    = JOscChannel(JOscChannel::MUON, JOscChannel::MUON, JOscChannel::PARTICLE);
    zap['d'] = make_field(debug)                                = 2;

    zap(argc, argv);
  }
  catch (const exception& error) {
    FATAL(error.what() << endl);
  }
  
  
  TH2D h0("h0", "; log_{10}(E); cos(#theta)",
          X.getNumberOfBins(), X.getLowerLimit(), X.getUpperLimit(),
          Y.getNumberOfBins(), Y.getLowerLimit(), Y.getUpperLimit());

  const JOscProbTable6D<> g6(inputFile.c_str());
  const JOscProbTable2D<> g2(g6, parameters);
  
  const JOscProbInterpolator2D<> interpolator(&g2);

  NOTICE(RIGHT(15) << "Energy [GeV]" <<
         RIGHT(15) << "cos(theta)"   <<
         RIGHT(15) << "P"            << endl);
  
  for (Int_t ix = 1; ix <= h0.GetNbinsX(); ++ix) {
    for (Int_t iy = 1; iy <= h0.GetNbinsY(); ++iy) {
      
      const Double_t log10E = h0.GetXaxis()->GetBinCenter(ix);
      const Double_t costh  = h0.GetYaxis()->GetBinCenter(iy);
      const Double_t P      = interpolator(channel, log10E, costh);

      NOTICE(FIXED(15,3) << pow(10, log10E) <<
             FIXED(15,3) << costh           <<
             FIXED(15,3) << P               << '\r'); DEBUG(endl);

      h0.SetBinContent(ix, iy, P);
    }
  }

  TFile out(outputFile.c_str(), "recreate");

  h0.Write();

  out.Write();
  out.Close();
}