Vertex.h

Go to the documentation of this file.

 
#ifndef _VERTEX_H_
#define _VERTEX_H_
 
 
#include <vector>
 
#include <TObject.h>
#include <TRefArray.h>
#include <TRef.h>
#include <TVector3.h>
#include <TLorentzVector.h>
 
//TODO make float/doubles accordingly.
 
class Vertex : public TObject {
 
    public:
 
        Vertex();
 
        ~Vertex();
 
        void Clear(Option_t *option="");
 
        void addParticle(TObject* part);
 
        //TODO unify
        void setChi2(const double chi2) {chi2_ = chi2;}
 
        void setNdf(const double ndf) {ndf_ = ndf;}
 
        void setX(const double x) {pos_.SetX(x);}
 
        void setY(const double y) {pos_.SetY(y);}
 
        void setZ(const double z) {pos_.SetZ(z);}
 
        void setPos(const float* pos, bool rotate = false);
 
        void setPos(const TVector3& pos);
 
        void setVtxParameters(const std::vector<float>& parameters);
 
        void setVtxParameters(const TLorentzVector& p1,
                  const TLorentzVector& p2);
  
        void setVtxParameters(const TVector3& p1,
                  const TVector3& p2,
                  const double m) ;
    
        void setType(const std::string& type) {type_ = type;}
 
        void setID (const int id) {id_=id;}
 
        void setCovariance( const std::vector<float>& vec);
 
        void setProbability(const float probability) {probability_ = probability;}
 
        TRefArray getParticles(){return parts_;}; 
 
        const std::vector<float>& getCovariance() const {return covariance_;}
 
        float getProbability() const {return probability_;}
 
        double getChi2() const {return chi2_;}
 
        double getNdf() const {return ndf_;}
 
        double getX() const {return pos_.X();}
 
        double getY() const {return pos_.Y();}
 
        double getZ() const {return pos_.Z();}
 
        TVector3 getPos() const {return pos_;}
 
        std::string getType() const {return type_;}
 
        int getID () {return id_;}
 
        float getInvMass     () const {return invM_;}
 
        float getInvMassErr  () const {return invMerr_;}
 
        //TODO - Change these to use the track references after filling the
        //track refs
        TVector3 getP1       () const {return p1_;}
        double   getP1X      () const {return p1_.X();}
        double   getP1Y      () const {return p1_.Y();}
        double   getP1Z      () const {return p1_.Z();}
 
        TVector3 getP2       () const {return p2_;}
        double   getP2X      () const {return p2_.X();}
        double   getP2Y      () const {return p2_.Y();}
        double   getP2Z      () const {return p2_.Z();}
 
        TVector3 getP        () const {return p_;}
        
        double getTgtConstrSigmaX() const {return parameters_[9];}
        
        double getTgtConstrX() const {return parameters_[7];}
        
        double getTgtConstrSigmaY() const {return parameters_[11];}
        
        double getTgtConstrY() const {return parameters_[20];}
        
        ClassDef(Vertex,1);
 
    private:
 
        double chi2_{-999};
        int ndf_{-999};  
        TVector3 pos_,p1_,p2_,p_;
 
        float invM_{-999};
        float invMerr_{-999};
 
        std::vector<float> covariance_{};
        float probability_{-999};
        int id_;
        std::string type_{""};
        TRefArray parts_;
        int n_parts_{0};
        std::vector<float> parameters_;
 
}; // Vertex
class Vertex : public TObject {…};
 
#endif // __VERTEX_H__