JPlotData.cc

Go to the documentation of this file.

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

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

#include "JFit/JEvt.hh"
#include "JFit/JEvtToolkit.hh"
#include "JFit/JFitParameters.hh"
#include "JFit/JFitApplications.hh"

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

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

#include "JGeometry3D/JTrack3D.hh"
#include "JGeometry3D/JGeometry3DToolkit.hh"
#include "JGeometry3D/JCylinder3D.hh"

#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JSupport.hh"

/**
 * \file
 *
 * Example program to produce some standard plots of reconstructed data
 * \author lquinn
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  JMultipleFileScanner<JEvt>     inputFile;                               // Class to scan for reconstructed information only
  JLimit_t&                      numberOfEvents = inputFile.getLimit();
  string                         outputFile;
  string                         detectorFile;
  size_t                         numberOfPrefits;                         // The number of reconstructed tracks to plot per event (1 by default)
  JQualitySorter                 quality_sorter;                          // The criterion by which to pick the "best" reconstructed track for each event.
  int                            application;
  int                            debug;

  try {

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

    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile)          = "postfit.root";
    zap['a'] = make_field(detectorFile);
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['N'] = make_field(numberOfPrefits)     = 1;
    zap['A'] = make_field(application)         = JMUONPREFIT, JMUONSIMPLEX, JMUONGANDALF, JMUONENERGY, JMUONSTART, JMUONPATH; 
    zap['L'] = make_field(quality_sorter)      = JPARSER::initialised();
    zap['d'] = make_field(debug)               = 2;

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



  JDetector detector;

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

  JCylinder3D cylinder(detector.begin(), detector.end());                           // Get detector geometry

  const double R_m  =  2.0 * cylinder.getRadius();
  const double zmin = -2.0 * cylinder.getZmin();
  const double zmax =  2.0 * cylinder.getZmax();

  TFile out(outputFile.c_str(), "recreate");
  TH1D job("job", "", 2,   -0.5, 1.5);
  TH1D hz ("hz ", "", 21 , -1.0, 1.0);                                              // Zenith angle
  TH2D hxy("hxy", "", 201, -R_m, R_m, 201, -200.0, 200.0);                          // X and Y position distribution
  TH2D hzR("hzR", "", 201,  0.0, R_m, 201, zmin, zmax);                             // Z and R position distribution
  TH1D hq1("hq1", "", 200, -50.0, 150);                                             // Reconstruction quality (JPrefit and JSimplex)
  TH1D hq2("hq2", "", 200, -5.0,  5.0);                                             // Reconstruction quality (JGandalf)
  TH1D hmu("hmu", "", 100, -50,  50);                                               // Atmospheric muon rejection parameter
  TH2D hmz("hmz", "", 100, -50,  50, 21, -1.0, 1.0);                                // Zenith angle vs atmospheric muon rejection parameter
  TH2D hqz("hqz", "", 200,  -5,   5, 21, -1.0, 1.0);                                // Zenith angle vs reconstruction quality (JGandalf)

  JAtmosphericMuon isAtmosphericMuon;

  while (inputFile.hasNext()) {                                 // Loop over each reconstructed event

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

    JEvt*      evt = inputFile.next();

    if (!evt->empty()) {

      job.Fill(1.0);

      JEvt::iterator __end = partition(evt->begin(), evt->end(), JHistory::is_event(application));

      if (evt->begin() == __end) {
        continue;
      }
      
      if (numberOfPrefits > 0) {        

        JEvt::iterator __q = __end;

        advance(__end = evt->begin(), min(numberOfPrefits, (size_t) distance(evt->begin(), __q)));
        
        partial_sort(evt->begin(), __end, __q, quality_sorter);

      } else {

        sort(evt->begin(), __end, quality_sorter);
      }

      const double muon_probability = isAtmosphericMuon(evt->begin(), __end);

      hmu.Fill(muon_probability);
    
      if (numberOfPrefits > 0) {
        advance(__end = evt->begin(), min(numberOfPrefits, evt->size()));
      }
        
      for (JEvt::const_iterator fit = evt->begin(); fit != __end; ++fit) {

        JTrack3D track = getTrack(*fit);
        track.sub(cylinder.getCenter());// Move the coordinate system so that (0,0,0) is the centre of the detector

        hz.Fill(track.getDZ());
        hxy.Fill(track.getX(), track.getY());
        hzR.Fill(sqrt(track.getX()*track.getX() + track.getY()*track.getY()), track.getZ());
        hq1.Fill(fit->getQ());
        hmz.Fill(muon_probability, track.getDZ());

        if (fit->hasW(JGANDALF_CHI2) && fit->hasW(JGANDALF_NUMBER_OF_HITS)) {

          const double lambda = fit->getW(JGANDALF_CHI2)/fit->getW(JGANDALF_NUMBER_OF_HITS);

          hq2.Fill(lambda);
          hqz.Fill(lambda, track.getDZ());
        }
      }
    }
    else {
      job.Fill(0.0);
    }
  }

  NOTICE("Number of events input             " << setw(8) << right << job.GetBinContent(1) << endl);
  NOTICE("Number of events with direction fit" << setw(8) << right << job.GetBinContent(2) << endl);

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