ParametrizationHelper.h

Go to the documentation of this file.

#ifndef DataFormats_PatCandidates_interface_ParametrizationHelper_h
#define DataFormats_PatCandidates_interface_ParametrizationHelper_h

#include "DataFormats/PatCandidates/interface/CandKinResolution.h"
#include <string>

namespace pat {
  namespace helper {
    namespace ParametrizationHelper {

      inline uint32_t dimension(pat::CandKinResolution::Parametrization parametrization) {
        return (static_cast<uint32_t>(parametrization) & 0x0F);
      }

      pat::CandKinResolution::Parametrization fromString(const std::string &name);

      const char *name(pat::CandKinResolution::Parametrization param);

      math::PtEtaPhiMLorentzVector polarP4fromParameters(pat::CandKinResolution::Parametrization parametrization,
                                                         const AlgebraicVector4 &parameters,
                                                         const math::PtEtaPhiMLorentzVector &initialP4);

      math::XYZTLorentzVector p4fromParameters(pat::CandKinResolution::Parametrization parametrization,
                                               const AlgebraicVector4 &parameters,
                                               const math::XYZTLorentzVector &initialP4);

      AlgebraicVector4 parametersFromP4(pat::CandKinResolution::Parametrization parametrization,
                                        const math::XYZTLorentzVector &p4);

      AlgebraicVector4 parametersFromP4(pat::CandKinResolution::Parametrization parametrization,
                                        const math::PtEtaPhiMLorentzVector &p4);

      void setParametersFromP4(pat::CandKinResolution::Parametrization parametrization,
                               AlgebraicVector4 &pars,
                               const math::XYZTLorentzVector &p4);

      void setParametersFromP4(pat::CandKinResolution::Parametrization parametrization,
                               AlgebraicVector4 &pars,
                               const math::PtEtaPhiMLorentzVector &p4);

      template <typename T>
      void setParametersFromAnyVector(pat::CandKinResolution::Parametrization parametrization,
                                      AlgebraicVector4 &pars,
                                      const T &p4);


      AlgebraicVector4 diffToParameters(pat::CandKinResolution::Parametrization parametrization,
                                        const math::XYZTLorentzVector &p4ini,
                                        const math::XYZTLorentzVector &p4fin);


      AlgebraicVector4 diffToParameters(pat::CandKinResolution::Parametrization parametrization,
                                        const math::PtEtaPhiMLorentzVector &p4ini,
                                        const math::PtEtaPhiMLorentzVector &p4fin);

      bool isAlwaysMassless(pat::CandKinResolution::Parametrization parametrization);

      bool isAlwaysMassive(pat::CandKinResolution::Parametrization parametrization);

      bool isMassConstrained(pat::CandKinResolution::Parametrization parametrization);

      bool isPhysical(pat::CandKinResolution::Parametrization parametrization,
                      const AlgebraicVector4 &v4,
                      const math::PtEtaPhiMLorentzVector &initialP4);
    }  // namespace ParametrizationHelper
  }    // namespace helper
}  // namespace pat

#endif