FdEnergyDepositFinder.h

Go to the documentation of this file.

#ifndef _FdEnergyDepositFinderKG_FdEnergyDepositFinder_h_
#define _FdEnergyDepositFinderKG_FdEnergyDepositFinder_h_

#include "CFMatrixCalculator.h"
#include "ProfileCalculator.h"
#include "ProfileFitter.h"
#include "GHShapeParameters.h"
#include <fwk/VModule.h>
#include <utl/PhysicalFunctions.h>
#include <evt/GaisserHillas4Parameter.h>

#include <map>


namespace fevt {
  class Eye;
}
namespace evt {
  class ShowerFRecData;
}

namespace FdEnergyDepositFinderKG {

  class FdEnergyDepositFinder : public fwk::VModule {

  public:
    fwk::VModule::ResultFlag Init() override;
    fwk::VModule::ResultFlag Run(evt::Event& event) override;
    fwk::VModule::ResultFlag Finish() override { return eSuccess; }

  private:
    enum EErrorPropStage {
      eDefaultReco,
      eRp,
      eChi0,
      eT0,
      eSDPTheta,
      eSDPPhi,
      eMie,
      eNErrorPropagation,
      eFirstGeometryProp = eRp,
      eFirstErrProp = eRp,
      eLastGeometryProp = eSDPPhi
    };

    enum EErrorPropType {
      eMinus = -1,
      ePlus = 1
    };

    bool IsEventCandidate(const fevt::Eye& eye) const;
    bool FitProfile(fevt::Eye& eye,
                    const EFitFunctionType type);
    bool GuessGHParameters(evt::ShowerFRecData& shower) const;
    void DumpCurrentParameters(const evt::ShowerFRecData&, bool,
                               const EFitFunctionType) const;
    bool PrepareEyeCopy(fevt::Eye& eye, const EErrorPropType type) const;
    void PropagateUncertainties(fevt::Eye& eye) const;

  private:
    // two versions of CFM are stored - first for ProfileCalculator, second for ProfileFitter
    CFMatrixCalculator fCFMatrixCalculator;
    CFMatrixCalculator fCFMatrixCalculatorDense;
    ProfileCalculator fProfileCalculator;
    ProfileFitter fProfileFitter;

    EErrorPropStage fErrPropStage;

    typedef std::map<EErrorPropStage, evt::GaisserHillas4Parameter> RecResults;
    RecResults fRecResultsPlus;
    RecResults fRecResultsMinus;

    // GaisserHillas shape parameters, constraints etc
    GHShapeParameters fGHShapePars;

    // energy threshold for electrons
    double fElectronEnergyThreshold = 1*utl::MeV;

    // options for invisible energy
    utl::InvisibleEnergy::ECompositionModel fComposition = utl::InvisibleEnergy::eData;
    utl::InvisibleEnergy::EInteractionModel fInvisibleEnergyModel = utl::InvisibleEnergy::eDATA;

    // options for error propagation
    int fErrPropVerbosity = 0;
    bool fRescaleErrors = false;
    int fMaxErrPropPoints = 0;

    // options for event selection
    double fMaxZenith = 0;
    bool fOnlyNonMono = false;

    unsigned int fDenseMatrixDim = 0;
    int fAafCorrection = 0;
    bool fUseNoiseBins = false;

  private:
    REGISTER_MODULE("FdEnergyDepositFinderKG", FdEnergyDepositFinder);

  };

}


#endif