MdMuonEstimator.h

Go to the documentation of this file.

#ifndef _MdMuonEstimatorAG_MdMuonEstimator_h_
#define _MdMuonEstimatorAG_MdMuonEstimator_h_

#include <utl/config.h> // before all

#include <fwk/VModule.h>
#include <mevt/MEvent.h>
#include <mevt/Counter.h>
#include <mdet/MDetector.h>
#include <evt/ShowerRecData.h>

#include <utl/Trace-fwd.h>
#include <utl/MessageLoggerConfig.h>
#include <utl/UnitsConfig.h>

#include <string>
#include <iostream>
#include <memory>

#include <boost/utility.hpp>


namespace MdMuonEstimatorAG {

  class VMuonEstimatorFunctor;


  class MdMuonEstimator : public boost::noncopyable, public fwk::VModule {

  public:
    enum CountingStrategyType {
      eOneBin,
      eNBin,
      eNBinCentered,
      eInfiniteWindow
    };
    static const char* const kCountingStrategyTags[];

    MdMuonEstimator() { }
    ~MdMuonEstimator() { }
    VModule::ResultFlag Init();
    VModule::ResultFlag Run(evt::Event& event);
    VModule::ResultFlag Finish();

  private:
    utl::MessageLoggerConfig fLog;

    unsigned int fCountingWindowSize;
    
    unsigned int fNBinWindowSize;

    CountingStrategyType fCountingStrategy;

    std::unique_ptr<VMuonEstimatorFunctor> fEstimator;

    utl::UnitsConfig fUnits;
    
    unsigned int GetTraceLength(evt::Event& event) const; 
    
    double GetSamplingTime(evt::Event& event) const; 
    
    std::vector<std::pair<unsigned int, unsigned int>> GetModulePatternMatches(mevt::Module& m) const;

    void FillChannelsOn(mevt::Module& m, const std::vector<std::pair<unsigned int, unsigned int>>&, const mdet::Module&) const;
    
    void LoadPatternMatchesInModuleRecData(mevt::Module& m, const std::vector<std::pair<unsigned int, unsigned int>>&, const mdet::Module&) const;
    
    void SetSegmentationAndArea(mevt::Module& module, const mdet::Module& mdetModule) const;
    
    void EstimateNumberOfMuons(mevt::Module& module, std::vector<std::pair<unsigned int, unsigned int>> vModulePatternMatches) const;
    
    REGISTER_MODULE("MdMuonEstimatorAG", MdMuonEstimator);

  };

}


#endif