GGSPsdHitsReader.h

Go to the documentation of this file.

/*
 * GGSPsdHitsReader.h
 *
 *  Created on: 31 May 2021
 *      Author: Valerio Formato
 */
 
#ifndef HERD_GGSPSDHITSREADER_H_
#define HERD_GGSPSDHITSREADER_H_
 
#ifndef GGSSMARTLOG_H_
#define HERD_GGSPSDHITSREADER_H_UNDEF_LOGGING
#endif
 
// GGS headers
#include "montecarlo/readers/GGSTHitsReader.h"
#include "montecarlo/readers/GGSTRootReader.h"
 
#ifdef HERD_GGSPSDHITSREADER_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/PsdGeoParamsColl.h"
#include "herd/analysis/dataobjects/PsdHitsColl.h"
#include "herd/analysis/dataobjects/PsdParticleHitsColl.h"
 
// c++ headers
#include <functional>
#include <string>
#include <utility>
 
namespace Herd::Readers {
// clang-format off
// clang-format on
class GGSPsdHitsReader {
public:
  explicit GGSPsdHitsReader(std::shared_ptr<GGSTRootReader> ggsReader);
 
  bool IsPsdHitsCollAvailable();
 
  EA::RetrievalResult GetPsdHitsColl(EA::ObjectWrapper &wrapper);
 
  bool IsPsdParticleHitsCollAvailable();
 
  EA::RetrievalResult GetPsdParticleHitsColl(EA::ObjectWrapper &wrapper);
 
  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, EA::Memory::Status memStatus);
 
  void SetPsdGeoParamsColl(EA::observer_ptr<PsdGeoParamsColl> psdGeoParamsColl);
 
  void SetFileName(const std::string &fileName) { m_fileName = fileName; };
 
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();
  std::pair<unsigned, unsigned> getLayerAndLayerID(RefFrame::Side side, int GGSPsdID);
 
  struct PsdIDVolInfo {
    unsigned int ID;
    unsigned int layer;
    RefFrame::Side side;
  };
 
  bool m_readPsdParticleHits = false;
  bool m_pre04 = false; // Constructor sets this to true if we are reading a MC file produced with HS < 0.4.0
  std::vector<int> m_nElemPerLayer; // Total number of tiles in a single layer of a GGS detector
  std::vector<int> m_nElemPerSide;  // Total number of tiles per GGS detector on a single side
 
  EA::observer_ptr<PsdGeoParamsColl> m_psdGeoParamsColl = nullptr;
  std::vector<int> m_topPsdDetIndexes;
  std::vector<int> m_sidePsdDetIndexes;
  bool SetupPsdReadout(bool getParticleHits);
  bool m_setupToBeDone = true;
  PsdIDVolInfo SidePsdHitIDLayerSideFromGGSID(int iGGSPsdID, int evenOddIndex);
 
  PsdHitsColl m_psdHitsColl;
  PsdParticleHitsColl m_psdParticleHitsColl;
 
  std::string m_fileName;
};
} // namespace Herd::Readers
 
#endif