ProcOptional.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     MVAComputer
// Class  :     ProcOptional
//
 
// Implementation:
//     Variable processor to set empty input variables to a default value.
//
// Author:      Christophe Saout
// Created:     Sat Apr 24 15:18 CEST 2007
//
 
#include "FWCore/Utilities/interface/Exception.h"
 
#include "PhysicsTools/MVAComputer/interface/VarProcessor.h"
#include "PhysicsTools/MVAComputer/interface/Calibration.h"
 
using namespace PhysicsTools;
 
namespace {  // anonymous
 
  class ProcOptional : public VarProcessor {
  public:
    typedef VarProcessor::Registry::Registry<ProcOptional, Calibration::ProcOptional> Registry;
 
    ProcOptional(const char *name, const Calibration::ProcOptional *calib, const MVAComputer *computer);
    ~ProcOptional() override {}
 
    void configure(ConfIterator iter, unsigned int n) override;
    void eval(ValueIterator iter, unsigned int n) const override;
    std::vector<double> deriv(ValueIterator iter, unsigned int n) const override;
 
  private:
    std::vector<double> neutralPos;
  };
 
  ProcOptional::Registry registry("ProcOptional");
 
  ProcOptional::ProcOptional(const char *name, const Calibration::ProcOptional *calib, const MVAComputer *computer)
      : VarProcessor(name, calib, computer), neutralPos(calib->neutralPos) {}
 
  void ProcOptional::configure(ConfIterator iter, unsigned int n) {
    if (n != neutralPos.size())
      return;
 
    while (iter)
      iter++(Variable::FLAG_OPTIONAL) << Variable::FLAG_NONE;
  }
 
  void ProcOptional::eval(ValueIterator iter, unsigned int n) const {
    for (std::vector<double>::const_iterator pos = neutralPos.begin(); pos != neutralPos.end(); pos++, ++iter) {
      switch (iter.size()) {
        case 0:
          iter(*pos);
          break;
        case 1:
          iter(*iter);
          break;
        default:
          throw cms::Exception("ProcOptional") << "Multiple input variables encountered." << std::endl;
      }
    }
  }
 
  std::vector<double> ProcOptional::deriv(ValueIterator iter, unsigned int n) const {
    unsigned int size = 0;
    for (ValueIterator iter2 = iter; iter2; ++iter2)
      size += iter2.size();
 
    std::vector<double> result;
 
    unsigned int column = 0;
    for (std::vector<double>::const_iterator pos = neutralPos.begin(); pos != neutralPos.end(); pos++, ++iter) {
      unsigned int row = result.size();
      result.resize(row + size);
      switch (iter.size()) {
        case 0:
          break;
        case 1:
          result[row + column++] = 1.0;
          break;
        default:
          throw cms::Exception("ProcOptionalError") << "Multiple input variables encountered." << std::endl;
      }
    }
 
    return result;
  }
 
}  // anonymous namespace