SurveyPxbImageLocalFit.h

Go to the documentation of this file.

#ifndef GUARD_surveypxbimagelocalfit_h
#define GUARD_surveypxbimagelocalfit_h

#include <sstream>
#include <vector>
#include <utility>
#include "DataFormats/GeometryVector/interface/LocalPoint.h"
#include "Alignment/SurveyAnalysis/interface/SurveyPxbImage.h"
#include "Math/SMatrix.h"
#include "Math/SVector.h"
#include <iostream>

class SurveyPxbImageLocalFit : public SurveyPxbImage
{
public:
    typedef std::vector<value_t> localpars_t;
    typedef std::vector<coord_t> fidpoint_t;
    typedef unsigned int count_t;
    static const count_t nGlD = 3; // no of global derivs
    static const count_t nLcD = 4; // no of local derivs
    static const count_t nMsrmts = 8; // no of measurements per image
    static const count_t nLcPars = 4; // no of local parameters
    static const count_t nFidpoints = 4; // no of fiducial points
    // Typedefs for pede
    typedef int pede_label_t;
    typedef float pede_deriv_t;

    // Constructors
    SurveyPxbImageLocalFit(): 
        SurveyPxbImage(), a_(nLcPars,0), fidpoints_(nFidpoints), fitValidFlag_(false), derivsValidFlag_(false)
    {
        initFidPoints();
    };

    SurveyPxbImageLocalFit(std::istringstream &iss): 
        SurveyPxbImage(iss), a_(nLcPars,0), fidpoints_(nFidpoints), fitValidFlag_(false), derivsValidFlag_(false)
    {
        initFidPoints();
    };

    void doFit(const fidpoint_t &fidpointvec);
    void doFit(const fidpoint_t &fidpointvec, const pede_label_t &label1, const pede_label_t &label2);
    void doFit(value_t x1, value_t y1, value_t g1, value_t x2, value_t y2, value_t g2);

    bool isFitValid() { return fitValidFlag_; };

    localpars_t getLocalParameters();

    value_t getChi2();

    const pede_label_t getLocalDerivsSize()                {return nLcD; };
    const pede_label_t getGlobalDerivsSize()               {return nGlD; };
    const pede_deriv_t* getLocalDerivsPtr(count_t i)       {return localDerivsMatrix_.Array()+i*nLcD; };
    const pede_deriv_t* getGlobalDerivsPtr(count_t i)      {return globalDerivsMatrix_.Array()+i*nGlD;};
    const pede_label_t* getGlobalDerivsLabelPtr(count_t i) {return i<4 ? &labelVec1_[0] : &labelVec2_[0];};
    const pede_deriv_t getResiduum(count_t i) { return (pede_deriv_t) r(i); };
    const pede_deriv_t getSigma(count_t i) { return i%2 ? sigma_x_ : sigma_y_ ; };

    void setLocalDerivsToZero(count_t i);
    void setGlobalDerivsToZero(count_t i);

private:
    localpars_t a_;
    
    ROOT::Math::SVector<value_t, nMsrmts> r;

    //std::vector<localDerivs_t> globalDerivsVec_;
    ROOT::Math::SMatrix<pede_deriv_t,nMsrmts,nGlD> globalDerivsMatrix_;

    //std::vector<globalDerivs_t> localDerivsVec_;
    ROOT::Math::SMatrix<pede_deriv_t,nMsrmts,nLcD> localDerivsMatrix_;

    std::vector<pede_label_t> labelVec1_, labelVec2_;

    std::vector<coord_t> fidpoints_;

    value_t chi2_;

    bool fitValidFlag_;

    bool derivsValidFlag_;

    void initFidPoints()
    {
        fidpoints_[0] = coord_t(-0.91,-3.30);
        fidpoints_[1] = coord_t(+0.91,-3.30);
        fidpoints_[2] = coord_t(-0.91,+3.30);
        fidpoints_[3] = coord_t(+0.91,+3.30);
    }

    value_t dist(coord_t p1, coord_t p2)
    {
        value_t dx = p1.x()-p2.x();
        value_t dy = p1.y()-p2.y();
        return sqrt(dx*dx+dy*dy);
    }
    
};

#endif