RdPreWaveFitter.h

Go to the documentation of this file.

#ifndef _RdPreWaveFitter_RdPreWaveFitter_h_
#define _RdPreWaveFitter_RdPreWaveFitter_h_


#include <fwk/VModule.h>

#include <revt/REvent.h>

#include <utl/Point.h>
#include <utl/CoordinateSystem.h>

#include "Chi2ForPlaneWaveFit.h"


namespace RdPreWaveFitter {

  struct FitParameters {
    // !!! all angles in rad !!!
    double r = 1;
    double theta = 0;
    double thetaError = 0;
    double phi = 0;
    double phiError = 0;

    double c00 = 0;
    double c11 = 0;
    double c22 = 0;
    double c01 = 0;
    double c02 = 0;
    double c12 = 0;

    bool corrected = false;
    double chiSquareNDF = 0;
    int status = 0;
    int nCalls = 0;
    double minChi = 0;
    int NDF = 0;
    int nStations = 0;
    double recStage = 0;    // RecStage defined in trunk/Framework/Shower/ShowerRRecData.cc

    // type defines which minimization mode is used: 0=migrad, 1=simplex, 2=combined, 3=scan
    // see Minuit2 Documentation for further information
    int type = 0;
  };

  class RdPreWaveFitter: public fwk::VModule {

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

  private:
    fwk::VModule::ResultFlag ComputeSignalPosition(revt::REvent& rEvent, bool hilbert = false);

    fwk::VModule::ResultFlag ComputeRMS(revt::REvent& rEvent);

    fwk::VModule::ResultFlag ComputeBaryCenter(const revt::REvent& rEvent, utl::Point& baryCenter);

    fwk::VModule::ResultFlag PlaneWaveFit(const revt::REvent& rEvent, FitParameters& fit, int type) const;

    fwk::VModule::ResultFlag ScanPWF(const revt::REvent& rEvent, FitParameters& sFitResults) const;

    fwk::VModule::ResultFlag CallPlaneWaveFit(const revt::REvent& rEvent, FitParameters& sFitResults,
                                              bool reuseFit = false) const;

    // xml settings
    int fMinNumberOfStations = 0;
    int fNumberOfStations = 0;

    double fThetaMax = 0;
    int fNInitsTheta = 0;
    int fNInitsPhi = 0;

    bool fMaxOnComboTrace = false;
    bool fRMSonBestTrace = false;
    bool fcontinueAfterFailedFit = false;

    bool fComputeRMS = false;
    double fSNRCut = 0;

    int fNFails = 0;  // number of times where the prewavefit fails but returns eSuccess anyway

    utl::Point fCoordinateOrigin;
    utl::CoordinateSystemPtr fLocalCS;

    unsigned int nSNRCut = 0; // number of stations that are rejected due to bad SNR ratio

    Chi2ForPlaneWaveFit* ChiPWF = nullptr;

    REGISTER_MODULE("RdPreWaveFitter", RdPreWaveFitter);
  };

}

#endif