ScannerHelpers.h

Go to the documentation of this file.

#ifndef PhysicsTools_FWLite_ScannerHelpers_h
#define PhysicsTools_FWLite_ScannerHelpers_h

#include <string>
#include <FWCore/Reflection/interface/TypeWithDict.h>
#include <TH1.h>
#include <TH2.h>
#include <TProfile.h>
#include <TGraph.h>

// AFAIK These includes are needed to build the dictionary
// but must be kept hidden if CINT sees this class
#if !defined(__CINT__) && !defined(__MAKECINT__)
#include "CommonTools/Utils/src/ExpressionPtr.h"
#include "CommonTools/Utils/src/SelectorPtr.h"
#include "CommonTools/Utils/src/ExpressionBase.h"
#include "CommonTools/Utils/src/SelectorBase.h"
#endif

namespace helper {
  class Parser {
  public:
    Parser() {}
    static reco::parser::ExpressionPtr makeExpression(const std::string &expr, const edm::TypeWithDict &type);
    static reco::parser::SelectorPtr makeSelector(const std::string &expr, const edm::TypeWithDict &type);
    static edm::TypeWithDict elementType(const edm::TypeWithDict &wrapperType);

    //--- we define also dictionaries for these two trivial functions that should be callable even by CINT
    //    because otherwise sometimes CINT crashes even on the creation and destruction of edm::ObjectWithDict
    static bool test(const reco::parser::SelectorPtr &sel, const edm::TypeWithDict type, const void *obj);
    static double eval(const reco::parser::ExpressionPtr &sel, const edm::TypeWithDict type, const void *obj);
  };

  class ScannerBase {
  public:
    ScannerBase() {}
    ScannerBase(const edm::TypeWithDict &objType) : objType_(objType), cuts_(1), ignoreExceptions_(false) {}

    bool addExpression(const char *expr);
    void clearExpressions() { exprs_.clear(); }
    size_t numberOfExpressions() const { return exprs_.size(); }

    bool setCut(const char *cut);
    void clearCut();
    bool addExtraCut(const char *cut);
    void clearExtraCuts();
    size_t numberOfExtraCuts() const { return cuts_.size() - 1; }

    bool test(const void *obj, size_t icut = 0) const;

    double eval(const void *obj, size_t iexpr = 0) const;

    void print(const void *obj) const;

    void fill1D(const void *obj, TH1 *hist) const;

    void fill2D(const void *obj, TH2 *hist2d) const;

    void fillGraph(const void *obj, TGraph *graph) const;

    void fillProf(const void *obj, TProfile *prof) const;

    void setIgnoreExceptions(bool ignoreThem) { ignoreExceptions_ = ignoreThem; }

  private:
    edm::TypeWithDict objType_;
    std::vector<reco::parser::ExpressionPtr> exprs_;

    std::vector<reco::parser::SelectorPtr> cuts_;

    bool ignoreExceptions_;
  };
}  // namespace helper

#endif