ParametricGeoCaloReader.h

Go to the documentation of this file.

/*
 * ParametricGeoCaloReader.h
 *
 *  Created on: 24 May 2021
 *      Author: Valerio Formato
 */
 
#ifndef HERD_PARAMETRICGEOCALOREADER_H_
#define HERD_PARAMETRICGEOCALOREADER_H_
 
#ifndef GGSSMARTLOG_H_
#define HERD_PARAMETRICGEOCALOREADER_H_UNDEF_LOGGING
#endif
 
// GGS headers
#include "montecarlo/readers/GGSTRootReader.h"
 
#ifdef HERD_PARAMETRICGEOCALOREADER_H_UNDEF_LOGGING
// Remove the definitions of GGS output macros (would conflict with the same macros defined in EventAnalysis)
// We only do this if GGSSmartLog.h is being parsed in this TU, otherwise there would be nothing to undef
// (or we could even be undefining the EA macros)
#undef COUT
#undef CCOUT
#undef ENDL
#endif
 
// EA headers
#include "core/ObjectWrapper.h"
#include "core/RetrievalResult.h"
#include "utils/Memory.h"
 
// HerdSoftware headers
#include "herd/analysis/common/DefaultValues.h"
#include "herd/analysis/dataobjects/CaloGeoParams.h"
 
// c++ headers
#include <functional>
#include <string>
 
namespace Herd::Readers {
// clang-format off
// clang-format on
class ParametricGeoCaloReader {
public:
  ParametricGeoCaloReader(std::shared_ptr<GGSTRootReader> ggsReader) : m_ggsReader{ggsReader}, m_sameIDs{false} {};
 
  bool IsCaloGeoParamsAvailable();
  EA::RetrievalResult CaloGeoParams(EA::ObjectWrapper &wrapper);
 
  std::function<unsigned int(const float *)> CaloHitIDFromGGSPos() { return m_caloVolIDFromGGSPos; };
  std::function<unsigned int(const float *)> CaloLpdHitIDFromGGSPos() { return m_caloLpdVolIDFromGGSPos; };
  std::function<unsigned int(const float *)> CaloSpdHitIDFromGGSPos() { return m_caloSpdVolIDFromGGSPos; };
 
  bool SameIDs() { return m_sameIDs; };
 
private:
  struct CaloGeoParamsHelpValues;
 
  std::shared_ptr<GGSTRootReader> m_ggsReader;
 
  Herd::CaloGeoParams m_caloGeoParams;
  std::function<unsigned int(const float *)> m_caloVolIDFromGGSPos;
  std::function<unsigned int(const float *)> m_caloLpdVolIDFromGGSPos;
  std::function<unsigned int(const float *)> m_caloSpdVolIDFromGGSPos;
 
  bool m_geoParamsRead = false;
  bool m_sameIDs; // If true then the GGS volume IDs are equal to HS IDs
 
  // Methods used for cube mask, which contains the GGS ID
  unsigned int CaloHitIDFromGGSID(int iGGSCubeID, CaloGeoParamsHelpValues caloValues);
  unsigned int CaloLpdHitIDFromGGSID(int iGGSLpdID, CaloGeoParamsHelpValues caloValues);
  unsigned int CaloSpdHitIDFromGGSID(int iGGSSpdID, CaloGeoParamsHelpValues caloValues);
};
} // namespace Herd::Readers
 
#endif