DefaultBranchNamer.h

Go to the documentation of this file.

#ifndef RDFAnalysis_BranchNamer_H
#define RDFAnalysis_BranchNamer_H

// STL includes
#include <map>

// package includes
#include <RDFAnalysis/IBranchNamer.h>

namespace RDFAnalysis {
  class DefaultBranchNamer : public IBranchNamer {
    public:
      DefaultBranchNamer(
          const std::vector<std::string>& systematics,
          bool systNameFirst = true,
          bool inputFromFriends = false,
          const std::string& nominalName = "NOSYS") :
        m_systematics(systematics),
        m_systNameFirst(systNameFirst),
        m_inputFromFriendTrees(inputFromFriends),
        m_nominalName(nominalName) {}

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

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

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

      std::string newBranchName(
          const std::string& branch,
          const std::string& systName = "") const;

      const std::string& nominalName() const override { return m_nominalName; }

      std::vector<std::string> systematics() const override
      { return m_systematics; }

      std::set<std::string> systematicsAffecting(
          const std::string& branch) const override;

      std::vector<std::string> branches() const override;

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

       std::unique_ptr<IBranchNamer> copy() const override
       { return std::make_unique<DefaultBranchNamer>(*this); }
    private:

      std::map<std::string, std::map<std::string, std::string>> m_branches;

      std::vector<std::string> m_systematics;

      bool m_systNameFirst{true};

      bool m_inputFromFriendTrees{true};

      std::string m_nominalName{"NOSYS"};

  }; //> end class DefaultBranchNamer
} //> end namespace RDFAnalysis
#endif //gb> !RDFAnalysis_BranchNamer_H