/data/git/CMSSW_5_3_11_patch5/src/RecoVertex/BeamSpotProducer/interface/BSFitter.h

Go to the documentation of this file.

#ifndef BeamSpotProducer_BSFitter_h
#define BeamSpotProducer_BSFitter_h

// CMS
#include "RecoVertex/BeamSpotProducer/interface/BSpdfsFcn.h"
#include "RecoVertex/BeamSpotProducer/interface/BSTrkParameters.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"

// ROOT
#include "TMatrixD.h"
#include "TMath.h"
#include "Minuit2/VariableMetricMinimizer.h"
#include "TH1F.h"

// C++ standard
#include <vector>
#include <string>

class BSFitter {
  public:

        //typedef std::vector <BSTrkParameters> BSTrkCollection;
        
        BSFitter();
        BSFitter( std::vector< BSTrkParameters > BSvector);
        
        virtual ~BSFitter();

        void SetFitType(std::string type) {
                ffit_type = type;
        }

        void SetFitVariable(std::string name) {
                ffit_variable = name;
        }

        reco::BeamSpot Fit();
        
        reco::BeamSpot Fit(double *inipar);
                
        // Fit Z distribution with a gaussian
        reco::BeamSpot Fit_z(std::string type, double *inipar);

        reco::BeamSpot Fit_z_chi2(double *inipar);
        reco::BeamSpot Fit_z_likelihood(double *inipar);
        
        // Fit only d0-phi distribution with a chi2
        reco::BeamSpot Fit_d0phi();
        void SetMaximumZ(double z) { fMaxZ = z; }
        void SetConvergence(double val) { fconvergence = val; }
        void SetMinimumNTrks(int n) { fminNtrks = n; }
        void Setd0Cut_d0phi(double d0cut);
        void SetChi2Cut_d0phi(double chi2cut);
        void SetInputBeamWidth(double val) { finputBeamWidth = val; }
        int GetAcceptedTrks() { return ftmprow; }
        void d0phi_Init() {
                ftmprow = 0;
                ftmp.ResizeTo(4,1);
                ftmp.Zero();
                fnthite=0;
                goodfit=true;
        }
        std::vector < BSTrkParameters > GetData() { return fBSvector; }
        
        reco::BeamSpot Fit_ited0phi();
                
        reco::BeamSpot Fit_d_likelihood(double *inipar);
        reco::BeamSpot Fit_d_z_likelihood(double *inipar, double *error_par);
        reco::BeamSpot Fit_dres_z_likelihood(double *inipar);

    double scanPDF(double *init_pars,int & tracksFailed,int option);
    
        double GetMinimum() {
                return ff_minimum;
        }
        double GetResPar0() {
                return fresolution_c0;
        }
        double GetResPar1() {
                return fresolution_c1;
        }
        double GetResPar0Err() {
                return fres_c0_err;
        }
        double GetResPar1Err() {
                return fres_c1_err;
        }

        reco::BeamSpot::ResCovMatrix GetResMatrix() {
                return fres_matrix;
        }

        TH1F * GetVzHisto() { return h1z; }
        
  private:

        ROOT::Minuit2::ModularFunctionMinimizer* theFitter;
        //BSzFcn* theGausszFcn;
        BSpdfsFcn* thePDF;
        
        reco::BeamSpot::BeamType fbeamtype;
        std::string ffit_type;
        std::string ffit_variable;

        double ff_minimum;
        
        static const int fdim         = 7;
        
        std::string fpar_name[fdim];

        Double_t fsqrt2pi;
                
        std::vector < BSTrkParameters > fBSvector;
    std::vector < BSTrkParameters > fBSvectorBW;
    
        double fresolution_c0;
        double fresolution_c1;
        double fres_c0_err;
        double fres_c1_err;
        reco::BeamSpot::ResCovMatrix fres_matrix;
        //reco::BeamSpot fBSforCuts;
        TMatrixD ftmp;
        bool fapplyd0cut;
        bool fapplychi2cut;
        double fd0cut;
        double fchi2cut;
        int ftmprow;
        int fnthite;
        bool goodfit;
        double fMaxZ;
        double fconvergence;
        int fminNtrks;
        double finputBeamWidth;
        TH1F *h1z;
};

#endif