TRefFinder.hpp

Go to the documentation of this file.

#ifndef CAFUTIL_TREFFINDER_HPP__
#define CAFUTIL_TREFFINDER_HPP__

#include "cafe/Event.hpp"
#include "cafe/Variables.hpp"
#include "TRef.h"
#include <string>

namespace caf_util {

  class TRefFinder {
  private:
    std::string _trackBranch ;
    std::string _vertexBranch ;
    cafe::Variables _trvars ;
    cafe::Variables _vtxvars ;
    int _debug ;
  public :    
    TRefFinder() ;
    virtual ~TRefFinder() {}

    void setDebug(int debug) {_debug = debug;}

    void setTrackBranchName(const std::string name) {_trackBranch = name ;}
    void setVertexBranchName(const std::string name) {_vertexBranch = name ;}

    void SetVariablesTrack(const cafe::Variables& vars) {_trvars = vars ;}
    void addTrackVariable(const std::string& name) {_trvars.add(name) ;}
    void SetVariablesVertex(const cafe::Variables& vars) {_vtxvars = vars ;}    
    void addVertexVariable(const std::string& name) {_vtxvars.add(name) ;}

    const TMBTrack*  FindTrack(cafe::Event &event, const TRef& ref) const ;
    const TMBPrimaryVertex*  FindVertex(cafe::Event &event, const TRef& ref) const ;

    ClassDef(TRefFinder, 0) ;
  } ;

}
#endif // CAFUTIL_TREFFINDER_HPP__

---