JFitEfficiency.cc

Go to the documentation of this file.

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

#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"

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

#include "JAAnet/JHead.hh"
#include "JAAnet/JAAnetToolkit.hh"

#include "JDAQ/JDAQEventIO.hh"
#include "JTools/JConstants.hh"
#include "JTools/JRange.hh"
#include "JSupport/JTriggeredFileScanner.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"
#include "JSupport/JSupport.hh"
#include "JGeometry3D/JCylinder3D.hh"
#include "JGizmo/JVolume.hh"

#include "JReconstruction/JEvt.hh"
#include "JReconstruction/JEvtToolkit.hh"

#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"

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


/**
 * \file
 *
 * Example program to histogram fit efficiency.
 * \author lquinn
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;

  typedef JTriggeredFileScanner<JEvt>                     JTriggeredFileScanner_t;
  typedef JTriggeredFileScanner_t::multi_pointer_type     multi_pointer_type;

  JTriggeredFileScanner_t inputFile;
  JLimit_t&               numberOfEvents = inputFile.getLimit();
  string                  outputFile;  
  string                  detectorFile;
  size_t                  numberOfPrefits;
  JRange<double>          E_GeV;
  double                  quality;
  bool                    logx;
  bool                    containment;
  int                     debug;

  try {

    JParser<> zap("Example program to histogram fit results.");

    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile)          = "fitefficiency.root";
    zap['a'] = make_field(detectorFile);
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['N'] = make_field(numberOfPrefits)     = 1;
    zap['E'] = make_field(E_GeV)               = JRange<double>(1.0, -1.0);
    zap['Q'] = make_field(quality)             = numeric_limits<double>::min();
    zap['X'] = make_field(logx);
    zap['C'] = make_field(containment);
    zap['d'] = make_field(debug)               = 1;

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




  Head head;

  try {
    head = getHeader(inputFile);
  }
  catch(const JException& error) {
    FATAL(error);
  }

  JHead   buffer(head);
  JVolume volume(head, logx);

  const double Xmax = E_GeV.is_valid() ? volume.getX(E_GeV.getMaximum()) :  volume.getXmax();
  const double Xmin = E_GeV.is_valid() ? volume.getX(E_GeV.getMinimum()) :  volume.getXmin();

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

  TH1D* hN = new TH1D("hN", NULL, 25, Xmin, Xmax);
  TH1D* he = new TH1D("he", NULL, 25, Xmin, Xmax);

  hN->Sumw2();
  he->Sumw2();

  JDetector detector;

  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }

  JCylinder3D cylinder(detector.begin(), detector.end());

  STATUS("Detector Geometry:" << endl);
  STATUS("Zmin: "     << cylinder.getZmin() << endl);
  STATUS("Zmax: "     << cylinder.getZmax() << endl);
  STATUS("Z origin: " << 0.5 * (cylinder.getZmax() + cylinder.getZmin()) << endl);
  STATUS("Radius: "   << cylinder.getRadius() << endl);

  while (inputFile.hasNext()) {

    STATUS("event: " << setw(10) << inputFile.getCounter() << '\r'); DEBUG(endl);

    multi_pointer_type ps = inputFile.next();

    //JDAQEvent* tev   = ps;
    JEvt*      evt   = ps;
    Evt*       event = ps;

    if (has_neutrino(*event)) {

      Trk neutrino = get_neutrino(*event);
      const double X = volume.getX(neutrino.E);

      hN->Fill(X);

      if (!evt->empty()) {

        JEvt::iterator __end = evt->end();
    
        if (numberOfPrefits > 0) {      

          advance(__end = evt->begin(), min(numberOfPrefits, evt->size()));

          partial_sort(evt->begin(), __end, evt->end(), qualitySorter);
        }


        for(JEvt::const_iterator track = evt->begin(); track != __end; ++track) {

          JPosition3D pos = getPosition(*track);

          const double R = sqrt(pos.getX() * pos.getX() + pos.getY() * pos.getY());

          if ((pos.getZ() > cylinder.getZmin()
          &&   pos.getZ() < cylinder.getZmax()
          &&   R < cylinder.getRadius())
          || !containment) {

            double Q = 0.0;

            if(track->hasW(JGANDALF_CHI2) && track->hasW(JGANDALF_NUMBER_OF_HITS)) {
              Q = -track->getW(JGANDALF_CHI2)/track->getW(JGANDALF_NUMBER_OF_HITS);
            }
            else {
              Q = track->getQ();
            }

            if (Q > quality) {
              he->Fill(X);
            }
          }
        }
      }
    }
    else {
      WARNING("No neutrino." << endl);
    }
  }
  STATUS(endl);

  he->Divide(hN);

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