GGSMCTruthReader.h

Go to the documentation of this file.

/*
 * GGSMCTruthReader.h
 *
 *  Created on: 19 May 2021
 *      Author: Valerio Formato
 */

#ifndef HERD_MCTRUTHREADER_H_
#define HERD_MCTRUTHREADER_H_

#ifndef GGSSMARTLOG_H_
#define HERD_MCTRUTHREADER_H_UNDEF_LOGGING
#endif

// GGS headers
#include "montecarlo/readers/GGSTHadrIntReader.h"
#include "montecarlo/readers/GGSTMCTruthReader.h"
#include "montecarlo/readers/GGSTPrimaryDisReader.h"
#include "montecarlo/readers/GGSTRootReader.h"

#ifdef HERD_MCTRUTHREADER_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/MCTruth.h"

// c++ headers
#include <string>

using namespace EA;

namespace Herd {

// TODO: write doxygen doc

namespace Readers {
// clang-format off
// clang-format on
class GGSMCTruthReader {
public:
  GGSMCTruthReader(std::shared_ptr<GGSTRootReader> ggsReader)
      : m_ggsReader{ggsReader}, m_mcReader{m_ggsReader->GetReader<GGSTMCTruthReader>()} {};

  bool IsMCTruthAvailable();

  RetrievalResult GetMCTruth(ObjectWrapper &wrapper);

private:
  std::shared_ptr<GGSTRootReader> m_ggsReader;
  GGSTMCTruthReader *m_mcReader = nullptr;
  GGSTHadrIntReader *m_hadrIntReader = nullptr;
  GGSTPrimaryDisReader *m_primDisReader = nullptr;
  long long int m_lastReadEvent = std::numeric_limits<unsigned long long>::max();

  MCTruth m_mcTruth;
  bool m_mcTruthRead = false;
  bool m_hadrIntChecked = false;
  bool m_PrimDisAvailable = false;
};
} // namespace Readers
} // namespace Herd

#endif