SelectBranches.hpp

Go to the documentation of this file.

#ifndef CAFE_SELECTBRANCHES_HPP_
#define CAFE_SELECTBRANCHES_HPP_

#include <string>
#include <vector>

class TBranch;
class TTree;

namespace cafe {

    class Event;
    class SelectBranches;

    class BranchNames {
    public:
        BranchNames();
        BranchNames(const std::string& name);
        BranchNames(const std::string& name1, const std::string& name2);
        BranchNames(const std::string& name1, const std::string& name2, const std::string& name3);
        BranchNames(const std::string& name1, const std::string& name2, const std::string& name3, const std::string& name4);
        BranchNames(const std::vector<std::string>& names);

        BranchNames(const BranchNames& other);
        BranchNames& operator=(const BranchNames& other);

    private:
        friend class SelectBranches;
        void disable(TTree *tree, int cookie) const;
        void enable() const;
        
        std::vector<std::string>      _branchNames;
        mutable std::vector<TBranch*> _branches;
        mutable std::vector<bool>     _oldStatus;
        mutable int                   _cookie;
    };


    class SelectBranches {
    public:
        SelectBranches(cafe::Event& event, const BranchNames& names);
        ~SelectBranches();
    private:
        // not implemented
        SelectBranches(const SelectBranches& );
        SelectBranches operator=(const SelectBranches& );

        const BranchNames& _names;
    };
    

}

#endif // CAFE_SELECTBRANCHES_HPP_

---