GGSFitHitsReader.h

Go to the documentation of this file.

/*
 * GGSFitHitsReader.h
 *
 *  Created on: 31 May 2021
 *      Author: Valerio Formato
 */
 
#ifndef HERD_GGSFITHITSREADER_H_
#define HERD_GGSFITHITSREADER_H_
 
#ifndef GGSSMARTLOG_H_
#define HERD_GGSFITHITSREADER_H_UNDEF_LOGGING
#endif
 
// GGS headers
#include "montecarlo/readers/GGSTHitsReader.h"
#include "montecarlo/readers/GGSTRootReader.h"
 
#ifdef HERD_GGSFITHITSREADER_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/FitGeoParamsColl.h"
#include "herd/analysis/dataobjects/FitHitsColl.h"
#include "herd/analysis/dataobjects/FitParticleHitsColl.h"
 
// c++ headers
#include <functional>
#include <string>
#include <utility>
 
namespace Herd::Readers {
// clang-format off
// clang-format on
class GGSFitHitsReader {
public:
  explicit GGSFitHitsReader(std::shared_ptr<GGSTRootReader> ggsReader)
      : m_ggsReader{std::move(ggsReader)}, m_hitReader{m_ggsReader->GetReader<GGSTHitsReader>()} {};
 
  bool IsFitHitsCollAvailable();
 
  EA::RetrievalResult GetFitHitsColl(EA::ObjectWrapper &wrapper) { return GetHitsHelper(wrapper, true); }
 
  bool IsFitParticleHitsCollAvailable();
 
  EA::RetrievalResult GetFitParticleHitsColl(EA::ObjectWrapper &wrapper) { return GetHitsHelper(wrapper, false); }
 
  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, EA::Memory::Status memStatus);
 
  void SetFitGeoParamsColl(EA::observer_ptr<FitGeoParamsColl> fitGeoParamsColl);
 
  void SetSideDetectorFirstVolumeID(std::array<int, 4> sideFitFirstVolumeID) {
    m_sideFitFirstVolumeID = sideFitFirstVolumeID;
  }
 
private:
  std::shared_ptr<GGSTRootReader> m_ggsReader;
  GGSTHitsReader *m_hitReader = nullptr;
  long long int m_lastHitsReadEvent = std::numeric_limits<long long int>::max();
  long long int m_lastPartHitsReadEvent = std::numeric_limits<long long int>::max();
 
  EA::observer_ptr<FitGeoParamsColl> m_fitGeoParamsColl = nullptr;
  std::array<int, 2> m_topFitDetIndex = {std::numeric_limits<int>::lowest(), std::numeric_limits<int>::lowest()};
  std::array<int, 2> m_sideFitDetIndex = {std::numeric_limits<int>::lowest(), std::numeric_limits<int>::lowest()};
  std::array<int, 4>
      m_sideFitFirstVolumeID{}; // ID of the first even/odd layer on each lateral FIT, indexed as in GGS files.
 
  FitHitsColl m_fitHitsColl;
  FitParticleHitsColl m_fitParticleHitsColl;
 
  bool m_pre04 = false;
 
  bool SetupFitReadout();
  // Helper function that reads both integrated and particle hits depending on the value of the intHits parameter (true:
  // integrated, false: particle).
  EA::RetrievalResult GetHitsHelper(EA::ObjectWrapper &wrapper, bool intHits);
};
} // namespace Herd::Readers
 
#endif