ConstraintFitOutput.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef CONSTRAINTFITOUTPUT_H
#define CONSTRAINTFITOUTPUT_H
#include "EventPrimitives/EventPrimitives.h"
#include "TLorentzVector.h"
 
namespace ZMassConstraint 
{
    class ConstraintFitOutput
    {
    public:

        ConstraintFitOutput(){}
        ~ConstraintFitOutput() {}

        unsigned int             getNConstituents() const { return m_particleList.size(); }

        const TLorentzVector&    getConstituentFourVector(int index) const { return m_particleList.at(index); }

        void                     getCompositeFourVector(TLorentzVector& lv) const
            { lv = TLorentzVector(); for( auto lv1 : m_particleList ) lv += lv1; }

        Amg::MatrixX             getConstituentCovariancePhiThetaP(int index) const 
            { return m_covariancePhiThetaP.block(5*index +2, 5*index+2, 3, 3); }

        void                     getConstituentCovariancePhiThetaP(int index, AmgMatrix(3,3)& outMatrix) const 
            { outMatrix = m_covariancePhiThetaP.block(5*index +2, 5*index+2, 3, 3); }

        Amg::MatrixX             getConstituentCovarianced0z0PhiThetaP(int index) const
            { return m_covariancePhiThetaP.block(5*index, 5*index, 5, 5); }

        void                     getConstituentCovarianced0z0PhiThetaP(int index, AmgMatrix(5,5)& outMatrix) const 
            { outMatrix = m_covariancePhiThetaP.block(5*index, 5*index, 5, 5); }

        const Amg::MatrixX&      getConstituentCovariancePhiThetaP() const 
            { return m_covariancePhiThetaP; }

        void setFitOutput(const std::vector<TLorentzVector>& particleList,
                          const Amg::MatrixX& covarXYZ,
                          const Amg::MatrixX& covard0z0PhiThetaP)
            { m_particleList = particleList; m_covariance   = covarXYZ; m_covariancePhiThetaP = covard0z0PhiThetaP; }
        
    protected:
        std::vector<TLorentzVector> m_particleList;
        Amg::MatrixX                m_covariance;
        Amg::MatrixX                m_covariancePhiThetaP;
    };
}
 
#endif