IBranchNamer.h

Go to the documentation of this file.

#ifndef RDFAnalysis_IBranchNamer_H
#define RDFAnalysis_IBranchNamer_H

#include <string>
#include <vector>
#include <set>
#include <memory>
#include <map>

#include <ROOT/RDataFrame.hxx>

namespace RDFAnalysis {
  class IBranchNamer {
    public:
      virtual ~IBranchNamer() {}

      virtual std::string nameBranch(
          const std::string& branch,
          const std::string& systName = "") const = 0;

      virtual std::vector<std::string> nameBranches(
          const std::vector<std::string>& branches,
          const std::string& systName = "") const;

      virtual std::string createBranch(
          const std::string& branch,
          const std::string& systName = "") = 0;

      virtual bool exists(
          const std::string& branch,
          const std::string& systName = "") const = 0;

      virtual const std::string& nominalName() const = 0;

      virtual std::vector<std::string> systematics() const = 0;

      virtual std::set<std::string> systematicsAffecting(
          const std::string& branch) const = 0;

      virtual std::set<std::string> systematicsAffecting(
          const std::vector<std::string>& branches) const;

      virtual std::vector<std::string> branches() const = 0;

      virtual void readBranchList(
          const std::map<std::string, ROOT::RDF::RNode>& rnodes) = 0;

      virtual std::unique_ptr<IBranchNamer> copy() const = 0;

      virtual std::pair<std::string, std::vector<std::string>> expandExpression(
          const std::string& expression) const;

      virtual std::string interpretExpression(
          const std::string& expression,
          const std::vector<std::string>& branches,
          const std::string& systematic);

  }; //> end class IBranchNamer
} //> end namespace RDFAnalysis

#endif //> !RDFAnalysis_IBranchNamer_H