JFitToT.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 "TF1.h"
#include "TFitResult.h"
#include "TMath.h"

#include "JLang/JLangToolkit.hh"
#include "JTools/JConstants.hh"
#include "km3net-dataformat/online/JDAQ.hh"

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

#include "JROOT/JRootToolkit.hh"
#include "JTools/JRange.hh"
#include "JSupport/JMeta.hh"

#include "JCalibrate/JCalibrateToT.hh"
#include "JCalibrate/JFitToT.hh"

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

namespace {

  /**
   * Wild card for histogram naming.\n
   * The wild card will replaced by the module identifier.
   */
  static const char* const WILDCARD = "%";
}


/**
 * \file
 *
 * Auxiliary program to fit time-over-threshold distributions.
 * The fitted parameters are the PMT gain and gain spread.
 * \author mkarel
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;

  typedef JRange<double>      JRange_t;

  string    inputFile;
  string    outputFile;
  string    detectorFile;
  string    pmtFile;
  bool      writeFits;
  JRange_t  fitRange;
  string    option;
  double    WMin;
  string    regexp;
  int       debug;

  try { 

    JParser<> zap("Auxiliary program to fit time-over-threshold distributions.");
    
    zap['f'] = make_field(inputFile,         "input file (output from JCalibrateToT).");
    zap['o'] = make_field(outputFile,        "output file.")                                                   = "fit.root";
    zap['a'] = make_field(detectorFile,      "detector file.");
    zap['P'] = make_field(pmtFile,           "specify PMT file name that can be given to JTriggerEfficiency.") = "";
    zap['w'] = make_field(writeFits,         "write fit results.");
    zap['x'] = make_field(fitRange,          "ROOT fit range (time-over-threshold).")                          = JRange_t(0.0, 35.0);
    zap['O'] = make_field(option,            "ROOT fit options, see TH1::Fit.")                                = "";
    zap['W'] = make_field(WMin,              "minimal histogram contents.")                                    = 1000.0;
    zap['R'] = make_field(regexp,            "regular expression for histogram name.")                         = MAKE_CSTRING(WILDCARD << _2SToT);
    zap['d'] = make_field(debug,             "debug.")                                                         = 1;

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


  //----------------------------------------------------------
  // load detector file and PMT parameters
  //----------------------------------------------------------

  JDetector detector;

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


  JPMTParametersMap parameters;

  if (pmtFile != "") {
    try { 
      parameters.load(pmtFile.c_str());
    }
    catch(const JException& error) {}
  }

  parameters.comment.add(JMeta(argc, argv));
  
  // Set ROOT fit options
  if (option.find('R') == string::npos) { option += 'R'; }
  if (option.find('S') == string::npos) { option += 'S'; }
  if (option.find('Q') == string::npos && debug < JEEP::debug_t) { option += 'Q'; }


  TFile* in = TFile::Open(inputFile.c_str(), "exist");

  if (in == NULL || !in->IsOpen()) {
    FATAL("File: " << inputFile << " not opened." << endl);
  }


  TH2D gain("gain", NULL,
            100,  0.0, 2.0,
            100,  0.0, 1.0);
  TH1D chi2("chi2", NULL, 100,  0.0, 5.0);


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

  //----------------------------------------------------------
  // loop over modules in detector file to fit histogram
  //----------------------------------------------------------

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

    DEBUG("Module " << module->getID() << endl);

    //----------------------------------------------------------
    // get histogram for this module
    //----------------------------------------------------------

    TH2D* h2s = (TH2D*) in->Get(replace(regexp, WILDCARD, MAKE_STRING(module->getID())).c_str());

    if (h2s == NULL) {

      WARNING("No histogram " << module->getID() << "; skip fit." << endl);

      continue;
    }

    const int    ny   = h2s->GetYaxis()->GetNbins();
    const double ymin = h2s->GetYaxis()->GetXmin();
    const double ymax = h2s->GetYaxis()->GetXmax();

    TH1D chi2_1d      (MAKE_CSTRING(module->getID() << ".1chi2"),       NULL, NUMBER_OF_PMTS, -0.5, NUMBER_OF_PMTS-0.5);
    TH1D gain_1d      (MAKE_CSTRING(module->getID() << ".1gain"),       NULL, NUMBER_OF_PMTS, -0.5, NUMBER_OF_PMTS-0.5);
    TH1D gainspread_1d(MAKE_CSTRING(module->getID() << ".1gainspread"), NULL, NUMBER_OF_PMTS, -0.5, NUMBER_OF_PMTS-0.5);

    for (int ix = 1; ix <= h2s->GetXaxis()->GetNbins(); ++ix) {

      const JPMTIdentifier id(module->getID(), ix-1);

      parameters[id].setPMTParameters(parameters.getDefaultPMTParameters());

      TH1D h1(MAKE_CSTRING(module->getID() << "." << id.getPMTAddress() << ".1ToT"), NULL, ny, ymin, ymax); 

      h1.Sumw2();

      for (int iy = h2s->GetNbinsY(); iy >= 1; --iy) {

        const double w = h1.GetBinWidth (iy);
        const double y = h2s->GetBinContent(ix, iy);

        h1.SetBinContent(iy, y/w);
        h1.SetBinError  (iy, sqrt(y)/w);
      }

      // Normalize histogram
      const double Wtot = h1.Integral();
      h1.Scale(1.0/Wtot);


      //----------------------------------------------------------
      // set-up fitting procedure
      //----------------------------------------------------------

      JFitToT fit(parameters.getPMTParameters(id), fitRange);
      const TFitResultPtr result = fit(h1, WMin/Wtot, option.c_str());

      if ((int(result) < 0) || (!result->IsValid())) { // Fit failed
        
        WARNING("Fit failure for PMT " << id << "; set default values." << endl);

        h1.GetListOfFunctions()->Clear();

        const double initG  = fit.getInitGain();
        const double initGS = fit.getInitGainSpread();

        parameters[id].gain       = ( initG < FITTOT_GAIN_MIN ? FITTOT_GAIN_MIN :
                                     (initG > FITTOT_GAIN_MAX ? FITTOT_GAIN_MAX : initG) );
        parameters[id].gainSpread = ( initGS < FITTOT_GAINSPREAD_MIN ? FITTOT_GAINSPREAD_MIN :
                                     (initGS > FITTOT_GAINSPREAD_MAX ? FITTOT_GAINSPREAD_MAX : initGS) );

        if (writeFits) {
          out << h1;
        }

        continue;
      }

      if ((ymin < fitRange.getLowerLimit()  || ymax > fitRange.getUpperLimit()) &&
          (option.find("0") == string::npos && option.find("N") == string::npos)) {
        
        // add function with full range
        
        TF1* f1 = new TF1(MAKE_CSTRING(FITTOT_FNAME << "_fullRange"),
                          &fit,
                          &JFitToT::getValue,
                          ymin,
                          ymax,
                          JFitToT::getNumberOfModelParameters());
        f1->SetParameters(fit.GetParameters());
        f1->SetLineStyle(kDotted);
        f1->SetNpx(1000);

        h1.GetListOfFunctions()->Add(f1);
      }

      // store fit results

      const double reducedChi2 = result->Chi2() / result->Ndf();

      const JFitToTParameters values(fit.GetParameters());
      const JFitToTParameters errors(fit.GetParErrors());
      
      chi2_1d.      SetBinContent(ix, reducedChi2);
      gain_1d.      SetBinContent(ix, values.gain);
      gain_1d.      SetBinError  (ix, errors.gain);
      gainspread_1d.SetBinContent(ix, values.gainSpread);
      gainspread_1d.SetBinError  (ix, errors.gainSpread);
      
      gain.Fill(fit.gain, fit.gainSpread);
      chi2.Fill(TMath::Log10(reducedChi2));
      
      NOTICE("PMT " << id << " chi2 / NDF " << result->Chi2() << ' ' << result->Ndf() << endl);
      
      parameters[id].gain       = fit.gain;
      parameters[id].gainSpread = fit.gainSpread;
        
      if (writeFits) {
        out << h1;
      }
    }

    if (writeFits) {
      out << chi2_1d << gain_1d << gainspread_1d;
    }
  }
    
  if (pmtFile != "") {
    parameters.store(pmtFile.c_str());
  }

  out << gain << chi2;

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