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 "common/DirectionsArray.h"
#include "dataobjects/TrackInfoForCalo.h"

using namespace EA;

namespace Herd {

struct MCAccOption {
  MCAccOption() : _trackLengthCaloCm(0), _trackLengthCaloX0(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; 
  DirectionsArray<int>
      _selectSurfCalo; 
};

class CaloAcceptanceCut : public Algorithm {
public:
  CaloAcceptanceCut(const std::string &name);

  bool Initialize() override;

  bool Process() override;

  bool Finalize() override;

private:
  
  bool _GetAcceptanceType(const TrackInfoForCalo &trackInfo);

  observer_ptr<EventDataStore> _evStore; 

  MCAccOption _mcAccOpt; 
};

} // namespace Herd

#endif /* HERD_CALOACCEPTANCECUT_H_ */