eclAutocovarianceCalibrationC3Algorithm.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/eclAutocovarianceCalibrationC3Algorithm.h>
 
/* ECL headers. */
#include <ecl/dataobjects/ECLElementNumbers.h>
#include <ecl/dbobjects/ECLAutoCovariance.h>
 
/* ROOT headers. */
#include <TFile.h>
#include <TGraph.h>
#include <TH2I.h>
#include"TMatrixDSym.h"
#include"TDecompChol.h"
 
using namespace Belle2;
using namespace ECL;
 
eclAutocovarianceCalibrationC3Algorithm::eclAutocovarianceCalibrationC3Algorithm():
  CalibrationAlgorithm("eclAutocovarianceCalibrationC3Collector")
{
  setDescription(
    "Computes the covariance matrix for each crystal"
  );
}
 
CalibrationAlgorithm::EResult eclAutocovarianceCalibrationC3Algorithm::calibrate()
{
 
 
  gROOT->SetBatch();
 
  std::vector<double> cryIDs;
  std::vector<double> invertStatusVector;
  std::vector<double> noiseMatrix00Vector;
  std::vector<double> totalCountsVector;
 
  auto CovarianceMatrixInfoVsCrysID = getObjectPtr<TH2F>("CovarianceMatrixInfoVsCrysID");
 
  ECLAutoCovariance* Autocovariances = new ECLAutoCovariance();
 
  for (int ID = 0; ID < ECLElementNumbers::c_NCrystals; ID++) {
 
    float totalCounts = CovarianceMatrixInfoVsCrysID->GetBinContent(CovarianceMatrixInfoVsCrysID->GetBin(ID + 1,
                        m_numberofADCPoints + 1));
 
    if (totalCounts < m_TotalCountsThreshold) {
      B2INFO("eclAutocovarianceCalibrationC3Algorithm: warning total entries for cell ID " << ID + 1 << " is only: " << totalCounts <<
             " Requirement is m_TotalCountsThreshold: " << m_TotalCountsThreshold);
      return c_NotEnoughData;
    }
 
    TMatrixDSym NoiseMatrix;
    NoiseMatrix.ResizeTo(m_numberofADCPoints, m_numberofADCPoints);
    for (int i = 0; i < m_numberofADCPoints; i++) {
      for (int j = 0; j < m_numberofADCPoints; j++) {
        int index = abs(i - j);
        NoiseMatrix(i, j) = float(CovarianceMatrixInfoVsCrysID->GetBinContent(CovarianceMatrixInfoVsCrysID->GetBin(ID + 1,
                                  index + 1))) / (totalCounts - 1) / (float(m_numberofADCPoints - index));
      }
    }
 
    double tempAutoCov[m_numberofADCPoints];
    for (int i = 0; i < m_numberofADCPoints; i++) tempAutoCov[i] = NoiseMatrix(0, i);
 
    TDecompChol dc(NoiseMatrix);
    bool InvertStatus = dc.Invert(NoiseMatrix);
    if (InvertStatus == false && ID > 0) {
      B2INFO("eclAutocovarianceCalibrationC3Algorithm iD InvertStatus [0][0] totalCounts: " << ID << " " << InvertStatus << " " <<
             NoiseMatrix(0, 0) << " " << totalCounts);
      B2INFO("eclAutocovarianceCalibrationC3Algorithm Setting ID " << ID << " Autocovariance to Autocovariance from ID " << ID - 1);
      B2INFO("eclAutocovarianceCalibrationC3Algorithm B: " << tempAutoCov[0]);
      Autocovariances->getAutoCovariance(ID + 1 - 1, tempAutoCov);
      B2INFO("eclAutocovarianceCalibrationC3Algorithm A: " << tempAutoCov[0]);
    }
 
    Autocovariances->setAutoCovariance(ID + 1, tempAutoCov);
 
    cryIDs.push_back(ID + 1);
    invertStatusVector.push_back(InvertStatus);
    noiseMatrix00Vector.push_back(NoiseMatrix(0, 0));
    totalCountsVector.push_back(totalCounts);
 
  }
 
  saveCalibration(Autocovariances, "ECLAutoCovariance");
 
  auto ginvertStatusVector = new TGraph(cryIDs.size(), cryIDs.data(), invertStatusVector.data());
  ginvertStatusVector->SetName("ginvertStatusVector");
  ginvertStatusVector->SetMarkerStyle(20);
  auto gnoiseMatrix00Vector = new TGraph(cryIDs.size(), cryIDs.data(), noiseMatrix00Vector.data());
  gnoiseMatrix00Vector->SetName("gnoiseMatrix00Vector");
  gnoiseMatrix00Vector->SetMarkerStyle(20);
  auto gtotalCountsVector = new TGraph(cryIDs.size(), cryIDs.data(), totalCountsVector.data());
  gtotalCountsVector->SetName("gtotalCountsVector");
  gtotalCountsVector->SetMarkerStyle(20);
 
  TString fName = m_outputName;
  TDirectory::TContext context;
  TFile* histfile = new TFile(fName, "recreate");
  histfile->cd();
  ginvertStatusVector->Write();
  gnoiseMatrix00Vector->Write();
  gtotalCountsVector->Write();
  histfile->Close();
  delete histfile;
 
  return c_OK;
}