SdHorizontalReconstruction.h

Go to the documentation of this file.

#ifndef _SdHorizontalReconstructionNS_SdHorizontalReconstruction_h_
#define _SdHorizontalReconstructionNS_SdHorizontalReconstruction_h_

#include "FitInterface.h"

#include <fwk/VModule.h>

#include <evt/Event.h>

#include <utl/TimeStamp.h>
#include <utl/CoordinateSystemPtr.h>

#include <boost/shared_ptr.hpp>

#include <vector>

namespace sdet {
  class SDetector;
  class Station;
}

namespace evt {
  class Event;
}

namespace sevt {
  class SEvent;
  class Station;
}

namespace tls {
  class MuonProfile;
  class VTankResponse;
  class EMComponent;
}

namespace utl {
  class Point;
}

namespace SdHorizontalReconstructionNS{

  enum ExternalGeometry {
    eNone,
    eMC,
    eFixHybrid,
    eSoftHybrid
  };

  enum ShowerFrontModel
  {
    eSphere,
    eMuonArrival
  };

  enum ShowerFrontFitOption {
    eFixCurvature = 1 << 0,
    eFixDirection = 1 << 1
  };

  enum ShowerSizeFitOption {
    eFixAxis = 1 << 0,
    eFixCore = 1 << 1,
    eApproximate = 1 << 2
  };

  class SdHorizontalReconstruction : public fwk::VModule{
  public:

    SdHorizontalReconstruction() {}
    virtual ~SdHorizontalReconstruction() {}

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

    // config
    double fMaxSilentRadius;
    double fMaxSilentRelDistance;
    double fSilentSignalThreshold;
    bool   fDropStationsBelowThreshold;
    double fMinTheta;
    double fMaxTheta;
    double fMaxDistanceCoreToBarycenter;
    double fMaxThetaDifference;
    double fMaxTimeSigma;
    double fMaxDistanceStationToExternalCore;
    std::size_t fMinStationsForEnergyFit;
    std::size_t fMinStationsForProductionDistanceFit;
    boost::shared_ptr<tls::MuonProfile> fMuonProfile;
    tls::VTankResponse* fTankResponse;
    boost::shared_ptr<tls::EMComponent> fEMComponent;
    std::vector<double> fN19Correction;
    double fEnergyConstant;
    double fGamma;
    ExternalGeometry fExternalGeometry;
    ShowerFrontModel fShowerFrontModel;
    bool fPropagateAxisUncertainty;
    std::vector<double> fCurvModelPars;

    // local reference system
    utl::Point fBary;
    utl::TimeStamp fBaryTime;
    utl::CoordinateSystemPtr fBaryCS;
    utl::Point fEarthCenter;

    bool
    UpdateBarycenter(std::size_t& nCandidates,
                     const sevt::SEvent& sevent);

    bool
    GetExternalGeometry(ExternalGeometryData& gd,
                        const evt::Event& event);

    bool
    CleanEvent(std::size_t& nCandidates,
               sevt::SEvent& sEvent,
               const AxisData& axisData,
               const SizeData& sizeData) const;

    void
    PrepareStationData(StationList& list,
                       SilentStationList& slist,
                       const utl::Point& core,
                       const utl::Point& origin,
                       const sevt::SEvent& sEvent) const;

    bool
    FitShowerFront(const utl::Point& core,
                   AxisData& ad,
                   const ExternalGeometryData& gd,
                   const StationList& list,
                   const int option = 0) const;

    bool
    FitShowerSize(SizeData& sd,
                  const AxisData& ad,
                  const ExternalGeometryData& gd,
                  const StationList& list,
                  const SilentStationList& slist,
                  const int option = 0) const;

    void
    SetReconstructedValues(evt::Event& event,
                           const SizeData& sd,
                           const AxisData& ad,
                           const StationList& list,
                           const SilentStationList& slist,
                           const ExternalGeometryData& extGeom) const;

    // Utilities
    void
    GetShowerAxis(utl::Point& core,
                  utl::Point& origin,
                  const double coreX,
                  const double coreY,
                  const double theta,
                  const double phi,
                  const double distance) const;

    // convenience...
    void
    GetShowerAxis(utl::Point& core,
                  utl::Point& origin,
                  const SizeData& sd,
                  const AxisData& ad) const;

    void
    LocalCoordinates(double& xProjected, double& yProjected,
                     double& rho, double& theta,
                     const utl::Point& pos,
                     const utl::Point& origin,
                     const utl::Point& core) const;

    double
    ExpectedCurvatureRadius(const double theta) const;

    REGISTER_MODULE("SdHorizontalReconstruction", SdHorizontalReconstruction);
  };

} // SdHorizontalReconstruction2 namespace

#endif

// Configure (x)emacs for this file ...
// Local Variables:
// mode:c++
// compile-command: "make  -k"
// End: