Interceptor.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Discriminator
// Class  :     Interceptor
// 

// Implementation:
//     Pseudo variable processor that just intercepts all the incoming
//     variables and forwards them to the calibration class.
//
// Author:      Christophe Saout
// Created:     Sat May 05 09:05 CEST 2007
// $Id: Interceptor.cc,v 1.3 2009/03/27 14:33:39 saout Exp $
//

#include <vector>
#include <algorithm>

#include "PhysicsTools/MVAComputer/interface/VarProcessor.h"
#include "PhysicsTools/MVAComputer/interface/Variable.h"

#include "PhysicsTools/MVATrainer/interface/Interceptor.h"

namespace PhysicsTools {

class Interceptor : public VarProcessor {
    public:
    typedef VarProcessor::Registry::Registry<Interceptor,
                    Calibration::Interceptor> Registry;

    Interceptor(const char *name,
                const Calibration::Interceptor *calib,
                const MVAComputer *computer);
    virtual ~Interceptor();

    virtual void configure(ConfIterator iter, unsigned int n);
    virtual void eval(ValueIterator iter, unsigned int n) const;

    private:
    Calibration::Interceptor    *interceptor;
    std::vector<double>     *values;
};

static Interceptor::Registry registry("Interceptor");

Interceptor::Interceptor(const char *name,
                         const Calibration::Interceptor *calib,
                         const MVAComputer *computer) :
    VarProcessor(name, calib, computer),
    interceptor(const_cast<Calibration::Interceptor*>(calib)),
    values(0)
{
}

Interceptor::~Interceptor()
{
    delete[] values;
}

void Interceptor::configure(ConfIterator iter, unsigned int n)
{
    std::vector<Variable::Flags> flags;
    for(ConfIterator iter2 = iter; iter2; iter2++)
        flags.push_back(*iter2);

    flags = interceptor->configure(computer, n, flags);
    if (flags.size() != n)
        return;

    for(unsigned int i = 0; i < n; i++)
        iter++(flags[i]);

    iter << Variable::FLAG_NONE;

    values = new std::vector<double>[n];
}

void Interceptor::eval(ValueIterator iter, unsigned int n) const
{
    std::vector<double> *var = values;

    for(unsigned int i = 0; i < n; i++) {
        var->resize(iter.size());
        std::copy(iter.begin(), iter.end(), var->begin());
        iter++;
        var++;
    }

    interceptor->intercept(values);

    iter(0.0);
}

} // namespace PhysicsTools