JEffectiveMass1D.cc

Go to the documentation of this file.

#include <string>
#include <iostream>

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

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

#include "JDAQ/JDAQEventIO.hh"

#include "JGeometry3D/JCylinder3D.hh"
#include "JGeometry3D/JPosition3D.hh"

#include "JAAnet/JVolume.hh"
#include "JAAnet/JHeadToolkit.hh"
#include "JAAnet/JAAnetToolkit.hh"

#include "JSupport/JSupport.hh"
#include "JSupport/JTriggeredFileScanner.hh"
#include "JSupport/JEvtWeightFileScannerSet.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"

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

#include "JROOT/JManager.hh"

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


namespace {

  const char* const Mass_t    =  "Mass";
  const char* const Volume_t  =  "Volume";
}

/**
 * \file
 *
 * Example program to histogram neutrino effective mass for triggered events inside the can volume.\n
 * For other than genie/gSeaGen events, use application JVolume1D.\n
 * The unit of the contents of the histogram is \f$Mton\f$ or \f$km^{3}\f$.
 * \author mdejong, bstrandberg, bjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;

  JMultipleFileScanner_t    inputFiles;
  string                    detectorFile;
  string                    outputFile;
  double                    margin;
  bool                      logx;
  int                       numberOfBins;
  std::string               option;
  int                       debug;

  
  try {

    JParser<> zap("Example program to histogram neutrino effective mass for triggered events.");

    zap['f'] = make_field(inputFiles);
    zap['a'] = make_field(detectorFile, "Detector file: if not provided, instrumented volume is not calculated")                        = "";
    zap['o'] = make_field(outputFile)      = "Meff.root";
    zap['R'] = make_field(margin,       "Addition or subtraction margin around the volume in which the considered events must reside")  = 0.0;
    zap['X'] = make_field(logx,         "Use logarithm of energy");
    zap['N'] = make_field(numberOfBins, "Number of bins in the energy range of the MC simulation")                                      = 10;
    zap['O'] = make_field(option,       "Result option")                                                                                = Mass_t, Volume_t;
    zap['d'] = make_field(debug)           = 2;
    
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  JDetector detector;

  if (detectorFile != "") {

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

  try{

    double Xmin = numeric_limits<double>::max();
    double Xmax = numeric_limits<double>::lowest();

    JEvtWeightFileScannerSet<> scanners(inputFiles);

    for (JEvtWeightFileScannerSet<>::const_iterator scanner = scanners.cbegin(); scanner != scanners.cend(); ++scanner) {

      const JHead header = scanner->getHeader();

      if (!is_gseagen(header)) {
        ERROR("Not a gSeaGen file; you may want to use JVolume1D" << endl);
      }
    
      const JVolume volume(header, logx);

      if (volume.getXmin() < Xmin) { Xmin = volume.getXmin(); }
      if (volume.getXmax() > Xmax) { Xmax = volume.getXmax(); }
    }

    JManager<int, TH1D> hM (new TH1D("hM[%]",  MAKE_CSTRING(option << " can"),                        numberOfBins, Xmin, Xmax));
    JManager<int, TH1D> hm (new TH1D("hm[%]",  MAKE_CSTRING(option << " instrumented"),               numberOfBins, Xmin, Xmax));
    JManager<int, TH1D> hNM(new TH1D("hNM[%]", MAKE_CSTRING(option << " normalization can"),          numberOfBins, Xmin, Xmax));
    JManager<int, TH1D> hNm(new TH1D("hNm[%]", MAKE_CSTRING(option << " normalization instrumented"), numberOfBins, Xmin, Xmax));

  
    for (JEvtWeightFileScannerSet<>::iterator scanner = scanners.begin(); scanner != scanners.end(); ++scanner) {

      const JHead header = scanner->getHeader();

      NOTICE("Scanning file type " << distance(scanners.begin(), scanner) << endl);
      DEBUG (header << endl);

      //------------------------------------------------------------------------------
      // define selection volumes
      //------------------------------------------------------------------------------

      // can volume comes from the generator header
      // position is consistent with those of generated tracks (Evt::mc_trks) by construction

      JCylinder3D canvol = getCylinder(header);

      canvol.addMargin(margin);

      // detector geometry comes from detector file
      // position needs to be synced to that used in event generator

      const Vec   offset = getOffset(header);

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

      instvol -= getPosition(offset);
      instvol.addMargin(margin);

      NOTICE("Light simulation volume dimensions, including margin: " << endl << canvol  << endl);
      NOTICE("Instrumented volume dimensions, including margin: "     << endl << instvol << endl);

      //------------------------------------------------------------------------------
      // loop over generated events in the input files
      //------------------------------------------------------------------------------

      size_t N = 0;
    
      while (scanner->hasNext()) {
        
        STATUS("event: " << setw(10) << scanner->getCounter() << '\r'); DEBUG(endl);

        const Evt*       event   = scanner->next();
        const Trk&       primary = get_primary(*event);
        const JVertex3D& vertex  = getVertex(*event);
        const double     x       = logx ? log10(primary.E) : primary.E;

        if (event->mc_hits.empty()) {
          N += 1;
        }

        // events are filled to hM if they are inside the can
        if (canvol.is_inside(vertex)) {
          hNM[primary.type]->Fill(x);
        }

        // events are filled to hm if they are inside the instrumented volume
        if (instvol.is_inside(vertex)) {
          hNm[primary.type]->Fill(x);
        }
      }
      STATUS(endl);

      if (N == 0) {
        ERROR("No events with zero hits -> use application JEffectiveMassOffline1D." << endl);
      }

      //------------------------------------------------------------------------------
      // loop over triggered events in the input files
      //------------------------------------------------------------------------------

      JTriggeredFileScanner<> in(scanner->getFilelist());
    
      while (in.hasNext()) {

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

        JTriggeredFileScanner<>::multi_pointer_type mp = in.next();

        const Evt*   event   = mp;
        const Trk&   primary = get_primary(*event);
        const double x       = logx ? log10(primary.E) : primary.E;

        double yM = 0.0;
        double ym = 0.0;

        if        (option == Mass_t) {
          yM = canvol .getVolume() * DENSITY_SEA_WATER * 1e-6;    // Mton
          ym = instvol.getVolume() * DENSITY_SEA_WATER * 1e-6;    // Mton
        } else if (option == Volume_t) {
          yM = canvol .getVolume() * 1.0e-15;                     // km^3
          ym = instvol.getVolume() * 1.0e-15;                     // km^3
        }

        hM[primary.type]->Fill(x, yM);
        hm[primary.type]->Fill(x, ym);
    
      }
      STATUS(endl);
    }

    //------------------------------------------------------------------------------
    // convert 'generated' and 'triggered' histograms to effective mass
    //------------------------------------------------------------------------------

    for (int pdg : get_keys(hM)) {
      hM[pdg]->Divide(hNM[pdg]);
      hm[pdg]->Divide(hNm[pdg]);
    }

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

    out << hM << hm;
  
    out.Close();
  }
  catch(const JException& error) {
    FATAL(error << endl << "You may want to use \"offline\" formatted file.");
  }
}