GGSCaloSpdHitsReader.h

Go to the documentation of this file.

/*
 * GGSCaloSpdHitsReader.h
 *
 *  Created on: 25 June 2022
 *      Author: Pietro Betti
 */
 
#ifndef HERD_GGSSPDHITSREADER_H_
#define HERD_GGSSPDHITSREADER_H_
 
#ifndef GGSSMARTLOG_H_
#define HERD_GGSSPDHITSREADER_H_UNDEF_LOGGING
#endif
 
// GGS headers
#include "montecarlo/readers/GGSTHitsReader.h"
#include "montecarlo/readers/GGSTRootReader.h"
 
#ifdef HERD_GGSSPDHITSREADER_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/CaloSpdHits.h"
 
// c++ headers
#include <functional>
#include <string>
 
using namespace EA;
 
namespace Herd {
 
namespace Readers {
// clang-format off
// clang-format on
class GGSCaloSpdHitsReader {
public:
  GGSCaloSpdHitsReader(std::shared_ptr<GGSTRootReader> ggsReader)
      : m_ggsReader{ggsReader}, m_hitReader{m_ggsReader->GetReader<GGSTHitsReader>()}, m_sameIDs{false} {};
 
  bool IsCaloSpdHitsAvailable();
 
  RetrievalResult GetCaloSpdHits(ObjectWrapper &wrapper);
 
  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, Memory::Status memStatus);
 
  void SetCaloSpdHitIDFromGGSPos(std::function<unsigned int(const float *)> conversionFunction) {
    m_spdVolIDFromGGSPos = conversionFunction;
  };
 
  void SetSameIDs(bool value) { m_sameIDs = value; };
 
private:
  std::shared_ptr<GGSTRootReader> m_ggsReader;
  GGSTHitsReader *m_hitReader = nullptr;
  long long int m_lastReadEvent = std::numeric_limits<long long int>::max();
 
  std::function<unsigned int(const float *)> m_spdVolIDFromGGSPos;
  bool m_sameIDs;
  int m_caloDetIndex = std::numeric_limits<int>::lowest();
  CaloSpdHits m_CaloSpdHits;
};
} // namespace Readers
} // namespace Herd
 
#endif