eclWaveformTemplateCalibrationC4Algorithm.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
/* Own header. */
#include <ecl/calibration/eclWaveformTemplateCalibrationC4Algorithm.h>
 
/* ECL headers. */
#include <ecl/dataobjects/ECLElementNumbers.h>
#include <ecl/dbobjects/ECLDigitWaveformParameters.h>
 
/* Basf2 headers. */
#include <framework/database/DBObjPtr.h>
 
/* ROOT headers. */
#include <TFile.h>
#include <TGraph.h>
 
using namespace std;
using namespace Belle2;
using namespace ECL;
using namespace Calibration;

eclWaveformTemplateCalibrationC4Algorithm::eclWaveformTemplateCalibrationC4Algorithm():
  CalibrationAlgorithm("DummyCollector")
{
  setDescription(
    "Collects results from C3 to produce final payload, which contains new waveform templates"
  );
}
 
CalibrationAlgorithm::EResult eclWaveformTemplateCalibrationC4Algorithm::calibrate()
{
  B2INFO("eclWaveformTemplateCalibrationC4Algorithm m_firstCellID, m_lastCellID " << m_firstCellID << " " << m_lastCellID);
 
  //save to db
  ECLDigitWaveformParameters* PhotonHadronDiodeParameters = new ECLDigitWaveformParameters();
 
  std::vector<float> cellIDs;
  std::vector<float> photonNorms;
  std::vector<float> hadronNorms;
  std::vector<float> diodeNorms;
 
  //int experimentNumber = -999;
 
  for (int i = 0; i < m_numBatches; i++) {
 
    int first = (i * m_batchsize) + 1;
    int last = ((i + 1) * m_batchsize);
    if (last > ECLElementNumbers::c_NCrystals) last = ECLElementNumbers::c_NCrystals;
 
    B2INFO("eclWaveformTemplateCalibrationC4Algorithm i " << i << " " << first << " " << last);
    B2INFO("eclWaveformTemplateCalibrationC4Algorithm " << Form("PhotonParameters_CellID%d_CellID%d", first, last));
    B2INFO("eclWaveformTemplateCalibrationC4Algorithm " << Form("HadronDiodeParameters_CellID%d_CellID%d", first, last));
 
    DBObjPtr<ECLDigitWaveformParameters> tempexistingPhotonWaveformParameters(Form("PhotonParameters_CellID%d_CellID%d", first,
        last));
    DBObjPtr<ECLDigitWaveformParameters> tempexistingHadronDiodeWaveformParameters(Form("HadronDiodeParameters_CellID%d_CellID%d",
        first, last));
 
    auto runs = getRunList();
    // Take the first run
    ExpRun chosenRun = runs.front();
    B2INFO("merging using the ExpRun (" << chosenRun.second << "," << chosenRun.first << ")");
    // After here your DBObjPtrs are correct
    updateDBObjPtrs(1, chosenRun.second, chosenRun.first);
 
    for (int j = first; j <= last; j++) {
      B2INFO("Check Norm Params CellID " << j);
      B2INFO("P " << j << " " << tempexistingPhotonWaveformParameters->getPhotonParameters(j)[0]);
      B2INFO("H " << j << " " << tempexistingHadronDiodeWaveformParameters->getHadronParameters(j)[0]);
      B2INFO("D " << j << " " << tempexistingHadronDiodeWaveformParameters->getDiodeParameters(j)[0]);
 
      cellIDs.push_back(j);
      photonNorms.push_back(tempexistingPhotonWaveformParameters->getPhotonParameters(j)[0]);
      hadronNorms.push_back(tempexistingPhotonWaveformParameters->getHadronParameters(j)[0]);
      diodeNorms.push_back(tempexistingPhotonWaveformParameters->getDiodeParameters(j)[0]);
 
      float tempPhotonWaveformParameters[11];
      float tempHadronWaveformParameters[11];
      float tempDiodeWaveformParameters[11];
 
      for (int k = 0; k < 11; k++) {
        tempPhotonWaveformParameters[k] = tempexistingPhotonWaveformParameters->getPhotonParameters(j)[k];
        tempHadronWaveformParameters[k] = tempexistingHadronDiodeWaveformParameters->getHadronParameters(j)[k];
        tempDiodeWaveformParameters[k] = tempexistingHadronDiodeWaveformParameters->getDiodeParameters(j)[k];
      }
      PhotonHadronDiodeParameters->setTemplateParameters(j, tempPhotonWaveformParameters, tempHadronWaveformParameters,
                                                         tempDiodeWaveformParameters);
    }
  }
 
  auto gphotonNorms = new TGraph(cellIDs.size(), cellIDs.data(), photonNorms.data());
  gphotonNorms->SetName("gphotonNorms");
  auto ghadronNorms = new TGraph(cellIDs.size(), cellIDs.data(), hadronNorms.data());
  ghadronNorms->SetName("ghadronNorms");
  auto gdiodeNorms = new TGraph(cellIDs.size(), cellIDs.data(), diodeNorms.data());
  gdiodeNorms->SetName("gdiodeNorms");
 
  TString fName = m_outputName;
  TFile* histfile = new TFile(fName, "recreate");
  histfile->cd();
  gphotonNorms->Write();
  ghadronNorms->Write();
  gdiodeNorms->Write();
  histfile->Close();
  delete histfile;
 
 
  B2INFO("eclWaveformTemplateCalibrationC4Algorithm: Successful, now writing DB PAyload");
  saveCalibration(PhotonHadronDiodeParameters, "ECLDigitWaveformParameters");
 
  return c_OK;
}