PhotonConversion/truthconversion/VtxRegressor.h

0001 /**
0002  * @VtxRegressor.h
0003  * @author Francesco Vassalli <Francesco.Vassalli@colorado.edu>
0004  * @version 1.0
0005  *
0006  * @brief A wrapper class to interface with TMVA for regression of the VtxRegressor vtx
0007  */
0008 #ifndef VtxRegressor_H__
0009 #define VtxRegressor_H__
0010
0011 #include <GenFit/GFRaveVertex.h>
0012 #include <TMVA/Reader.h>
0013
0014 class SvtxTrack;
0015
0016 class VtxRegressor
0017 {
0018     public:
0019         VtxRegressor(std::string, std::string);
0020         ~VtxRegressor();
0021         float regress(SvtxTrack*,SvtxTrack*,genfit::GFRaveVertex*);
0022
0023     private:
0024         TMVA::Reader* reader=NULL;
0025     std::string _method;
0026         float pt1,pt2,phi,dphi,eta,deta,rin;
0027 };
0028 #endif //VtxRegressor_H__