StKFVertex.h

#ifndef __StKFVertex_h__
#define __StKFVertex_h__
// $Id: StKFVertex.h,v 2.4 2013/04/10 22:14:20 fisyak Exp $
#include "Riostream.h"
#include "TObject.h"
#include "StKFTrack.h"
#include "TObjArray.h"
#include "KFVertex.h"
#include "TVector3.h"

class StKFVertex;
std::ostream&  operator<<(std::ostream& os,  const StKFVertex& v);
class StKFVertex : public TObject  {
public:
  StKFVertex(Int_t id = -1) : fID(id) , fIdTruth(0), fQuality(0), fIdParentTk(0),
                              fTimeMc(0), fNoDaughtersMc(0), fgePidMc(0)
  {fKFTracks.SetOwner(kTRUE); Clear();}
  virtual     ~StKFVertex() {Clear();}
  void         AddTrack(const StKFTrack *track);
  virtual void Clear(Option_t *opt="")     {fKFTracks.Clear(opt);}
  StKFTrack*   Remove(Int_t k=0)           {return (StKFTrack *) fKFTracks.RemoveAt(k);}
  StKFTrack*   Remove(StKFTrack *track)    {return (StKFTrack *) fKFTracks.Remove(track);}
  StKFTrack*   Remove(KFParticle *particle); 
  Int_t        ID()                  const {return   fID;}
  KFVertex    &Vertex()                    {return *&fVertex;}
  KFVertex     Vertex()              const {return   fVertex;}
  TObjArray   &Tracks()                    {return *&fKFTracks;}
  Int_t        NoTracks()            const {return   fKFTracks.GetEntriesFast();}
  Int_t        Charge()              const {return   fVertex.GetQ();}
  StKFTrack*   Track(Int_t k = 0)          {return (StKFTrack* ) fKFTracks[k];}
  const StKFTrack*   Track(Int_t k = 0) const {return (const StKFTrack* ) fKFTracks[k];}
  Double_t     UpdateVertex2TrackChi2();
  void         Compress()                  {fKFTracks.Compress();}
  void         Fit(); 
  Int_t        IdTruth() const { return fIdTruth;}
  Int_t        QaTruth() const { return fQuality; }
  Int_t        IdParentTk() const {return fIdParentTk;}
  void         SetIdTruth(Int_t idtru,Int_t qatru=0) {fIdTruth = (UShort_t) idtru; fQuality = (UShort_t) qatru;}
  void         SetIdParentTk(Int_t id) {fIdParentTk = id;}
  Int_t        MultW() const {return MultWE(1);}
  Int_t        MultE() const {return MultWE(2);}
  Int_t        MultWE(Int_t k = 1) const;
  Int_t Q() const;
  void operator +=(StKFVertex &vtx);
  void Print(Option_t *option="") const {std::cout << option << *this << std::endl; }
  void PrintW(Option_t *option="") const;
  void SetMc(Float_t time, Float_t x, Float_t y, Float_t z, Int_t NoDaughters, Int_t gePid);
  Float_t   TimeMc()   const {return fTimeMc;}
  const TVector3 &XyzMc()    const {return *&fXyzMc;}
  Int_t     NoDaughtersMc() const {return fNoDaughtersMc;}
  Int_t     gePidMc()  const {return fgePidMc;}
  static void      SetDebug(Int_t k = 0) {_debug = k;}
  static Int_t     Debug() {return _debug;}
private:
  Int_t     fID;
  KFVertex  fVertex;
  TObjArray fKFTracks;
  UShort_t  fIdTruth; // MC vertex id 
  UShort_t  fQuality; // quality of this information (percentage of hits coming from the above MC track)
  Int_t     fIdParentTk;
  Float_t   fTimeMc;
  TVector3  fXyzMc;
  Int_t     fNoDaughtersMc;
  Int_t     fgePidMc;
 public:
  static Int_t _debug;
  static const Char_t *GeNames[52];
  ClassDef(StKFVertex,0)
};
// $Log: StKFVertex.h,v $
// Revision 2.4  2013/04/10 22:14:20  fisyak
// Roll back to version 04/04/2013
//
// Revision 2.2  2012/06/11 15:33:41  fisyak
// std namespace
//
// Revision 2.1  2012/05/07 14:56:14  fisyak
// Add StKFVertexMaker
//
// Revision 1.3  2012/03/29 23:35:47  fisyak
// Fix problem with multiple beam tracks
//
// Revision 1.2  2012/02/07 19:38:26  fisyak
// Repackage
//
#endif