CaloAcceptanceCut.h

Go to the documentation of this file.

/*
 * CaloAcceptanceCut.h
 *
 *  Created on: 03 June 2019
 *      Author: Lorenzo Pacini
 */
 
#ifndef HERD_CALOACCEPTANCECUT_H_
#define HERD_CALOACCEPTANCECUT_H_
 
#include "algorithm/Algorithm.h"
#include "herd/analysis/common/DirectionsArray.h"
#include "herd/analysis/dataobjects/CaloGeoParams.h"
#include "herd/analysis/dataobjects/Point.h"
#include "herd/analysis/dataobjects/Segment.h"
#include "herd/analysis/dataobjects/TrackInfoForCalo.h"
#include "herd/analysis/utils/geom/GeomUtils.h"
 
namespace Herd {
 
struct MCAccOption {
  MCAccOption()
      : trackLengthCaloCm(0), trackLengthCaloX0(0), trackLengthLYSOX0(0), exactTrackLengthCaloX0(0),
        exactTrackLengthLYSOX0(0), trackMaxDistanceFromEnvelope(0) {
    std::for_each(RefFrame::Directions.begin(), RefFrame::Directions.end(), [&](auto &itDir) {
      if (itDir == RefFrame::Direction::Zneg)
        selectSurfCalo[itDir] = 0;
      else
        selectSurfCalo[itDir] = 1;
    });
  };
 
  void Dump() {
    std::cout << trackLengthCaloCm << " ; " << trackLengthCaloCm << " ; ";
    auto itName = RefFrame::DirectionName.begin();
    std::for_each(RefFrame::Directions.begin(), RefFrame::Directions.end(), [&](auto &itDir) {
      std::cout << *itName << "_flag: " << selectSurfCalo[itDir] << " ; ";
      itName++;
    });
    std::cout << std::endl;
  };
 
  float trackLengthCaloCm;            
  float trackLengthCaloX0;            
  float trackLengthLYSOX0;            
  float exactTrackLengthCaloX0;       
  float exactTrackLengthLYSOX0;       
  float trackMaxDistanceFromEnvelope; 
  DirectionsArray<int>
      selectSurfCalo; 
};
// clang-format off
// clang-format on
class CaloAcceptanceCut : public EA::Algorithm {
public:
  CaloAcceptanceCut(const std::string &name);
 
  bool Initialize() override;
 
  bool Process() override;
 
  bool Finalize() override;
 
private:
  bool _GetAcceptanceType(const TrackInfoForCalo &trackInfo);
 
  EA::observer_ptr<EA::EventDataStore> m_evStore; 
 
  MCAccOption m_mcAccOpt; 
  EA::observer_ptr<CaloGeoParams> m_caloGeoParams;
};
 
} // namespace Herd
 
#endif /* HERD_CALOACCEPTANCECUT_H_ */