MultivariateFunctorScanner.h

Go to the documentation of this file.

#ifndef NPSTAT_MULTIVARIATEFUNCTORSCANNER_HH_
#define NPSTAT_MULTIVARIATEFUNCTORSCANNER_HH_

#include <vector>
#include <cassert>
#include "JetMETCorrections/InterpolationTables/interface/NpstatException.h"

#include "JetMETCorrections/InterpolationTables/interface/AbsMultivariateFunctor.h"

namespace npstat {
    template<class IndexMapper>
    class MultivariateFunctorScanner
    {
    public:
        inline MultivariateFunctorScanner(const AbsMultivariateFunctor& fcn,
                                          const std::vector<IndexMapper>& maps)
            : fcn_(fcn), mapping_(maps), buf_(fcn.minDim()), dim_(fcn.minDim())
        {
            if (!(dim_ && dim_ == maps.size())) throw npstat::NpstatInvalidArgument(
                "In npstat::MultivariateFunctorScanner constructor: "
                "incompatible arguments");
            if (dim_ != fcn.maxDim()) throw npstat::NpstatInvalidArgument(
                "In npstat::MultivariateFunctorScanner constructor: "
                "functors of variable dimensionality are not supported");
        }

        inline double operator()(const unsigned* index,
                                 const unsigned indexLen) const
        {
            if (dim_ != indexLen) throw npstat::NpstatInvalidArgument(
                "In npstat::MultivariateFunctorScanner::operator(): "
                "incompatible input point dimensionality");
            assert(index);
            double* x = &buf_[0];
            for (unsigned i=0; i<dim_; ++i)
                x[i] = mapping_[i](index[i]);
            return fcn_(x, dim_);
        }

    private:
        MultivariateFunctorScanner();

        const AbsMultivariateFunctor& fcn_;
        const std::vector<IndexMapper>& mapping_;
        mutable std::vector<double> buf_;
        unsigned dim_;
    };
}

#endif // NPSTAT_MULTIVARIATEFUNCTORSCANNER_HH_