RdStationAssociator.h

Go to the documentation of this file.

#ifndef _RdStationAssociator_RdStationAssociator_h_
#define _RdStationAssociator_RdStationAssociator_h_

#include <fwk/VModule.h>

#include <utl/CoordinateSystem.h>
#include <utl/AugerUnits.h>
#include <utl/Minou.h>

#include <vector>


namespace utl {
  class Point;
}

namespace rdet {
  class RDetector;
}


namespace RdStationAssociator {

  class RdStationAssociator : public fwk::VModule {
    public:

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

    private:
      /* This function is the core of the Module and search the closest RStation for a given point.
         The function will return the ID of the Closest station, if no station is with in a certain
         distance (masDistanceFactor) from the point -1 will be returned */
      int FindClosestStationFromPoint(const utl::Point& pt, const rdet::RDetector& Det, double masDistanceFactor);

      // Add virtual rdet::stations at the exact positions of the simulated pulses to the detector.
      void AddVirtualStations(evt::Event& event, utl::CoordinateSystemPtr corsys) const;

      double fMaximumDistance = 5 * utl::meter;
      double fTracePaddingAtFront = 2000 * utl::nanosecond;
      double fMinimumTraceLength = 12000 * utl::nanosecond;
      bool fAddNoSignalStations = false;
      double fMaximumDistanceOfNoSignalStations = 0 * utl::m;
      bool fCreateAllSDStationsWithRDTrace = false;

      std::vector<int> fIncludedStationIds;
      std::vector<int> fExcludedStationIds;

      bool fAddVirtualStations = false;
      int fFirstStationId = -1;
      bool fAddEmptyTriggeredRDStations = true;
      bool fExtrapolateTheTailOfThePulses = false;

    REGISTER_MODULE("RdStationAssociator", RdStationAssociator);
  };


  class ExpoTail : public utl::Minou::Base {

    /* This class is used to fit the tail of the sim. pulses in order to extrapolate them.
       This can be nessecary when the simulation was performed within a to short time window. */

    private:
      std::vector<std::pair<double, double>> fData;

      double fxO = 0;

    public:
      ExpoTail(std::vector<utl::Minou::ParameterDef> pars) { GetParameterDefs() = pars ; }

      void SetData(std::vector<std::pair<double, double>> data) { fData = data; }
      void Setx0(double x0) { fxO = x0; }

      double GetY(double x, const std::vector<double>& p) const { return p[0] * std::exp(p[1] * (x - fxO) / utl::ns) ; }

      double operator()(const std::vector<double>& p)
        const
      {
        double chi2 = 0;
        for (const auto& xy : fData)
          chi2 += std::pow(xy.second - GetY(xy.first, p), 2);
        return chi2;
      }
  };

}

#endif