ECLAutoCovariance.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Packed autocovariance database storage                                 *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Alexei Sibidanov                                         *
 *               Savino Longo                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#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 > 8736) return m_acov[0];
      return m_acov[cellID - 1];
    }
 
    void getAutoCovariance(const int cellID, double acov[31]) const
    {
      if (cellID < 1 || cellID > 8736) 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 > 8736) 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[8736] = {}; 
    //Initial Version
    ClassDef(ECLAutoCovariance, 1); 
  };
} // end namespace Belle2