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

// Dear emacs, this is -*- c++ -*-
// $Id: IConstraintFit.h 670486 2015-05-28 09:29:21Z schaffer $
//
// Z mass constraint fit
//
// original version by:
// Konstantinos Nikolopoulos
// Univ. of Athens
//
#ifndef ZMASSCONSTRAINT_ICONSTRAINTFIT_H
#define ZMASSCONSTRAINT_ICONSTRAINTFIT_H

// Framework include(s):
#include "AsgTools/IAsgTool.h"

// EDM includes
#include "xAODBase/IParticle.h"
#include "xAODMuon/Muon.h"
#include "xAODEgamma/Electron.h"

// ZMassConstraint I/O includes
#include "ZMassConstraint/ConstraintFitInput.h"
#include "ZMassConstraint/ConstraintFitOutput.h"

namespace ZMassConstraint 
{

    enum MassConstraintMuonType {
        isCombMCMT,      // 0
        isID_MCMT,       // 1
        isMS_MCMT        // 2
    };

    class IConstraintFit  : public virtual asg::IAsgTool 
    {
        ASG_TOOL_INTERFACE( ZMassConstraint::IConstraintFit )
        
        public:

        virtual StatusCode  doMassFit     (const ConstraintFitInput& input, 
                                           ConstraintFitOutput&      output) = 0;

        virtual double      getMassError  (const ConstraintFitInput& firstInput,
                                           const ConstraintFitInput& secondInput = ConstraintFitInput()) = 0;

        virtual double      getMassError  (const ConstraintFitOutput& fitOutput,
                                           const ConstraintFitInput&  extraInput = ConstraintFitInput()) = 0;

        virtual double      getMassError  (const ConstraintFitOutput& fitOutput,
                                           const ConstraintFitOutput& secondFitOutput) = 0;

        virtual void        addParticle   (const xAOD::Muon&       mu, 
                                           ConstraintFitInput&     input, 
                                           MassConstraintMuonType  muonType = isCombMCMT) = 0;

        virtual void        addParticle   (const xAOD::Electron&   el,
                                           float                   elEnergyRes,
                                           ConstraintFitInput&     input) = 0;

        //    The four vector is used in the mass fit and comes from the FsrTool
        virtual void        addFSRParticle(const xAOD::IParticle & part,
                                           const TLorentzVector&   fsr4Vec,
                                           ConstraintFitInput&     input) = 0;
        
    };

} // namespace ZMassConstraint 

#endif // ZMASSCONSTRAINT_ICONSTRAINTFIT_H

---