ECLAutoCovariance.h

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#pragma once
 
/* ECL headers. */
#include <ecl/dataobjects/ECLElementNumbers.h>
 
/* ROOT headers. */
#include <TObject.h>
 
namespace Belle2 {
  struct PackedAutoCovariance {
    float sigmaNoiseSq = -1; 
    short int packedCovMat[30] = {}; 
    ClassDef(PackedAutoCovariance, 1); 
  };
 
  class ECLAutoCovariance: public TObject {
  public:
 
    ECLAutoCovariance() {};
 
    ~ECLAutoCovariance() {}
 
    const PackedAutoCovariance& getPacked(const int cellID) const
    {
      if (cellID < 1 || cellID > ECLElementNumbers::c_NCrystals) return m_acov[0];
      return m_acov[cellID - 1];
    }
 
    void getAutoCovariance(const int cellID, double acov[31]) const
    {
      if (cellID < 1 || cellID > ECLElementNumbers::c_NCrystals) return;
      const PackedAutoCovariance& tempPacked = m_acov[cellID - 1];
      acov[0] = static_cast<double>(tempPacked.sigmaNoiseSq);
      const double norm = acov[0] * (1.0 / 32767);
      for (int i = 0; i < 30; i++) acov[i + 1] = norm * static_cast<double>(tempPacked.packedCovMat[i]);
    }
 
    void setAutoCovariance(const int cellID, const double acov[31])
    {
      if (cellID < 1 || cellID > ECLElementNumbers::c_NCrystals) return;
      const double norm = 32767 / acov[0];
      PackedAutoCovariance& tempPacked = m_acov[cellID - 1];
      tempPacked.sigmaNoiseSq = static_cast<float>(acov[0]);
      for (int i = 0; i < 30; i++)
        tempPacked.packedCovMat[i] = static_cast<short int>(std::max(-32767.0, std::min(acov[i + 1] * norm, 32767.0)));
    }
 
  private:
    PackedAutoCovariance m_acov[ECLElementNumbers::c_NCrystals] = {}; 
    //Initial Version
    ClassDef(ECLAutoCovariance, 1); 
  };
} // end namespace Belle2