ParametricGeoReader.h

Go to the documentation of this file.

/*
 * ParametricGeoReader.h
 *
 *  Created on: 07 Jan 2021
 *      Author: Valerio Formato
 */

#ifndef HERD_PARAMETRICGEOREADER_H_
#define HERD_PARAMETRICGEOREADER_H_

#ifndef GGSSMARTLOG_H_
#define HERD_PARAMETRICGEOREADER_H_UNDEF_LOGGING
#endif

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

#ifdef HERD_PARAMETRICGEOREADER_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

// HerdSoftware headers
#include "common/DefaultValues.h"
#include "dataobjects/CaloGeoParams.h"
#include "dataobjects/CssGeoParams.h"
#include "dataobjects/FitGeoParamsColl.h"
#include "dataobjects/PsdGeoParamsColl.h"
#include "dataobjects/ScdGeoParamsColl.h"
#include "dataobjects/StkGeoParamsColl.h"
#include "dataproviders/readers/DataReader.h"

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

using namespace EA;

namespace Herd {

// clang-format off
// clang-format on

class ParametricGeoReader : public DataReader {
public:
  ParametricGeoReader(std::shared_ptr<GGSTRootReader> reader);

  virtual ~ParametricGeoReader() = default;

  virtual bool SetCurrentEvent(unsigned int) final { return true; };

  virtual bool IsObjectAvailable(const std::string &name) final;

  virtual bool IsObjectKnown(const std::string &name) final;

  RetrievalResult GetObject(const std::string &name, ObjectCategory category, ObjectWrapper &wrapper,
                            std::string &actualName) final;

  virtual std::vector<std::string> FreeObjects(const std::vector<std::string> &, Memory::Status) final {
    return std::vector<std::string>();
  };

  std::function<unsigned int(const float *)> CaloHitIDFromGGSPos() { return m_caloVolIDFromGGSPos; };

  std::array<int, 4> SideFitFirstVolumeID() { return m_sideFitFirstVolumeID; }

  std::array<int, 4> SideStkFirstWaferID() { return m_sideStkFirstWaferID; }

  std::array<int, 4> SideStkFirstStripID() { return m_sideStkFirstStripID; }

  std::array<int, 4> SideScdFirstWaferID() { return m_sideScdFirstWaferID; }

  std::array<int, 4> SideScdFirstStripID() { return m_sideScdFirstStripID; }

  std::pair<bool, bool> StkHasStrips() { return {m_topStkHasStrips, m_sideStkHasStrips}; }

  std::pair<bool, bool> ScdHasStrips() { return {m_topScdHasStrips, m_sideScdHasStrips}; }

  enum class PsdType : int { NONE = -1, tiles = 0, bars = 1 };
  PsdType GetPsdType() { return m_psdType; }

private:
  struct ObjDescriptor {
    std::string name;
    std::vector<std::string> aliases;
    bool isOnSource;
    ObjectWrapper wrapper;
    std::function<ObjectWrapper(ParametricGeoReader *)> retriever;
    std::function<bool(ParametricGeoReader *)> checker;
  };

  struct CaloGeoParamsHelpValues; // implemented inside CaloReader.cpp

  std::vector<ObjDescriptor> m_knownObjects;

  std::shared_ptr<GGSTRootReader> m_ggsreader;

  void CheckAvailableObjects();

  // CALO
  CaloGeoParams m_caloGeoParams;
  std::function<unsigned int(const float *)> m_caloVolIDFromGGSPos;
  bool IsCaloGeoParamsAvailable();
  ObjectWrapper GetCaloGeoParams();
  // Methods used for cube mask, which constins the GGS ID
  int CaloTray(int iGGSCubeID, CaloGeoParamsHelpValues caloValues);
  int CaloLayer(int iGGSCubeID, CaloGeoParamsHelpValues caloValues);
  unsigned int CaloHitIDFromGGSID(int iGGSCubeID, CaloGeoParamsHelpValues caloValues);

  // PSD
  PsdGeoParamsColl m_psdGeoParamsColl;
  PsdType m_psdType = PsdType::NONE;
  bool IsPsdGeoParamsAvailable();
  ObjectWrapper GetPsdGeoParams();

  RetrievalResult BuildBarPsdGeoParams(const GGSTGeoParams *params, bool onSource);
  RetrievalResult BuildTilePsdGeoParams(const GGSTGeoParams *params, bool onSource);

  // FIT
  FitGeoParamsColl m_fitGeoParamsColl;
  std::array<int, 4> m_sideFitFirstVolumeID;
  bool IsFitGeoParamsAvailable();
  ObjectWrapper GetFitGeoParams();

  // STK
  StkGeoParamsColl m_stkGeoParamsColl;
  std::array<int, 4> m_sideStkFirstWaferID;
  std::array<int, 4> m_sideStkFirstStripID;
  bool m_topStkHasStrips = false;
  bool m_sideStkHasStrips = false;
  bool IsStkGeoParamsAvailable();
  ObjectWrapper GetStkGeoParams();

  // SCD
  ScdGeoParamsColl m_scdGeoParamsColl;
  std::array<int, 4> m_sideScdFirstWaferID;
  std::array<int, 4> m_sideScdFirstStripID;
  bool m_topScdHasStrips = false;
  bool m_sideScdHasStrips = false;
  bool IsScdGeoParamsAvailable();
  ObjectWrapper GetScdGeoParams();

  // CSS
  CssGeoParams m_cssGeoParams;
  bool IsCssGeoParamsAvailable();
  ObjectWrapper GetCssGeoParams();

  // TGeo
  bool IsGGSTGeoAvailable();
  ObjectWrapper GetGGSTGeo();
};
} // namespace Herd

#endif