vertex_lcfi/inc/vertex.h

#ifndef LCFIVERTEX_H
#define LCFIVERTEX_H

#include "../util/inc/vector3.h"
#include "../util/inc/matrix.h"
#include <vector>
#include <map>

namespace vertex_lcfi
{
        namespace ZVTOP
        {
                class CandidateVertex;
        }
        
        using namespace util;
        
        //Forward Declarations
        class Track;
        class Event;
        

        class Vertex
        {
        public:
                
                Vertex() {}
                

                Vertex(Event* Event, const std::vector<Track*> & Tracks, const Vector3 & Position, const SymMatrix3x3 & PosError, bool IsPrimary, double Chi2, double Probability, std::map<Track*,double> ChiTrack);
                

                Vertex(Event* Event, const std::vector<Track*> & Tracks, const Vector3 & Position, const SymMatrix3x3 & PosError, bool IsPrimary, double Chi2, double Probability);
                

                Vertex(ZVTOP::CandidateVertex* CandidateVertex, Event* Event);
                

                inline Event* event() const{return _Event;}
                

                inline const std::vector<Track*> & tracks() const {return _Tracks;}
                

                inline bool isPrimary() const {return _IsPrimary;}

                inline bool & isPrimary() {return _IsPrimary;}


                inline void addTrack(Track* AddTrack) {_Tracks.push_back(AddTrack);}
                

                bool removeTrack(Track* RTrack);
                

                bool hasTrack(Track* HTrack) const;
                

                double charge() const;
                

                Vector3 momentum() const;
                

                inline const Vector3 & position() const {return _Position;}
        
                

                inline const SymMatrix3x3 & positionError() const {return _PosError;}

                

                inline double chi2() const {return _Chi2;}
                

                inline std::map<Track*,double> chi2OfTracks() const {return _ChiSquaredOfTrack;}
                

                inline double probability() const {return _Probability;}
                

                double radius(Projection Proj) const;
                

                double radiusError(Projection Proj) const;
                

                double distanceToVertex(Vertex* FarVertex,Projection Proj) const;
                

                double distanceToVertexError(Vertex* FarVertex,Projection Proj) const;
                

        private:
                Event* _Event;
                std::vector<vertex_lcfi::Track*> _Tracks;
                Vector3 _Position;
                SymMatrix3x3 _PosError;
                bool _IsPrimary;
                double _Chi2;
                double _Probability;
                std::map<Track*,double> _ChiSquaredOfTrack;
        };

        }
#endif //LCFIVERTEX_H