DLECorrectionWG/DLECorrection.h

Go to the documentation of this file.

#ifndef _DLECorrectionWG_DLECorrection_h_
#define _DLECorrectionWG_DLECorrection_h_

#include <fwk/VModule.h>
#include <evt/Event.h>
#include <utl/Trace-fwd.h>
#include <utl/TabulatedFunction.h>

#include <vector>
#include <list>
#include <map>
//#include <utility>

#include <sevt/Station.h>
#include <sevt/StationConstants.h>
#include <sevt/PMTRecData.h>

// ROOT headers
#include <TFile.h>
#include <TFormula.h>
#include <TH1.h>
#include <TH1D.h>


namespace evt {
  class Event;
}

namespace sevt {
  class PMT;
  class PMTRecData;
  class Station;
}


namespace DLECorrectionWG {

  class DLECorrection : public fwk::VModule {

  public:
    DLECorrection() = default;
    virtual ~DLECorrection() = default;

    fwk::VModule::ResultFlag Init() override;
    fwk::VModule::ResultFlag Run(evt::Event& event) override;
    fwk::VModule::ResultFlag Finish() override;

  private:
    void CorrectDLE(evt::Event& event) const;
    void CorrectAverageDLE(sevt::Station& station,
                           const double theta, const double phi) const;
    void CorrectIndividualDLE(sevt::Station& station) const;
    double EstimateSignalFluct(const sevt::PMT& pmt,
                               const bool ishgsat, const int pos) const;
    double GetFCorr(const int pmtId,
                    const double theta, const double phi) const;

    bool FlagNegBins(const sevt::PMT& pmt) const;
    bool FlagOscBaselines(const sevt::PMT& pmt, const bool hgsat,
                          const int saturationValue) const;
    bool SelectPMT(const sevt::PMT& pmt, const bool lgsat, const bool hgsat,
                   const unsigned int saturationValue) const;

    void ComputeCleanedRiseFall(sevt::PMTRecData& pmtRec,
                                const unsigned int startIntegration,
                                const unsigned int endIntegration,
                                const double traceIntegral) const;

    unsigned int fTraceSize = 0;
    double fBinSize = 0;

    // Variables read from the xml
    bool fOverwriteVEMTrace = false;
    bool fCalcCorrSignalRiseFall = false;
    std::pair<double, double> fRiseTimeFractions = { 0, 0 };
    std::pair<double, double> fFallTimeFractions = { 0, 0 };
    bool fDoCorrectionIndividual = false;
    bool fDoCorrectionAverage = false;
    bool fCorrectHGSat = false;
    double fDLEFilterThreshold = 0;
    double fDLESignalThreshold = 0;
    int fDLEMinNPMTs = 0;
    double fShiftPMT1 = 0;
    double fShiftPMT2 = 0;
    double fShiftPMT3 = 0;
    std::string fAsymmParA;
    std::string fAsymmParB;
    std::vector<int> fExcludeStations;
    bool fExcludeBadTraces = false;
    bool fExcludeBadTracesCalc = false;
    double fOscThrHG = 0;
    double fOscThrLG = 0;
    int fOscOffset = 0;
    int fNegSignalThr = 0;

    // Derived quantities
    sevt::StationConstants::SignalComponent fComponent = sevt::StationConstants::eTotal;
    sevt::StationConstants::SignalComponent fCleaned = sevt::StationConstants::eTotal;
    TFormula* fFormulaDLEParA = nullptr;
    TFormula* fFormulaDLEParB = nullptr;

    REGISTER_MODULE("DLECorrectionWG", DLECorrection);

  };

}


#endif