ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitAna/DataTree/interface/Vertex.h
(Generate patch)

Comparing UserCode/MitAna/DataTree/interface/Vertex.h (file contents):
Revision 1.9 by loizides, Wed Mar 18 15:44:32 2009 UTC vs.
Revision 1.12 by loizides, Fri Dec 11 13:55:46 2009 UTC

# Line 3 | Line 3
3   //
4   // Vertex
5   //
6 < // Vertex class implemented as holding a 3d vector as a point in space with fit information.
6 > // Vertex class derived from BaseVertex holding additional fit information.
7   //
8   // Authors: J.Bendavid
9   //--------------------------------------------------------------------------------------------------
# Line 27 | Line 27 | namespace mithep
27        Vertex(const ThreeVector &pos) :
28          BaseVertex(pos), fChi2(0), fNdof(0), fNTracks(0) {}
29        
30 <      Double_t            Chi2()      const { return fChi2;                    }
31 <      UShort_t            Ndof()      const { return fNdof;                    }
32 <      UInt_t              NTracks()   const { return fNTracks;                 }
33 <      EObjType            ObjType()   const { return kVertex;                  }      
34 <      Double_t            Prob()      const { return TMath::Prob(fChi2,fNdof); }
30 >      Double_t            Chi2()       const { return fChi2;                    }
31 >      Int_t               Compare(const TObject *o) const;
32 >      Bool_t              IsSortable() const { return kTRUE;                    }
33 >      UShort_t            Ndof()       const { return fNdof;                    }
34 >      UInt_t              NTracks()    const { return fNTracks;                 }
35 >      EObjType            ObjType()    const { return kVertex;                  }      
36 >      Double_t            Prob()       const { return TMath::Prob(fChi2,fNdof); }
37        void                SetChi2(Double_t chi2)     { fChi2 = chi2;     }
38        void                SetNdof(UShort_t nDof)     { fNdof = nDof;     }
39        void                SetNTracks(UShort_t ntrks) { fNTracks = ntrks; }
# Line 44 | Line 46 | namespace mithep
46      ClassDef(Vertex, 1) // Vertex class
47    };
48   }
49 +
50 + //--------------------------------------------------------------------------------------------------
51 + inline Int_t mithep::Vertex::Compare(const TObject *o) const
52 + {
53 +  // Default compare function for sorting according to transverse momentum.
54 +  // Returns -1 if this object is smaller than given object, 0 if objects are
55 +  // equal and 1 if this is larger than given object.
56 +
57 +  const mithep::Vertex *v = dynamic_cast<const mithep::Vertex *>(o);
58 +  if (!v)
59 +    return 1;
60 +
61 +  Int_t myn = NTracks();
62 +  Int_t n   = v->NTracks();
63 +  if (myn>n)
64 +    return -1;
65 +  else if (n>myn)
66 +    return +1;
67 +
68 +  Double_t myd = Chi2();
69 +  Double_t d   = v->Chi2();
70 +  if (myd<d)
71 +    return -1;
72 +  else if (d<myd)
73 +    return +1;
74 +  
75 +  return 0;
76 + }
77   #endif

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines