JDrawDetector1D.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <limits>
#include <map>

#include "TROOT.h"
#include "TGraph.h"
#include "TH1D.h"
#include "JROOT/JGraph.hh"
#include "JROOT/JManager.hh"
#include "JMath/JMath.hh"
#include "JPhysics/JConstants.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"


/**
 * \file
 * Auxiliary program to draw the string of detector.
 * \author mdejong
 */
int main(int argc, char**argv)
{
  using namespace std;
  using namespace JPP;

  string              detectorFile;
  string              outputFile;
  double              step;
  int                 debug;

  try {

    JParser<> zap("Auxiliary program to draw the strigs of detector.");

    zap['a'] = make_field(detectorFile, "detector file")                      = JPARSER::initialised();
    zap['o'] = make_field(outputFile,   "graphics output")                    = "detector.root";
    zap['z'] = make_field(step)                                               = 0.0;
    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);
  }

  const double V = 0.61 * C;

  JManager<int, TH1D>   H1(new TH1D("H[%].s", NULL, 500, -1.0e3, 1.0e+3));
  map     <int, double> z1;

  map<int, JGraph_t>  H0;
  map<int, double>    z0;

  for (JDetector::const_iterator module = detector.begin();  module != detector.end(); ++module) {

    if (module->getFloor() == 1) {

      if (step != 0.0)
        z0[module->getString()] = module->getZ();
      else
        z0[module->getString()] = 0.0;

      z1[module->getString()] = module->getZ() - getAverage(make_array(module->begin(), module->end(), &JPMT::getT0)) * V;
    }
  }
  
  for (JDetector::const_iterator module = detector.begin();  module != detector.end(); ++module) {

    if (module->getFloor() != 0) {

      const double dz = module->getZ() - (module->getFloor() - 1) * step                                           -  z0[module->getString()];
      const double ds = module->getZ() - getAverage(make_array(module->begin(), module->end(), &JPMT::getT0)) * V  -  z1[module->getString()];

      H0[module->getString()].put((Double_t) module->getFloor(), dz);
      H1[module->getString()]->Fill(ds);

      DEBUG(module->getLocation() << ' ' << FIXED(7,3) << ds << endl);
    }
  }

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

  out << H1;

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

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

}