GGSSiliconDetectorHitsReader.h

Go to the documentation of this file.

/*
 * GGSSiliconDetectorHitsReader.h
 *
 *  Created on: 27 May 2021
 *      Author: Valerio Formato
 */

#ifndef HERD_GGSSILICONDETECTORHITSREADER_H_
#define HERD_GGSSILICONDETECTORHITSREADER_H_

#ifndef GGSSMARTLOG_H_
#define HERD_GGSSILICONDETECTORHITSREADER_H_UNDEF_LOGGING
#endif

// GGS headers
#include "montecarlo/readers/GGSTHitsReader.h"
#include "montecarlo/readers/GGSTRootReader.h"

#ifdef HERD_GGSSILICONDETECTORHITSREADER_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 "common/DefaultValues.h"
#include "dataobjects/SiliconDetectorGeoParamsColl.h"
#include "dataobjects/SiliconDetectorHitsColl.h"
#include "dataobjects/SiliconDetectorParticleHitsColl.h"

// c++ headers
#include <string>
#include <utility>

using namespace EA;

namespace Herd {

namespace Readers {
// clang-format off
// clang-format on
class GGSSiliconDetectorHitsReader {
public:
  enum class DetectorType { STK, SCD };

  GGSSiliconDetectorHitsReader(std::shared_ptr<GGSTRootReader> ggsReader, DetectorType type)
      : m_ggsReader{std::move(ggsReader)}, m_hitReader{m_ggsReader->GetReader<GGSTHitsReader>()}, m_type{type} {};

  bool IsSiliconDetectorHitsCollAvailable();
  RetrievalResult GetSiliconDetectorHitsColl(ObjectWrapper &wrapper);

  bool IsSiliconDetectorWaferParticleHitsCollAvailable() { return IsSiliconDetectorHitsCollAvailable(); };

  RetrievalResult GetSiliconDetectorWaferParticleHitsColl(ObjectWrapper &wrapper);

  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, Memory::Status memStatus);

  void SetSiliconDetectorGeoParamsColl(observer_ptr<SiliconDetectorGeoParamsColl> sdGeoParamsColl);

  void SetSideDetectorFirstWaferID(std::array<int, 4> sideSDFirstWaferID) { m_sideSDFirstWaferID = sideSDFirstWaferID; }

  //  void SetSideDetectorFirstStripID(std::array<int, 4> sideSDFirstStripID) { m_sideSDFirstStripID =
  //  sideSDFirstStripID; }

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();

  DetectorType m_type;

  observer_ptr<SiliconDetectorGeoParamsColl> m_sdGeoParamsColl = nullptr;
  int m_topSDDetIndex = std::numeric_limits<int>::lowest();
  int m_sideSDHDetIndex = std::numeric_limits<int>::lowest();
  int m_sideSDVDetIndex = std::numeric_limits<int>::lowest();
  std::array<int, 4> m_sideSDFirstWaferID{}; // ID of the first Wafer on each lateral SD, indexed as in GGS files

  SiliconDetectorHitsColl m_sdHitsColl;
  SiliconDetectorParticleHitsColl m_sdWaferParticleHitsColl;

  bool firstRun = true;
  bool SetupSDReadout();
};
} // namespace Readers
} // namespace Herd

#endif