SdSimMuonNumberFitter.h

Go to the documentation of this file.

#ifndef _SdSimMuonNumberFitter_h_
#define _SdSimMuonNumberFitter_h_

#include <fwk/VModule.h>
#include <evt/Event.h>

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

#include <vector>
#include <fstream>


namespace evt {
    class Event;
}

namespace sevt {
    class Station;
}

namespace sdet {
    class MuonProfile;
}

class TVirtualPad;

namespace SdSimMuonNumberFitterNS {

  struct StationData {
    int fId;
    utl::Point fPos;
    unsigned int fNMuon;
    utl::TimeStamp fTime;
  };


  class Likelihood {
  public:
    double fTheta;
    double fPhi;

    struct InternalStationData {
      int fId;
      double fX,fY;
      double fTheta;
      unsigned int fNMuon;
      InternalStationData(const int id, const double x, const double y, const double theta, const unsigned int nmu)
        : fId(id), fX(x), fY(y), fTheta(theta), fNMuon(nmu) { }
    };

    typedef std::vector<InternalStationData> InternalStationList;
    InternalStationList fStationList;

  public:
    void Clear() { InternalStationList().swap(fStationList); }

    void SetTheta(const double theta) { fTheta = theta; }
    void SetPhi(const double phi) { fPhi = phi; }

    void MinuitFnc(int& nPar, double* const grad, double& value, double* const par, const int flag);

    void AddStation(const int id, const utl::Point& sPos, const utl::Point& showerOrigin,
                    const utl::CoordinateSystemPtr& coreCS, const unsigned int nmuon);

    void GetChi2(double& chi2, unsigned int& ndof, const double n19, const double corex, const double corey);

    unsigned int GetNumberOfStations() { return fStationList.size(); }
  };


  class SdSimMuonNumberFitter : public fwk::VModule {

  public:
    SdSimMuonNumberFitter();
    virtual ~SdSimMuonNumberFitter();

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

    enum EnergyFitType { eUnknown = 0, eROOT, eMinuit };

  private:
    int fVerbosity;

    double fSilentRadius;

    bool fFixedCore;

    double fEnergyConstant;
    double fGamma;

    utl::CoordinateSystemPtr fSimCoreCS;

    utl::Point FindMuonDensityBaryCenter(const std::vector<StationData>& stationList) const;

    static void N19FitFnc(int& nPar, double* const grad, double& value, double* const par, const int flag);

    bool EnergyFit(const utl::CoordinateSystemPtr& localCS,
                   utl::Point& core, utl::Vector& coreError, double& n19, double& n19Error,
                   double& logLikelihoodMin, const bool fixedCore) const;

    void DrawLogLikelihood(TVirtualPad* const pad, const double n19, const char* const comment) const;

    void DrawMuonDensity(TVirtualPad* const pad, const double n19, const double theta, const double phi, const char* const comment) const;

    REGISTER_MODULE("SdSimMuonNumberFitter",SdSimMuonNumberFitter) ;

  };

}


#endif