JNarrabri.cc

Go to the documentation of this file.

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

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

#include "JROOT/JGraph.hh"
#include "JROOT/JRootToolkit.hh"
#include "JROOT/JManager.hh"

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

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

#include "JLang/JObjectMultiplexer.hh"

#include "JAcoustics/JEvt.hh"
#include "JAcoustics/JSupport.hh"

#include "JDynamics/JDynamics.hh"

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


/**
 * \file
 * Example program to check contents of acoustic events.
 *
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  JMultipleFileScanner<JACOUSTICS::JEvt> inputFile;
  JLimit_t&       numberOfEvents = inputFile.getLimit();
  string          detectorFile;
  string          outputFile;
  double          Tmax_s;
  string          text_file;
  int             debug;

  try {

    JParser<> zap("Example program to check contents of acoustic events.");

    zap['f'] = make_field(inputFile,       "output of JKatoomba[.sh]");
    zap['n'] = make_field(numberOfEvents)      = JLimit_t::max();
    zap['a'] = make_field(detectorFile);
    zap['o'] = make_field(outputFile)          = "narrabri.root";
    zap['T'] = make_field(Tmax_s);
    zap['+'] = make_field(text_file)           = "";
    zap['d'] = make_field(debug)               = 1;

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


  JDetector detector;

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


  JDynamics dynamics(detector, Tmax_s);

  STATUS("loading input from file(s) " << inputFile << "... " << flush);

  dynamics.load(inputFile);

  STATUS("OK" << endl);


  map<int, JGraph_t>  ZO;
  map<int, JGraph_t>  ZA;

  Double_t xmin = numeric_limits<Double_t>::max();
  Double_t xmax = numeric_limits<Double_t>::lowest();

  for (JDynamics::JPosition::const_iterator string = dynamics.position.begin(); string != dynamics.position.end(); ++string) {

    if (!string->second.empty()) {
      xmin = min(xmin, string->second.getXmin());
      xmax = max(xmax, string->second.getXmax());
    }
  }

  const JFormat_t format(4, 0, std::ios_base::fmtflags(), '0');

  JManager<int, TH2D> H2(new TH2D  ("[%].tilt", NULL, 100, -1.0, +1.0, 100, -1.0, +1.0), '%', format);
  JManager<int, TH1D> HT(new TH1D  ("[%].time", NULL, 200,  0.0,  4.0),                  '%', format);

  JManager<int, TH1D> HO(new TH1D  ("H[%].orientation", NULL, 1000, xmin, xmax), '%', format);
  JManager<int, TH1D> HA(new TH1D  ("H[%].amplitude",   NULL, 1000, xmin, xmax), '%', format);

  for (JDynamics::JPosition::const_iterator string = dynamics.position.begin(); string != dynamics.position.end(); ++string) {

    TH1D* ho = HO[string->first];
    TH1D* ha = HA[string->first];

    for (Int_t i = 1; i <= HO->GetXaxis()->GetNbins(); ++i) {

      const Double_t x = HO->GetXaxis()->GetBinCenter(i);

      const JMODEL::JString tilt = string->second(x);

      ho->SetBinContent(i, tilt.getAngle());
      ha->SetBinContent(i, tilt.getLength());
    }
  }


  for (JDynamics::JPosition::const_iterator string = dynamics.position.begin(); string != dynamics.position.end(); ++string) {

    if (string->second.size() > 1) {

      TH2D* h2 = H2[string->first];
      TH1D* hc = HT[string->first];

      for (typename JDynamics::JPosition::function_type::const_iterator q = string->second.begin(), p = q++;  q != string->second.end(); ++p, ++q) {

        const double t1 = q->getX() - p->getX();
        
        hc->Fill(log10(t1));

        if (t1 > 0 && t1 < Tmax_s) {
          h2->Fill(600.0e3 * (q->getY().tx - p->getY().tx) / t1,     // [mrad/10 min]
                   600.0e3 * (q->getY().ty - p->getY().ty) / t1);    // [mrad/10 min]
        }
      }

      for (typename JDynamics::JPosition::function_type::const_iterator i = string->second.begin(); i != string->second.end(); ++i) {
        ZO[string->first].put(i->getX(), i->getY().getAngle());
        ZA[string->first].put(i->getX(), i->getY().getLength());
      }
    }
  }

  
  if (text_file != "") {

    ofstream out(text_file.c_str());

    JComment comment;

    comment.add("documentation: https://common.pages.km3net.de/jpp/Position_calibration.PDF");
    comment.add("format: string number; UTC [s]; Tx; Ty");

    comment.add(JMeta(argc, argv));

    JMultipleFileScanner<JMetaTypes_t> in(inputFile);
    JObjectMultiplexer  <JMetaTypes_t> abc(in);

    while (abc.hasNext()) {
      comment.add(*abc.next());
    }

    out << comment;
          
    for (JDynamics::JPosition::const_iterator string = dynamics.position.begin(); string != dynamics.position.end(); ++string) {
      for (typename JDynamics::JPosition::function_type::const_iterator i = string->second.begin(); i != string->second.end(); ++i) {
        out << setw(4)     << string->first << ';'
            << FIXED(20,5) << i->getX()     << ';'
            << FIXED( 9,6) << i->getY().tx  << ';'
            << FIXED( 9,6) << i->getY().ty  << endl;
      }
    }

    out.close();
  }

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

  out << H2 << HT << HO << HA;

  for (map<int, JGraph_t>::const_iterator i = ZO.begin(); i != ZO.end(); ++i) {
    out << JGraph(i->second, MAKE_CSTRING("G[" << i->first << "].orientation"));
  }

  for (map<int, JGraph_t>::const_iterator i = ZA.begin(); i != ZA.end(); ++i) {
    out << JGraph(i->second, MAKE_CSTRING("G[" << i->first << "].amplitude"));
  }

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