/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/ZMassConstraint/ZMassConstraint/ConstraintFitInput.h

#ifndef CONSTRAINTFITINPUT_H
#define CONSTRAINTFITINPUT_H
#include "EventPrimitives/EventPrimitives.h"
#include "TLorentzVector.h"

namespace ZMassConstraint 
{
    class ConstraintFitInput {

      
    public:

        ConstraintFitInput() : m_isOK(true) {}
        ~ConstraintFitInput()               {}

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

        const TLorentzVector&   getConstituentFourVector(int i) const { return m_particles[i]; }

        const AmgMatrix(5,5)&   getConstituentCovariance(int i) const { return m_covariances[i]; }

        void                    getConstituentCovariancePhiThetaP(int i, AmgMatrix(3,3)& outMatrix) const 
            { outMatrix = m_covariances[i].block(2, 2, 3, 3); }

        const AmgMatrix(5,5)&   getCovarianceCartesian(int ipart) const { return m_cartCovariances[ipart]; }

        void addConstituent_FourVector_d0z0PhiThetaP(const TLorentzVector& vector,
                                                     const AmgMatrix(5,5)& covard0z0PhiThetaP,
                                                     const AmgMatrix(5,5)& covarXYZ,
                                                     bool isOK)
            {   m_particles.push_back(vector); m_covariances.push_back(covard0z0PhiThetaP);
                m_cartCovariances.push_back(covarXYZ); if (m_isOK) m_isOK = isOK; }

        inline bool isOK() { return m_isOK; }

    protected:
        std::vector<TLorentzVector> m_particles;
        std::vector<AmgMatrix(5,5)> m_covariances;    
        std::vector<AmgMatrix(5,5)> m_cartCovariances; 
        bool m_isOK;

    };
}


#endif

---