PufitUtils.h

Go to the documentation of this file.

/*
 * Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
 */
 
/******************************************************************************
 * @package Trigger/TrigAlgorithms/TrigEFMissingET
 *
 * @brief Helper functions for pufit calculations
 * @author Bob Kowalewski, Jon Burr
 *****************************************************************************/
 
#ifndef TRIGEFMISSINGET_PUFITUTILS_H
#define TRIGEFMISSINGET_PUFITUTILS_H
 
#include <Eigen/Dense>
#include "TrigEFMissingET/SignedKinematics.h"
#include "TrigEFMissingET/PufitGrid.h"
#include "TrigEFMissingET/PufitMultiGrid.h"
#include <vector>
 
namespace HLT
{
    namespace MET
    {
        namespace PufitUtils
        {
            struct CovarianceSum
            {
                CovarianceSum();
                CovarianceSum(const Eigen::Vector2d &sum, const Eigen::Matrix2d &covariance);
                CovarianceSum &add(const SignedKinematics &kin, double sigma);
                Eigen::Vector2d sum;
                Eigen::Matrix2d covariance;
            };

            void trimmedMeanAndVariance(
                const std::vector<double> &sorted,
                double trimFraction,
                double &mean,
                double &variance);

            void trimmedMeanAndVariance(
                const PufitGrid &grid,
                double trimFraction,
                double &mean,
                double &variance);

            template <std::size_t N>
            void trimmedMeanAndVariance(
                const PufitMultiGrid<N> &grid,
                std::size_t type,
                double trimFraction,
                double &mean,
                double &variance);

            void unmaskedMeanAndVariance(
                const PufitGrid &grid, double &mean, double &variance);

            template <std::size_t N>
            void unmaskedMeanAndVariance(
                const PufitMultiGrid<N> &grid, int type, double &mean, double &variance);

            GridDisplacement selectGrid(const PufitGridSet &grids);

            template <typename Grid>
            GridDisplacement selectGrid(
                const PufitMultiGridSet<Grid> &grids,
                std::size_t type);

            Eigen::VectorXd pufit(
                const Eigen::Vector2d &pileupSum,
                const Eigen::Matrix2d &pileupCovariance,
                const Eigen::VectorXd &towerExpectations,
                const Eigen::VectorXd &towerVariances,
                const Eigen::VectorXd &correctionDirections,
                double constraintImportance = 1);

            Eigen::VectorXd pufit(
                const Eigen::Vector2d &pileupSum,
                const Eigen::Matrix2d &pileupCovariance,
                const Eigen::VectorXd &towerExpectations,
                const Eigen::VectorXd &towerVariances,
                const Eigen::Matrix<double, 2, Eigen::Dynamic> &cosSin,
                double constraintImportance = 1);

            std::vector<SignedKinematics> pufit(
                const Eigen::Vector2d &pileupSum,
                const Eigen::Matrix2d &pileupCovariance,
                const std::vector<double> &towerExpectations,
                const std::vector<double> &towerVariances,
                const std::vector<SignedKinematics> &toCorrect,
                double constraintImportance = 1);

            Eigen::VectorXd pufit(
                const Eigen::Vector2d &pileupSum,
                const Eigen::Matrix2d &pileupCovariance,
                double towerMean,
                double towerVariance,
                const Eigen::VectorXd &correctionDirections,
                double constraintImportance = 1);

            Eigen::VectorXd pufit(
                const Eigen::Vector2d &pileupSum,
                const Eigen::Matrix2d &pileupCovariance,
                double towerMean,
                double towerVariance,
                const Eigen::Matrix<double, 2, Eigen::Dynamic> &cosSin,
                double constraintImportance = 1);

            std::vector<SignedKinematics> pufit(
                const Eigen::Vector2d &pileupSum,
                const Eigen::Matrix2d &pileupCovariance,
                double towerMean,
                double towerVariance,
                const std::vector<SignedKinematics> &toCorrect,
                double constraintImportance = 1);
        } // namespace PufitUtils
    }     // namespace MET
} // namespace HLT
 
#include "TrigEFMissingET/PufitUtils.icc"
 
#endif //> !TRIGEFMISSINGET_PUFITUTILS_H